Merge pull request #4 from ai-made-approachable/return_graph_output

Optionally return graph output (non streaming)

config/default.json

@@ -1,7 +1,11 @@
 {
     "file": "./example.rivet-project",
     "graphName": "chat-history",
-    "nodeType": "chat",
-    "nodeName": "Output (Chat)",
-    "graphInput": "input"
-}
+    "graphInputName": "input",
+    "streamingOutput": {
+      "nodeType": "chat",
+      "nodeName": "Output (Chat)"
+    },
+    "returnGraphOutput": false,
+    "graphOutputName": "output"
+  }

example.rivet-data

@@ -1 +1 @@
-{"datasets":[{"meta":{"id":"J-3PhzWvBMkJHEu6dQLX1","projectId":"vHpwmvjUYbE-MQJ7H_sV_","name":"New Dataset","description":""},"data":{"id":"J-3PhzWvBMkJHEu6dQLX1","rows":[]}}]}
+{"datasets":[]}

src/graphManager.ts

@@ -32,7 +32,7 @@ class GraphManager {
         this.startDebuggerServerIfNeeded();
         const projectContent = await fs.readFile(config.get('file'), 'utf8');
         const project = loadProjectFromString(projectContent);
-        const graphInput = config.get('graphInput') as string;
+        const graphInput = config.get('graphInputName') as string;
 
         const options = {
           graph: config.get('graphName'),
@@ -65,8 +65,8 @@ class GraphManager {
             for await (const event of processor.events()) {
                 if (
                     event.type === 'partialOutput' &&
-                    event.node.type === config.get('nodeType') &&
-                    event.node.title === config.get('nodeName')
+                    event.node.type === config.get('streamingOutput.nodeType') &&
+                    event.node.title === config.get('streamingOutput.nodeName')
                 ) {
                     const content = (event.outputs as any).response.value;
                     this.output = content; // Update the output variable with the content
@@ -81,7 +81,11 @@ class GraphManager {
 
             console.log('Finished processing events'); // Debugging line
 
-            await runPromise;
+            const finalOutputs = await runPromise;
+            // Also return the graph output if returnGraphOutput is configured as true
+            if(config.get('returnGraphOutput')) {
+                yield finalOutputs[config.get('graphOutputName')].value;
+            }
             this.isRunning = false;
 
             console.log('runGraph finished'); // Debugging line
@@ -91,7 +95,7 @@ class GraphManager {
             this.isRunning = false;
         }
     }
-
+    
     getOutput() {
         return this.output;
     }