@mayijuntuan/cnb-chat 0.0.0-stage → 1.1.1

package/README.md

@@ -1,3 +1,62 @@
-# Temporary Holding Version
-
-This version is a temporary placeholder for this package. An operational version to replace this has been submitted for review and is awaiting a staged release.
+# cnb-chat
+
+CNB 流水线工具集（npx 包）。
+
+## 命令
+
+### `chat` — AI 对话流式代理（纯对话）
+
+在 CNB 流水线容器内运行（AI 接口仅允许流水线 `CNB_TOKEN` 调用，**本地不可用**）。
+读取环境变量（由平台 `build/start` 的 `env` 注入），
+调用 CNB AI 对话接口（`/{repo}/-/ai/chat/completions`，stream），
+将模型流式输出逐 delta POST 回平台 relay 接口，结束时发送 `done`。
+
+| 环境变量 | 说明 |
+|---|---|
+| `CHAT_CALLBACK` | 平台 relay 回调地址 |
+| `CHAT_SECRET` | 回调鉴权密钥（`X-Chat-Secret` 头） |
+| `CHAT_ID` | 会话 ID（随每个回调携带） |
+| `CHAT_MODEL` | 模型名，可空 |
+| `CHAT_MESSAGES` | JSON 数组 `[{role, content}, ...]` |
+| `CNB_REPO_SLUG` / `CNB_TOKEN` | CNB 流水线自动注入 |
+
+### `theme` — CodeBuddy 代理开发（可编辑代码）
+
+调用 `@tencent-ai/agent-sdk` 驱动 CodeBuddy 在仓库工作目录中编辑代码，
+文本块逐段回调 `delta`、工具调用回调 `progress`，完成后自动 `git commit + push`。
+
+| 环境变量 | 说明 |
+|---|---|
+| `CHAT_MESSAGES` | JSON 数组（取用户/助手轮次组装上下文，以最后一条用户消息为准） |
+| `CHAT_CB_API_KEY` | CodeBuddy API Key（copilot.tencent.com 创建） |
+| `CHAT_MODEL` | CodeBuddy 模型名，可空 |
+| `CHAT_CALLBACK` / `CHAT_SECRET` / `CHAT_ID` | 平台 relay 回调（缺失时进入本地模式） |
+| `CHAT_BRANCH` | push 目标分支，默认 main |
+| `CHAT_NOTIFY_URL` | push 后的主题更新通知地址，可空 |
+| `CNB_REPO_SLUG` / `CNB_TOKEN` | CNB 流水线自动注入（git push 凭据） |
+
+## 流水线接入
+
+平台触发构建时通过 `build/start` 的 `config` 参数内联传入流水线配置
+（`CnbPipeline::chatConfig($mode)` 生成，仓库无需 `.cnb.yml`）：
+
+```yaml
+$:
+  api_trigger_chat:
+    - name: AI 对话
+      docker:
+        image: node:20
+      stages:
+        - name: chat-relay
+          script: |
+            npm i -g @tencent-ai/codebuddy-code --no-audit --no-fund
+            npx -y @mayijuntuan/cnb-chat@latest theme
+```
+
+## 发布
+
+```bash
+npm publish
+```
+
+发布后流水线中的 `npx -y @mayijuntuan/cnb-chat@latest` 即可拉取最新版本。

package/bin/cli.mjs

@@ -0,0 +1,19 @@
+#!/usr/bin/env node
+// cnb-chat CLI：CNB 流水线工具集
+// 用法: npx @mayijuntuan/cnb-chat <command>
+//   chat   AI 对话流式代理（纯对话，不读写文件，调用 CNB AI 接口并逐 delta 回调）
+//   theme  CodeBuddy 代理开发（调用 @tencent-ai/agent-sdk 编辑主题代码并 push）
+
+const [cmd] = process.argv.slice(2);
+
+switch (cmd) {
+  case 'chat':
+    await import('../lib/chat-relay.mjs');
+    break;
+  case 'theme':
+    await import('../lib/theme-agent.mjs');
+    break;
+  default:
+    console.error('Usage: cnb-chat <chat|theme>');
+    process.exit(1);
+}

package/lib/chat-relay.mjs

