@kevisual/cnb 0.0.76 → 0.0.78

package/agent/routes/knowledge/embedding.ts

@@ -0,0 +1,166 @@
+import { app, cnbManager } from '../../app.ts'
+import z from 'zod'
+import fs from 'node:fs';
+import crypto from 'node:crypto';
+
+// pnpm cli embedding getEmbedding -- repo=kevisual/starred-auto text="hello world"
+app.route({
+  path: 'embedding',
+  key: 'getEmbedding',
+  middleware: ['auth'],
+  description: '获取文本的向量表示',
+  metadata: {
+    args: {
+      repo: z.string().describe('知识库所属的仓库'),
+      text: z.string().describe('需要获取向量表示的文本')
+    }
+  }
+}).define(async (ctx) => {
+  const cnb = await cnbManager.getContext(ctx)
+  const { text, repo } = ctx.args
+
+  const result = await cnb.knowledgeBase.getEmbedding(repo, text)
+  console.log('embedding result', result)
+}).addTo(app)
+
+// pnpm cli embedding list-docs -- repo=kevisual/starred-auto
+// pnpm cli embedding list-docs -- repo=kevisual/cnb       
+app.route({
+  path: 'embedding',
+  key: 'list-docs',
+  middleware: ['auth'],
+  description: '获取知识库中的文档列表',
+  metadata: {
+    args: {
+      repo: z.string().describe('知识库所属的仓库'),
+    }
+  }
+}).define(async (ctx) => {
+  const cnb = await cnbManager.getContext(ctx)
+  const { repo } = ctx.args
+
+  const result = await cnb.knowledgeBase.listDocuments(repo)
+  console.log('list docs result', result)
+}).addTo(app)
+
+// pnpm cli embedding add-doc -- repo=kevisual/starred-auto content="hello world"
+// pnpm cli embedding add-doc -- repo=kevisual/cnb content="hello world"
+
+app.route({
+  path: 'embedding',
+  key: 'add-doc',
+  middleware: ['auth'],
+  description: '向知识库中添加文档，test',
+  metadata: {
+    args: {
+      repo: z.string().describe('知识库所属的仓库'),
+      content: z.string().describe('文档内容')
+    }
+  }
+}).define(async (ctx) => {
+  const cnb = await cnbManager.getContext(ctx)
+  let { repo, content } = ctx.args
+  const embeddingResult = await cnb.knowledgeBase.getEmbedding(repo, content)
+  const embedding = embeddingResult.data.embeddings;
+  const path = process.cwd() + '/docs/readme.md'
+  const read = fs.readFileSync(path, 'utf-8')
+  content = read;
+  const size = Buffer.byteLength(content, 'utf-8')
+  const gitBlobHashContent = (content: string | Buffer) => {
+    const bytes = Buffer.isBuffer(content) ? content : Buffer.from(content);
+    return crypto.createHash('md5').update(bytes).digest('hex');
+  }
+  const result = await cnb.knowledgeBase.addDocument(repo, {
+    path: 'docs/readme.md',
+    name: 'readme',
+    extension: 'md',
+    size,
+    hash: gitBlobHashContent(content),
+    type: 'code',
+    url: `https://cnb.cool/kevisual/cnb/-/blob/38f5e7ef1baf583/docs/readme.md`,
+    chunks: [
+      {
+        content,
+        hash: gitBlobHashContent(content),
+        position: 0,
+        embedding: embedding,
+      }
+    ],
+    batch_index: 0,
+    batch_count: 1,
+    is_last_batch: true,
+  })
+  console.log('add doc result', result)
+}).addTo(app)
+
+// pnpm cli cnb-base query -- repo=kevisual/starred-auto  query="openlist"
+app.route({
+  path: 'cnb-base',
+  key: 'query',
+  middleware: ['auth'],
+  description: '获取知识库基本信息',
+  metadata: {
+    args: {
+      repo: z.string().describe('知识库所属的仓库'),
+      query: z.string().describe('查询文本'),
+      score_threshold: z.number().optional().describe('相似度阈值，范围0-1，默认为0.5'),
+      top_k: z.number().optional().describe('返回最相似的前K条结果，默认为10')
+    }
+  }
+}).define(async (ctx) => {
+  const cnb = await cnbManager.getContext(ctx)
+  const { repo, query, score_threshold, top_k } = ctx.args
+
+  const result = await cnb.knowledgeBase.queryKnowledgeBase(repo, { query, score_threshold, top_k })
+  console.log('get knowledge base result', result)
+}).addTo(app)
+
+// pnpm cli cnb-base create-knowledge-base -- repo=kevisual/cnb
+app.route({
+  path: 'cnb-base',
+  key: 'create-knowledge-base',
+  middleware: ['auth'],
+  description: '创建知识库',
+  metadata: {
+    args: {
+      repo: z.string().describe('知识库所属的仓库'),
+    }
+  }
+}).define(async (ctx) => {
+  const cnb = await cnbManager.getContext(ctx)
+  const repo = 'kevisual/cnb'
+  const model_name = 'hunyuan'
+  const include = 'docs/**.md'
+  const exclude = ''
+  const chunk_size = 300
+  const chunk_overlap = 50
+  const text_separator = '\n'
+  const issue_sync_enabled = false
+  const issue_labels = ''
+  const issue_exclude_labels = ''
+  const issue_state = 'open'
+  const baseRes = await cnb.knowledgeBase.getBase(repo)
+  console.log(JSON.stringify(baseRes, null, 2))
+  if (baseRes.code === 200) {
+    // console.log('知识库已存在，删除后重新创建')
+    // const res = await cnb.knowledgeBase.deleteBase(repo)
+    // console.log('delete knowledge base result', res)
+    ctx.body = {
+      content: '知识库已存',
+    }
+    return;
+  }
+  const result = await cnb.knowledgeBase.createKnowledgeBase(repo, {
+    model_name,
+    include,
+    exclude,
+    chunk_size,
+    chunk_overlap,
+    text_separator,
+    issue_sync_enabled,
+    issue_labels,
+    issue_exclude_labels,
+    issue_state
+  })
+  console.log('create knowledge base result', result)
+}).addTo(app)

