@zhin.js/ai 0.0.1 → 0.0.2

package/lib/index.d.ts

@@ -0,0 +1,167 @@
+/**
+ * @zhin.js/ai - AI Service for Zhin.js
+ *
+ * 多模型 AI 服务插件，支持：
+ * - 多模型提供商（OpenAI、Claude、DeepSeek、Ollama 等）
+ * - 工具调用（Function Calling）
+ * - 流式输出
+ * - 会话管理
+ * - Agent 能力
+ * - 独立的 AI 触发中间件（@机器人、前缀触发、私聊直接对话）
+ */
+import { type Plugin, type Tool, type ToolContext, type AITriggerConfig } from '@zhin.js/core';
+import type { AIProvider, AIConfig, ChatCompletionRequest, ChatCompletionResponse, ChatCompletionChunk, AgentTool } from './types.js';
+import { SessionManager } from './session.js';
+import { Agent } from './agent.js';
+import { ContextManager, type ContextConfig } from './context-manager.js';
+declare module '@zhin.js/core' {
+    namespace Plugin {
+        interface Contexts {
+            ai: AIService;
+        }
+    }
+}
+/**
+ * AI 服务
+ * 统一管理多个模型提供商，提供会话和 Agent 能力
+ */
+export declare class AIService {
+    private providers;
+    private defaultProvider;
+    sessions: SessionManager;
+    contextManager?: ContextManager;
+    private builtinTools;
+    private sessionConfig;
+    private contextConfig;
+    private triggerConfig;
+    private plugin?;
+    /** 额外注册的自定义工具 */
+    private customTools;
+    constructor(config?: AIConfig);
+    /**
+     * 检查 AI 服务是否就绪
+     */
+    isReady(): boolean;
+    /**
+     * 处理 AI 请求
+     * 这是 AI 触发中间件调用的主入口
+     *
+     * 采用三步处理架构：
+     * 1. 意图分析：分析用户意图，筛选相关工具
+     * 2. 工具执行：用筛选的工具让模型决策和执行
+     * 3. 结果总结：基于结果生成友好回复
+     */
+    process(content: string, context: ToolContext, tools: Tool[]): Promise<string | AsyncIterable<string>>;
+    /**
+     * 简单对话（不使用会话历史）
+     */
+    private simpleChat;
+    /**
+     * 第一步：意图分析，筛选相关工具
+     * 使用 AI 分析用户意图，选择最相关的工具
+     */
+    private analyzeIntentAndSelectTools;
+    /**
+     * 基于关键词匹配工具（降级方案）
+     */
+    private matchToolsByKeywords;
+    /**
+     * 第三步：总结工具调用结果
+     */
+    private summarizeToolResults;
+    /**
+     * 收集所有工具（包括外部传入的）
+     * 注意：过滤掉命令转换的工具（cmd_xxx），避免工具过多影响模型性能
+     */
+    private collectAllToolsWithExternal;
+    /**
+     * 将 Tool 转换为 AgentTool
+     */
+    private convertToolToAgentTool;
+    /**
+     * 设置会话管理器（用于切换到数据库存储）
+     */
+    setSessionManager(manager: SessionManager): void;
+    /**
+     * 设置上下文管理器
+     */
+    setContextManager(manager: ContextManager): void;
+    /**
+     * 设置插件引用（用于收集工具）
+     */
+    setPlugin(plugin: Plugin): void;
+    /**
+     * 注册自定义工具到 AI 服务
+     */
+    registerTool(tool: AgentTool): () => void;
+    /**
+     * 收集所有可用工具
+     */
+    collectAllTools(): AgentTool[];
+    getContextConfig(): ContextConfig;
+    getSessionConfig(): {
+        maxHistory?: number;
+        expireMs?: number;
+    };
+    getTriggerConfig(): AITriggerConfig;
+    registerProvider(provider: AIProvider): void;
+    getProvider(name?: string): AIProvider;
+    listProviders(): string[];
+    listModels(providerName?: string): Promise<{
+        provider: string;
+        models: string[];
+    }[]>;
+    chat(request: ChatCompletionRequest, providerName?: string): Promise<ChatCompletionResponse>;
+    chatStream(request: ChatCompletionRequest, providerName?: string): AsyncIterable<ChatCompletionChunk>;
+    ask(question: string, options?: {
+        provider?: string;
+        model?: string;
+        systemPrompt?: string;
+        temperature?: number;
+    }): Promise<string>;
+    chatWithSession(sessionId: string, message: string, options?: {
+        provider?: string;
+        model?: string;
+        systemPrompt?: string;
+        stream?: boolean;
+    }): Promise<string | AsyncIterable<string>>;
+    createAgent(options?: {
+        provider?: string;
+        model?: string;
+        systemPrompt?: string;
+        tools?: AgentTool[];
+        useBuiltinTools?: boolean;
+        collectExternalTools?: boolean;
+        maxIterations?: number;
+    }): Agent;
+    runAgent(task: string, options?: {
+        provider?: string;
+        model?: string;
+        tools?: AgentTool[];
+        systemPrompt?: string;
+    }): Promise<{
+        content: string;
+        toolCalls: any[];
+        usage: any;
+    }>;
+    healthCheck(): Promise<Record<string, boolean>>;
+    dispose(): void;
+}
+/**
+ * 创建 AI 服务 Context
+ * 可在 setup.ts 中直接使用：provide(createAIService())
+ */
+export declare function createAIService(): {
+    name: "ai";
+    description: string;
+    mounted(p: Plugin): Promise<any>;
+    dispose(service: AIService | null): Promise<void>;
+};
+export { Agent, createAgent } from './agent.js';
+export { SessionManager, createMemorySessionManager, createDatabaseSessionManager } from './session.js';
+export { ContextManager, createContextManager, CHAT_MESSAGE_MODEL, CONTEXT_SUMMARY_MODEL } from './context-manager.js';
+export type * from './types.js';
+export * from './providers/index.js';
+export * from './tools.js';
+export { createToolService, defineTool, ZhinTool, isZhinTool, shouldTriggerAI, inferSenderPermissions, parseRichMediaContent, extractTextContent, mergeAITriggerConfig, type ToolService, type AITriggerConfig, } from '@zhin.js/core';
+//# sourceMappingURL=index.d.ts.map