@@ -0,0 +1,83 @@
+// AI 对话流式代理
+// 在 CNB 流水线容器内运行（依赖容器注入的 CNB_TOKEN）
+// 环境变量:
+//   CHAT_CALLBACK  回调地址（平台 relay 接口）
+//   CHAT_SECRET    回调鉴权密钥（放入 X-Chat-Secret 头）
+//   CHAT_ID        会话 ID（随每个回调携带）
+//   CHAT_MODEL     模型名（可空，使用默认模型）
+//   CHAT_MESSAGES  JSON 数组 [{role, content}, ...]
+//   CNB_REPO_SLUG  CNB 自动注入：仓库路径
+//   CNB_TOKEN      CNB 自动注入：流水线临时令牌
+
+const CALLBACK = process.env.CHAT_CALLBACK || '';
+const CHAT_ID = process.env.CHAT_ID || '';
+const SECRET = process.env.CHAT_SECRET || '';
+const MODEL = process.env.CHAT_MODEL || '';
+const MESSAGES = process.env.CHAT_MESSAGES || '[]';
+const API = 'https://api.cnb.cool/' + process.env.CNB_REPO_SLUG + '/-/ai/chat/completions';
+
+const post = (payload) => fetch(CALLBACK, {
+  method: 'POST',
+  headers: { 'Content-Type': 'application/json', 'Chat-Id': CHAT_ID,'X-Chat-Secret': SECRET },
+  body: JSON.stringify(payload)
+}).catch(() => {});
+
+(async () => {
+  let messages = [];
+  try { messages = JSON.parse(MESSAGES); } catch (e) { messages = []; }
+  if (!CALLBACK || !messages.length) {
+    await post({ type: 'error', content: '流水线参数缺失' });
+    process.exit(1);
+  }
+  const body = { stream: true, messages };
+  if (MODEL) body.model = MODEL;
+
+  try {
+    const res = await fetch(API, {
+      method: 'POST',
+      headers: {
+        'Authorization': 'Bearer ' + process.env.CNB_TOKEN,
+        'Content-Type': 'application/json',
+        'Accept': 'text/event-stream'
+      },
+      body: JSON.stringify(body)
+    });
+    if (!res.ok) {
+      const t = await res.text();
+      await post({ type: 'error', content: 'AI 接口错误 ' + res.status + ': ' + t.slice(0, 500) });
+      process.exit(1);
+    }
+    const reader = res.body.getReader();
+    const decoder = new TextDecoder();
+    let buf = '';
+    let full = '';
+    while (true) {
+      const { done, value } = await reader.read();
+      if (done) break;
+      buf += decoder.decode(value, { stream: true });
+      let idx;
+      while ((idx = buf.indexOf('\n')) >= 0) {
+        const line = buf.slice(0, idx).trim();
+        buf = buf.slice(idx + 1);
+        if (!line.startsWith('data:')) continue;
+        const data = line.slice(5).trim();
+        if (data === '[DONE]') {
+          await post({ type: 'done', content: full });
+          process.exit(0);
+        }
+        try {
+          const j = JSON.parse(data);
+          const delta = (j.choices && j.choices[0] && j.choices[0].delta && j.choices[0].delta.content) || '';
+          if (delta) {
+            full += delta;
+            await post({ type: 'delta', content: delta });
+          }
+        } catch (e) {}
+      }
+    }
+    await post({ type: 'done', content: full });
+  } catch (e) {
+    await post({ type: 'error', content: String((e && e.message) || e) });
+    process.exit(1);
+  }
+})();

package/lib/theme-agent.mjs

