@dudousxd/nestjs-agent-react 0.3.3 → 0.4.1

package/dist/index.cjs

@@ -42,6 +42,7 @@ __export(src_exports, {
   formatRelativeTime: () => formatRelativeTime,
   loadStoredSpeechLang: () => loadStoredSpeechLang,
   persistSpeechLang: () => persistSpeechLang,
+  storedMessageToUiMessage: () => storedMessageToUiMessage,
   useAgentChat: () => useAgentChat,
   useSpeechRecognition: () => useSpeechRecognition
 });
@@ -223,6 +224,14 @@ var AgentChatTransport = class {
               break;
             case "step-finish":
               closeStep();
+              if (event.costUsd !== void 0) {
+                controller.enqueue({
+                  type: "message-metadata",
+                  messageMetadata: {
+                    costUsd: event.costUsd
+                  }
+                });
+              }
               break;
             case "text":
               ensureStep();
@@ -259,7 +268,15 @@ var AgentChatTransport = class {
               controller.enqueue({
                 type: "tool-input-start",
                 toolCallId: event.id,
-                toolName: event.name
+                toolName: event.name,
+                // `toolKind` is absent on older backends — omitted (not `undefined`-valued) so
+                // `toolMetadata` itself is only present when there's something to say, letting a
+                // UI gate approval affordances on `kind === 'action'` without hardcoding tool names.
+                ...event.toolKind !== void 0 ? {
+                  toolMetadata: {
+                    toolKind: event.toolKind
+                  }
+                } : {}
               });
               break;
             case "tool-input-delta":
@@ -276,7 +293,12 @@ var AgentChatTransport = class {
                 type: "tool-input-available",
                 toolCallId: event.id,
                 toolName: event.name,
-                input: event.input
+                input: event.input,
+                ...event.toolKind !== void 0 ? {
+                  toolMetadata: {
+                    toolKind: event.toolKind
+                  }
+                } : {}
               });
               break;
             case "tool-output":
@@ -458,10 +480,39 @@ var AgentClient = class {
     return this.request("POST", `/agent/threads/${encodeURIComponent(threadId)}/fork-from/${encodeURIComponent(messageId)}`);
   }
   renameThread(id, title) {
-    return this.request("PATCH", `/agent/threads/${encodeURIComponent(id)}`, {
+    return this.updateThread(id, {
       title
     });
   }
+  /** General `PATCH /agent/threads/:threadId` — title and/or the thread's pinned default agent. */
+  updateThread(id, patch) {
+    return this.request("PATCH", `/agent/threads/${encodeURIComponent(id)}`, patch);
+  }
+  /**
+  * Uploads a file (image/PDF) for a vision-capable model turn. Multipart, field name `file` —
+  * mirrors the backend's `POST /agent/attachments`. The returned {@link MessageAttachment} is
+  * what a caller then rides on `sendMessage({ text }, { body: { attachments: [...] } })`.
+  */
+  async uploadAttachment(file) {
+    const fetchImpl = this.options.fetch ?? globalThis.fetch;
+    const baseUrl = (this.options.baseUrl ?? "").replace(/\/$/, "");
+    const dynamic = await this.options.getHeaders?.() ?? {};
+    const formData = new FormData();
+    formData.append("file", file);
+    const response = await fetchImpl(`${baseUrl}/agent/attachments`, {
+      method: "POST",
+      headers: {
+        accept: "application/json",
+        ...this.options.headers,
+        ...dynamic
+      },
+      body: formData,
+      ...this.options.credentials !== void 0 ? {
+        credentials: this.options.credentials
+      } : {}
+    });
+    return this.handleResponse(response, "POST", "/agent/attachments");
+  }
   promoteThread(id) {
     return this.request("POST", `/agent/threads/${encodeURIComponent(id)}/promote`);
   }
@@ -501,6 +552,9 @@ var AgentClient = class {
         credentials: this.options.credentials
       } : {}
     });
+    return this.handleResponse(response, method, path);
+  }
+  async handleResponse(response, method, path) {
     if (!response.ok) {
       throw new AgentHttpError(response.status, method, path, response.statusText);
     }
@@ -511,6 +565,52 @@ var AgentClient = class {
   }
 };
 
