@n0ts123/mcplink-core 0.0.2 → 0.0.3

package/dist/index.d.ts

@@ -1,4 +1,6 @@
 import { LanguageModel } from 'ai';
+export { createOpenAI } from '@ai-sdk/openai';
+export { createAnthropic } from '@ai-sdk/anthropic';
 
 /** MCP 服务器配置 - stdio 模式 */
 interface MCPServerConfigStdio {
@@ -18,8 +20,29 @@ interface MCPServerConfigSSE {
     /** 请求头 */
     headers?: Record<string, string>;
 }
+/** MCP 服务器配置 - Streamable HTTP 模式 */
+interface MCPServerConfigStreamableHTTP {
+    type: 'streamable-http';
+    /** HTTP 服务地址 */
+    url: string;
+    /** 请求头 */
+    headers?: Record<string, string>;
+}
 /** MCP 服务器配置 */
-type MCPServerConfig = MCPServerConfigStdio | MCPServerConfigSSE;
+type MCPServerConfig = MCPServerConfigStdio | MCPServerConfigSSE | MCPServerConfigStreamableHTTP;
+/**
+ * 即时结果匹配器
+ * 用于检测 MCP 工具返回的特殊格式数据
+ * 当工具返回的 JSON 包含匹配器中所有的 key-value 时，会立即发送 IMMEDIATE_RESULT 事件
+ *
+ * @example
+ * // 匹配包含 { type: "card" } 的返回结果
+ * { type: "card" }
+ *
+ * // 匹配包含 { type: "product_list", isCard: true } 的返回结果
+ * { type: "product_list", isCard: true }
+ */
+type ImmediateResultMatcher = Record<string, unknown>;
 /** MCPLink 配置 */
 interface MCPLinkConfig {
     /** AI 模型实例 (Vercel AI SDK) */
@@ -41,6 +64,37 @@ interface MCPLinkConfig {
      * - 'auto': 自动检测（默认）
      */
     usePromptBasedTools?: boolean | 'auto';
+    /**
+     * 即时结果匹配器列表
+     * 当 MCP 工具返回的结果匹配任意一个匹配器时，会立即发送 IMMEDIATE_RESULT 事件
+     * 匹配规则：结果 JSON 中包含匹配器的所有 key 且对应的 value 相等
+     *
+     * @example
+     * immediateResultMatchers: [
+     *   { type: "card" },           // 匹配 { type: "card", ... }
+     *   { type: "product_list" },   // 匹配 { type: "product_list", ... }
+     * ]
+     */
+    immediateResultMatchers?: ImmediateResultMatcher[];
+    /**
+     * 是否启用思考阶段（两阶段调用）
+     * - true: 每次迭代先让 AI 思考分析，再执行工具调用（默认，推荐）
+     * - false: 直接调用 AI 并执行工具（更快但可能不够准确）
+     *
+     * 启用后的流程：
+     * 1. 思考阶段：AI 分析需求，输出思考过程，决定调用什么工具
+     * 2. 执行阶段：根据思考结果执行工具调用
+     *
+     * 优点：
+     * - 任何模型都能看到思考过程
+     * - Chain-of-Thought 效应，显著提高复杂任务准确性
+     *
+     * 缺点：
+     * - 每次迭代多一次 API 调用，增加延迟和成本
+     *
+     * @default true
+     */
+    enableThinkingPhase?: boolean;
 }
 /** 工具调用 */
 interface ToolCall {
@@ -85,12 +139,12 @@ declare enum MCPLinkEventType {
     TOOL_CALL_START = "tool_call_start",
     /** 工具参数 (流式) */
     TOOL_CALL_DELTA = "tool_call_delta",
-    /** 工具调用参数完成 */
-    TOOL_CALL_END = "tool_call_end",
     /** 工具正在执行 */
     TOOL_EXECUTING = "tool_executing",
     /** 工具返回结果 */
     TOOL_RESULT = "tool_result",
+    /** 工具返回的即时结果（匹配 immediateResultMatchers 时触发） */
+    IMMEDIATE_RESULT = "immediate_result",
     /** 开始新一轮迭代 */
     ITERATION_START = "iteration_start",
     /** 迭代结束 */
@@ -98,15 +152,7 @@ declare enum MCPLinkEventType {
     /** 全部完成 */
     COMPLETE = "complete",
     /** 发生错误 */
-    ERROR = "error",
-    /** TODO 列表开始 */
-    TODO_START = "todo_start",
-    /** 添加 TODO 项 */
-    TODO_ITEM_ADD = "todo_item_add",
-    /** 更新 TODO 项状态 */
-    TODO_ITEM_UPDATE = "todo_item_update",
-    /** TODO 列表完成 */
-    TODO_END = "todo_end"
+    ERROR = "error"
 }
 /** 事件数据 */
 interface MCPLinkEventData {
@@ -123,12 +169,8 @@ interface MCPLinkEventData {
     argsTextDelta?: string;
     isError?: boolean;
     error?: Error;
-    todoId?: string;
-    todoTitle?: string;
-    todoItemId?: string;
-    todoItemContent?: string;
-    todoItemStatus?: TodoItemStatus;
-    todoItemResult?: string;
+    /** 即时结果数据（原封不动的工具返回结果） */
+    immediateResult?: unknown;
 }
 /** 事件 */
 interface MCPLinkEvent {
@@ -152,14 +194,6 @@ interface ChatCallbacks {
     onIterationEnd?: (iteration: number) => void;
     /** 发生错误 */
     onError?: (error: Error) => void;
-    /** TODO 列表开始 */
-    onTodoStart?: (todoId: string, title: string) => void;
-    /** TODO 项添加 */
-    onTodoItemAdd?: (todoId: string, item: TodoItem) => void;
-    /** TODO 项更新 */
-    onTodoItemUpdate?: (todoId: string, itemId: string, status: TodoItemStatus, result?: string) => void;
-    /** TODO 列表完成 */
-    onTodoEnd?: (todoId: string) => void;
 }
 /** 对话结果 */
 interface ChatResult {
@@ -204,30 +238,6 @@ interface MCPServerStatus {
     tools: MCPTool[];
     error?: string;
 }
-/** TODO 项状态 */
-type TodoItemStatus = 'pending' | 'in_progress' | 'completed' | 'failed';
-/** TODO 项 */
-interface TodoItem {
-    /** 唯一标识 */
-    id: string;
-    /** 任务内容 */
-    content: string;
-    /** 当前状态 */
-    status: TodoItemStatus;
-    /** 完成后的结果摘要 */
-    result?: string;
-}
-/** TODO 列表 */
-interface TodoList {
-    /** 唯一标识 */
-    id: string;
-    /** 整体任务标题 */
-    title: string;
-    /** 任务项列表 */
-    items: TodoItem[];
-    /** 创建时间 */
-    createdAt: number;
-}
 
 /**
  * MCPLink 主类
@@ -358,7 +368,7 @@ declare class MCPManager {
  * 默认用户提示词
  * 这只是用户自定义的部分，核心工具调用逻辑已内置到代码中
  */
-declare const DEFAULT_SYSTEM_PROMPT = "\u4F60\u662F\u4E00\u4E2A\u667A\u80FD\u52A9\u624B\uFF0C\u8BF7\u4F7F\u7528\u4E2D\u6587\u56DE\u590D\u3002";
+declare const DEFAULT_SYSTEM_PROMPT = "\u4F60\u662F\u4E00\u4E2A\u4E13\u4E1A\u3001\u53CB\u597D\u7684\u667A\u80FD\u52A9\u624B\u3002\n\n## \u56DE\u590D\u8981\u6C42\n- \u7B80\u6D01\u6E05\u6670\uFF0C\u91CD\u70B9\u7A81\u51FA\n- \u7528\u5217\u8868\u5448\u73B0\u5173\u952E\u4FE1\u606F\n- \u8BED\u6C14\u793C\u8C8C\u81EA\u7136\uFF0C\u50CF\u4E13\u4E1A\u52A9\u624B\n- \u6709\u7ED3\u8BBA\u65F6\u76F4\u63A5\u7ED9\u51FA\uFF0C\u9700\u8981\u8865\u5145\u4FE1\u606F\u65F6\u7B80\u5355\u8BE2\u95EE";
 /**
  * Agent 引擎
  * 负责执行 AI 对话循环，处理工具调用
@@ -368,18 +378,43 @@ declare class Agent {
     private mcpManager;
     private systemPrompt;
     private maxIterations;
+    private immediateResultMatchers;
+    private parallelToolCalls;
+    private enableThinkingPhase;
     constructor(model: LanguageModel, mcpManager: MCPManager, options?: {
         systemPrompt?: string;
         maxIterations?: number;
+        immediateResultMatchers?: ImmediateResultMatcher[];
+        parallelToolCalls?: boolean;
+        enableThinkingPhase?: boolean;
     });
+    /**
+     * 生成工具描述文本（用于思考阶段）
+     */
+    private generateToolsDescription;
+    /**
+     * 思考阶段的系统提示词
+     */
+    private readonly THINKING_PHASE_PROMPT;
+    /**
+     * 检查工具返回结果是否匹配即时结果匹配器
+     * @param result 工具返回的结果
+     * @returns 如果匹配返回 true，否则返回 false
+     */
+    private matchImmediateResult;
     /**
      * 将 MCP 工具转换为 Vercel AI SDK 格式
      */
     private convertMCPToolsToAITools;
     /**
-     * 简单的 JSON Schema 到 Zod 转换
+     * JSON Schema 到 Zod 的完整递归转换
+     * 支持嵌套对象、对象数组、枚举等所有常见类型
      */
     private jsonSchemaToZod;
+    /**
+     * 递归转换 JSON Schema 节点为 Zod 类型
+     */
+    private convertSchemaToZod;
     /**
      * 执行对话
      */
@@ -414,10 +449,22 @@ declare class PromptBasedAgent {
     private mcpManager;
     private systemPrompt;
     private maxIterations;
+    private immediateResultMatchers;
+    private parallelToolCalls;
+    private enableThinkingPhase;
     constructor(model: LanguageModel, mcpManager: MCPManager, options?: {
         systemPrompt?: string;
         maxIterations?: number;
+        immediateResultMatchers?: ImmediateResultMatcher[];
+        parallelToolCalls?: boolean;
+        enableThinkingPhase?: boolean;
     });
+    /**
+     * 检查工具返回结果是否匹配即时结果匹配器
+     * @param result 工具返回的结果
+     * @returns 如果匹配返回 true，否则返回 false
+     */
+    private matchImmediateResult;
     /**
      * 生成工具列表描述
      */
@@ -453,4 +500,4 @@ declare class PromptBasedAgent {
     }): AsyncGenerator<MCPLinkEvent>;
 }
 
-export { Agent, type ChatCallbacks, type ChatResult, DEFAULT_SYSTEM_PROMPT, MCPLink, type MCPLinkConfig, type MCPLinkEvent, type MCPLinkEventData, MCPLinkEventType, MCPManager, type MCPServerConfig, type MCPServerConfigSSE, type MCPServerConfigStdio, type MCPServerStatus, type MCPTool, type Message, type MessageRole, PromptBasedAgent, type TodoItem, type TodoItemStatus, type TodoList, type ToolCall, type ToolResult };
+export { Agent, type ChatCallbacks, type ChatResult, DEFAULT_SYSTEM_PROMPT, type ImmediateResultMatcher, MCPLink, type MCPLinkConfig, type MCPLinkEvent, type MCPLinkEventData, MCPLinkEventType, MCPManager, type MCPServerConfig, type MCPServerConfigSSE, type MCPServerConfigStdio, type MCPServerStatus, type MCPTool, type Message, type MessageRole, PromptBasedAgent, type ToolCall, type ToolResult };