智能体分权

frontend-web/src/core/threads/hooks.ts

@@ -28,6 +28,7 @@ export type ThreadStreamOptions = {
   threadId?: string | null | undefined;
   context: LocalSettings["context"];
   isMock?: boolean;
+  assistantId?: string;
   onStart?: (threadId: string) => void;
   onFinish?: (state: AgentThreadState) => void;
   onToolEnd?: (event: ToolEndEvent) => void;
@@ -139,6 +140,7 @@ export function useThreadStream({
   threadId,
   context,
   isMock,
+  assistantId,
   onStart,
   onFinish,
   onToolEnd,
@@ -204,7 +206,7 @@ export function useThreadStream({
 
   const thread = useStream<AgentThreadState>({
     client: getAPIClient(isMock),
-    assistantId: "lead_agent",
+    assistantId: assistantId ?? "lead_agent",
     threadId: onStreamThreadId,
     reconnectOnMount: runMetadataStorageRef.current
       ? () => runMetadataStorageRef.current!
@@ -213,10 +215,11 @@ export function useThreadStream({
     onCreated(meta) {
       handleStreamStart(meta.thread_id);
       setOnStreamThreadId(meta.thread_id);
-      if (context.agent_name && !isMock) {
+      const metadataAgentId = context.agent_id ?? context.agent_name;
+      if (metadataAgentId && !isMock) {
         void getAPIClient()
           .threads.update(meta.thread_id, {
-            metadata: { agent_name: context.agent_name },
+            metadata: { agent_id: metadataAgentId },
           })
           .catch(() => ({}));
       }