@workglow/ai 0.2.33 → 0.2.34

package/dist/bun.js

@@ -1471,7 +1471,7 @@ var AiChatOutputSchema = {
     text: {
       type: "string",
       title: "Text",
-      description: "Last assistant response",
+      description: "Full streamed transcript across all assistant turns",
       "x-stream": "append"
     },
     messages: {
@@ -1556,6 +1556,7 @@ class AiChatTask extends StreamingAiTask {
         await getAiProviderRegistry().disposeSession(model.provider, sessionId);
       });
     }
+    let completedTurns = 0;
     yield {
       type: "object-delta",
       port: "messages",
@@ -1581,6 +1582,7 @@ class AiChatTask extends StreamingAiTask {
         content: [{ type: "text", text: assistantText }]
       };
       history.push(assistantMsg);
+      completedTurns = turn + 1;
       yield {
         type: "object-delta",
         port: "messages",
@@ -1620,7 +1622,10 @@ class AiChatTask extends StreamingAiTask {
         objectDelta: [userMsg]
       };
     }
-    yield { type: "finish" };
+    yield {
+      type: "finish",
+      data: { iterations: completedTurns }
+    };
   }
 }
 
@@ -1847,7 +1852,7 @@ var AiChatWithKbOutputSchema = {
     text: {
       type: "string",
       title: "Text",
-      description: "Last assistant response",
+      description: "Full streamed transcript across all assistant turns",
       "x-stream": "append"
     },
     messages: {
@@ -1939,6 +1944,7 @@ class AiChatWithKbTask extends StreamingAiTask {
     const topK = input.topKPerKb ?? 4;
     const minScore = input.minScore ?? 0.3;
     const maxRefs = input.maxReferences ?? 6;
+    let completedTurns = 0;
     let lastNonEmptyRefs = [];
     for (let turn = 0;turn < maxIterations; turn++) {
       const lastUserText = extractLastUserText(history);
@@ -1965,12 +1971,13 @@ class AiChatWithKbTask extends StreamingAiTask {
         }));
       }
       const allChunks = perKbResults.flatMap(({ kbId, kbLabel, kb, results }) => results.filter((r) => r.score >= minScore).map((r) => ({ kbId, kbLabel, kb, r }))).sort((a, b) => b.r.score - a.r.score).slice(0, maxRefs);
+      const docUrlKey = (kbId, docId) => `${kbId}:${docId}`;
       const docUrls = new Map;
       const docFetches = [];
-      for (const { kb, r } of allChunks) {
+      for (const { kbId, kb, r } of allChunks) {
         if (!kb)
           continue;
-        const key = `${r.doc_id}`;
+        const key = docUrlKey(kbId, r.doc_id);
         if (docUrls.has(key))
           continue;
         docUrls.set(key, undefined);
@@ -1986,7 +1993,7 @@ class AiChatWithKbTask extends StreamingAiTask {
         kbId: entry.kbId,
         kbLabel: entry.kbLabel,
         result: entry.r,
-        url: docUrls.get(entry.r.doc_id)
+        url: docUrls.get(docUrlKey(entry.kbId, entry.r.doc_id))
       }));
       const effectiveRefs = refs.length > 0 ? refs : lastNonEmptyRefs;
       if (refs.length > 0) {
@@ -2046,6 +2053,7 @@ class AiChatWithKbTask extends StreamingAiTask {
         content: [{ type: "text", text: assistantText }]
       };
       history.push(assistantMsg);
+      completedTurns = turn + 1;
       yield {
         type: "object-delta",
         port: "messages",
@@ -2085,7 +2093,10 @@ class AiChatWithKbTask extends StreamingAiTask {
         objectDelta: [userMsg]
       };
     }
-    yield { type: "finish" };
+    yield {
+      type: "finish",
+      data: { iterations: completedTurns }
+    };
   }
 }
 function extractLastUserText(messages) {
@@ -8028,4 +8039,4 @@ export {
   AI_PROVIDER_REGISTRY
 };
 
-//# debugId=CA4BEA1CEF63FF6F64756E2164756E21
+//# debugId=79B4A8A40215A1E864756E2164756E21