opencode-codebuddy-external-auth 1.0.9 → 1.0.11

package/dist/plugin.js

@@ -8,9 +8,13 @@ const PROVIDER_ID = "codebuddy-external";
 const CONFIG = {
     // IOA 版本使用 copilot.tencent.com 进行认证
     serverUrl: "https://copilot.tencent.com",
+    // 本地 codebuddy --serve 代理地址
+    localServeUrl: "http://127.0.0.1:3000",
     // 平台标识
     platform: "CLI",
     appVersion: "2.37.20",
+    // 使用 HTTP API 模式还是 CLI 模式
+    useHttpApi: true,
 };
 // ============================================================================
 // Utility Functions
@@ -147,7 +151,8 @@ async function executeCodeBuddyCLI(prompt, model, systemPrompt) {
             args.push("--system-prompt", systemPrompt);
         }
         args.push(prompt);
-        console.log(`[codebuddy-external] Executing: codebuddy ${args.join(" ").substring(0, 100)}...`);
+        // Debug log (可注释掉)
+        // console.log(`[codebuddy-external] Executing: codebuddy ${args.join(" ").substring(0, 100)}...`);
         const child = spawn("codebuddy", args, {
             env: { ...process.env },
             stdio: ["pipe", "pipe", "pipe"],
@@ -199,29 +204,17 @@ function createProxyFetch() {
                 const openaiRequest = JSON.parse(typeof body === "string" ? body : await new Response(body).text());
                 // Convert to CodeBuddy format
                 const { prompt, systemPrompt } = convertMessagesToPrompt(openaiRequest.messages);
-                console.log(`[codebuddy-external] Invoking CodeBuddy CLI`);
-                console.log(`[codebuddy-external] Model: ${openaiRequest.model}, Prompt length: ${prompt.length}`);
-                // Execute codebuddy CLI
-                const resultText = await executeCodeBuddyCLI(prompt, openaiRequest.model, systemPrompt);
-                // Convert response to OpenAI format
-                if (openaiRequest.stream) {
-                    return new Response(createOpenAIStreamResponse(resultText, openaiRequest.model), {
-                        headers: {
-                            "Content-Type": "text/event-stream",
-                            "Cache-Control": "no-cache",
-                            "Connection": "keep-alive",
-                        },
-                    });
+                // 根据配置选择 HTTP API 模式或 CLI 模式
+                if (CONFIG.useHttpApi) {
+                    return await executeViaHttpApi(openaiRequest, prompt, systemPrompt);
                 }
                 else {
-                    return new Response(createOpenAIResponse(resultText, openaiRequest.model), {
-                        headers: { "Content-Type": "application/json" },
-                    });
+                    return await executeViaCli(openaiRequest, prompt, systemPrompt);
                 }
             }
             catch (error) {
-                console.error(`[codebuddy-external] CLI error:`, error);
-                return new Response(JSON.stringify({ error: `CodeBuddy CLI error: ${error}` }), {
+                console.error(`[codebuddy-external] Error:`, error);
+                return new Response(JSON.stringify({ error: `CodeBuddy error: ${error}` }), {
                     status: 500,
                     headers: { "Content-Type": "application/json" },
                 });
@@ -231,6 +224,172 @@ function createProxyFetch() {
         return fetch(url, init);
     };
 }
+/**
+ * Execute via HTTP API (codebuddy --serve)
+ * 支持流式响应
+ */
+async function executeViaHttpApi(openaiRequest, prompt, systemPrompt) {
+    // 构建 CodeBuddy /agent 请求
+    const agentRequest = {
+        prompt,
+        model: openaiRequest.model,
+        outputFormat: openaiRequest.stream ? "stream-json" : "json",
+    };
+    if (systemPrompt) {
+        agentRequest.systemPrompt = systemPrompt;
+    }
+    // 调用本地 codebuddy --serve
+    const response = await fetch(`${CONFIG.localServeUrl}/agent`, {
+        method: "POST",
+        headers: {
+            "Content-Type": "application/json",
+        },
+        body: JSON.stringify(agentRequest),
+    });
+    if (!response.ok) {
+        const errorText = await response.text();
+        throw new Error(`HTTP API error: ${response.status} - ${errorText}`);
+    }
+    // 处理流式响应
+    if (openaiRequest.stream && response.body) {
+        return new Response(transformCodeBuddyStreamToOpenAI(response.body, openaiRequest.model), {
+            headers: {
+                "Content-Type": "text/event-stream",
+                "Cache-Control": "no-cache",
+                "Connection": "keep-alive",
+            },
+        });
+    }
+    // 非流式响应
+    const responseText = await response.text();
+    let resultText = responseText;
+    // 尝试解析 JSON 响应
+    try {
+        const jsonResponse = JSON.parse(responseText);
+        if (jsonResponse.output) {
+            resultText = jsonResponse.output;
+        }
+    }
+    catch {
+        // 不是 JSON，直接使用原文
+    }
+    return new Response(createOpenAIResponse(resultText, openaiRequest.model), {
+        headers: { "Content-Type": "application/json" },
+    });
+}
+/**
+ * Transform CodeBuddy SSE stream to OpenAI SSE stream
+ *
+ * CodeBuddy 格式:
+ *   event: next
+ *   data: {"type": "assistant", "message": {"content": [{"type": "text", "text": "..."}]}}
+ *
+ * OpenAI 格式:
+ *   data: {"choices": [{"delta": {"content": "..."}}]}
+ */
+function transformCodeBuddyStreamToOpenAI(sourceStream, model) {
+    const encoder = new TextEncoder();
+    const decoder = new TextDecoder();
+    const id = `chatcmpl-${Date.now()}`;
+    const created = Math.floor(Date.now() / 1000);
+    let buffer = "";
+    let sentRole = false;
+    return new ReadableStream({
+        async start(controller) {
+            const reader = sourceStream.getReader();
+            // 先发送 role chunk
+            const roleChunk = {
+                id,
+                object: "chat.completion.chunk",
+                created,
+                model,
+                choices: [{ index: 0, delta: { role: "assistant" }, finish_reason: null }],
+            };
+            controller.enqueue(encoder.encode(`data: ${JSON.stringify(roleChunk)}\n\n`));
+            sentRole = true;
+            try {
+                while (true) {
+                    const { done, value } = await reader.read();
+                    if (done)
+                        break;
+                    buffer += decoder.decode(value, { stream: true });
+                    // 处理 SSE 事件
+                    const lines = buffer.split("\n");
+                    buffer = lines.pop() || ""; // 保留不完整的行
+                    for (const line of lines) {
+                        if (line.startsWith("data: ")) {
+                            const dataStr = line.slice(6).trim();
+                            if (!dataStr || dataStr === "{}")
+                                continue;
+                            try {
+                                const data = JSON.parse(dataStr);
+                                // 提取文本内容
+                                let text = "";
+                                if (data.type === "assistant" && data.message?.content) {
+                                    for (const part of data.message.content) {
+                                        if (part.type === "text" && part.text) {
+                                            text += part.text;
+                                        }
+                                    }
+                                }
+                                if (text) {
+                                    const chunk = {
+                                        id,
+                                        object: "chat.completion.chunk",
+                                        created,
+                                        model,
+                                        choices: [{ index: 0, delta: { content: text }, finish_reason: null }],
+                                    };
+                                    controller.enqueue(encoder.encode(`data: ${JSON.stringify(chunk)}\n\n`));
+                                }
+                            }
+                            catch {
+                                // 忽略解析错误
+                            }
+                        }
+                        else if (line.startsWith("event: done")) {
+                            // 发送结束 chunk
+                            const doneChunk = {
+                                id,
+                                object: "chat.completion.chunk",
+                                created,
+                                model,
+                                choices: [{ index: 0, delta: {}, finish_reason: "stop" }],
+                            };
+                            controller.enqueue(encoder.encode(`data: ${JSON.stringify(doneChunk)}\n\n`));
+                            controller.enqueue(encoder.encode("data: [DONE]\n\n"));
+                        }
+                    }
+                }
+            }
+            finally {
+                reader.releaseLock();
+                controller.close();
+            }
+        },
+    });
+}
+/**
+ * Execute via CLI (codebuddy -p)
+ * 不支持流式，需等待完整响应
+ */
+async function executeViaCli(openaiRequest, prompt, systemPrompt) {
+    const resultText = await executeCodeBuddyCLI(prompt, openaiRequest.model, systemPrompt);
+    if (openaiRequest.stream) {
+        return new Response(createOpenAIStreamResponse(resultText, openaiRequest.model), {
+            headers: {
+                "Content-Type": "text/event-stream",
+                "Cache-Control": "no-cache",
+                "Connection": "keep-alive",
+            },
+        });
+    }
+    else {
+        return new Response(createOpenAIResponse(resultText, openaiRequest.model), {
+            headers: { "Content-Type": "application/json" },
+        });
+    }
+}
 // ============================================================================
 // OAuth Flow Implementation (IOA Login)
 // ============================================================================

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-codebuddy-external-auth",
-  "version": "1.0.9",
+  "version": "1.0.11",
   "description": "OpenCode plugin for CodeBuddy External (IOA) authentication",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",