package/agent/routes/knowledge/index.ts

@@ -1 +1,2 @@
-import './ai.ts'
+import './ai.ts'
+import './embedding.ts'

package/agent/routes/labels/tags.ts

@@ -2,10 +2,10 @@ import { z } from 'zod';
 import { app, cnbManager } from '../../app.ts';
 
 // 查询仓库的标签列表，并导入到新仓库
-// pnpm cli cnb importTags -- source=kevisual/cnb target=abearxiong/jour-2026-04-13
+// pnpm cli cnb importLabels -- source=kevisual/cnb target=abearxiong/jour-2026-04-12
 app.route({
   path: 'cnb',
-  key: 'importTags',
+  key: 'importLabels',
   description: '从一个仓库导入标签到另一个仓库',
   middleware: ['auth'],
   metadata: {
@@ -33,4 +33,44 @@ app.route({
     labels: res.data,
   });
   ctx.forward(importRes);
-}).addTo(app);    
+}).addTo(app);
+
+// pnpm cli cnb removeLabels -- repo=abearxiong/jour-2026-04-12 all=true
+app.route({
+  path: 'cnb',
+  key: 'removeLabels',
+  description: '从仓库中删除标签',
+  middleware: ['auth'],
+  metadata: {
+    args: {
+      repo: z.string().describe('仓库路径, 如 my-user/my-repo'),
+      labels: z.array(z.string()).describe('要删除的标签名称列表'),
+      all: z.boolean().describe('是否删除所有标签，默认为 false'),
+    },
+  }
+}).define(async (ctx) => {
+  const cnb = await cnbManager.getContext(ctx);
+  const repo = ctx.query?.repo;
+  let labels = ctx.query?.labels;
+  const all = ctx.query?.all === true;
+  if (!repo) {
+    ctx.throw(400, '缺少参数 repo');
+  }
+  if (all) {
+    const res = await cnb.labels.repoLabel.getAll(repo);
+    if (res.code !== 200) {
+      return ctx.forward(res)
+    }
+    labels = res.data.map((label: { name: string }) => label.name);
+  }
+  if (!labels || !Array.isArray(labels)) {
+    ctx.throw(400, '缺少参数 labels 或 labels 不是数组');
+  }
+
+  // 删除标签列表
+  const removeRes = await cnb.labels.repoLabel.removeLabels({
+    repo,
+    labels,
+  });
+  ctx.forward(removeRes);
+}).addTo(app);

package/agent/routes/repo/download.ts

@@ -0,0 +1,23 @@
+import { downloadByCNBApi } from "@/repo/download.ts";
+
+import { app } from '../../app.ts';
+import { z } from 'zod';
+
+// pnpm cli cnb download -- url=https://
+app.route({
+  path: 'cnb',
+  key: 'download',
+  description: '下载文件',
+  middleware: ['auth'],
+  metadata: {
+    args: {
+      url: z.string().describe('文件下载链接'),
+    },
+  }
+}).define(async (ctx) => {
+  const url = ctx.query?.url;
+  if (!url) {
+    ctx.throw(400, '缺少 url 参数');
+  }
+  // 
+}).addTo(app);

