More events

metaskills · May 1, 2024 · 81837e9 · 81837e9
1 parent 6325103
commit 81837e9
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 1 deletion.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "experts",
-  "version": "0.3.0",
+  "version": "0.3.1",
   "description": "An opinionated panel of experts implementation using OpenAI's Assistants API",
   "type": "module",
   "main": "./src/index.js",

diff --git a/src/experts/assistant.js b/src/experts/assistant.js
@@ -85,6 +85,14 @@ class Assistant {
     this.eventEmitter.emit("toolCallDelta", delta, snapshot);
   }
 
+  onRunStepDone(runStep) {
+    this.eventEmitter.emit("runStepDone", runStep);
+  }
+
+  onToolCallDone(toolCall) {
+    this.eventEmitter.emit("toolCallDone", toolCall);
+  }
+
   // Tool Assistant
 
   addAssistantTool(toolClass) {

diff --git a/src/experts/run.js b/src/experts/run.js
@@ -22,6 +22,8 @@ class Run {
     stream.on("event", (e) => this.onEvent(e));
     stream.on("textDelta", (td, s) => this.onTextDelta(td, s));
     stream.on("toolCallDelta", (tcd, s) => this.onToolCallDelta(tcd, s));
+    stream.on("runStepDone", (rs) => this.onRunStepDone(rs));
+    stream.on("toolCallDone", (tc) => this.onToolCallDone(tc));
     this.toolOutputs = [];
     this.isToolOuputs = false;
   }
@@ -84,6 +86,14 @@ class Run {
     this.assistant.onToolCallDelta(delta, snapshot);
   }
 
+  onRunStepDone(runStep) {
+    this.assistant.onRunStepDone(runStep);
+  }
+
+  onToolCallDone(toolCall) {
+    this.assistant.onToolCallDone(toolCall);
+  }
+
   // Private (Tools)
 
   async callTools() {