Function chat

test/e2e-ctf-proxy.mjs:30–94

(messages, { stream = true, label = '' } = {})

// ─── 请求辅助 ────────────────────────────────────────────────────
async function chat(messages, { stream = true, label = '' } = {}) {
    const body = {
        model: MODEL,
        max_tokens: 4096,
        messages: messages.map(m => ({
            role: m.role,
            content: m.text,
        })),
        stream,
    };

    const t0 = Date.now();
    process.stdout.write(`  ${C.dim}[${label || 'request'}]${C.reset} `);

    const resp = await fetch(`${BASE_URL}/v1/messages`, {
        method: 'POST',
        headers: { 'Content-Type': 'application/json', 'x-api-key': 'dummy' },
        body: JSON.stringify(body),
    });

    if (!resp.ok) {
        const text = await resp.text();
        throw new Error(`HTTP ${resp.status}: ${text.substring(0, 200)}`);
    }

    if (stream) {
        // 流式读取
        const reader = resp.body.getReader();
        const decoder = new TextDecoder();
        let buffer = '';
        let fullText = '';
        let chunkCount = 0;

        while (true) {
            const { done, value } = await reader.read();
            if (done) break;
            buffer += decoder.decode(value, { stream: true });
            const lines = buffer.split('\n');
            buffer = lines.pop() || '';

            for (const line of lines) {
                if (!line.startsWith('data: ')) continue;
                const data = line.slice(6).trim();
                if (!data) continue;
                try {
                    const evt = JSON.parse(data);
                    if (evt.type === 'content_block_delta' && evt.delta?.type === 'text_delta') {
                        fullText += evt.delta.text;
                        chunkCount++;
                        if (chunkCount % 20 === 0) process.stdout.write('.');
                    }
                } catch { /* ignore */ }
            }
        }
        const elapsed = ((Date.now() - t0) / 1000).toFixed(1);
        process.stdout.write(` ${C.dim}(${elapsed}s, ${fullText.length} chars)${C.reset}\n`);
        return { text: fullText, elapsed };
    } else {