+// src/stored-message-to-ui-message.ts
+function storedMessageToUiMessage(message) {
+  const parts = [];
+  if (message.content) {
+    parts.push({
+      type: "text",
+      text: message.content
+    });
+  }
+  for (const attachment of message.attachments ?? []) {
+    parts.push({
+      type: "file",
+      mediaType: attachment.contentType,
+      filename: attachment.name,
+      url: attachment.url
+    });
+  }
+  for (const call of message.toolCalls ?? []) {
+    const result = message.toolResults?.find((candidate) => candidate.id === call.id);
+    const toolKind = call.kind;
+    parts.push({
+      type: `tool-${call.name}`,
+      toolCallId: call.id,
+      ...toolKind !== void 0 ? {
+        toolMetadata: {
+          toolKind
+        }
+      } : {},
+      ...result !== void 0 ? {
+        state: "output-available",
+        input: call.input,
+        output: result.output
+      } : {
+        state: "input-available",
+        input: call.input
+      }
+    });
+  }
+  return {
+    id: message.id,
+    role: message.role,
+    parts
+  };
+}
+__name(storedMessageToUiMessage, "storedMessageToUiMessage");
+
 // src/use-agent-chat.ts
 var import_react = require("@ai-sdk/react");
 var import_react2 = require("react");
@@ -520,6 +620,9 @@ function useAgentChat(options) {
   const [runId, setRunId] = (0, import_react2.useState)(options.resumeRunId);
   const runIdRef = (0, import_react2.useRef)(runId);
   runIdRef.current = runId;
+  const autoResumeRunIdRef = (0, import_react2.useRef)(void 0);
+  const [autoResume, setAutoResume] = (0, import_react2.useState)(false);
+  const [activeRunId, setActiveRunId] = (0, import_react2.useState)(null);
   const regenerateNext = (0, import_react2.useRef)(false);
   const createdThreadId = (0, import_react2.useRef)(void 0);
   const client = (0, import_react2.useMemo)(() => {
@@ -537,6 +640,26 @@ function useAgentChat(options) {
       getHeaders: /* @__PURE__ */ __name(async () => mergeHeaders(latest.current), "getHeaders")
     });
   }, []);
+  (0, import_react2.useEffect)(() => {
+    if (!options.resume || options.threadId === void 0) return;
+    let cancelled = false;
+    const threadId = options.threadId;
+    void client.getThread(threadId).then((thread) => {
+      if (cancelled) return;
+      setActiveRunId(thread.activeRunId ?? null);
+      if (thread.activeRunId != null) {
+        autoResumeRunIdRef.current = thread.activeRunId;
+        setAutoResume(true);
+      }
+    });
+    return () => {
+      cancelled = true;
+    };
+  }, [
+    client,
+    options.resume,
+    options.threadId
+  ]);
   const transport = (0, import_react2.useMemo)(() => {
     function onMeta(meta) {
       setRunId(meta.runId);
@@ -578,13 +701,13 @@ function useAgentChat(options) {
           } : {}
         };
       }, "getBody"),
-      getResumeRunId: /* @__PURE__ */ __name(() => latest.current.resumeRunId, "getResumeRunId"),
+      getResumeRunId: /* @__PURE__ */ __name(() => latest.current.resumeRunId ?? autoResumeRunIdRef.current, "getResumeRunId"),
       onMeta
     });
   }, []);
   const chat = (0, import_react.useChat)({
     transport,
-    resume: options.resumeRunId !== void 0,
+    resume: options.resumeRunId !== void 0 || autoResume,
     ...options.threadId !== void 0 ? {
       id: options.threadId
     } : {},
@@ -651,10 +774,10 @@ function useAgentChat(options) {
   ]);
   const cancel = (0, import_react2.useCallback)(async () => {
     chatRef.current.stop();
-    const activeRunId = runIdRef.current;
-    if (activeRunId) {
+    const activeRunId2 = runIdRef.current;
+    if (activeRunId2) {
       try {
-        await client.cancelStream(activeRunId);
+        await client.cancelStream(activeRunId2);
       } catch {
       }
     }
@@ -686,6 +809,8 @@ function useAgentChat(options) {
     ...chat,
     addToolResult,
     runId,
+    /** The `resume`-fetched thread's active run id, or `null` once resolved with none. */
+    activeRunId,
     client,
     threads,
     loadThreads,
@@ -1295,6 +1420,7 @@ __name(persistSpeechLang, "persistSpeechLang");
   formatRelativeTime,
   loadStoredSpeechLang,
   persistSpeechLang,
+  storedMessageToUiMessage,
   useAgentChat,
   useSpeechRecognition
 });