Merge pull request #32 from receptron/fixStreaming

fix streaming
receptron · May 11, 2024 · 00abd08 · 00abd08
2 parents d477514 + 33f4da5
commit 00abd08
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/utils/graph_data.ts b/src/utils/graph_data.ts
@@ -265,7 +265,7 @@ const messages = [
 
 Array.from(messages.keys()).forEach((k) => {
   const message = messages[k];
-  const inputs = k > 2 ? ["streamMockAgent" + (k - 3)] : ["echo"];
+  const inputs = k > 2 ? [":streamMockAgent" + (k - 3)] : [":echo"];
   graph_data_stream["nodes"]["streamMockAgent" + k] = {
     agent: "streamMockAgent",
     inputs,

diff --git a/src/utils/stream.ts b/src/utils/stream.ts
@@ -11,7 +11,7 @@ export const useStreamData = () => {
     streamData.value[nodeId] = (streamData.value[nodeId] || "") + token;
   };
   const streamAgentFilter: AgentFilterFunction = async (context, next) => {
-    context.params.streamCallback = (token: string) => {
+    context.filterParams.streamTokenCallback = (token: string) => {
       outSideFunciton(context.debugInfo.nodeId, token);
     };
     return next(context);