feat(workbench): redesign investigation reports and diagrams

src/features/workbench/hooks/use-workbench-agent.ts

@@ -10,6 +10,8 @@ import { showMapNotice } from "@/features/map/core";
 import { findOwnedWorkbenchProject } from "../api/project-client";
 import {
   FrontendActionExecutor,
+  agentTaskDiscussionContextSchema,
+  type AgentTaskDiscussionContext,
   agentBootstrapKey,
   agentSessionsKey,
   createAgentApiClient,
@@ -90,6 +92,7 @@ export function useWorkbenchAgent({
   const [sessionId, setSessionId] = useState<string | null>(null);
   const [sessionPurpose, setSessionPurpose] = useState<AgentSessionPurpose>("ordinary_chat");
   const [sessionTaskId, setSessionTaskId] = useState<string | null>(null);
+  const taskDiscussionRef = useRef<{ sessionId: string; context: AgentTaskDiscussionContext } | null>(null);
   const [registry, setRegistry] = useState<UIRegistry | null>(null);
   const [sessionTitle, setSessionTitle] = useState("新对话");
   const [statusLabel, setStatusLabel] = useState("Agent 后端连接中");
@@ -231,7 +234,11 @@ export function useWorkbenchAgent({
         api: `${env.TJWATER_AGENT_API_BASE_URL.replace(/\/$/, "")}/api/v1/agent/chat/stream`,
         headers: async (): Promise<Record<string, string>> => {
           const accessToken = await getAccessToken?.();
-          return accessToken ? { Authorization: `Bearer ${accessToken}` } : {};
+          const projectId = await getProjectId();
+          return {
+            ...(projectId ? { "X-Project-Id": projectId } : {}),
+            ...(accessToken ? { Authorization: `Bearer ${accessToken}` } : {})
+          };
         },
         prepareSendMessagesRequest({ id, messages, body, trigger, messageId }) {
           return {
@@ -248,7 +255,7 @@ export function useWorkbenchAgent({
           };
         }
       }),
-    [getAccessToken]
+    [getAccessToken, getProjectId]
   );
 
   const chat = useChat<AgentUiMessage>({
@@ -575,7 +582,7 @@ export function useWorkbenchAgent({
     [applySessionId, chat, refreshSessionHistory, startSessionStreamSubscription, stopSessionStreamSubscription, streaming]
   );
 
-  const openTaskCollaboration = useCallback(async (taskId: string) => {
+  const openTaskCollaboration = useCallback(async (taskId: string, context?: AgentTaskDiscussionContext) => {
     const normalizedTaskId = taskId.trim();
     if (!normalizedTaskId) {
       throw new Error("工况任务 ID 不能为空。");
@@ -586,6 +593,7 @@ export function useWorkbenchAgent({
       throw new Error("Agent 未返回任务协作会话。");
     }
     await loadHistorySession(collaborationSessionId);
+    taskDiscussionRef.current = { sessionId: collaborationSessionId, context: agentTaskDiscussionContextSchema.parse(context ?? {}) };
     return collaborationSessionId;
   }, [loadHistorySession]);
 
@@ -709,7 +717,8 @@ export function useWorkbenchAgent({
           body: {
             session_id: sessionIdRef.current ?? undefined,
             model: selectedModel || undefined,
-            approval_mode: approvalMode
+            approval_mode: approvalMode,
+            task_context: taskDiscussionRef.current?.sessionId === sessionIdRef.current ? taskDiscussionRef.current.context : undefined
           }
         }
       );