Claude: Streamed tool calls parser

db04fff3df342e7c035c7dccccb88df5dd2fc75c

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

Parents: ca1f6a56c34fce6b9b5726e0f2d98915be00a436

public/scripts/tool-calling.js+46 -19 Viewed

      */
     static #tools = new Map();
 +    static #INPUT_DELTA_KEY = '__input_json_delta';
++
     /**
      * Returns an Array of all tools that have been registered.
      * @type {ToolDefinition[]}

+        }
         if (typeof parsed?.content_block === 'object') {
             const choiceIndex = 0;
 +            const toolCallIndex = parsed?.index ?? 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];
                 ObjectToolManager.assign#applyToolCallDelta(targetToolCall, parsed.content_block);
 +            }
 +        }
 +        if (typeof parsed?.delta === 'object') {
 +            const choiceIndex = 0;
 +            const toolCallIndex = parsed?.index ?? 0;
 +            const targetToolCall = toolCalls[choiceIndex]?.[toolCallIndex];
 +            if (targetToolCall){
 +                if (parsed?.delta?.type === 'input_json_delta') {
 +                    const jsonDelta = parsed?.delta?.partial_json;
 +                    if (!targetToolCall[this.#INPUT_DELTA_KEY]) {
 +                        targetToolCall[this.#INPUT_DELTA_KEY] = '';
 +                    }
 +                    targetToolCall[this.#INPUT_DELTA_KEY] += jsonDelta;
 +                }
 +            }
 +        }
 +        if (parsed?.type === 'content_block_stop') {
 +            const choiceIndex = 0;
 +            const toolCallIndex = parsed?.index ?? 0;
 +            const targetToolCall = toolCalls[choiceIndex]?.[toolCallIndex];
 +            if (targetToolCall) {
 +                const jsonDeltaString = targetToolCall[this.#INPUT_DELTA_KEY];
 +                if (jsonDeltaString) {
 +                    try {
 +                        const jsonDelta = { input: JSON.parse(jsonDeltaString) };
 +                        delete targetToolCall[this.#INPUT_DELTA_KEY];
 +                        ToolManager.#applyToolCallDelta(targetToolCall, jsonDelta);
 +                    } catch (error) {
 +                        console.warn('Failed to apply input JSON delta:', error);
 +                    }
 +                }
+            }
+        }
+    }

      * @returns {any[]} Tool calls from the response data
      */
     static #getToolCallsFromData(data) {
 +        const isClaudeToolCall = c => Array.isArray(c) ? c.filter(x => x).every(isClaudeToolCall) : c?.input && c?.name && c?.id;
 +        const convertClaudeToolCall = c => ({ id: c.id, function: { name: c.name, arguments: c.input } });
++
         // Parsed tool calls from streaming data
         if (Array.isArray(data) && data.length > 0 && Array.isArray(data[0])) {
 -            return data[0];
 +            return isClaudeToolCall(data[0]) ? data[0].filter(x => x).map(convertClaudeToolCall) : data[0];
+        }
         // Parsed tool calls from non-streaming data

+            }
+        }
 +        // Claude tool calls to OpenAI tool calls
         if (Array.isArray(data?.content)) {
 -            // Claude tool calls to OpenAI tool calls
 +            const content = data.content.filter(c => c.type === 'tool_use').map(convertClaudeToolCall);
 -            const content = data.content.filter(c => c.type === 'tool_use').map(c => {
++
 -                return {
 +            if (content) {
 -                    id: c.id,
 +                return content;
 -                    function: {
 +            }
 -                        name: c.name,
 -                        arguments: c.input,
 -                    },
 -                };
 -            });
+-
 -            return content;
+        }
+    }