@kevisual/ai 0.0.26 → 0.0.27

package/dist/agent.d.ts

@@ -0,0 +1,32 @@
+import { QueryRouterServer, App, RouteInfo } from '@kevisual/router';
+import { ModelMessage, LanguageModel, GenerateTextResult } from 'ai';
+
+declare const createTool: (app: QueryRouterServer | App, message: {
+    path: string;
+    key: string;
+    token?: string;
+}) => Promise<any>;
+declare const createTools: (opts: {
+    app: QueryRouterServer | App;
+    token?: string;
+}) => Promise<Record<string, any>>;
+type Route = Partial<RouteInfo>;
+type AgentResult = {
+    result: GenerateTextResult<Record<string, any>, any>;
+    messages: ModelMessage[];
+};
+declare const reCallAgent: (opts: {
+    messages?: ModelMessage[];
+    tools?: Record<string, any>;
+    languageModel: LanguageModel;
+}) => Promise<AgentResult>;
+declare const runAgent: (opts: {
+    app: QueryRouterServer | App;
+    messages?: ModelMessage[];
+    routes?: Route[];
+    query?: string;
+    languageModel: LanguageModel;
+    token: string;
+}) => Promise<AgentResult>;
+
+export { createTool, createTools, reCallAgent, runAgent };

package/dist/agent.js

@@ -0,0 +1,64 @@
+// src/agent/index.ts
+import { generateText, tool } from "ai";
+import z from "zod";
+import { filter } from "@kevisual/js-filter";
+var createTool = async (app, message) => {
+  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
+    }),
+    execute: async (args) => {
+      const res = await app.run({ path: message.path, key: message.key, payload: args, token: message.token });
+      return res;
+    }
+  });
+  return _tool;
+};
+var createTools = async (opts) => {
+  const { app, token } = opts;
+  const tools = {};
+  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;
+};
+var reCallAgent = async (opts) => {
+  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 });
+  }
+  return { result, messages };
+};
+var runAgent = async (opts) => {
+  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 });
+};
+export {
+  runAgent,
+  reCallAgent,
+  createTools,
+  createTool
+};

package/dist/ai-provider-browser.d.ts

@@ -1,5 +1,3 @@
-import * as _kevisual_permission from '@kevisual/permission';
-import { Permission } from '@kevisual/permission';
 import { Result } from '@kevisual/query';
 
 type ChatMessage = {
@@ -609,141 +607,5 @@ type RerankOptions = {
     overlap_tokens?: number;
 };
 
-type AIModel = {
-    /**
-     * 提供商
-     */
-    provider: string;
-    /**
-     * 模型名称
-     */
-    model: string;
-    /**
-     * 模型组
-     */
-    group: string;
-    /**
-     * 每日请求频率限制
-     */
-    dayLimit?: number;
-    /**
-     * 总的token限制
-     */
-    tokenLimit?: number;
-};
-type SecretKey = {
-    /**
-     * 组
-     */
-    group: string;
-    /**
-     * API密钥
-     */
-    apiKey: string;
-    /**
-     * 解密密钥
-     */
-    decryptKey?: string;
-};
-type AIConfig = {
-    title?: string;
-    description?: string;
-    models: AIModel[];
-    secretKeys: SecretKey[];
-    permission?: Permission;
-    filter?: {
-        objectKey: string;
-        type: 'array' | 'object';
-        operate: 'removeAttribute' | 'remove';
-        attribute: string[];
-    }[];
-};
-
-declare function encryptAES(plainText: string, secretKey: string): string;
-declare function decryptAES(cipherText: string, secretKey: string): string;
-type GetProviderOpts = {
-    model: string;
-    group: string;
-    decryptKey?: string;
-};
-type ProviderResult = {
-    provider: string;
-    model: string;
-    group: string;
-    apiKey: string;
-    dayLimit?: number;
-    tokenLimit?: number;
-    baseURL?: string;
-    /**
-     * 解密密钥
-     */
-    decryptKey?: string;
-};
-declare class AIConfigParser {
-    private config;
-    result: ProviderResult;
-    constructor(config: AIConfig);
-    /**
-     * 获取模型配置
-     * @param opts
-     * @returns
-     */
-    getProvider(opts: GetProviderOpts): ProviderResult;
-    /**
-     * 获取解密密钥
-     * @param opts
-     * @returns
-     */
-    getSecretKey(opts?: {
-        getCache?: (key: string) => Promise<string>;
-        setCache?: (key: string, value: string) => Promise<void>;
-        providerResult?: ProviderResult;
-    }): Promise<string>;
-    /**
-     * 加密
-     * @param plainText
-     * @param secretKey
-     * @returns
-     */
-    encrypt(plainText: string, secretKey: string): string;
-    /**
-     * 解密
-     * @param cipherText
-     * @param secretKey
-     * @returns
-     */
-    decrypt(cipherText: string, secretKey: string): string;
-    /**
-     * 获取模型配置
-     * @returns
-     */
-    getSelectOpts(): {
-        group: string;
-        apiKey: string;
-        decryptKey?: string;
-        provider: string;
-        model: string;
-        dayLimit?: number;
-        tokenLimit?: number;
-    }[];
-    getConfig(keepSecret?: boolean, config?: AIConfig): AIConfig | {
-        secretKeys: {
-            apiKey: any;
-            decryptKey: any;
-            group: string;
-        }[];
-        title?: string;
-        description?: string;
-        models?: AIModel[];
-        permission?: _kevisual_permission.Permission;
-        filter?: {
-            objectKey: string;
-            type: "array" | "object";
-            operate: "removeAttribute" | "remove";
-            attribute: string[];
-        }[];
-    };
-}
-
-export { AIConfigParser, AIUtils, BailianChat, BailianProvider, BaseChat, CNBChat, ChatProviderMap, Custom, CustomProvider, DeepSeek, DeepSeekProvider, Kevisual, KevisualProvider, Kimi, KimiProvider, KnowledgeBase, ModelScope, ModelScopeProvider, Ollama, OllamaProvider, ProviderManager, SiliconFlow, SiliconFlowKnowledge, SiliconFlowProvider, Volces, VolcesProvider, Zhipu, ZhipuProvider, decryptAES, encryptAES, readStream };
-export type { AIConfig, AIModel, BaseChatInterface, BaseChatOptions, ChatMessage, ChatMessageComplete, ChatMessageOptions, ChatStream, EmbeddingMessage, EmbeddingMessageComplete, EmbeddingObject, GetProviderOpts, KnowledgeOptions, ProviderResult, RerankOptions, SecretKey, Usage };
+export { AIUtils, BailianChat, BailianProvider, BaseChat, CNBChat, ChatProviderMap, Custom, CustomProvider, DeepSeek, DeepSeekProvider, Kevisual, KevisualProvider, Kimi, KimiProvider, KnowledgeBase, ModelScope, ModelScopeProvider, Ollama, OllamaProvider, ProviderManager, SiliconFlow, SiliconFlowKnowledge, SiliconFlowProvider, Volces, VolcesProvider, Zhipu, ZhipuProvider, readStream };
+export type { BaseChatInterface, BaseChatOptions, ChatMessage, ChatMessageComplete, ChatMessageOptions, ChatStream, EmbeddingMessage, EmbeddingMessageComplete, EmbeddingObject, KnowledgeOptions, RerankOptions, Usage };