package/agent/routes/repo/index.ts

@@ -1,3 +1,4 @@
 import './list.ts'
 import './repo.ts'
-import './repo-label.ts'
+import './repo-label.ts'
+import './download.ts'

package/dist/cli-live.js

@@ -46,7 +46,7 @@ var __export = (target, all) => {
 var __esm = (fn, res) => () => (fn && (res = fn(fn = 0)), res);
 var __require = /* @__PURE__ */ createRequire(import.meta.url);
 
-// node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/constants.js
+// ../../node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/constants.js
 var require_constants = __commonJS((exports, module) => {
   var BINARY_TYPES = ["nodebuffer", "arraybuffer", "fragments"];
   var hasBlob = typeof Blob !== "undefined";
@@ -66,7 +66,7 @@ var require_constants = __commonJS((exports, module) => {
   };
 });
 
-// node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/buffer-util.js
+// ../../node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/buffer-util.js
 var require_buffer_util = __commonJS((exports, module) => {
   var { EMPTY_BUFFER } = require_constants();
   var FastBuffer = Buffer[Symbol.species];
@@ -127,7 +127,7 @@ var require_buffer_util = __commonJS((exports, module) => {
   };
 });
 
-// node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/limiter.js
+// ../../node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/limiter.js
 var require_limiter = __commonJS((exports, module) => {
   var kDone = Symbol("kDone");
   var kRun = Symbol("kRun");
@@ -159,7 +159,7 @@ var require_limiter = __commonJS((exports, module) => {
   module.exports = Limiter;
 });
 
-// node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/permessage-deflate.js
+// ../../node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/permessage-deflate.js
 var require_permessage_deflate = __commonJS((exports, module) => {
   var zlib = __require("zlib");
   var bufferUtil = require_buffer_util();
@@ -423,7 +423,7 @@ var require_permessage_deflate = __commonJS((exports, module) => {
   }
 });
 
-// node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/validation.js
+// ../../node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/validation.js
 var require_validation = __commonJS((exports, module) => {
   var { isUtf8 } = __require("buffer");
   var { hasBlob } = require_constants();
@@ -603,7 +603,7 @@ var require_validation = __commonJS((exports, module) => {
   }
 });
 
-// node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/receiver.js
+// ../../node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/receiver.js
 var require_receiver = __commonJS((exports, module) => {
   var { Writable } = __require("stream");
   var PerMessageDeflate = require_permessage_deflate();
@@ -984,7 +984,7 @@ var require_receiver = __commonJS((exports, module) => {
   module.exports = Receiver;
 });
 
-// node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/sender.js
+// ../../node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/sender.js
 var require_sender = __commonJS((exports, module) => {
   var { Duplex } = __require("stream");
   var { randomFillSync } = __require("crypto");
@@ -1338,7 +1338,7 @@ var require_sender = __commonJS((exports, module) => {
   }
 });
 
-// node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/event-target.js
+// ../../node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/event-target.js
 var require_event_target = __commonJS((exports, module) => {
   var { kForOnEventAttribute, kListener } = require_constants();
   var kCode = Symbol("kCode");
@@ -1489,7 +1489,7 @@ var require_event_target = __commonJS((exports, module) => {
   }
 });
 
-// node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/extension.js
+// ../../node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/extension.js
 var require_extension = __commonJS((exports, module) => {
   var { tokenChars } = require_validation();
   function push(dest, name, elem) {
@@ -1654,7 +1654,7 @@ var require_extension = __commonJS((exports, module) => {
   module.exports = { format, parse };
 });
 
-// node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/websocket.js
+// ../../node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/websocket.js
 var require_websocket = __commonJS((exports, module) => {
   var EventEmitter = __require("events");
   var https = __require("https");
@@ -2416,7 +2416,7 @@ var require_websocket = __commonJS((exports, module) => {
   }
 });
 
-// node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/stream.js
+// ../../node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/stream.js
 var require_stream = __commonJS((exports, module) => {
   var WebSocket = require_websocket();
   var { Duplex } = __require("stream");
@@ -2519,7 +2519,7 @@ var require_stream = __commonJS((exports, module) => {
   module.exports = createWebSocketStream;
 });
 
-// node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/subprotocol.js
+// ../../node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/subprotocol.js
 var require_subprotocol = __commonJS((exports, module) => {
   var { tokenChars } = require_validation();
   function parse(header) {
@@ -2564,7 +2564,7 @@ var require_subprotocol = __commonJS((exports, module) => {
   module.exports = { parse };
 });
 
-// node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/websocket-server.js
+// ../../node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/lib/websocket-server.js
 var require_websocket_server = __commonJS((exports, module) => {
   var EventEmitter = __require("events");
   var http = __require("http");
@@ -2866,7 +2866,7 @@ var require_websocket_server = __commonJS((exports, module) => {
   }
 });
 
-// node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/wrapper.mjs
+// ../../node_modules/.pnpm/@kevisual+ws@8.19.0/node_modules/@kevisual/ws/wrapper.mjs
 var exports_wrapper = {};
 __export(exports_wrapper, {
   default: () => wrapper_default,
@@ -2886,7 +2886,7 @@ var init_wrapper = __esm(() => {
   wrapper_default = import_websocket.default;
 });
 
-// node_modules/.pnpm/commander@14.0.3/node_modules/commander/lib/error.js
+// ../../node_modules/.pnpm/commander@14.0.3/node_modules/commander/lib/error.js
 var require_error = __commonJS((exports) => {
   class CommanderError extends Error {
     constructor(exitCode, code, message) {
@@ -2910,7 +2910,7 @@ var require_error = __commonJS((exports) => {
   exports.InvalidArgumentError = InvalidArgumentError;
 });
 
-// node_modules/.pnpm/commander@14.0.3/node_modules/commander/lib/argument.js
+// ../../node_modules/.pnpm/commander@14.0.3/node_modules/commander/lib/argument.js
 var require_argument = __commonJS((exports) => {
   var { InvalidArgumentError } = require_error();
 
@@ -2990,7 +2990,7 @@ var require_argument = __commonJS((exports) => {
   exports.humanReadableArgName = humanReadableArgName;
 });
 
-// node_modules/.pnpm/commander@14.0.3/node_modules/commander/lib/help.js
+// ../../node_modules/.pnpm/commander@14.0.3/node_modules/commander/lib/help.js
 var require_help = __commonJS((exports) => {
   var { humanReadableArgName } = require_argument();
 
@@ -3347,7 +3347,7 @@ ${itemIndentStr}`);
   exports.stripColor = stripColor;
 });
 
-// node_modules/.pnpm/commander@14.0.3/node_modules/commander/lib/option.js
+// ../../node_modules/.pnpm/commander@14.0.3/node_modules/commander/lib/option.js
 var require_option = __commonJS((exports) => {
   var { InvalidArgumentError } = require_error();
 
@@ -3531,7 +3531,7 @@ var require_option = __commonJS((exports) => {
   exports.DualOptions = DualOptions;
 });
 
-// node_modules/.pnpm/commander@14.0.3/node_modules/commander/lib/suggestSimilar.js
+// ../../node_modules/.pnpm/commander@14.0.3/node_modules/commander/lib/suggestSimilar.js
 var require_suggestSimilar = __commonJS((exports) => {
   var maxDistance = 3;
   function editDistance(a, b) {
@@ -3604,7 +3604,7 @@ var require_suggestSimilar = __commonJS((exports) => {
   exports.suggestSimilar = suggestSimilar;
 });
 
-// node_modules/.pnpm/commander@14.0.3/node_modules/commander/lib/command.js
+// ../../node_modules/.pnpm/commander@14.0.3/node_modules/commander/lib/command.js
 var require_command = __commonJS((exports) => {
   var EventEmitter = __require("node:events").EventEmitter;
   var childProcess = __require("node:child_process");
@@ -4959,7 +4959,7 @@ Expecting one of '${allowedValues.join("', '")}'`);
   exports.useColor = useColor;
 });
 
-// node_modules/.pnpm/commander@14.0.3/node_modules/commander/index.js
+// ../../node_modules/.pnpm/commander@14.0.3/node_modules/commander/index.js
 var require_commander = __commonJS((exports) => {
   var { Argument } = require_argument();
   var { Command } = require_command();
@@ -5176,7 +5176,7 @@ function createKeepAlive(config) {
 import { readFileSync } from "node:fs";
 import path from "node:path";
 
-// node_modules/.pnpm/commander@14.0.3/node_modules/commander/esm.mjs
+// ../../node_modules/.pnpm/commander@14.0.3/node_modules/commander/esm.mjs
 var import__ = __toESM(require_commander(), 1);
 var {
   program,