@kevisual/ai 0.0.26 → 0.0.28

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kevisual/ai",
-  "version": "0.0.26",
+  "version": "0.0.28",
   "description": "AI Center Services",
   "main": "index.js",
   "basename": "/root/ai-center-services",
@@ -14,9 +14,8 @@
     "types"
   ],
   "scripts": {
-    "build": "npm run clean && bun bun.config.mjs",
-    "dev": "bun run --watch bun.config.mjs",
-    "test": "tsx  test/**/*.ts",
+    "build": "npm run clean && bun bun.config.ts",
+    "dev": "bun run --watch bun.config.ts",
     "clean": "rm -rf dist",
     "pub": "envision pack -p -u"
   },
@@ -27,7 +26,7 @@
   ],
   "author": "abearxiong <xiongxiao@xiongxiao.me> (https://www.xiongxiao.me)",
   "license": "MIT",
-  "packageManager": "pnpm@10.30.3",
+  "packageManager": "pnpm@10.32.1",
   "type": "module",
   "publishConfig": {
     "registry": "https://registry.npmjs.org/",
@@ -37,30 +36,38 @@
     ".": "./dist/ai-provider-browser.js",
     "./ai-provider": "./dist/ai-provider.js",
     "./browser": "./dist/ai-provider-browser.js",
+    "./agent": "./dist/agent.js",
     "./src/*": "./src/*"
   },
   "devDependencies": {
-    "@kevisual/router": "0.0.84",
+    "@kevisual/code-builder": "^0.0.6",
+    "@kevisual/router": "0.1.1",
     "@kevisual/types": "^0.0.12",
     "@kevisual/use-config": "^1.0.30",
-    "@types/bun": "^1.3.9",
+    "@types/bun": "^1.3.10",
     "@types/crypto-js": "^4.2.2",
-    "@types/formidable": "^3.4.7",
-    "@types/node": "^25.3.3",
+    "@types/formidable": "^3.5.0",
+    "@types/node": "^25.4.0",
     "cross-env": "^10.1.0",
     "crypto-js": "^4.2.0",
     "dayjs": "^1.11.19",
     "dotenv": "^17.3.1",
     "formidable": "^3.5.4",
-    "openai": "6.25.0",
+    "openai": "6.27.0",
     "pm2": "^6.0.14",
     "rimraf": "^6.1.3",
     "typescript": "^5.9.3",
     "vite": "^7.3.1"
   },
   "dependencies": {
+    "@ai-sdk/anthropic": "^3.0.58",
+    "@ai-sdk/openai": "^3.0.41",
+    "@ai-sdk/openai-compatible": "^2.0.35",
+    "@kevisual/js-filter": "^0.0.6",
     "@kevisual/logger": "^0.0.4",
     "@kevisual/permission": "^0.0.4",
-    "@kevisual/query": "^0.0.52"
+    "@kevisual/query": "^0.0.53",
+    "ai": "^6.0.116",
+    "zod": "^4.3.6"
   }
 }

package/src/agent/index.ts

@@ -0,0 +1,67 @@
+import { type QueryRouterServer, type App, type RouteInfo } from '@kevisual/router'
+import { generateText, tool, type ModelMessage, type LanguageModel, type GenerateTextResult } from 'ai';
+import z from 'zod';
+import { filter } from '@kevisual/js-filter'
+export const createTool = async (app: QueryRouterServer | App, message: { path: string, key: string, token?: string }) => {
+  const route = app.findRoute({ path: message.path, key: message.key });
+  if (!route) {
+    console.error(`未找到路径 ${message.path} 和 key ${message.key} 的路由`);
+    return null;
+  }
+  const _tool = tool({
+    description: route?.metadata?.summary || route?.description || '无描述',
+    inputSchema: z.object({
+      ...route.metadata?.args
+    }), // 这里可以根据实际需要定义输入参数的 schema
+    execute: async (args: any) => {
+      const res = await app.run({ path: message.path, key: message.key, payload: args, token: message.token });
+      return res;
+    }
+  });
+  return _tool;
+}
+
+export const createTools = async (opts: { app: QueryRouterServer | App, token?: string }) => {
+  const { app, token } = opts;
+  const tools: Record<string, any> = {};
+  for (const route of app.routes) {
+    const id = route.id!;
+    const _tool = await createTool(app, { path: route.path!, key: route.key!, token });
+    if (_tool && id) {
+      tools[id] = _tool;
+    }
+  }
+  return tools;
+}
+type Route = Partial<RouteInfo>
+type AgentResult = {
+  result: GenerateTextResult<Record<string, any>, any>,
+  messages: ModelMessage[],
+  outputText: string,
+}
+export const reCallAgent = async (opts: { messages?: ModelMessage[], tools?: Record<string, any>, languageModel: LanguageModel }): Promise<AgentResult> => {
+  const { messages = [], tools = {}, languageModel } = opts;
+  const result = await generateText({
+    model: languageModel,
+    messages,
+    tools,
+  });
+  const step = result.steps[0]!;
+  if (step.finishReason === 'tool-calls') {
+    messages.push(...result.response.messages);
+    return reCallAgent({ messages, tools, languageModel });
+  }
+  let outputText = result.output
+  return { result, messages, outputText };
+}
+export const runAgent = async (opts: { app: QueryRouterServer | App, messages?: ModelMessage[], routes?: Route[], query?: string, languageModel: LanguageModel, token: string }) => {
+  const { app, languageModel } = opts;
+  let messages = opts.messages || [];
+
+  let routes = opts?.routes || app.routes;
+  if (opts.query) {
+    routes = filter(routes, opts.query);
+  };
+  const tools = await createTools({ app, token: opts.token });
+  return await reCallAgent({ messages, tools, languageModel });
+}

