@dudousxd/nestjs-agent-react 0.4.2 → 0.5.0

package/dist/index.cjs

@@ -43,6 +43,7 @@ __export(src_exports, {
   loadStoredSpeechLang: () => loadStoredSpeechLang,
   persistSpeechLang: () => persistSpeechLang,
   storedMessageToUiMessage: () => storedMessageToUiMessage,
+  storedThreadToUiMessages: () => storedThreadToUiMessages,
   useAgentChat: () => useAgentChat,
   useSpeechRecognition: () => useSpeechRecognition
 });
@@ -70,6 +71,7 @@ var AgentChatTransport = class {
     return this.currentThreadId;
   }
   async sendMessages(options) {
+    this.options.onAttemptStart?.();
     const lastMessage = options.messages.at(-1);
     const message = lastMessage ? extractText(lastMessage) : "";
     const body = {
@@ -102,6 +104,7 @@ var AgentChatTransport = class {
     return this.toChunkStream(response.body);
   }
   async reconnectToStream(options) {
+    this.options.onAttemptStart?.();
     const runId = this.options.getResumeRunId?.();
     if (runId === void 0) return null;
     const response = await this.fetchImpl()(`${this.baseUrl()}/agent/chat/${encodeURIComponent(runId)}/stream`, {
@@ -611,6 +614,59 @@ function storedMessageToUiMessage(message) {
 }
 __name(storedMessageToUiMessage, "storedMessageToUiMessage");
 
+// src/stored-thread-to-ui-messages.ts
+function storedThreadToUiMessages(messages) {
+  const turns = [];
+  for (const message of messages) {
+    const currentTurn = turns.at(-1);
+    const previousRow = currentTurn?.at(-1);
+    if (message.role === "assistant" && previousRow?.role === "assistant" && currentTurn) {
+      currentTurn.push(message);
+    } else {
+      turns.push([
+        message
+      ]);
+    }
+  }
+  return turns.map(mergeTurn);
+}
+__name(storedThreadToUiMessages, "storedThreadToUiMessages");
+function mergeTurn(rows) {
+  const lastRow = rows.at(-1);
+  if (lastRow === void 0) {
+    throw new Error("storedThreadToUiMessages: encountered an empty turn");
+  }
+  if (rows.length === 1) {
+    return storedMessageToUiMessage(lastRow);
+  }
+  const parts = rows.flatMap((row) => storedMessageToUiMessage(row).parts);
+  const usage = sumUsage(rows);
+  return {
+    id: lastRow.id,
+    role: lastRow.role,
+    parts,
+    ...usage !== void 0 ? {
+      metadata: {
+        usage
+      }
+    } : {}
+  };
+}
+__name(mergeTurn, "mergeTurn");
+function sumUsage(rows) {
+  const usages = rows.map((row) => row.usage).filter((usage) => usage !== void 0);
+  if (usages.length === 0) {
+    return void 0;
+  }
+  const allCostsUnknown = usages.every((usage) => usage.costUsd === null || usage.costUsd === void 0);
+  return {
+    inputTokens: usages.reduce((sum, usage) => sum + usage.inputTokens, 0),
+    outputTokens: usages.reduce((sum, usage) => sum + usage.outputTokens, 0),
+    costUsd: allCostsUnknown ? null : usages.reduce((sum, usage) => sum + (usage.costUsd ?? 0), 0)
+  };
+}
+__name(sumUsage, "sumUsage");
+
 // src/use-agent-chat.ts
 var import_react = require("@ai-sdk/react");
 var import_react2 = require("react");
@@ -620,6 +676,7 @@ function useAgentChat(options) {
   const [runId, setRunId] = (0, import_react2.useState)(options.resumeRunId);
   const runIdRef = (0, import_react2.useRef)(runId);
   runIdRef.current = runId;
+  const settlingRunIdRef = (0, import_react2.useRef)(void 0);
   const autoResumeRunIdRef = (0, import_react2.useRef)(void 0);
   const [autoResume, setAutoResume] = (0, import_react2.useState)(false);
   const [activeRunId, setActiveRunId] = (0, import_react2.useState)(null);
@@ -662,6 +719,7 @@ function useAgentChat(options) {
   ]);
   const transport = (0, import_react2.useMemo)(() => {
     function onMeta(meta) {
+      settlingRunIdRef.current = meta.runId;
       setRunId(meta.runId);
       if (latest.current.threadId === void 0 && meta.threadId && createdThreadId.current !== meta.threadId) {
         createdThreadId.current = meta.threadId;
@@ -702,7 +760,10 @@ function useAgentChat(options) {
         };
       }, "getBody"),
       getResumeRunId: /* @__PURE__ */ __name(() => latest.current.resumeRunId ?? autoResumeRunIdRef.current, "getResumeRunId"),
-      onMeta
+      onMeta,
+      onAttemptStart: /* @__PURE__ */ __name(() => {
+        settlingRunIdRef.current = void 0;
+      }, "onAttemptStart")
     });
   }, []);
   const chat = (0, import_react.useChat)({
@@ -714,8 +775,15 @@ function useAgentChat(options) {
     ...options.initialMessages !== void 0 ? {
       messages: options.initialMessages
     } : {},
-    onFinish: /* @__PURE__ */ __name(() => {
+    onFinish: /* @__PURE__ */ __name(({ isError }) => {
       latest.current.onFinish?.();
+      const settledRunId = settlingRunIdRef.current;
+      if (settledRunId !== void 0) {
+        latest.current.onRunSettled?.({
+          runId: settledRunId,
+          status: isError ? "failed" : "completed"
+        });
+      }
     }, "onFinish")
   });
   const chatRef = (0, import_react2.useRef)(chat);
@@ -1421,6 +1489,7 @@ __name(persistSpeechLang, "persistSpeechLang");
   loadStoredSpeechLang,
   persistSpeechLang,
   storedMessageToUiMessage,
+  storedThreadToUiMessages,
   useAgentChat,
   useSpeechRecognition
 });