package/lib/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,EAaL,KAAK,MAAM,EACX,KAAK,IAAI,EACT,KAAK,WAAW,EAChB,KAAK,eAAe,EACrB,MAAM,eAAe,CAAC;AAGvB,OAAO,KAAK,EACV,UAAU,EACV,QAAQ,EAER,qBAAqB,EACrB,sBAAsB,EACtB,mBAAmB,EACnB,SAAS,EACV,MAAM,YAAY,CAAC;AASpB,OAAO,EACL,cAAc,EAKf,MAAM,cAAc,CAAC;AACtB,OAAO,EAAE,KAAK,EAAe,MAAM,YAAY,CAAC;AA2ChD,OAAO,EACL,cAAc,EAKd,KAAK,aAAa,EACnB,MAAM,sBAAsB,CAAC;AAM9B,OAAO,QAAQ,eAAe,CAAC;IAC7B,UAAU,MAAM,CAAC;QACf,UAAU,QAAQ;YAChB,EAAE,EAAE,SAAS,CAAC;SACf;KACF;CACF;AAMD;;;GAGG;AACH,qBAAa,SAAS;IACpB,OAAO,CAAC,SAAS,CAAsC;IACvD,OAAO,CAAC,eAAe,CAAS;IACzB,QAAQ,EAAE,cAAc,CAAC;IACzB,cAAc,CAAC,EAAE,cAAc,CAAC;IACvC,OAAO,CAAC,YAAY,CAAc;IAClC,OAAO,CAAC,aAAa,CAA6C;IAClE,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,aAAa,CAAkB;IACvC,OAAO,CAAC,MAAM,CAAC,CAAS;IACxB,iBAAiB;IACjB,OAAO,CAAC,WAAW,CAAqC;gBAE5C,MAAM,GAAE,QAAa;IAwCjC;;OAEG;IACH,OAAO,IAAI,OAAO;IAIlB;;;;;;;;OAQG;IACG,OAAO,CACX,OAAO,EAAE,MAAM,EACf,OAAO,EAAE,WAAW,EACpB,KAAK,EAAE,IAAI,EAAE,GACZ,OAAO,CAAC,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;IAiF1C;;OAEG;YACW,UAAU;IAaxB;;;OAGG;YACW,2BAA2B;IA6EzC;;OAEG;IACH,OAAO,CAAC,oBAAoB;IA4D5B;;OAEG;YACW,oBAAoB;IA4ClC;;;OAGG;IACH,OAAO,CAAC,2BAA2B;IA4BnC;;OAEG;IACH,OAAO,CAAC,sBAAsB;IAa9B;;OAEG;IACH,iBAAiB,CAAC,OAAO,EAAE,cAAc,GAAG,IAAI;IAKhD;;OAEG;IACH,iBAAiB,CAAC,OAAO,EAAE,cAAc,GAAG,IAAI;IAQhD;;OAEG;IACH,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAI/B;;OAEG;IACH,YAAY,CAAC,IAAI,EAAE,SAAS,GAAG,MAAM,IAAI;IAOzC;;OAEG;IACH,eAAe,IAAI,SAAS,EAAE;IAgB9B,gBAAgB,IAAI,aAAa;IAIjC,gBAAgB,IAAI;QAAE,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;KAAE;IAI9D,gBAAgB,IAAI,eAAe;IAInC,gBAAgB,CAAC,QAAQ,EAAE,UAAU,GAAG,IAAI;IAI5C,WAAW,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,UAAU;IAStC,aAAa,IAAI,MAAM,EAAE;IAInB,UAAU,CAAC,YAAY,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,EAAE,CAAA;KAAE,EAAE,CAAC;IAiBpF,IAAI,CACR,OAAO,EAAE,qBAAqB,EAC9B,YAAY,CAAC,EAAE,MAAM,GACpB,OAAO,CAAC,sBAAsB,CAAC;IAK3B,UAAU,CACf,OAAO,EAAE,qBAAqB,EAC9B,YAAY,CAAC,EAAE,MAAM,GACpB,aAAa,CAAC,mBAAmB,CAAC;IAK/B,GAAG,CACP,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE;QACP,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,WAAW,CAAC,EAAE,MAAM,CAAC;KACjB,GACL,OAAO,CAAC,MAAM,CAAC;IAoBZ,eAAe,CACnB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,EACf,OAAO,GAAE;QACP,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,MAAM,CAAC,EAAE,OAAO,CAAC;KACb,GACL,OAAO,CAAC,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;IAqD1C,WAAW,CAAC,OAAO,GAAE;QACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,KAAK,CAAC,EAAE,SAAS,EAAE,CAAC;QACpB,eAAe,CAAC,EAAE,OAAO,CAAC;QAC1B,oBAAoB,CAAC,EAAE,OAAO,CAAC;QAC/B,aAAa,CAAC,EAAE,MAAM,CAAC;KACnB,GAAG,KAAK;IAgCR,QAAQ,CACZ,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE;QACP,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,KAAK,CAAC,EAAE,SAAS,EAAE,CAAC;QACpB,YAAY,CAAC,EAAE,MAAM,CAAC;KAClB,GACL,OAAO,CAAC;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,GAAG,EAAE,CAAC;QAAC,KAAK,EAAE,GAAG,CAAA;KAAE,CAAC;IAKvD,WAAW,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAcrD,OAAO,IAAI,IAAI;CAIhB;AAwiBD;;;GAGG;AACH,wBAAgB,eAAe;;;eAIV,MAAM;qBAsBA,SAAS,GAAG,IAAI;EAM1C;AAOD,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAChD,OAAO,EAAE,cAAc,EAAE,0BAA0B,EAAE,4BAA4B,EAAE,MAAM,cAAc,CAAC;AACxG,OAAO,EAAE,cAAc,EAAE,oBAAoB,EAAE,kBAAkB,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AACvH,mBAAmB,YAAY,CAAC;AAChC,cAAc,sBAAsB,CAAC;AACrC,cAAc,YAAY,CAAC;AAG3B,OAAO,EACL,iBAAiB,EACjB,UAAU,EACV,QAAQ,EACR,UAAU,EAEV,eAAe,EACf,sBAAsB,EACtB,qBAAqB,EACrB,kBAAkB,EAClB,oBAAoB,EACpB,KAAK,WAAW,EAChB,KAAK,eAAe,GACrB,MAAM,eAAe,CAAC"}