@kevisual/cnb 0.0.42 → 0.0.43

package/agent/app.ts

@@ -3,6 +3,7 @@ import { useContextKey } from '@kevisual/context'
 import { useKey } from '@kevisual/use-config'
 import { CNB } from '../src/index.ts';
 import { CNBManager } from './modules/cnb-manager.ts'
+
 export const cnbManager = new CNBManager()
 
 // CNB_TOKEN是降级兼容变量，推荐使用CNB_API_KEY
@@ -18,9 +19,9 @@ try {
     cnb: new CNB({ token: token, cookie: cookie })
   })
 } catch (error) {
-
+  process.exit(1)
 }
-export const cnb = (await cnbManager.getCNB({ username: 'default' })).cnb
+await new Promise(resolve => setTimeout(resolve, 1000))
 export const app = await useContextKey<App>('app', () => {
   return new App({})
 })
@@ -28,7 +29,7 @@ export const app = await useContextKey<App>('app', () => {
 export const notCNBCheck = (ctx: any) => {
   const isCNB = useKey('CNB');
   if (!isCNB) {
-    ctx.throw(400, '当前环境非 cnb-board 环境，无法获取 live 内容');
+    ctx.throw(400, '当前环境非 cnb-board 环境，无法获取内容');
   }
   return false;
-}
+}

package/agent/modules/cnb-manager.ts

@@ -1,6 +1,7 @@
 import { Result } from '@kevisual/query';
 import { CNB } from '../../src/index.ts';
 import { useKey } from '@kevisual/context';
+import { createOpenAICompatible } from '@ai-sdk/openai-compatible';
 export const getConfig = async (opts: { token?: string }) => {
   const kevisualEnv = useKey('KEVISUAL_ENV')
   const baseUrl = kevisualEnv === 'production' ? 'https://kevisual.cn/api/router' : 'https://kevisual.xiongxiao.me/api/router';
@@ -32,7 +33,14 @@ type CNBItem = {
   runAt?: number
   owner?: boolean
   cnb: CNB
+  cnbAi: ReturnType<typeof createOpenAICompatible>
 }
+// const repo = useKey('CNB_REPO_SLUG_LOWERCASE') as string || 'kevision/kevision';
+// export const cnbAi = createOpenAICompatible({
+//   baseURL: `https://api.cnb.cool/${repo}/-/ai/`,
+//   name: 'custom-cnb',
+//   apiKey: token,
+// });
 export class CNBManager {
   cnbMap: Map<string, CNBItem> = new Map()
   constructor() {
@@ -71,20 +79,24 @@ export class CNBManager {
    * @returns CNB 实例
    */
   async getContext(ctx: any) {
+    const item = await this.getCNBItem(ctx)
+    return item.cnb
+  }
+  async getCNBItem(ctx: any) {
     const tokenUser = ctx?.state?.tokenUser
     const username = tokenUser?.username
     if (!username) {
       ctx.throw(403, 'Unauthorized')
     }
     if (username === 'default') {
-      return this.getDefaultCNB().cnb
+      return this.getDefaultCNB()
     }
     const kevisualToken = ctx.query?.token;
     const item = await this.getCNB({ username, kevisualToken });
     if (!item) {
       ctx.throw(400, '不存在的 CNB 配置项，请检查 登录 Token 是否正确，或添加 CNB 配置')
     }
-    return item.cnb
+    return item;
   }
   addCNB(opts: Partial<CNBItem>) {
     if (!opts.username || !opts.token) {
@@ -98,6 +110,12 @@ export class CNBManager {
     const cnb = opts?.cnb || new CNB({ token: opts.token, cookie: opts.cookie });
     opts.cnb = cnb;
     opts.runAt = Date.now()
+    const repoSlug = useKey('CNB_REPO_SLUG_LOWERCASE') as string || 'kevision/kevision';
+    opts.cnbAi = createOpenAICompatible({
+      baseURL: `https://api.cnb.cool/${repoSlug}/-/ai/`,
+      name: `custom-cnb-${opts.username}`,
+      apiKey: opts.token,
+    })
     this.cnbMap.set(opts.username, opts as CNBItem)
     return opts as CNBItem
   }

package/agent/npc.ts

@@ -0,0 +1,94 @@
+import { app } from './index.ts';
+
+import { useIssueEnv, useCommentEnv, useRepoInfoEnv, IssueLabel } from '../src/index.ts'
+import { pick } from 'es-toolkit';
+
+const writeToProcess = (message: string) => {
+  if (process.send) {
+    process.send(message);
+  } else {
+    console.log(message);
+  }
+}
+const getIssuesLabels = async () => {
+  const issueEnv = useIssueEnv();
+  const repoInfoEnv = useRepoInfoEnv();
+  const issueId = issueEnv.issueId;
+  const repoSlug = repoInfoEnv.repoSlug;
+  if (!issueId || !repoSlug) {
+    return [];
+  }
+  const res = await app.run({
+    path: 'cnb',
+    key: 'getIssue',
+    payload: {
+      repo: repoSlug,
+      issueNumber: issueId
+    }
+  });
+  if (res.code === 200) {
+    const issueData = res.data as any;
+    const labels = issueData.labels || [];
+    return labels as IssueLabel[];
+  }
+  console.error('获取 Issue 详情失败', res);
+  return []
+
+}
+
+const main = async () => {
+  const repoInfoEnv = useRepoInfoEnv();
+  const commentEnv = useCommentEnv();
+  const issueEnv = useIssueEnv();
+  const pickCommentEnv = pick(commentEnv, ['commentId', 'commentIdLabel']);
+  const pickIssueEnv = pick(issueEnv, ['issueId', 'issueIdLabel', 'issueIid', 'issueIidLabel', 'issueTitle', 'issueTitleLabel', 'issueDescription', 'issueDescriptionLabel']);
+  const pickRepoInfoEnv = pick(repoInfoEnv, ['repoId', 'repoIdLabel', 'repoName', 'repoNameLabel', 'repoSlug', 'repoSlugLabel']);
+  // const issueLabels = issueEnv.issueLabels || [];
+  const isComment = !!commentEnv.commentId;
+  const envList = [
+    ...Object.entries(pickRepoInfoEnv).map(([key, value]) => `${key}: ${value}`),
+    ...Object.entries(issueEnv).map(([key, value]) => `${key}: ${value}`),
+    ...Object.entries(pickCommentEnv).map(([key, value]) => `${key}: ${value}`),
+  ]
+  writeToProcess('当前环境变量:');
+  const issueLabels = await getIssuesLabels();
+  const issueLabelsNames = issueLabels.map(label => label.name) || [];
+  envList.forEach(item => writeToProcess(item));
+  if (!isComment && !issueLabelsNames.includes('Run')) {
+    writeToProcess('当前 Issue 不包含 Run 标签，跳过执行');
+    process.exit(0);
+  }
+  const messages = [
+    {
+      role: 'system',
+      content: `你是一个智能的代码助手, 根据用户提供的上下文信息，提供有用的建议和帮助, 如果用户的要求和执行工具不一致，请说出你不能这么做。并把最后的结果提交一个评论到对应的issue中,提交的内容必须不能包含 @ 提及。用户提供的上下文信息如下：`
+    },
+    {
+      role: 'system',
+      content: `相关变量：${JSON.stringify({ ...pickCommentEnv, ...pickIssueEnv, ...pickRepoInfoEnv })}`
+    }, {
+      role: 'user',
+      content: commentEnv.commentBody || pickIssueEnv.issueDescription || '无'
+    }
+  ]
+  writeToProcess('输入消息:');
+  writeToProcess(JSON.stringify(messages, null, 2));
+  const result = await app.run({
+    path: 'cnb',
+    key: 'chat',
+    payload: {
+      messages
+    }
+  }, { appId: app.appId })
+  if (result.code === 200) {
+    let _message = result.data.message || []
+    writeToProcess('执行完成')
+    writeToProcess(JSON.stringify(_message, null, 2))
+    process.exit(0)
+  } else {
+    writeToProcess(result.message || '执行错误')
+    process.exit(1)
+  }
+}
+
+main();

package/agent/routes/chat/chat.ts

@@ -0,0 +1,41 @@
+import { runAgent } from '@kevisual/ai/agent'
+import { app, cnbManager } from '../../app.ts';
+import z from 'zod';
+
+app.route({
+  path: 'cnb',
+  key: 'chat',
+  description: 'cnb智能对话接口',
+  middleware: ['auth'],
+  metadata: {
+    args: {
+      question: z.string().describe('用户输入的问题'),
+      messages: z.array(z.object({
+        role: z.enum(['user', 'assistant']).describe('消息角色，user表示用户输入，assistant表示助手回复'),
+        content: z.string().describe('消息内容')
+      })).describe('对话消息列表，按照时间顺序排列，包含用户和助手的历史消息'),
+      model: z.string().optional().describe('默认auto')
+    }
+  }
+}).define(async (ctx) => {
+  // notCNBCheck(ctx);
+  if (!ctx.args.question && !ctx.args.messages) {
+    ctx.throw(400, '缺少必要参数，必须提供question或messages');
+    return;
+  }
+  const model = ctx.args?.model || 'auto'
+  const item = await cnbManager.getCNBItem(ctx);
+  const cnbAi = item.cnbAi;
+  const messages = ctx.args.messages || [{
+    role: 'user',
+    content: ctx.args.question
+  }]
+  const result = await runAgent({
+    app,
+    messages: messages,
+    languageModel: cnbAi(model),
+    token: '',
+    // token: ctx.query.token as string,
+  });
+  ctx.body = result;
+}).addTo(app);

package/agent/routes/index.ts

@@ -10,6 +10,7 @@ import './cnb-board/index.ts';
 import './share/index.ts';
 import './cnb-manager/index.ts';
 import './build/index.ts';
+import './chat/chat.ts';
 
 /**
  * 验证上下文中的 App ID 是否与指定的 App ID 匹配

package/agent/routes/issues/comments.ts

@@ -59,6 +59,7 @@ app.route({
         repo: tool.schema.string().optional().describe('代码仓库名称, 如 my-user/my-repo'),
         issueNumber: tool.schema.number().describe('Issue 编号'),
         body: tool.schema.string().describe('评论内容'),
+        clearAt: tool.schema.boolean().optional().describe('是否清除评论内容中的 @ 提及，默认: true'),
       },
       summary: '创建 Issue 评论',
     })
@@ -67,7 +68,8 @@ app.route({
   const cnb = await cnbManager.getContext(ctx);
   let repo = ctx.query?.repo || useKey('CNB_REPO_SLUG_LOWERCASE');
   const issueNumber = ctx.query?.issueNumber;
-  const body = ctx.query?.body;
+  let body = ctx.query?.body;
+  const clearAt = ctx.query?.clearAt ?? true;
 
   if (!repo) {
     ctx.throw(400, '缺少参数 repo');
@@ -78,6 +80,10 @@ app.route({
   if (!body) {
     ctx.throw(400, '缺少参数 body');
   }
+  if (clearAt && body) {
+    // 清除评论内容中的 @ 提及
+    body = body.replace(/@/g, '');
+  }
 
   const res = await cnb.issue.createComment(repo, issueNumber, body);
   ctx.forward(res);
@@ -138,6 +144,7 @@ app.route({
         issueNumber: tool.schema.number().describe('Issue 编号'),
         commentId: tool.schema.number().describe('评论 ID'),
         body: tool.schema.string().describe('评论内容'),
+        clearAt: tool.schema.boolean().optional().describe('是否清除评论内容中的 @ 提及，默认: true'),
       },
       summary: '修改 Issue 评论',
     })
@@ -147,8 +154,8 @@ app.route({
   let repo = ctx.query?.repo || useKey('CNB_REPO_SLUG_LOWERCASE');
   const issueNumber = ctx.query?.issueNumber;
   const commentId = ctx.query?.commentId;
-  const body = ctx.query?.body;
-
+  let body = ctx.query?.body;
+  const clearAt = ctx.query?.clearAt ?? true;
   if (!repo) {
     ctx.throw(400, '缺少参数 repo');
   }
@@ -161,7 +168,10 @@ app.route({
   if (!body) {
     ctx.throw(400, '缺少参数 body');
   }
-
+  if (clearAt && body) {
+    // 清除评论内容中的 @ 提及
+    body = body.replace(/@/g, '');
+  }
   const res = await cnb.issue.updateComment(repo, issueNumber, commentId, body);
   ctx.forward(res);
 }).addTo(app);

package/agent/routes/issues/list.ts

@@ -49,4 +49,37 @@ app.route({
 
   const res = await cnb.issue.getList(repo, params);
   ctx.forward(res);
+}).addTo(app);
+
+app.route({
+  path: 'cnb',
+  key: 'getIssue',
+  description: '获取 单个 Issue',
+  middleware: ['auth'],
+  metadata: {
+    tags: ['opencode'],
+    ...createSkill({
+      skill: 'getIssue',
+      title: '获取 单个 Issue',
+      args: {
+        repo: tool.schema.string().optional().describe('代码仓库名称, 如 my-user/my-repo'),
+        issueNumber: tool.schema.union([tool.schema.string(), tool.schema.number()]).describe('Issue 编号'),
+      },
+      summary: '获取 单个 Issue',
+    })
+  }
+}).define(async (ctx) => {
+  const cnb = await cnbManager.getContext(ctx);
+  let repo = ctx.query?.repo || useKey('CNB_REPO_SLUG_LOWERCASE');
+  const issueNumber = ctx.query?.issueNumber;
+
+  if (!repo) {
+    ctx.throw(400, '缺少参数 repo');
+  }
+  if (!issueNumber) {
+    ctx.throw(400, '缺少参数 issueNumber');
+  }
+
+  const res = await cnb.issue.getItem(repo, issueNumber);
+  ctx.forward(res);
 }).addTo(app);

package/agent/routes/opencode/index.ts

@@ -0,0 +1,12 @@
+import { createOpencodeClient } from "@opencode-ai/sdk"
+
+const client = await createOpencodeClient({
+  // baseUrl: "https://yccb64t1z-100.cnb.run",
+  // auth: async () => {
+  //   return 'cm9vdDozR0I2MDg5ZGpYOE5oMDFjM1FteE5DWDd0ZkI='
+  // }
+  baseUrl: "http://localhost:4096",
+})
+
+const sessionList = await client.session.list()
+console.log(sessionList.data)