fix: explicit return to mark stream done

run-llama · Oct 10, 2024 · 9530db4 · 9530db4
1 parent 828859a
commit 9530db4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/templates/components/multiagent/typescript/workflow/single-agent.ts b/templates/components/multiagent/typescript/workflow/single-agent.ts
@@ -143,13 +143,15 @@ export class FunctionCallingAgent extends Workflow {
         fullResponse = chunk;
       }
 
-      if (fullResponse) {
+      if (fullResponse?.options && Object.keys(fullResponse.options).length) {
         memory.put({
           role: "assistant",
           content: "",
           options: fullResponse.options,
         });
         yield fullResponse;
+      } else {
+        return; // Explicitly mark generator as done
       }
     };