Add Claude streamed tool parser

0cab91e0f8c0074d4bfdf33c3971b3dde1029412

Cohee <18619528+Cohee1207@users.noreply.github.com>

Parents: c3c10a629e000f3d7479f9a50fbb6d28316acaa0

public/scripts/tool-calling.js+39 -22 Viewed

      * @returns {void}
      */
     static parseToolCalls(toolCalls, parsed) {
-        if (!Array.isArray(parsed?.choices)) {
+        if (Array.isArray(parsed?.choices)) {
-            return;
+            for (const choice of parsed.choices) {
-        }
+                const choiceIndex = (typeof choice.index === 'number') ? choice.index : null;
-        for (const choice of parsed.choices) {
+                const choiceDelta = choice.delta;
-            const choiceIndex = (typeof choice.index === 'number') ? choice.index : null;
-            const choiceDelta = choice.delta;
-            if (choiceIndex === null || !choiceDelta) {
+                if (choiceIndex === null || !choiceDelta) {
-                continue;
+                    continue;
-            }
+                }
-            const toolCallDeltas = choiceDelta?.tool_calls;
+                const toolCallDeltas = choiceDelta?.tool_calls;
-            if (!Array.isArray(toolCallDeltas)) {
+                if (!Array.isArray(toolCallDeltas)) {
-                continue;
+                    continue;
-            }
+                }
-            if (!Array.isArray(toolCalls[choiceIndex])) {
+                if (!Array.isArray(toolCalls[choiceIndex])) {
-                toolCalls[choiceIndex] = [];
+                    toolCalls[choiceIndex] = [];
-            }
+                }
-            for (const toolCallDelta of toolCallDeltas) {
+                for (const toolCallDelta of toolCallDeltas) {
-                const toolCallIndex = (typeof toolCallDelta?.index === 'number') ? toolCallDelta.index : toolCallDeltas.indexOf(toolCallDelta);
+                    const toolCallIndex = (typeof toolCallDelta?.index === 'number') ? toolCallDelta.index : toolCallDeltas.indexOf(toolCallDelta);
-                if (isNaN(toolCallIndex) || toolCallIndex < 0) {
+                    if (isNaN(toolCallIndex) || toolCallIndex < 0) {
-                    continue;
+                        continue;
+                    }
+                    if (toolCalls[choiceIndex][toolCallIndex] === undefined) {
+                        toolCalls[choiceIndex][toolCallIndex] = {};
+                    }
+                    const targetToolCall = toolCalls[choiceIndex][toolCallIndex];
+                    ToolManager.#applyToolCallDelta(targetToolCall, toolCallDelta);
+                }
+            }
+        }
+        if (typeof parsed?.content_block === 'object') {
+            const choiceIndex = 0;
+            if (parsed?.content_block?.type === 'tool_use') {
+                if (!Array.isArray(toolCalls[choiceIndex])) {
+                    toolCalls[choiceIndex] = [];
+                }
+                const toolCallIndex = toolCalls[choiceIndex].length;
                 if (toolCalls[choiceIndex][toolCallIndex] === undefined) {
                     toolCalls[choiceIndex][toolCallIndex] = {};
+                }
                 const targetToolCall = toolCalls[choiceIndex][toolCallIndex];
+                Object.assign(targetToolCall, parsed.content_block);
-                ToolManager.#applyToolCallDelta(targetToolCall, toolCallDelta);
+            }
+        }
+    }