graphlit-client 1.0.20250716002 → 1.0.20250716003

package/dist/streaming/ui-event-adapter.d.ts

@@ -31,6 +31,8 @@ export declare class UIEventAdapter {
     private reasoningSignature?;
     private isInReasoning;
     private usageData?;
+    private hasToolCallsInProgress;
+    private hadToolCallsBeforeResume;
     constructor(onEvent: (event: AgentStreamEvent) => void, conversationId: string, options?: {
         smoothingEnabled?: boolean;
         chunkingStrategy?: ChunkingStrategy;

package/dist/streaming/ui-event-adapter.js

@@ -30,6 +30,8 @@ export class UIEventAdapter {
     reasoningSignature;
     isInReasoning = false;
     usageData;
+    hasToolCallsInProgress = false;
+    hadToolCallsBeforeResume = false;
     constructor(onEvent, conversationId, options = {}) {
         this.onEvent = onEvent;
         this.conversationId = conversationId;
@@ -99,6 +101,9 @@ export class UIEventAdapter {
         this.lastTokenTime = 0;
         this.tokenCount = 0;
         this.tokenDelays = [];
+        // Reset tool call tracking flags
+        this.hasToolCallsInProgress = false;
+        this.hadToolCallsBeforeResume = false;
         // Note: We only clear tool calls here if this is truly a new conversation start
         // For multi-round tool calling, handleStart is only called once at the beginning
         if (this.activeToolCalls.size > 0) {
@@ -124,6 +129,19 @@ export class UIEventAdapter {
         }
         this.lastTokenTime = now;
         this.tokenCount++;
+        // Check if we're resuming after tool calls and need to add newlines
+        if (this.hadToolCallsBeforeResume && this.hasToolCallsInProgress === false) {
+            // We had tool calls before and now we're receiving content again
+            // Add double newline to separate the content from tool results
+            if (this.currentMessage.length > 0 && !this.currentMessage.endsWith('\n\n')) {
+                if (process.env.DEBUG_GRAPHLIT_SDK_STREAMING) {
+                    console.log(`📝 [UIEventAdapter] Adding newlines after tool calls before resuming content`);
+                }
+                this.currentMessage += '\n\n';
+            }
+            // Reset the flag now that we've added the newlines
+            this.hadToolCallsBeforeResume = false;
+        }
         if (this.chunkBuffer) {
             const chunks = this.chunkBuffer.addToken(token);
             // Add chunks to queue for all chunking modes (character, word, sentence)
@@ -155,6 +173,9 @@ export class UIEventAdapter {
             toolCall: conversationToolCall,
             status: "preparing",
         });
+        // Mark that we have tool calls in progress
+        this.hasToolCallsInProgress = true;
+        this.hadToolCallsBeforeResume = true;
         if (process.env.DEBUG_GRAPHLIT_SDK_STREAMING) {
             console.log(`🔧 [UIEventAdapter] Active tool calls after: ${this.activeToolCalls.size}`);
         }
@@ -219,6 +240,9 @@ export class UIEventAdapter {
                 toolCall: conversationToolCall,
                 status: "ready",
             });
+            // Mark that we have tool calls
+            this.hasToolCallsInProgress = true;
+            this.hadToolCallsBeforeResume = true;
             this.emitUIEvent({
                 type: "tool_update",
                 toolCall: conversationToolCall,
@@ -246,6 +270,21 @@ export class UIEventAdapter {
         else {
             console.warn(`🔧 [UIEventAdapter] Tool call complete for unknown tool ID: ${toolCall.id}`);
         }
+        // Check if all tool calls are complete
+        let allComplete = true;
+        for (const [, data] of this.activeToolCalls) {
+            if (data.status !== "completed" && data.status !== "failed") {
+                allComplete = false;
+                break;
+            }
+        }
+        if (allComplete && this.activeToolCalls.size > 0) {
+            // All tool calls are complete, mark that we're no longer processing tools
+            this.hasToolCallsInProgress = false;
+            if (process.env.DEBUG_GRAPHLIT_SDK_STREAMING) {
+                console.log(`🔧 [UIEventAdapter] All tool calls complete, ready to resume content streaming`);
+            }
+        }
     }
     handleComplete(tokens) {
         if (process.env.DEBUG_GRAPHLIT_SDK_STREAMING) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "graphlit-client",
-  "version": "1.0.20250716002",
+  "version": "1.0.20250716003",
   "description": "Graphlit API Client for TypeScript",
   "type": "module",
   "main": "./dist/client.js",
@@ -28,11 +28,7 @@
     "format": "prettier --write .",
     "build": "tsc -p tsconfig.json",
     "prepublishOnly": "npm run build",
-    "test": "vitest",
-    "test:watch": "vitest --watch",
-    "test:coverage": "vitest --coverage",
-    "test:ui": "vitest --ui",
-    "test:streaming": "vitest --run src/tests/streaming"
+    "test": "vitest"
   },
   "keywords": [
     "Graphlit",