@amical/aiagent-sdk 0.1.0 → 0.1.2

package/src/agent/Agent.ts

@@ -1,566 +0,0 @@
-import { IAgentRuntime } from '../runtime';
-import { IApiHandler, ApiStreamChunk } from '../llm/index';
-import { IPromptManager } from '../prompt/index';
-import { IMemoryManager, AgentMessage } from '../memory/index';
-import { IMcpManager } from '../mcp/index';
-import { parseAssistantMessageV2, ToolUse, toolParamNames } from '../parser/index';
-import { AgentDefaultTool, setDynamicToolUseNames, TOOL_RESULT_BLOCK } from '../shared/tools';
-import matter from 'gray-matter';
-
-/**
- * Agent 初始化配置依赖
- */
-export interface AgentOptions {
-  runtime: IAgentRuntime;
-  api: IApiHandler;
-  promptManager: IPromptManager;
-  memory: IMemoryManager;
-  mcp?: IMcpManager;
-  customTools?: any[];
-  allowDefaultTools?: string[];
-  customInstructions?: string;
-  /** 最大循环次数，默认150 */
-  maxLoopCount?: number;
-}
-
-/**
- * Agent 状态
- */
-export enum AgentState {
-  IDLE = 'idle',
-  THINKING = 'thinking',
-  WAITING_FOR_USER = 'waiting_for_user',
-  FINISHED = 'finished',
-  ERROR = 'error',
-}
-
-/**
- * ClineAgent 核心逻辑循环状态机
- * 它独立于 VS Code 或 Node 环境，纯粹通过依赖注入的方式执行工作流。
- */
-export class ClineAgent {
-  private runtime: IAgentRuntime;
-  private api: IApiHandler;
-  private promptManager: IPromptManager;
-  private memory: IMemoryManager;
-  private mcp?: IMcpManager;
-
-  public state: AgentState = AgentState.IDLE;
-  public allToolParamNames: string[] = [...toolParamNames];
-  /** 循环计数器 */
-  private loopCount: number = 0;
-  /** 最大循环次数 */
-  private maxLoopCount: number;
-
-  skillDir: string;
-  customTools: any[];
-  customToolNames: Set<string>;
-  allowDefaultTools?: string[];
-  customInstructions?: string;
-
-  sysPrompt?: string;
-  noUseTollSysPrompt?: string;
-
-  // 简单的事件回调系统，供宿主环境注册以监听 Agent 内部事件
-  public onStateChange?: (state: AgentState) => void;
-  public onStreamChunk?: (chunk: ApiStreamChunk) => void;
-  public onToolCall?: (tool: ToolUse) => Promise<any | void>;
-
-  constructor(options: AgentOptions) {
-    this.runtime = options.runtime;
-    this.api = options.api;
-    this.promptManager = options.promptManager;
-    this.memory = options.memory;
-    this.mcp = options.mcp;
-    this.maxLoopCount = options.maxLoopCount ?? 150;
-    this.customTools = options.customTools || [];
-    this.allowDefaultTools = options.allowDefaultTools;
-    if (this.allowDefaultTools && !this.allowDefaultTools.includes(AgentDefaultTool.ATTEMPT)) {
-      this.allowDefaultTools.push(AgentDefaultTool.ATTEMPT);
-    }
-    this.customInstructions = options.customInstructions;
-    if (this.customTools.length > 0) {
-      const dynamicToolNames = new Set<string>();
-      const customToolParams = new Set<string>();
-      for (const customTool of this.customTools) {
-        dynamicToolNames.add(customTool.name);
-        if (customTool.params) {
-          for (const key in customTool.params) {
-            if (!Object.hasOwn(customTool.params, key)) continue;
-            customToolParams.add(key);
-          }
-        }
-      }
-
-      // 检测和内置工具的重复，抛出错误
-      const defaultToolNames = Object.values(AgentDefaultTool);
-      const overlappingTools = Array.from(dynamicToolNames).filter((name: any) => defaultToolNames.includes(name));
-      if (overlappingTools.length > 0) {
-        throw new Error(`Custom tools and default tools have overlapping names: ${overlappingTools.join(', ')}`);
-      }
-
-      if (dynamicToolNames.size > 0) {
-        setDynamicToolUseNames('customUser', Array.from(dynamicToolNames));
-      }
-      if (customToolParams.size > 0) {
-        this.allToolParamNames = Array.from(new Set([...this.allToolParamNames, ...Array.from(customToolParams)]));
-      }
-    }
-    this.customToolNames = new Set<string>(this.customTools.map((t: any) => t.name));
-    this.skillDir = this.runtime.path.join(this.runtime.env.cwd(), '.agent/skills');
-  }
-
-  private setState(newState: AgentState) {
-    this.state = newState;
-    if (this.onStateChange) {
-      this.onStateChange(newState);
-    }
-  }
-
-  /**
-   * 获取当前 Agent 的系统提示词 
-   * 如果 force 为 true，则强制重新获取系统提示词
-   */
-  public async getPrompt(force = false) {
-    if (!force && this.sysPrompt && this.noUseTollSysPrompt) {
-      return { sysPrompt: this.sysPrompt, noUseTollSysPrompt: this.noUseTollSysPrompt };
-    }
-
-    const context = {
-        cwd: this.runtime.env.cwd(),
-        os: this.runtime.env.osInfo(),
-        shell: this.runtime.env.shell(),
-        skills: await this.getSkills(),
-        customTools: this.customTools,
-        allowDefaultTools: this.allowDefaultTools,
-        customInstructions: this.customInstructions,
-      };
-      // 1. 获取动态系统提示词
-      this.sysPrompt = await this.promptManager.getSystemPrompt(context);
-      this.noUseTollSysPrompt = this.promptManager.getNoUseToolInstructions(context);
-
-      return { sysPrompt: this.sysPrompt, noUseTollSysPrompt: this.noUseTollSysPrompt };
-  }
-
-  /**
-   * 启动一个新的任务
-   */
-  public async startTask(taskDescription: AgentMessage['content']): Promise<void> {
-    this.setState(AgentState.THINKING);
-    try {
-      await this.memory.clear();
-      // 重置循环计数器
-      this.loopCount = 0;
-      await this.memory.appendMessage({
-        role: 'user',
-        content: taskDescription,
-      });
-
-      await this.loop();
-    } catch (err) {
-      this.setState(AgentState.ERROR);
-      throw err;
-    }
-  }
-
-  private async getSkills() {
-    const skillDir = this.skillDir;
-    const skillList: { name: string; description: string }[] = [];
-    if (!this.runtime.fs.fileExists(skillDir)) {
-      return skillList;
-    }
-    const files = await this.runtime.fs.readDir(skillDir);
-    const skills = files.filter((item) => item.isDirectory()).map((item) => item.name);
-
-    if (skills.length === 0) {
-      return [];
-    }
-
-    for (const skill of skills) {
-      try {
-        const skillPath = this.runtime.path.join(skillDir, skill);
-        const files = await this.runtime.fs.readDir(skillPath);
-        const skillFile = files.map((item) => item.isFile()).find((f) => f.toLowerCase() === 'skill.md');
-        if (skillFile) {
-          const mdPath = this.runtime.path.join(skillPath, skillFile);
-          const mdContent = await this.runtime.fs.readFile(mdPath);
-          const parsed = matter(mdContent);
-          if (parsed.data && parsed.data.description) {
-            let description = String(parsed.data.description).trim();
-            description = description.replace(/\s+/g, ' ').trim();
-            if (description) {
-              skillList.push({
-                name: skill,
-                description: description,
-              });
-            }
-          }
-        }
-      } catch (e) {
-        // SKILL.md might not exist or be invalid, ignore
-        return skillList;
-      }
-    }
-  }
-
-  /**
-   * 核心任务循环
-   */
-  private async loop(): Promise<void> {
-    const {sysPrompt, noUseTollSysPrompt} = await this.getPrompt(true);
-    const {customTools, customToolNames} = this;
-    while (this.state === AgentState.THINKING) {
-      this.loopCount++;
-      // 检查是否超过最大循环次数
-      if (this.loopCount > this.maxLoopCount) {
-        await this.memory.appendMessage({
-          role: 'user',
-          content: `[ERROR] You have exceeded the maximum loop limit of ${this.maxLoopCount} times. Please immediately call the agent_attempt_completion tool to return the final result. If you have pending user-specified tools to execute, you may run them first before calling agent_attempt_completion.`,
-        });
-      }
-      // 1.历史记忆
-      const history = await this.memory.getHistory();
-
-      // 2. 调用 API
-      const stream = this.api.createMessage(sysPrompt, history);
-      let fullResponseText = '';
-
-      // 3. 处理流式输出
-      for await (const chunk of stream) {
-        if (chunk.type === 'text' && chunk.text) {
-          fullResponseText += chunk.text;
-        }
-        if (this.onStreamChunk) {
-          this.onStreamChunk(chunk);
-        }
-      }
-
-      // 4. 将助手的完整回复记入 Memory
-      await this.memory.appendMessage({
-        role: 'assistant',
-        content: fullResponseText,
-      });
-
-      // 5. 解析回复中的工具调用
-      const parsedBlocks = parseAssistantMessageV2(fullResponseText, this.allToolParamNames);
-      const toolUses = parsedBlocks.filter((b) => b.type === 'tool_use') as ToolUse[];
-
-      if (toolUses.length === 0) {
-        // 如果没有工具调用，错误
-        await this.memory.appendMessage({
-          role: 'user',
-          content: noUseTollSysPrompt,
-        });
-      } else {
-        // 6. 顺序执行工具（为简化模型，暂时使用 await 顺序执行）
-        const toolResults = [];
-        for (const tool of toolUses) {
-          if (this.onToolCall) {
-            // 如果宿主接管了工具执行或需要拦截请求权限
-            const hookResult = await this.onToolCall(tool);
-            if (hookResult) {
-              toolResults.push({ tool, result: hookResult });
-              continue;
-            }
-          }
-
-          // 外部工具调用
-          if (customToolNames.has(tool.name)) {
-            const customTool = customTools.find((t: any) => t.name === tool.name);
-            if (customTool.onCall) {
-              try {
-                const res = await customTool.onCall(tool.params);
-                toolResults.push({ tool, result: res });
-              } catch (error: any) {
-                toolResults.push({ tool, result: `Error: ${error.message}` });
-              }
-            } else {
-              toolResults.push({ tool, result: `Error: Custom tool "${tool.name}" does not have an onCall method.` });
-            }
-            continue;
-          }
-
-          // 内置逻辑分发（可由 IAgentRuntime 支撑的核心工具）
-          if (tool.name === AgentDefaultTool.ATTEMPT) {
-            toolResults.push({ tool, result: 'Task marked as completed.' });
-            this.setState(AgentState.FINISHED);
-            break; // 中断循环
-          } else if (tool.name === AgentDefaultTool.FILE_READ) {
-            const path = tool.params.path;
-            if (path) {
-              try {
-                const content = await this.runtime.fs.readFile(path);
-                toolResults.push({ tool, result: content });
-              } catch (error: any) {
-                toolResults.push({ tool, result: `Error: ${error.message}` });
-              }
-            } else {
-              toolResults.push({ tool, result: 'Error: No path provided.' });
-            }
-          } else if (tool.name === AgentDefaultTool.BASH) {
-            const cmd = tool.params.command;
-            if (cmd) {
-              const result = await this.runtime.terminal.execCommand(cmd);
-              toolResults.push({ tool, result: `STDOUT: ${result.stdout}\nSTDERR: ${result.stderr}` });
-            }
-          } else if (tool.name === AgentDefaultTool.FILE_NEW) {
-            const path = tool.params.path;
-            const content = tool.params.content;
-            if (path && content !== undefined) {
-              try {
-                await this.runtime.fs.writeFile(path, content);
-                toolResults.push({ tool, result: `Successfully wrote to ${path}` });
-              } catch (error: any) {
-                toolResults.push({ tool, result: `Error: ${error.message}` });
-              }
-            }
-          } else if (tool.name === AgentDefaultTool.LIST_FILES) {
-            const path = tool.params.path || this.runtime.env.cwd();
-            try {
-              const files = await this.runtime.fs.readDir(path);
-              // 输出是 文件还是目录
-              toolResults.push({ tool, result: files.map((f) => f.name + (f.isDirectory() ? '/' : '')).join('\n') });
-            } catch (e: any) {
-              toolResults.push({ tool, result: `Error: ${e.message}` });
-            }
-          } else if (tool.name === AgentDefaultTool.FILE_EDIT) {
-            const path = tool.params.path;
-            const diff = tool.params.diff;
-            if (path && diff) {
-              try {
-                const content = await this.runtime.fs.readFile(path);
-                const newContent = this.applyDiff(content, diff);
-                await this.runtime.fs.writeFile(path, newContent);
-                toolResults.push({ tool, result: `Successfully replaced content in ${path}` });
-              } catch (e: any) {
-                toolResults.push({ tool, result: `Error: ${e.message}` });
-              }
-            } else {
-              toolResults.push({ tool, result: 'Error: Missing path or diff.' });
-            }
-          } else if (tool.name === AgentDefaultTool.FILE_DELETE) {
-            const path = tool.params.path;
-            if (path) {
-              try {
-                await this.runtime.fs.deleteFile(path);
-                toolResults.push({ tool, result: `Successfully deleted ${path}` });
-              } catch (e: any) {
-                toolResults.push({ tool, result: `Error: ${e.message}` });
-              }
-            } else {
-              toolResults.push({ tool, result: 'Error: Missing path.' });
-            }
-          } else if (tool.name === AgentDefaultTool.DIR_DELETE) {
-            const path = tool.params.path;
-            if (path) {
-              try {
-                await this.runtime.fs.deleteDirectory(path);
-                toolResults.push({ tool, result: `Successfully deleted ${path}` });
-              } catch (e: any) {
-                toolResults.push({ tool, result: `Error: ${e.message}` });
-              }
-            } else {
-              toolResults.push({ tool, result: 'Error: Missing path.' });
-            }
-          } else if (tool.name === AgentDefaultTool.ASK) {
-            const question = tool.params.question;
-            toolResults.push({ tool, result: `Asking user: ${question}` });
-            this.setState(AgentState.WAITING_FOR_USER);
-            // 遇到主动提问，立刻跳出循环，等待外部应用 resume()
-            break;
-          } else if (tool.name === AgentDefaultTool.SEARCH) {
-            const path = tool.params.path || this.runtime.env.cwd();
-            const regexStr = tool.params.regex;
-            const filePattern = tool.params.file_pattern;
-            if (regexStr && this.runtime.fs.searchFiles) {
-              try {
-                const result = await this.runtime.fs.searchFiles(path, regexStr, filePattern);
-                toolResults.push({ tool, result: `${result}` });
-              } catch (e: any) {
-                toolResults.push({ tool, result: `Error: ${e.message}` });
-              }
-            } else {
-              toolResults.push({ tool, result: `Error: searchFiles not supported by runtime or missing regex.` });
-            }
-          } else if (tool.name === AgentDefaultTool.MCP_USE) {
-            const serverName = tool.params.server_name;
-            const toolName = tool.params.tool_name;
-            const args = tool.params.arguments;
-            if (this.mcp && serverName && toolName) {
-              try {
-                let parsedArgs = args;
-                if (typeof args === 'string') {
-                  parsedArgs = JSON.parse(args);
-                }
-                const result = await this.mcp.callTool(serverName, toolName, parsedArgs);
-                if (result.isError) {
-                  toolResults.push({ tool, result: `Error executing MCP tool: ${JSON.stringify(result.content)}` });
-                } else {
-                  toolResults.push({ tool, result: `${JSON.stringify(result.content)}` });
-                }
-              } catch (e: any) {
-                toolResults.push({ tool, result: `Error: ${e.message}` });
-              }
-            } else {
-              toolResults.push({ tool, result: `Error: MCP Manager not available or missing parameters.` });
-            }
-          } else if (tool.name === AgentDefaultTool.MCP_ACCESS) {
-            const serverName = tool.params.server_name;
-            const uri = tool.params.uri;
-            if (this.mcp && this.mcp.readResource && serverName && uri) {
-              try {
-                const result = await this.mcp.readResource(serverName, uri);
-                toolResults.push({ tool, result: `${JSON.stringify(result)}` });
-              } catch (e: any) {
-                toolResults.push({ tool, result: `Error: ${e.message}` });
-              }
-            } else {
-              toolResults.push({ tool, result: `Error: MCP Manager not available, readResource not implemented, or missing parameters.` });
-            }
-          } else if (tool.name === AgentDefaultTool.MCP_DOCS) {
-            // 提供一个简短的关于创建和使用 MCP 的静态文档，供 LLM 了解如何构建和提供 MCP
-            const docs = `MCP (Model Context Protocol) is used to connect to external tools/resources. To build one, typically you create an HTTP/stdio server responding to MCP defined JSON-RPC messages...`;
-            toolResults.push({ tool, result: docs });
-          } else if (tool.name === AgentDefaultTool.LIST_CODE_DEF) {
-            // 简单的 LIST_CODE_DEF 实现
-            const path = tool.params.path || this.runtime.env.cwd();
-            try {
-              const result = await this.runtime.fs.searchFiles?.(path, '^(class|function|const|let|var|export|interface|type)\\s+\\w+', '*.ts|*.js');
-              toolResults.push({ tool, result: `${result ? result : 'No definitions found or searchFiles not supported.'}` });
-            } catch (e: any) {
-              toolResults.push({ tool, result: `Error: ${e.message}` });
-            }
-          } else if (tool.name === AgentDefaultTool.WEB_FETCH) {
-            const url = tool.params.url;
-            if (url) {
-              try {
-                const res = await fetch(url);
-                const text = await res.text();
-                // 简单截断防止超长
-                toolResults.push({ tool, result: text.substring(0, 3000) });
-              } catch (e: any) {
-                toolResults.push({ tool, result: `Error: ${e.message}` });
-              }
-            }
-          } else if (tool.name === AgentDefaultTool.TODO) {
-            toolResults.push({ tool, result: `Focus chain updated successfully.` });
-          } else if (tool.name === AgentDefaultTool.PLAN_MODE || tool.name === AgentDefaultTool.ACT_MODE) {
-            toolResults.push({ tool, result: `Mode switch acknowledged.` });
-          } else if (tool.name === AgentDefaultTool.NEW_RULE) {
-            toolResults.push({ tool, result: `Rule recorded in current context.` });
-          } else if (tool.name === AgentDefaultTool.NEW_TASK) {
-            toolResults.push({ tool, result: `New task request received. Terminating current task.` });
-            this.setState(AgentState.FINISHED);
-            break;
-          } else if (tool.name === AgentDefaultTool.LOAD_SKILL) {
-            const skillName = (tool.params as any).skill_name || (tool.params as any).skill;
-            if (!skillName) {
-              toolResults.push({ tool, result: `"Error: Missing required parameter 'skill_name'. Please provide the name of the skill to activate."` });
-            } else {
-              const skillDir = this.skillDir;
-              // 按照约定从本地硬盘查找 skillDir/SKILL.md
-              const skillFilePath = this.runtime.path.join(skillDir, skillName, 'SKILL.md');
-              try {
-                const fileExists = await this.runtime.fs.fileExists(skillFilePath);
-                if (!fileExists) {
-                  toolResults.push({ tool, result: `Error: Skill "${skillName}" not found at ${skillFilePath}.` });
-                } else {
-                  const content = await this.runtime.fs.readFile(skillFilePath);
-                  toolResults.push({
-                    tool,
-                    result: `# Skill "${skillName}" is now active\n\n${content}\n\n---\nIMPORTANT: The skill is now loaded. Do NOT call agent_load_skill again for this task. Simply follow the instructions above to complete the user's request.`,
-                  });
-                }
-              } catch (err: any) {
-                toolResults.push({ tool, result: `Error loading skill "${skillName}": ${err.message}` });
-              }
-            }
-          } else if (tool.name === AgentDefaultTool.BROWSER || tool.name === AgentDefaultTool.WEB_SEARCH || tool.name === AgentDefaultTool.APPLY_PATCH) {
-            // 这些高级工具通常需要宿主环境的深度集成（如 Puppeteer、真实搜索引擎、Patch 引擎等）
-            // 如果宿主没有通过 onToolCall 拦截，默认返回提示信息
-            toolResults.push({ tool, result: `This tool requires specific host integration. If you see this, the host environment has not implemented it yet.` });
-          } else if (
-            tool.name === AgentDefaultTool.CONDENSE ||
-            tool.name === AgentDefaultTool.SUMMARIZE_TASK ||
-            tool.name === AgentDefaultTool.REPORT_BUG ||
-            tool.name === AgentDefaultTool.GENERATE_EXPLANATION
-          ) {
-            // 这些主要起记录/输出作用的工具
-            toolResults.push({ tool, result: `Action logged successfully.` });
-          } else {
-            // 未知或新增工具兜底
-            toolResults.push({ tool, result: `Executed by host or unsupported.` });
-          }
-        }
-
-        // 7. 如果任务未完成，把执行结果告诉 LLM
-        if (this.state === AgentState.THINKING && toolResults.length > 0) {
-          const toolResultsContext = toolResults
-            .map((tr) => `<${TOOL_RESULT_BLOCK} tool_name="${tr.tool.name}">\n${typeof tr.result === 'string' ? tr.result.trim() : JSON.stringify(tr.result)}\n</${TOOL_RESULT_BLOCK}>`)
-            .join('\n\n');
-          // const lastResult = toolResults[toolResults.length - 1];
-          // if (lastResult && lastResult.tool.params.task_progress) {
-          //   toolResultsContext += `\n<task_progress>\n${lastResult.tool.params.task_progress}\n</task_progress>\n`;
-          // }
-          if (this.onStreamChunk) {
-            this.onStreamChunk({ type: 'text', text: toolResultsContext });
-          }
-          await this.memory.appendMessage({
-            role: 'user',
-            content: toolResultsContext,
-          });
-          // 循环将继续，发送新的请求
-        }
-      }
-    }
-  }
-
-  /**
-   * 简单的 Diff 解析与替换辅助方法
-   */
-  private applyDiff(content: string, diff: string): string {
-    const regex = /-------\s*SEARCH\r?\n(.*?)\r?\n=======\r?\n(.*?)\r?\n\+\+\+\+\+\+\+\s*REPLACE/gs;
-    let newContent = content;
-    let match;
-    let matchCount = 0;
-    while ((match = regex.exec(diff)) !== null) {
-      matchCount++;
-      const searchBlock = match[1];
-      const replaceBlock = match[2];
-      const idx = newContent.indexOf(searchBlock);
-      if (idx !== -1) {
-        newContent = newContent.substring(0, idx) + replaceBlock + newContent.substring(idx + searchBlock.length);
-      } else {
-        throw new Error(`Search block not found in file:\n${searchBlock.substring(0, 100)}...`);
-      }
-    }
-    if (matchCount === 0) {
-      throw new Error(`No valid SEARCH/REPLACE blocks found in the diff.`);
-    }
-    return newContent;
-  }
-
-  /**
-   * 恢复被中断的循环（例如等待用户输入/确认后继续）
-   */
-  public async resume(userInput?: string): Promise<void> {
-    if (userInput) {
-      await this.memory.appendMessage({
-        role: 'user',
-        content: userInput,
-      });
-    }
-    this.setState(AgentState.THINKING);
-    await this.loop();
-  }
-
-  public async appendHistory(history: AgentMessage[]) {
-    for (const message of history) {
-      await this.memory.appendMessage(message);
-    }
-  }
-
-  public async getHistory() {
-    return this.memory.getHistory();
-  }
-}

package/src/agent/index.ts

@@ -1 +0,0 @@
-export * from "./Agent";

package/src/index.ts

@@ -1,15 +0,0 @@
-export const VERSION = '0.1.0';
-
-globalThis.AI_SDK_LOG_WARNINGS = false;
-
-export * from './runtime';
-export * from './shared/tools';
-export * from './parser';
-export * from './prompt';
-export * from './llm';
-export * from './mcp';
-export * from './agent';
-export * from './memory';
-
-export { createNodeAgent } from './runtime/nodeRuntime';
-export { createWebAgent } from './runtime/webRuntime';