@@ -0,0 +1,124 @@
+// CodeBuddy 代理开发：编辑主题代码 + 流式回传 + git push
+// 在 CNB 流水线容器内运行。工作目录为 CNB 检出的仓库代码。
+// 详见 README.md 的环境变量表
+
+import { execFileSync } from 'node:child_process';
+
+const CALLBACK = process.env.CHAT_CALLBACK || '';
+const CHAT_ID = process.env.CHAT_ID || '';
+const SECRET = process.env.CHAT_SECRET || '';
+const MODEL = process.env.CHAT_MODEL || '';
+const MESSAGES = process.env.CHAT_MESSAGES || '[]';
+const CB_API_KEY = process.env.CHAT_CB_API_KEY || '';
+const BRANCH = process.env.CHAT_BRANCH || 'main';
+
+const post = (payload) => fetch(CALLBACK, {
+  method: 'POST',
+  headers: { 'Content-Type': 'application/json', 'X-Chat-Secret': SECRET },
+  body: JSON.stringify({ chat_id: CHAT_ID, ...payload })
+}).catch(() => {});
+
+const sh = (cmd, args) => execFileSync(cmd, args, { encoding: 'utf8', stdio: ['ignore', 'pipe', 'pipe'] });
+const text = (e) => String((e && e.message) || e).slice(0, 300);
+
+(async () => {
+  let messages = [];
+  try { messages = JSON.parse(MESSAGES); } catch (e) { messages = []; }
+  const turns = messages.filter((m) => m.role === 'user' || m.role === 'assistant');
+  const lastUser = [...turns].reverse().find((m) => m.role === 'user');
+  if (!CALLBACK || !lastUser) {
+    await post({ type: 'error', content: '流水线参数缺失' });
+    process.exit(1);
+  }
+
+  // 多轮上下文 + 最新需求
+  const prompt = turns.map((m) => (m.role === 'user' ? '【用户】' : '【AI】') + m.content).join('\n\n')
+    + '\n\n请以上面的对话为上下文（以最后的用户消息为准），直接在当前仓库中修改代码并保存。';
+
+  // CodeBuddy 认证（SDK 要求；中国版需 internal 环境标识）
+  if (CB_API_KEY) {
+    process.env.CODEBUDDY_API_KEY = CB_API_KEY;
+    process.env.CODEBUDDY_INTERNET_ENVIRONMENT = process.env.CODEBUDDY_INTERNET_ENVIRONMENT || 'internal';
+  }
+
+  let query;
+  try {
+    ({ query } = await import('@tencent-ai/agent-sdk'));
+  } catch (e) {
+    await post({ type: 'error', content: 'agent-sdk 加载失败: ' + text(e) });
+    process.exit(1);
+  }
+
+  const full = [];
+
+  // 优先使用环境预装的 codebuddy CLI（如 cnbcool/default-dev-env，已由 CNB 注入认证），
+  // 未预装时 SDK 自动回退到包内捆绑的 CLI
+  let cliPath = '';
+  try { cliPath = sh('which', ['codebuddy']).trim(); } catch (e) { /* 未预装 */ }
+
+  try {
+    const q = query({
+      prompt,
+      options: {
+        cwd: process.cwd(),
+        permissionMode: 'acceptEdits',
+        allowedTools: ['Read', 'Write', 'Edit', 'Glob', 'Grep', 'Bash'],
+        ...(MODEL ? { model: MODEL } : {}),
+        ...(cliPath ? { pathToCodebuddyCode: cliPath } : {}),
+        maxTurns: 30,
+      },
+    });
+
+    for await (const message of q) {
+      if (message.type === 'assistant' && Array.isArray(message.message?.content)) {
+        for (const block of message.message.content) {
+          if (block.type === 'text' && block.text) {
+            full.push(block.text);
+            await post({ type: 'delta', content: block.text });
+          } else if (block.type === 'tool_use') {
+            const name = block.name || 'tool';
+            const file = block.input?.file_path || block.input?.path || '';
+            await post({ type: 'progress', content: file ? `🔧 ${name} ${file}` : `🔧 ${name}` });
+          }
+        }
+      } else if (message.type === 'result' && message.subtype !== 'success') {
+        throw new Error('CodeBuddy 执行失败: ' + (message.subtype || 'unknown'));
+      }
+    }
+
+    // 提交并推送到仓库
+    try {
+      sh('git', ['config', 'user.email', 'codebuddy@cnb.build']);
+      sh('git', ['config', 'user.name', 'CodeBuddy']);
+      sh('git', ['add', '-A']);
+      try { sh('git', ['commit', '-m', 'AI: ' + String(lastUser.content).slice(0, 80)]); } catch (e) { /* 无变更 */ }
+      try {
+        // 优先用容器内已配置凭据的 origin（Prepare 阶段克隆时注入）
+        sh('git', ['push', 'origin', 'HEAD:' + BRANCH]);
+      } catch (e) {
+        // 兜底：显式 token URL
+        sh('git', ['push', 'https://' + process.env.CNB_TOKEN + '@cnb.cool/' + process.env.CNB_REPO_SLUG + '.git', 'HEAD:' + BRANCH]);
+      }
+      await post({ type: 'progress', content: '✅ 代码已推送到 ' + BRANCH + ' 分支' });
+    } catch (e) {
+      await post({ type: 'progress', content: '⚠️ git push 失败: ' + text(e) });
+    }
+
+    // 通知平台刷新主题（沿用现有回调）
+    const notify = process.env.CHAT_NOTIFY_URL || '';
+    if (notify) {
+      await fetch(notify, { method: 'POST' }).catch(() => {});
+    }
+
+    await post({ type: 'done', content: full.join('\n') });
+    process.exit(0);
+  } catch (e) {
+    const msg = text(e);
+    // SDK preview 版在 CLI 立即退出（认证失败/CLI 缺失）时仅抛 Transport closed
+    const hint = /Transport closed/i.test(msg)
+      ? 'CodeBuddy CLI 启动失败，请检查 CODEBUDDY_API_KEY 是否有效、CLI 是否已安装'
+      : msg;
+    await post({ type: 'error', content: hint });
+    process.exit(1);
+  }
+})();

package/package.json

@@ -1,6 +1,29 @@
-{
-  "name": "@mayijuntuan/cnb-chat",
-  "version": "0.0.0-stage",
-  "stub": true,
-  "description": "Temporary package placeholder for staged publishing"
-}
+{
+  "name": "@mayijuntuan/cnb-chat",
+  "version": "1.1.1",
+  "description": "CNB pipeline for AI chat streaming relay and CodeBuddy code editing",
+  "type": "module",
+  "bin": {
+    "cnb-chat": "bin/cli.mjs"
+  },
+  "dependencies": {
+    "@tencent-ai/agent-sdk": "^0.1.0"
+  },
+  "engines": {
+    "node": ">=18.20"
+  },
+  "files": [
+    "bin",
+    "lib"
+  ],
+  "publishConfig": {
+    "access": "public",
+    "registry": "https://registry.npmjs.org/"
+  },
+  "keywords": [
+    "cnb",
+    "chat",
+    "streaming",
+    "codebuddy"
+  ]
+}