chore: update version to 0.0.24 and router dependency to 0.0.62

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kevisual/ai",
-  "version": "0.0.22",
+  "version": "0.0.24",
   "description": "AI Center Services",
   "main": "index.js",
   "basename": "/root/ai-center-services",
@@ -27,7 +27,7 @@
   ],
   "author": "abearxiong <xiongxiao@xiongxiao.me> (https://www.xiongxiao.me)",
   "license": "MIT",
-  "packageManager": "pnpm@10.28.1",
+  "packageManager": "pnpm@10.28.2",
   "type": "module",
   "publishConfig": {
     "registry": "https://registry.npmjs.org/",
@@ -48,7 +48,7 @@
     }
   },
   "devDependencies": {
-    "@kevisual/router": "0.0.60",
+    "@kevisual/router": "0.0.62",
     "@kevisual/types": "^0.0.12",
     "@kevisual/use-config": "^1.0.28",
     "@types/bun": "^1.3.6",

pnpm-lock.yaml

@@ -19,8 +19,8 @@ importers:
         version: 0.0.38
     devDependencies:
       '@kevisual/router':
-        specifier: 0.0.60
-        version: 0.0.60
+        specifier: 0.0.62
+        version: 0.0.62
       '@kevisual/types':
         specifier: ^0.0.12
         version: 0.0.12
@@ -251,8 +251,8 @@ packages:
   '@kevisual/query@0.0.38':
     resolution: {integrity: sha512-bfvbSodsZyMfwY+1T2SvDeOCKsT/AaIxlVe0+B1R/fNhlg2MDq2CP0L9HKiFkEm+OXrvXcYDMKPUituVUM5J6Q==}
 
-  '@kevisual/router@0.0.60':
-    resolution: {integrity: sha512-2v/ZzUstsaq+Uqo+tZX9ys5E+/2erPggCtljv9jTb3NA88ZdHsYUAsd5wUFvLtf9QucpJCzyWEt+InDV/98FKw==}
+  '@kevisual/router@0.0.62':
+    resolution: {integrity: sha512-kKQFYkJ/qemGAygGSKkM/TujvreoU9HxL7/2vx2RYDMRyRuZOYUuvrTF92XIffFnZHugFbs8uEt+Z8I11SrFUg==}
 
   '@kevisual/types@0.0.12':
     resolution: {integrity: sha512-zJXH2dosir3jVrQ6QG4i0+iLQeT9gJ3H+cKXs8ReWboxBSYzUZO78XssVeVrFPsJ33iaAqo4q3DWbSS1dWGn7Q==}
@@ -322,56 +322,67 @@ packages:
     resolution: {integrity: sha512-k9oD15soC/Ln6d2Wv/JOFPzZXIAIFLp6B+i14KhxAfnq76ajt0EhYc5YPeX6W1xJkAdItcVT+JhKl1QZh44/qw==}
     cpu: [arm]
     os: [linux]
+    libc: [glibc]
 
   '@rollup/rollup-linux-arm-musleabihf@4.53.3':
     resolution: {integrity: sha512-vTNlKq+N6CK/8UktsrFuc+/7NlEYVxgaEgRXVUVK258Z5ymho29skzW1sutgYjqNnquGwVUObAaxae8rZ6YMhg==}
     cpu: [arm]
     os: [linux]
+    libc: [musl]
 
   '@rollup/rollup-linux-arm64-gnu@4.53.3':
     resolution: {integrity: sha512-RGrFLWgMhSxRs/EWJMIFM1O5Mzuz3Xy3/mnxJp/5cVhZ2XoCAxJnmNsEyeMJtpK+wu0FJFWz+QF4mjCA7AUQ3w==}
     cpu: [arm64]
     os: [linux]
+    libc: [glibc]
 
   '@rollup/rollup-linux-arm64-musl@4.53.3':
     resolution: {integrity: sha512-kASyvfBEWYPEwe0Qv4nfu6pNkITLTb32p4yTgzFCocHnJLAHs+9LjUu9ONIhvfT/5lv4YS5muBHyuV84epBo/A==}
     cpu: [arm64]
     os: [linux]
+    libc: [musl]
 
   '@rollup/rollup-linux-loong64-gnu@4.53.3':
     resolution: {integrity: sha512-JiuKcp2teLJwQ7vkJ95EwESWkNRFJD7TQgYmCnrPtlu50b4XvT5MOmurWNrCj3IFdyjBQ5p9vnrX4JM6I8OE7g==}
     cpu: [loong64]
     os: [linux]
+    libc: [glibc]
 
   '@rollup/rollup-linux-ppc64-gnu@4.53.3':
     resolution: {integrity: sha512-EoGSa8nd6d3T7zLuqdojxC20oBfNT8nexBbB/rkxgKj5T5vhpAQKKnD+h3UkoMuTyXkP5jTjK/ccNRmQrPNDuw==}
     cpu: [ppc64]
     os: [linux]
+    libc: [glibc]
 
   '@rollup/rollup-linux-riscv64-gnu@4.53.3':
     resolution: {integrity: sha512-4s+Wped2IHXHPnAEbIB0YWBv7SDohqxobiiPA1FIWZpX+w9o2i4LezzH/NkFUl8LRci/8udci6cLq+jJQlh+0g==}
     cpu: [riscv64]
     os: [linux]
+    libc: [glibc]
 
   '@rollup/rollup-linux-riscv64-musl@4.53.3':
     resolution: {integrity: sha512-68k2g7+0vs2u9CxDt5ktXTngsxOQkSEV/xBbwlqYcUrAVh6P9EgMZvFsnHy4SEiUl46Xf0IObWVbMvPrr2gw8A==}
     cpu: [riscv64]
     os: [linux]
+    libc: [musl]
 
   '@rollup/rollup-linux-s390x-gnu@4.53.3':
     resolution: {integrity: sha512-VYsFMpULAz87ZW6BVYw3I6sWesGpsP9OPcyKe8ofdg9LHxSbRMd7zrVrr5xi/3kMZtpWL/wC+UIJWJYVX5uTKg==}
     cpu: [s390x]
     os: [linux]
+    libc: [glibc]
 
   '@rollup/rollup-linux-x64-gnu@4.53.3':
     resolution: {integrity: sha512-3EhFi1FU6YL8HTUJZ51imGJWEX//ajQPfqWLI3BQq4TlvHy4X0MOr5q3D2Zof/ka0d5FNdPwZXm3Yyib/UEd+w==}
     cpu: [x64]
     os: [linux]
+    libc: [glibc]
 
   '@rollup/rollup-linux-x64-musl@4.53.3':
     resolution: {integrity: sha512-eoROhjcc6HbZCJr+tvVT8X4fW3/5g/WkGvvmwz/88sDtSJzO7r/blvoBDgISDiCjDRZmHpwud7h+6Q9JxFwq1Q==}
     cpu: [x64]
     os: [linux]
+    libc: [musl]
 
   '@rollup/rollup-openharmony-arm64@4.53.3':
     resolution: {integrity: sha512-OueLAWgrNSPGAdUdIjSWXw+u/02BRTcnfw9PN41D2vq/JSEPnJnVuBgw18VkN8wcd4fjUs+jFHVM4t9+kBSNLw==}
@@ -1213,7 +1224,7 @@ snapshots:
     dependencies:
       tslib: 2.8.1
 
-  '@kevisual/router@0.0.60':
+  '@kevisual/router@0.0.62':
     dependencies:
       hono: 4.11.5
 

src/provider/chat-adapter/cnb.ts

@@ -1,18 +1,30 @@
 import { BaseChat, BaseChatOptions } from '../core/chat.ts';
-
+import { Result } from '@kevisual/query'
 export type CNBOptions = Partial<BaseChatOptions>;
 export class CNBChat extends BaseChat {
   static BASE_URL = 'https://api.cnb.cool/{repo}/-/ai';
+  repo: string;
   constructor(options: CNBOptions & { repo: string }) {
     const baseURL = CNBChat.BASE_URL.replace('{repo}', options.repo);
-    super({ ...(options as BaseChatOptions), baseURL: baseURL });
+    super({ model: "hunyuan-a13b", ...(options as BaseChatOptions), baseURL: baseURL });
   }
-  query({ repo, ...rest }: CNBQueryParam & { repo: string }) {
-    const baseURL = 'https://api.cnb.cool/{repo}/-/knowledge/base/query'
-      .replace('{repo}', repo);
-    this.post(baseURL, {
-      data: rest,
-    });
+  async query(params: CNBQueryParam): Promise<Result<QueryRag[]>> {
+    const url = this.baseURL.replace('/ai', '/knowledge/base/query');
+    const response = await this.post(url, {
+      data: {
+        score_threshold: 0.62,
+        top_k: 10,
+        ...params
+      },
+    })
+    if (!response.ok) {
+      throw new Error(`query API error: ${response.status} ${response.statusText}`);
+    }
+    const res = await response.json() as QueryRag[];
+    return {
+      code: 200,
+      data: res || [],
+    }
   }
 }
 
@@ -31,4 +43,17 @@ export type CNBQueryParam = {
   },
   score_threshold?: number;
   top_k?: number;
+}
+
+type QueryRag = {
+  chunk: string;
+  score: number;
+  metadata: {
+    hash: string;
+    name: string;
+    path: string;
+    position: string;
+    type: string; // code, text
+    url: string;
+  }
 }