Merge pull request #36 from metaskills/add-file-search-debug

Enhanced DEBUG Output for Tool Call Events. Ex: file search
metaskills · Sep 9, 2024 · 50ccd73 · 50ccd73
2 parents 1597cc3 + 1063cbb
commit 50ccd73
Show file tree

Hide file tree

Showing 4 changed files with 41 additions and 9 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -2,6 +2,12 @@
 
 See this http://keepachangelog.com link for information on how we want this document formatted.
 
+## v1.6.1
+
+### Added
+
+- Enhanced debug output for tool call events, including file search.
+
 ## v1.6.0
 
 ### Changed

diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "experts",
-  "version": "1.6.0",
+  "version": "1.6.1",
   "description": "An opinionated panel of experts implementation using OpenAI's Assistants API",
   "type": "module",
   "scripts": {

diff --git a/src/helpers.js b/src/helpers.js
@@ -11,8 +11,7 @@ const debug = (message) => {
 
 const debugEvent = (event) => {
   if (!DEBUG) return;
-  if (!DEBUG_EVENTS) return;
-  if (event.event.includes("delta") && !DEBUG_DELTAS) return;
+
   const eventCopy = JSON.parse(JSON.stringify(event));
   if (eventCopy.data && eventCopy.data.instructions) {
     eventCopy.data.instructions = "[INSTRUCTIONS REMOVED]";
@@ -29,10 +28,37 @@ const debugEvent = (event) => {
       return tool;
     });
   }
-  const jsonOutput = DEBUG_PRETTY_JSON
-    ? JSON.stringify(eventCopy, null, 2)
-    : JSON.stringify(eventCopy);
-  debug(`📡 Event: ${jsonOutput}`);
+
+  const isToolCallEvent = eventCopy.data?.step_details?.type === "tool_calls";
+
+  if (isToolCallEvent) {
+    const toolCalls = eventCopy.data.step_details.tool_calls;
+    toolCalls.forEach((toolCall) => {
+      const emoji = getToolCallEmoji(toolCall.type);
+      const toolCallData = toolCall[toolCall.type];
+      const jsonOutput = DEBUG_PRETTY_JSON
+        ? JSON.stringify(toolCallData, null, 2)
+        : JSON.stringify(toolCallData);
+      debug(`${emoji} Tool Call (${toolCall.type}): ${jsonOutput}`);
+    });
+  } else if (DEBUG_EVENTS) {
+    if (event.event.includes("delta") && !DEBUG_DELTAS) return;
+    const jsonOutput = DEBUG_PRETTY_JSON
+      ? JSON.stringify(eventCopy, null, 2)
+      : JSON.stringify(eventCopy);
+    debug(`📡 Event: ${jsonOutput}`);
+  }
+};
+
+const getToolCallEmoji = (toolType) => {
+  switch (toolType) {
+    case "file_search":
+      return "🧰 🔍";
+    case "code_interpreter":
+      return "🧰 💻";
+    default:
+      return "🧰";
+  }
 };
 
 const messagesContent = (messages) => {