import { createHmac } from 'node:crypto';

function issueAgentJwt(userId, orgId) {
  const now = Math.floor(Date.now() / 1000);
  const header = { alg: 'HS256', typ: 'JWT' };
  const payload = { sub: userId, org: orgId, iat: now, exp: now + 4 * 3600 };
  const h = Buffer.from(JSON.stringify(header)).toString('base64url');
  const p = Buffer.from(JSON.stringify(payload)).toString('base64url');
  const sig = createHmac('sha256', process.env.TENSOR_AGENT_S2S_KEY)
    .update(`${h}.${p}`)
    .digest('base64url');
  return `${h}.${p}.${sig}`;
}

app.get('/api/agent-token', requireAuth, (req, res) => {
  res.json({ jwt: issueAgentJwt(req.user.id, req.user.orgId) });
});

import base64, hmac, hashlib, json, os, time

def issue_agent_jwt(user_id: str, org_id: str) -> str:
    now = int(time.time())
    header = {"alg": "HS256", "typ": "JWT"}
    payload = {"sub": user_id, "org": org_id, "iat": now, "exp": now + 4 * 3600}
    b64 = lambda d: base64.urlsafe_b64encode(
        json.dumps(d, separators=(",", ":")).encode()
    ).rstrip(b"=").decode()
    h, p = b64(header), b64(payload)
    sig = base64.urlsafe_b64encode(
        hmac.new(os.environ["TENSOR_AGENT_S2S_KEY"].encode(),
                 f"{h}.{p}".encode(), hashlib.sha256).digest()
    ).rstrip(b"=").decode()
    return f"{h}.{p}.{sig}"

func issueAgentJWT(userID, orgID string) string {
    now := time.Now().Unix()
    header := map[string]string{"alg": "HS256", "typ": "JWT"}
    payload := map[string]any{"sub": userID, "org": orgID, "iat": now, "exp": now + 4*3600}
    h, _ := json.Marshal(header); p, _ := json.Marshal(payload)
    hB := base64.RawURLEncoding.EncodeToString(h)
    pB := base64.RawURLEncoding.EncodeToString(p)
    mac := hmac.New(sha256.New, []byte(os.Getenv("TENSOR_AGENT_S2S_KEY")))
    mac.Write([]byte(hB + "." + pB))
    sig := base64.RawURLEncoding.EncodeToString(mac.Sum(nil))
    return hB + "." + pB + "." + sig
}

npm install @nexusgpu/agent-sdk-react

import { TensorAgentProvider, useTensorAgent } from '@nexusgpu/agent-sdk-react';

function App() {
  const [jwt, setJwt] = useState('');
  useEffect(() => {
    fetch('/api/agent-token').then(r => r.json()).then(d => setJwt(d.jwt));
  }, []);
  if (!jwt) return <Loading />;

  return (
    <TensorAgentProvider
      jwt={jwt}
      agentId="my-agent"
      onDataChange={(e) => queryClient.invalidateQueries([e.resource])}
      onAuthFailed={() => {
        // 极少触发：续期彻底失败（比如你轮换了 S2S 密钥）。
        // 重新调用你的后端拿一个新 JWT。
        fetch('/api/agent-token').then(r => r.json()).then(d => setJwt(d.jwt));
      }}
    >
      <YourApp />
    </TensorAgentProvider>
  );
}

function CustomerPage({ customer }) {
  const { setContextPrompts, sendPrompt, open } = useTensorAgent();
  useEffect(() => {
    setContextPrompts([`正在查看客户 ${customer.name}`]);
  }, [customer]);
  return <button onClick={() => { open(); sendPrompt('分析这个客户'); }}>AI 分析</button>;
}

npm install @nexusgpu/agent-sdk-vue

<template>
  <TensorAgentLayout
    :jwt="jwt"
    agent-id="my-agent"
    @data-change="onDataChange"
    @auth-failed="refetchJwt"
  >
    <router-view />
  </TensorAgentLayout>
</template>

<script setup lang="ts">
import { TensorAgentLayout } from '@nexusgpu/agent-sdk-vue';
</script>

npm install @nexusgpu/agent-sdk-js

import { TensorAgent } from '@nexusgpu/agent-sdk-js';

const res = await fetch('/api/agent-token');
const { jwt } = await res.json();

const agent = TensorAgent.init({
  jwt,
  agentId: 'my-agent',
  onDataChange: (e) => refreshList(e.resource),
  onAuthFailed: async () => {
    const r = await fetch('/api/agent-token');
    const { jwt } = await r.json();
    agent.updateToken(jwt);
  },
});

<script src="https://agent.tos.run/sdk/agent-embed.iife.js"></script>
<script>
  fetch('/api/agent-token').then(r => r.json()).then(({ jwt }) => {
    TensorAgent.init({ jwt, agentId: 'my-agent' });
  });
</script>