package/src/provider/chat-adapter/cnb.ts

@@ -6,7 +6,7 @@ export class CNBChat extends BaseChat {
   repo: string;
   constructor(options: CNBOptions & { repo: string }) {
     const baseURL = CNBChat.BASE_URL.replace('{repo}', options.repo);
-    super({ model: "hunyuan-a13b", ...(options as BaseChatOptions), baseURL: baseURL });
+    super({ model: "CNB-Models", ...(options as BaseChatOptions), baseURL: baseURL });
   }
   async query(params: CNBQueryParam): Promise<Result<QueryRag[]>> {
     const url = this.baseURL.replace('/ai', '/knowledge/base/query');

package/src/provider/index.ts

@@ -2,6 +2,4 @@ export * from './chat.ts';
 
 export * from './knowledge.ts';
 
-export * from './utils/index.ts';
-
 export { AIUtils } from './core/utils/index.ts';

package/src/provider/utils/ai-config-type.ts

@@ -1,52 +0,0 @@
-import type { Permission } from '@kevisual/permission';
-
-export type AIModel = {
-  /**
-   * 提供商
-   */
-  provider: string;
-  /**
-   * 模型名称
-   */
-  model: string;
-  /**
-   * 模型组
-   */
-  group: string;
-  /**
-   * 每日请求频率限制
-   */
-  dayLimit?: number;
-  /**
-   * 总的token限制
-   */
-  tokenLimit?: number;
-};
-export type SecretKey = {
-  /**
-   * 组
-   */
-  group: string;
-  /**
-   * API密钥
-   */
-  apiKey: string;
-  /**
-   * 解密密钥
-   */
-  decryptKey?: string;
-};
-
-export type AIConfig = {
-  title?: string;
-  description?: string;
-  models: AIModel[];
-  secretKeys: SecretKey[];
-  permission?: Permission;
-  filter?: {
-    objectKey: string;
-    type: 'array' | 'object';
-    operate: 'removeAttribute' | 'remove';
-    attribute: string[];
-  }[];
-};

package/src/provider/utils/index.ts

@@ -1,2 +0,0 @@
-export * from './ai-config-type.ts';
-export * from './parse-config.ts';

package/src/provider/utils/parse-config.ts

@@ -1,192 +0,0 @@
-import AES from 'crypto-js/aes.js';
-import Utf8 from 'crypto-js/enc-utf8.js';
-import type { AIConfig } from './ai-config-type.js';
-const CryptoJS = { AES, enc: { Utf8 } };
-// 加密函数
-export function encryptAES(plainText: string, secretKey: string) {
-  return CryptoJS.AES.encrypt(plainText, secretKey).toString();
-}
-
-// 解密函数
-export function decryptAES(cipherText: string, secretKey: string) {
-  const bytes = CryptoJS.AES.decrypt(cipherText, secretKey);
-  return bytes.toString(CryptoJS.enc.Utf8);
-}
-
-type AIModel = {
-  /**
-   * 提供商
-   */
-  provider: string;
-  /**
-   * 模型名称
-   */
-  model: string;
-  /**
-   * 模型组
-   */
-  group: string;
-  /**
-   * 每日请求频率限制
-   */
-  dayLimit?: number;
-  /**
-   * 总的token限制
-   */
-  tokenLimit?: number;
-};
-
-type SecretKey = {
-  /**
-   * 组
-   */
-  group: string;
-  /**
-   * API密钥
-   */
-  apiKey: string;
-  /**
-   * 解密密钥
-   */
-  decryptKey?: string;
-};
-
-export type GetProviderOpts = {
-  model: string;
-  group: string;
-  decryptKey?: string;
-};
-export type ProviderResult = {
-  provider: string;
-  model: string;
-  group: string;
-  apiKey: string;
-  dayLimit?: number;
-  tokenLimit?: number;
-  baseURL?: string;
-  /**
-   * 解密密钥
-   */
-  decryptKey?: string;
-};
-
-export class AIConfigParser {
-  private config: AIConfig;
-  result: ProviderResult;
-  constructor(config: AIConfig) {
-    this.config = config;
-  }
-  /**
-   * 获取模型配置
-   * @param opts
-   * @returns
-   */
-  getProvider(opts: GetProviderOpts): ProviderResult {
-    const { model, group, decryptKey } = opts;
-    const modelConfig = this.config.models.find((m) => m.model === model && m.group === group);
-    const groupConfig = this.config.secretKeys.find((m) => m.group === group);
-    if (!modelConfig) {
-      throw new Error(`在模型组 ${group} 中未找到模型 ${model}`);
-    }
-    const mergeConfig = {
-      ...modelConfig,
-      ...groupConfig,
-      decryptKey: decryptKey || groupConfig?.decryptKey,
-    };
-    // 验证模型配置
-    if (!mergeConfig.provider) {
-      throw new Error(`模型 ${model} 未配置提供商`);
-    }
-    if (!mergeConfig.model) {
-      throw new Error(`模型 ${model} 未配置模型名称`);
-    }
-    if (!mergeConfig.apiKey) {
-      throw new Error(`组 ${group} 未配置 API 密钥`);
-    }
-    if (!mergeConfig.group) {
-      throw new Error(`组 ${group} 未配置`);
-    }
-    this.result = mergeConfig;
-    return mergeConfig;
-  }
-  /**
-   * 获取解密密钥
-   * @param opts
-   * @returns
-   */
-  async getSecretKey(opts?: {
-    getCache?: (key: string) => Promise<string>;
-    setCache?: (key: string, value: string) => Promise<void>;
-    providerResult?: ProviderResult;
-  }) {
-    const { getCache, setCache, providerResult } = opts || {};
-    const { apiKey, decryptKey, group = '', model } = providerResult || this.result;
-    const cacheKey = `${group}--${model}`;
-    if (!decryptKey) {
-      return apiKey;
-    }
-    if (getCache) {
-      const cache = await getCache(cacheKey);
-      if (cache) {
-        return cache;
-      }
-    }
-    const secretKey = decryptAES(apiKey, decryptKey);
-    if (setCache) {
-      await setCache(cacheKey, secretKey);
-    }
-    return secretKey;
-  }
-  /**
-   * 加密
-   * @param plainText
-   * @param secretKey
-   * @returns
-   */
-  encrypt(plainText: string, secretKey: string) {
-    return encryptAES(plainText, secretKey);
-  }
-  /**
-   * 解密
-   * @param cipherText
-   * @param secretKey
-   * @returns
-   */
-  decrypt(cipherText: string, secretKey: string) {
-    return decryptAES(cipherText, secretKey);
-  }
-
-  /**
-   * 获取模型配置
-   * @returns
-   */
-  getSelectOpts() {
-    const { models, secretKeys = [] } = this.config;
-
-    return models.map((model) => {
-      const selectOpts = secretKeys.find((m) => m.group === model.group);
-      return {
-        ...model,
-        ...selectOpts,
-      };
-    });
-  }
-  getConfig(keepSecret?: boolean, config?: AIConfig) {
-    const chatConfig = config ?? this.config;
-    if (keepSecret) {
-      return chatConfig;
-    }
-    // 过滤掉secret中的所有apiKey，移除掉并返回chatConfig
-    const { secretKeys = [], ...rest } = chatConfig || {};
-    return {
-      ...rest,
-      secretKeys: secretKeys.map((item) => {
-        return {
-          ...item,
-          apiKey: undefined,
-          decryptKey: undefined,
-        };
-      }),
-    };
-  }
-}