Function runMultiTurn

test/e2e-prompt-ab.mjs:179–235

(userMessage, { tools = TOOLS, systemPrompt = '', toolChoice, maxTurns = MAX_TURNS } = {})

// ─── 多轮 Agentic 循环（用于完整任务分析） ─────────────────────────────
async function runMultiTurn(userMessage, { tools = TOOLS, systemPrompt = '', toolChoice, maxTurns = MAX_TURNS } = {}) {
    const messages = [{ role: 'user', content: userMessage }];
    const system = systemPrompt || 'You are an AI coding assistant. Working directory: /project.';

    let totalToolCalls = 0;
    let totalTextChars = 0;
    let turns = 0;
    let firstTurnHasToolCall = false;
    const toolCallLog = [];

    while (turns < maxTurns) {
        turns++;
        const resp = await fetch(`${BASE_URL}/v1/messages`, {
            method: 'POST',
            headers: { 'Content-Type': 'application/json', 'x-api-key': 'dummy' },
            body: JSON.stringify({
                model: MODEL,
                max_tokens: 4096,
                system,
                tools,
                ...(toolChoice ? { tool_choice: toolChoice } : {}),
                messages,
            }),
        });

        if (!resp.ok) throw new Error(`HTTP ${resp.status}`);
        const data = await resp.json();

        const textBlocks = data.content?.filter(b => b.type === 'text') || [];
        const toolUseBlocks = data.content?.filter(b => b.type === 'tool_use') || [];

        totalTextChars += textBlocks.reduce((s, b) => s + (b.text?.length || 0), 0);
        totalToolCalls += toolUseBlocks.length;

        if (turns === 1 && toolUseBlocks.length > 0) firstTurnHasToolCall = true;

        for (const tb of toolUseBlocks) {
            toolCallLog.push({ turn: turns, tool: tb.name, input: tb.input });
        }

        if (data.stop_reason === 'end_turn' || toolUseBlocks.length === 0) break;

        messages.push({ role: 'assistant', content: data.content });

        const toolResults = toolUseBlocks.map(tb => ({
            type: 'tool_result',
            tool_use_id: tb.id,
            content: mockExecute(tb.name, tb.input),
        }));
        messages.push({ role: 'user', content: toolResults });

        // Check for completion signal
        if (toolResults.some(r => r.content.startsWith('__DONE__'))) break;
    }

    return { totalToolCalls, totalTextChars, turns, firstTurnHasToolCall, toolCallLog };
}