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)) {
 -            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) {
                                     continue;
+                            }
                             const toolCallDeltas = choiceDelta?.tool_calls;
                             if (!Array.isArray(toolCallDeltas)) {
                                     continue;
+                            }
                             if (!Array.isArray(toolCalls[choiceIndex])) {
                                     toolCalls[choiceIndex] = [];
+                            }
                             for (const toolCallDelta of toolCallDeltas) {
                                     const toolCallIndex = (typeof toolCallDelta?.index === 'number') ? toolCallDelta.index : toolCallDeltas.indexOf(toolCallDelta);
                                     if (isNaN(toolCallIndex) || toolCallIndex < 0) {
                                             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);
+            }
+        }
+    }