sema-core 1.0.2
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -0
- package/README.md +150 -0
- package/dist/constants/config.d.ts +20 -0
- package/dist/constants/config.d.ts.map +1 -0
- package/dist/constants/config.js +31 -0
- package/dist/constants/config.js.map +1 -0
- package/dist/constants/message.d.ts +10 -0
- package/dist/constants/message.d.ts.map +1 -0
- package/dist/constants/message.js +25 -0
- package/dist/constants/message.js.map +1 -0
- package/dist/constants/product.d.ts +5 -0
- package/dist/constants/product.d.ts.map +1 -0
- package/dist/constants/product.js +8 -0
- package/dist/constants/product.js.map +1 -0
- package/dist/core/Conversation.d.ts +16 -0
- package/dist/core/Conversation.d.ts.map +1 -0
- package/dist/core/Conversation.js +226 -0
- package/dist/core/Conversation.js.map +1 -0
- package/dist/core/RunTools.d.ts +11 -0
- package/dist/core/RunTools.d.ts.map +1 -0
- package/dist/core/RunTools.js +244 -0
- package/dist/core/RunTools.js.map +1 -0
- package/dist/core/SemaCore.d.ts +49 -0
- package/dist/core/SemaCore.d.ts.map +1 -0
- package/dist/core/SemaCore.js +94 -0
- package/dist/core/SemaCore.js.map +1 -0
- package/dist/core/SemaEngine.d.ts +54 -0
- package/dist/core/SemaEngine.d.ts.map +1 -0
- package/dist/core/SemaEngine.js +335 -0
- package/dist/core/SemaEngine.js.map +1 -0
- package/dist/events/EventSystem.d.ts +42 -0
- package/dist/events/EventSystem.d.ts.map +1 -0
- package/dist/events/EventSystem.js +134 -0
- package/dist/events/EventSystem.js.map +1 -0
- package/dist/events/types.d.ts +300 -0
- package/dist/events/types.d.ts.map +1 -0
- package/dist/events/types.js +4 -0
- package/dist/events/types.js.map +1 -0
- package/dist/index.d.ts +2 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +6 -0
- package/dist/index.js.map +1 -0
- package/dist/manager/ConfManager.d.ts +76 -0
- package/dist/manager/ConfManager.d.ts.map +1 -0
- package/dist/manager/ConfManager.js +329 -0
- package/dist/manager/ConfManager.js.map +1 -0
- package/dist/manager/ModelManager.d.ts +51 -0
- package/dist/manager/ModelManager.d.ts.map +1 -0
- package/dist/manager/ModelManager.js +291 -0
- package/dist/manager/ModelManager.js.map +1 -0
- package/dist/manager/PermissionManager.d.ts +16 -0
- package/dist/manager/PermissionManager.d.ts.map +1 -0
- package/dist/manager/PermissionManager.js +308 -0
- package/dist/manager/PermissionManager.js.map +1 -0
- package/dist/manager/StateManager.d.ts +164 -0
- package/dist/manager/StateManager.d.ts.map +1 -0
- package/dist/manager/StateManager.js +357 -0
- package/dist/manager/StateManager.js.map +1 -0
- package/dist/services/agents/agentsManager.d.ts +112 -0
- package/dist/services/agents/agentsManager.d.ts.map +1 -0
- package/dist/services/agents/agentsManager.js +394 -0
- package/dist/services/agents/agentsManager.js.map +1 -0
- package/dist/services/agents/defaultBuiltInAgentsConfs.d.ts +3 -0
- package/dist/services/agents/defaultBuiltInAgentsConfs.d.ts.map +1 -0
- package/dist/services/agents/defaultBuiltInAgentsConfs.js +102 -0
- package/dist/services/agents/defaultBuiltInAgentsConfs.js.map +1 -0
- package/dist/services/agents/genSystemPrompt.d.ts +23 -0
- package/dist/services/agents/genSystemPrompt.d.ts.map +1 -0
- package/dist/services/agents/genSystemPrompt.js +189 -0
- package/dist/services/agents/genSystemPrompt.js.map +1 -0
- package/dist/services/agents/prompt.d.ts +12 -0
- package/dist/services/agents/prompt.d.ts.map +1 -0
- package/dist/services/agents/prompt.js +162 -0
- package/dist/services/agents/prompt.js.map +1 -0
- package/dist/services/api/adapt/anthropic.d.ts +10 -0
- package/dist/services/api/adapt/anthropic.d.ts.map +1 -0
- package/dist/services/api/adapt/anthropic.js +233 -0
- package/dist/services/api/adapt/anthropic.js.map +1 -0
- package/dist/services/api/adapt/openai.d.ts +8 -0
- package/dist/services/api/adapt/openai.d.ts.map +1 -0
- package/dist/services/api/adapt/openai.js +412 -0
- package/dist/services/api/adapt/openai.js.map +1 -0
- package/dist/services/api/adapt/util.d.ts +7 -0
- package/dist/services/api/adapt/util.d.ts.map +1 -0
- package/dist/services/api/adapt/util.js +24 -0
- package/dist/services/api/adapt/util.js.map +1 -0
- package/dist/services/api/apiUtil.d.ts +10 -0
- package/dist/services/api/apiUtil.d.ts.map +1 -0
- package/dist/services/api/apiUtil.js +256 -0
- package/dist/services/api/apiUtil.js.map +1 -0
- package/dist/services/api/cache.d.ts +11 -0
- package/dist/services/api/cache.d.ts.map +1 -0
- package/dist/services/api/cache.js +128 -0
- package/dist/services/api/cache.js.map +1 -0
- package/dist/services/api/queryLLM.d.ts +18 -0
- package/dist/services/api/queryLLM.d.ts.map +1 -0
- package/dist/services/api/queryLLM.js +147 -0
- package/dist/services/api/queryLLM.js.map +1 -0
- package/dist/services/command/runCommand.d.ts +18 -0
- package/dist/services/command/runCommand.d.ts.map +1 -0
- package/dist/services/command/runCommand.js +149 -0
- package/dist/services/command/runCommand.js.map +1 -0
- package/dist/services/mcp/MCPClient.d.ts +35 -0
- package/dist/services/mcp/MCPClient.d.ts.map +1 -0
- package/dist/services/mcp/MCPClient.js +166 -0
- package/dist/services/mcp/MCPClient.js.map +1 -0
- package/dist/services/mcp/MCPManager.d.ts +140 -0
- package/dist/services/mcp/MCPManager.d.ts.map +1 -0
- package/dist/services/mcp/MCPManager.js +650 -0
- package/dist/services/mcp/MCPManager.js.map +1 -0
- package/dist/services/mcp/MCPToolAdapter.d.ts +22 -0
- package/dist/services/mcp/MCPToolAdapter.d.ts.map +1 -0
- package/dist/services/mcp/MCPToolAdapter.js +177 -0
- package/dist/services/mcp/MCPToolAdapter.js.map +1 -0
- package/dist/services/plugins/customCommands.d.ts +29 -0
- package/dist/services/plugins/customCommands.d.ts.map +1 -0
- package/dist/services/plugins/customCommands.js +227 -0
- package/dist/services/plugins/customCommands.js.map +1 -0
- package/dist/services/skill/skillLoader.d.ts +17 -0
- package/dist/services/skill/skillLoader.d.ts.map +1 -0
- package/dist/services/skill/skillLoader.js +122 -0
- package/dist/services/skill/skillLoader.js.map +1 -0
- package/dist/services/skill/skillParser.d.ts +15 -0
- package/dist/services/skill/skillParser.d.ts.map +1 -0
- package/dist/services/skill/skillParser.js +66 -0
- package/dist/services/skill/skillParser.js.map +1 -0
- package/dist/services/skill/skillRegistry.d.ts +34 -0
- package/dist/services/skill/skillRegistry.d.ts.map +1 -0
- package/dist/services/skill/skillRegistry.js +98 -0
- package/dist/services/skill/skillRegistry.js.map +1 -0
- package/dist/tools/AskUserQuestion/AskUserQuestion.d.ts +179 -0
- package/dist/tools/AskUserQuestion/AskUserQuestion.d.ts.map +1 -0
- package/dist/tools/AskUserQuestion/AskUserQuestion.js +165 -0
- package/dist/tools/AskUserQuestion/AskUserQuestion.js.map +1 -0
- package/dist/tools/AskUserQuestion/prompt.d.ts +3 -0
- package/dist/tools/AskUserQuestion/prompt.d.ts.map +1 -0
- package/dist/tools/AskUserQuestion/prompt.js +17 -0
- package/dist/tools/AskUserQuestion/prompt.js.map +1 -0
- package/dist/tools/Bash/Bash.d.ts +75 -0
- package/dist/tools/Bash/Bash.d.ts.map +1 -0
- package/dist/tools/Bash/Bash.js +200 -0
- package/dist/tools/Bash/Bash.js.map +1 -0
- package/dist/tools/Bash/prompt.d.ts +8 -0
- package/dist/tools/Bash/prompt.d.ts.map +1 -0
- package/dist/tools/Bash/prompt.js +167 -0
- package/dist/tools/Bash/prompt.js.map +1 -0
- package/dist/tools/Bash/utils.d.ts +5 -0
- package/dist/tools/Bash/utils.d.ts.map +1 -0
- package/dist/tools/Bash/utils.js +21 -0
- package/dist/tools/Bash/utils.js.map +1 -0
- package/dist/tools/Edit/Edit.d.ts +110 -0
- package/dist/tools/Edit/Edit.d.ts.map +1 -0
- package/dist/tools/Edit/Edit.js +213 -0
- package/dist/tools/Edit/Edit.js.map +1 -0
- package/dist/tools/Edit/prompt.d.ts +3 -0
- package/dist/tools/Edit/prompt.d.ts.map +1 -0
- package/dist/tools/Edit/prompt.js +15 -0
- package/dist/tools/Edit/prompt.js.map +1 -0
- package/dist/tools/Edit/utils.d.ts +10 -0
- package/dist/tools/Edit/utils.d.ts.map +1 -0
- package/dist/tools/Edit/utils.js +51 -0
- package/dist/tools/Edit/utils.js.map +1 -0
- package/dist/tools/ExitPlanMode/ExitPlanMode.d.ts +42 -0
- package/dist/tools/ExitPlanMode/ExitPlanMode.d.ts.map +1 -0
- package/dist/tools/ExitPlanMode/ExitPlanMode.js +157 -0
- package/dist/tools/ExitPlanMode/ExitPlanMode.js.map +1 -0
- package/dist/tools/ExitPlanMode/prompt.d.ts +3 -0
- package/dist/tools/ExitPlanMode/prompt.d.ts.map +1 -0
- package/dist/tools/ExitPlanMode/prompt.js +28 -0
- package/dist/tools/ExitPlanMode/prompt.js.map +1 -0
- package/dist/tools/Glob/Glob.d.ts +44 -0
- package/dist/tools/Glob/Glob.d.ts.map +1 -0
- package/dist/tools/Glob/Glob.js +113 -0
- package/dist/tools/Glob/Glob.js.map +1 -0
- package/dist/tools/Glob/prompt.d.ts +3 -0
- package/dist/tools/Glob/prompt.d.ts.map +1 -0
- package/dist/tools/Glob/prompt.js +11 -0
- package/dist/tools/Glob/prompt.js.map +1 -0
- package/dist/tools/Grep/Grep.d.ts +106 -0
- package/dist/tools/Grep/Grep.d.ts.map +1 -0
- package/dist/tools/Grep/Grep.js +301 -0
- package/dist/tools/Grep/Grep.js.map +1 -0
- package/dist/tools/Grep/prompt.d.ts +3 -0
- package/dist/tools/Grep/prompt.d.ts.map +1 -0
- package/dist/tools/Grep/prompt.js +17 -0
- package/dist/tools/Grep/prompt.js.map +1 -0
- package/dist/tools/NotebookEdit/NotebookEdit.d.ts +98 -0
- package/dist/tools/NotebookEdit/NotebookEdit.d.ts.map +1 -0
- package/dist/tools/NotebookEdit/NotebookEdit.js +228 -0
- package/dist/tools/NotebookEdit/NotebookEdit.js.map +1 -0
- package/dist/tools/NotebookEdit/prompt.d.ts +3 -0
- package/dist/tools/NotebookEdit/prompt.d.ts.map +1 -0
- package/dist/tools/NotebookEdit/prompt.js +6 -0
- package/dist/tools/NotebookEdit/prompt.js.map +1 -0
- package/dist/tools/Read/Read.d.ts +107 -0
- package/dist/tools/Read/Read.d.ts.map +1 -0
- package/dist/tools/Read/Read.js +175 -0
- package/dist/tools/Read/Read.js.map +1 -0
- package/dist/tools/Read/prompt.d.ts +4 -0
- package/dist/tools/Read/prompt.d.ts.map +1 -0
- package/dist/tools/Read/prompt.js +21 -0
- package/dist/tools/Read/prompt.js.map +1 -0
- package/dist/tools/Skill/Skill.d.ts +62 -0
- package/dist/tools/Skill/Skill.d.ts.map +1 -0
- package/dist/tools/Skill/Skill.js +142 -0
- package/dist/tools/Skill/Skill.js.map +1 -0
- package/dist/tools/Skill/prompt.d.ts +3 -0
- package/dist/tools/Skill/prompt.d.ts.map +1 -0
- package/dist/tools/Skill/prompt.js +58 -0
- package/dist/tools/Skill/prompt.js.map +1 -0
- package/dist/tools/Task/Task.d.ts +59 -0
- package/dist/tools/Task/Task.d.ts.map +1 -0
- package/dist/tools/Task/Task.js +213 -0
- package/dist/tools/Task/Task.js.map +1 -0
- package/dist/tools/Task/prompt.d.ts +6 -0
- package/dist/tools/Task/prompt.d.ts.map +1 -0
- package/dist/tools/Task/prompt.js +75 -0
- package/dist/tools/Task/prompt.js.map +1 -0
- package/dist/tools/TodoWrite/TodoWrite.d.ts +92 -0
- package/dist/tools/TodoWrite/TodoWrite.d.ts.map +1 -0
- package/dist/tools/TodoWrite/TodoWrite.js +101 -0
- package/dist/tools/TodoWrite/TodoWrite.js.map +1 -0
- package/dist/tools/TodoWrite/prompt.d.ts +3 -0
- package/dist/tools/TodoWrite/prompt.d.ts.map +1 -0
- package/dist/tools/TodoWrite/prompt.js +187 -0
- package/dist/tools/TodoWrite/prompt.js.map +1 -0
- package/dist/tools/Write/Write.d.ts +77 -0
- package/dist/tools/Write/Write.d.ts.map +1 -0
- package/dist/tools/Write/Write.js +189 -0
- package/dist/tools/Write/Write.js.map +1 -0
- package/dist/tools/Write/prompt.d.ts +3 -0
- package/dist/tools/Write/prompt.d.ts.map +1 -0
- package/dist/tools/Write/prompt.js +13 -0
- package/dist/tools/Write/prompt.js.map +1 -0
- package/dist/tools/base/Tool.d.ts +36 -0
- package/dist/tools/base/Tool.d.ts.map +1 -0
- package/dist/tools/base/Tool.js +3 -0
- package/dist/tools/base/Tool.js.map +1 -0
- package/dist/tools/base/tools.d.ts +9 -0
- package/dist/tools/base/tools.d.ts.map +1 -0
- package/dist/tools/base/tools.js +105 -0
- package/dist/tools/base/tools.js.map +1 -0
- package/dist/types/agent.d.ts +30 -0
- package/dist/types/agent.d.ts.map +1 -0
- package/dist/types/agent.js +3 -0
- package/dist/types/agent.js.map +1 -0
- package/dist/types/command.d.ts +44 -0
- package/dist/types/command.d.ts.map +1 -0
- package/dist/types/command.js +6 -0
- package/dist/types/command.js.map +1 -0
- package/dist/types/config.d.ts +14 -0
- package/dist/types/config.d.ts.map +1 -0
- package/dist/types/config.js +3 -0
- package/dist/types/config.js.map +1 -0
- package/dist/types/errors.d.ts +18 -0
- package/dist/types/errors.d.ts.map +1 -0
- package/dist/types/errors.js +33 -0
- package/dist/types/errors.js.map +1 -0
- package/dist/types/index.d.ts +91 -0
- package/dist/types/index.d.ts.map +1 -0
- package/dist/types/index.js +18 -0
- package/dist/types/index.js.map +1 -0
- package/dist/types/mcp.d.ts +127 -0
- package/dist/types/mcp.d.ts.map +1 -0
- package/dist/types/mcp.js +6 -0
- package/dist/types/mcp.js.map +1 -0
- package/dist/types/message.d.ts +31 -0
- package/dist/types/message.d.ts.map +1 -0
- package/dist/types/message.js +3 -0
- package/dist/types/message.js.map +1 -0
- package/dist/types/model.d.ts +21 -0
- package/dist/types/model.d.ts.map +1 -0
- package/dist/types/model.js +3 -0
- package/dist/types/model.js.map +1 -0
- package/dist/types/notebook.d.ts +57 -0
- package/dist/types/notebook.d.ts.map +1 -0
- package/dist/types/notebook.js +3 -0
- package/dist/types/notebook.js.map +1 -0
- package/dist/types/skill.d.ts +34 -0
- package/dist/types/skill.d.ts.map +1 -0
- package/dist/types/skill.js +8 -0
- package/dist/types/skill.js.map +1 -0
- package/dist/types/uuid.d.ts +2 -0
- package/dist/types/uuid.d.ts.map +1 -0
- package/dist/types/uuid.js +3 -0
- package/dist/types/uuid.js.map +1 -0
- package/dist/util/adapter.d.ts +11 -0
- package/dist/util/adapter.d.ts.map +1 -0
- package/dist/util/adapter.js +75 -0
- package/dist/util/adapter.js.map +1 -0
- package/dist/util/agentStats.d.ts +11 -0
- package/dist/util/agentStats.d.ts.map +1 -0
- package/dist/util/agentStats.js +60 -0
- package/dist/util/agentStats.js.map +1 -0
- package/dist/util/cacheLLM.d.ts +47 -0
- package/dist/util/cacheLLM.d.ts.map +1 -0
- package/dist/util/cacheLLM.js +154 -0
- package/dist/util/cacheLLM.js.map +1 -0
- package/dist/util/commands.d.ts +16 -0
- package/dist/util/commands.d.ts.map +1 -0
- package/dist/util/commands.js +216 -0
- package/dist/util/commands.js.map +1 -0
- package/dist/util/compact.d.ts +25 -0
- package/dist/util/compact.d.ts.map +1 -0
- package/dist/util/compact.js +451 -0
- package/dist/util/compact.js.map +1 -0
- package/dist/util/cwd.d.ts +44 -0
- package/dist/util/cwd.d.ts.map +1 -0
- package/dist/util/cwd.js +84 -0
- package/dist/util/cwd.js.map +1 -0
- package/dist/util/diff.d.ts +10 -0
- package/dist/util/diff.d.ts.map +1 -0
- package/dist/util/diff.js +127 -0
- package/dist/util/diff.js.map +1 -0
- package/dist/util/directory.d.ts +37 -0
- package/dist/util/directory.d.ts.map +1 -0
- package/dist/util/directory.js +137 -0
- package/dist/util/directory.js.map +1 -0
- package/dist/util/env.d.ts +5 -0
- package/dist/util/env.d.ts.map +1 -0
- package/dist/util/env.js +39 -0
- package/dist/util/env.js.map +1 -0
- package/dist/util/exec.d.ts +6 -0
- package/dist/util/exec.d.ts.map +1 -0
- package/dist/util/exec.js +43 -0
- package/dist/util/exec.js.map +1 -0
- package/dist/util/file.d.ts +61 -0
- package/dist/util/file.d.ts.map +1 -0
- package/dist/util/file.js +365 -0
- package/dist/util/file.js.map +1 -0
- package/dist/util/filePermission.d.ts +14 -0
- package/dist/util/filePermission.d.ts.map +1 -0
- package/dist/util/filePermission.js +45 -0
- package/dist/util/filePermission.js.map +1 -0
- package/dist/util/fileReference.d.ts +40 -0
- package/dist/util/fileReference.d.ts.map +1 -0
- package/dist/util/fileReference.js +343 -0
- package/dist/util/fileReference.js.map +1 -0
- package/dist/util/format.d.ts +9 -0
- package/dist/util/format.d.ts.map +1 -0
- package/dist/util/format.js +61 -0
- package/dist/util/format.js.map +1 -0
- package/dist/util/frontmatter.d.ts +35 -0
- package/dist/util/frontmatter.d.ts.map +1 -0
- package/dist/util/frontmatter.js +119 -0
- package/dist/util/frontmatter.js.map +1 -0
- package/dist/util/git.d.ts +4 -0
- package/dist/util/git.d.ts.map +1 -0
- package/dist/util/git.js +55 -0
- package/dist/util/git.js.map +1 -0
- package/dist/util/history.d.ts +27 -0
- package/dist/util/history.d.ts.map +1 -0
- package/dist/util/history.js +248 -0
- package/dist/util/history.js.map +1 -0
- package/dist/util/log.d.ts +23 -0
- package/dist/util/log.d.ts.map +1 -0
- package/dist/util/log.js +246 -0
- package/dist/util/log.js.map +1 -0
- package/dist/util/logLLM.d.ts +20 -0
- package/dist/util/logLLM.d.ts.map +1 -0
- package/dist/util/logLLM.js +414 -0
- package/dist/util/logLLM.js.map +1 -0
- package/dist/util/message.d.ts +12 -0
- package/dist/util/message.d.ts.map +1 -0
- package/dist/util/message.js +156 -0
- package/dist/util/message.js.map +1 -0
- package/dist/util/model.d.ts +22 -0
- package/dist/util/model.d.ts.map +1 -0
- package/dist/util/model.js +61 -0
- package/dist/util/model.js.map +1 -0
- package/dist/util/notebook.d.ts +14 -0
- package/dist/util/notebook.d.ts.map +1 -0
- package/dist/util/notebook.js +104 -0
- package/dist/util/notebook.js.map +1 -0
- package/dist/util/ripgrep.d.ts +3 -0
- package/dist/util/ripgrep.d.ts.map +1 -0
- package/dist/util/ripgrep.js +187 -0
- package/dist/util/ripgrep.js.map +1 -0
- package/dist/util/rules.d.ts +6 -0
- package/dist/util/rules.d.ts.map +1 -0
- package/dist/util/rules.js +111 -0
- package/dist/util/rules.js.map +1 -0
- package/dist/util/savePath.d.ts +73 -0
- package/dist/util/savePath.d.ts.map +1 -0
- package/dist/util/savePath.js +189 -0
- package/dist/util/savePath.js.map +1 -0
- package/dist/util/secureFile.d.ts +136 -0
- package/dist/util/secureFile.d.ts.map +1 -0
- package/dist/util/secureFile.js +496 -0
- package/dist/util/secureFile.js.map +1 -0
- package/dist/util/session.d.ts +37 -0
- package/dist/util/session.d.ts.map +1 -0
- package/dist/util/session.js +120 -0
- package/dist/util/session.js.map +1 -0
- package/dist/util/shell.d.ts +40 -0
- package/dist/util/shell.d.ts.map +1 -0
- package/dist/util/shell.js +876 -0
- package/dist/util/shell.js.map +1 -0
- package/dist/util/time.d.ts +24 -0
- package/dist/util/time.d.ts.map +1 -0
- package/dist/util/time.js +53 -0
- package/dist/util/time.js.map +1 -0
- package/dist/util/tokens.d.ts +9 -0
- package/dist/util/tokens.d.ts.map +1 -0
- package/dist/util/tokens.js +79 -0
- package/dist/util/tokens.js.map +1 -0
- package/dist/util/topic.d.ts +9 -0
- package/dist/util/topic.d.ts.map +1 -0
- package/dist/util/topic.js +52 -0
- package/dist/util/topic.js.map +1 -0
- package/package.json +83 -0
|
@@ -0,0 +1,147 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.MAIN_QUERY_TEMPERATURE = void 0;
|
|
4
|
+
exports.queryLLM = queryLLM;
|
|
5
|
+
exports.queryQuick = queryQuick;
|
|
6
|
+
const crypto_1 = require("crypto");
|
|
7
|
+
const ModelManager_1 = require("../../manager/ModelManager");
|
|
8
|
+
const ConfManager_1 = require("../../manager/ConfManager");
|
|
9
|
+
const log_1 = require("../../util/log");
|
|
10
|
+
const logLLM_1 = require("../../util/logLLM");
|
|
11
|
+
const EventSystem_1 = require("../../events/EventSystem");
|
|
12
|
+
const cache_1 = require("./cache");
|
|
13
|
+
// 导入适配层
|
|
14
|
+
const openai_1 = require("./adapt/openai");
|
|
15
|
+
const anthropic_1 = require("./adapt/anthropic");
|
|
16
|
+
const adapter_1 = require("../../util/adapter");
|
|
17
|
+
exports.MAIN_QUERY_TEMPERATURE = 0.7;
|
|
18
|
+
// ============================================================================
|
|
19
|
+
// 主入口函数
|
|
20
|
+
// ============================================================================
|
|
21
|
+
async function queryLLM(messages, systemPromptContent, signal, tools, modelPointer = 'main', disableChunkEvents = false) {
|
|
22
|
+
const modelProfile = (0, ModelManager_1.getModelManager)().getModel(modelPointer);
|
|
23
|
+
if (!modelProfile) {
|
|
24
|
+
throw new Error(`解析模型失败: ${modelPointer}`);
|
|
25
|
+
}
|
|
26
|
+
try {
|
|
27
|
+
const coreConfig = (0, ConfManager_1.getConfManager)().getCoreConfig();
|
|
28
|
+
const shouldUseCache = coreConfig?.enableLLMCache; // 默认不启用
|
|
29
|
+
const shouldStream = coreConfig?.stream !== false; // 默认启用
|
|
30
|
+
const enableThinking = coreConfig?.thinking !== false; // 默认启用
|
|
31
|
+
const emitChunkEvents = !disableChunkEvents && shouldStream !== false; // 默认启用
|
|
32
|
+
if (shouldUseCache) {
|
|
33
|
+
const cachedResponse = await (0, cache_1.tryGetCachedResponse)(messages, systemPromptContent, modelProfile.modelName, shouldStream, enableThinking, emitChunkEvents, signal);
|
|
34
|
+
if (cachedResponse) {
|
|
35
|
+
// 缓存命中时也记录请求和响应
|
|
36
|
+
(0, logLLM_1.logLLMRequest)({ cached: true, model: modelProfile.modelName, messages });
|
|
37
|
+
(0, logLLM_1.logLLMResponse)(cachedResponse);
|
|
38
|
+
return cachedResponse;
|
|
39
|
+
}
|
|
40
|
+
}
|
|
41
|
+
// 根据模型配置的 adapt 字段路由到不同的查询函数,若为空则自动解析
|
|
42
|
+
const adapt = modelProfile.adapt || (0, adapter_1.resolveAdapter)(modelProfile.provider, modelProfile.modelName);
|
|
43
|
+
let result;
|
|
44
|
+
switch (adapt) {
|
|
45
|
+
case 'anthropic':
|
|
46
|
+
result = await (0, anthropic_1.queryAnthropic)(messages, systemPromptContent, tools, signal, modelProfile, enableThinking, emitChunkEvents);
|
|
47
|
+
break;
|
|
48
|
+
case 'openai':
|
|
49
|
+
default:
|
|
50
|
+
result = await (0, openai_1.queryOpenAI)(messages, systemPromptContent, tools, signal, modelProfile, enableThinking, emitChunkEvents);
|
|
51
|
+
break;
|
|
52
|
+
}
|
|
53
|
+
(0, logLLM_1.logLLMResponse)(result);
|
|
54
|
+
// 保存到缓存(仅当内容或工具调用非空时,且未被中断)
|
|
55
|
+
if (shouldUseCache && !signal.aborted) {
|
|
56
|
+
const hasContent = result.message.content.some(block => (block.type === 'text' && block.text.trim().length > 0) ||
|
|
57
|
+
block.type === 'tool_use');
|
|
58
|
+
if (hasContent) {
|
|
59
|
+
(0, cache_1.setCachedResponse)(messages, systemPromptContent, modelProfile.modelName, result, enableThinking);
|
|
60
|
+
(0, log_1.logDebug)(`LLM响应已缓存,当前缓存条目数: ${(0, cache_1.getCacheSize)()}`);
|
|
61
|
+
}
|
|
62
|
+
}
|
|
63
|
+
return result;
|
|
64
|
+
}
|
|
65
|
+
catch (error) {
|
|
66
|
+
if (error instanceof Error && error.name !== 'InterruptedException') {
|
|
67
|
+
emitSessionError(error);
|
|
68
|
+
}
|
|
69
|
+
throw error;
|
|
70
|
+
}
|
|
71
|
+
}
|
|
72
|
+
// ============================================================================
|
|
73
|
+
// 错误处理辅助函数
|
|
74
|
+
// ============================================================================
|
|
75
|
+
function emitSessionError(error, type = 'api_error') {
|
|
76
|
+
const eventBus = (0, EventSystem_1.getEventBus)();
|
|
77
|
+
let errorCode = 'UNKNOWN_ERROR';
|
|
78
|
+
let errorMessage = '未知错误';
|
|
79
|
+
if (error instanceof Error) {
|
|
80
|
+
errorMessage = error.message;
|
|
81
|
+
// 根据错误消息确定错误类型和代码
|
|
82
|
+
if (error.message.includes('cancelled') || error.message.includes('aborted')) {
|
|
83
|
+
return; // 用户取消的情况不需要触发错误事件
|
|
84
|
+
}
|
|
85
|
+
// 检测 API request failed (xxx) 格式的错误
|
|
86
|
+
const apiErrorMatch = error.message.match(/API request failed \((\d{3})\)/);
|
|
87
|
+
if (apiErrorMatch) {
|
|
88
|
+
const statusCode = apiErrorMatch[1];
|
|
89
|
+
errorCode = `API_ERROR_${statusCode}`;
|
|
90
|
+
errorMessage = error.message;
|
|
91
|
+
type = 'api_error';
|
|
92
|
+
}
|
|
93
|
+
else if (error.message.includes('JSON')) {
|
|
94
|
+
errorCode = 'API_RESPONSE_ERROR';
|
|
95
|
+
errorMessage = 'API响应格式错误,无法解析数据';
|
|
96
|
+
type = 'api_error';
|
|
97
|
+
}
|
|
98
|
+
else if (error.message.includes('fetch') || error.message.includes('network')) {
|
|
99
|
+
errorCode = 'NETWORK_ERROR';
|
|
100
|
+
errorMessage = '网络连接错误,请检查网络连接';
|
|
101
|
+
type = 'api_error';
|
|
102
|
+
}
|
|
103
|
+
else if (error.message.includes('401') || error.message.includes('auth')) {
|
|
104
|
+
errorCode = 'AUTH_ERROR';
|
|
105
|
+
errorMessage = 'API认证失败,请检查API密钥';
|
|
106
|
+
type = 'api_error';
|
|
107
|
+
}
|
|
108
|
+
else if (error.message.includes('429') || error.message.includes('rate limit')) {
|
|
109
|
+
errorCode = 'RATE_LIMIT';
|
|
110
|
+
errorMessage = 'API调用频率超限,请稍后重试';
|
|
111
|
+
type = 'api_error';
|
|
112
|
+
}
|
|
113
|
+
else if (error.message.includes('context') || error.message.includes('token')) {
|
|
114
|
+
errorCode = 'CONTEXT_TOO_LONG';
|
|
115
|
+
errorMessage = '上下文长度超出限制';
|
|
116
|
+
type = 'context_length_exceeded';
|
|
117
|
+
}
|
|
118
|
+
}
|
|
119
|
+
const sessionError = {
|
|
120
|
+
type,
|
|
121
|
+
error: {
|
|
122
|
+
code: errorCode,
|
|
123
|
+
message: errorMessage,
|
|
124
|
+
details: error
|
|
125
|
+
}
|
|
126
|
+
};
|
|
127
|
+
eventBus.emit('session:error', sessionError);
|
|
128
|
+
(0, log_1.logError)(`会话错误 [${errorCode}]: ${errorMessage}`);
|
|
129
|
+
}
|
|
130
|
+
// ============================================================================
|
|
131
|
+
// 导出的便捷函数
|
|
132
|
+
// ============================================================================
|
|
133
|
+
async function queryQuick({ systemPrompt = [], userPrompt, signal, }) {
|
|
134
|
+
const messages = [
|
|
135
|
+
{
|
|
136
|
+
message: {
|
|
137
|
+
role: 'user',
|
|
138
|
+
content: [{ type: 'text', text: userPrompt }]
|
|
139
|
+
},
|
|
140
|
+
type: 'user',
|
|
141
|
+
uuid: (0, crypto_1.randomUUID)(),
|
|
142
|
+
},
|
|
143
|
+
];
|
|
144
|
+
return queryLLM(messages, systemPrompt, signal || new AbortController().signal, [], 'quick', true // 禁用流式事件
|
|
145
|
+
);
|
|
146
|
+
}
|
|
147
|
+
//# sourceMappingURL=queryLLM.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"queryLLM.js","sourceRoot":"","sources":["../../../src/services/api/queryLLM.ts"],"names":[],"mappings":";;;AAuBA,4BA0EC;AAmED,gCA6BC;AAjMD,mCAAmC;AAGnC,6DAA4D;AAC5D,2DAA0D;AAC1D,wCAAmD;AACnD,8CAAiE;AACjE,0DAAsD;AAEtD,mCAA+E;AAG/E,QAAQ;AACR,2CAA4C;AAC5C,iDAAkD;AAClD,gDAAmD;AAEtC,QAAA,sBAAsB,GAAG,GAAG,CAAA;AAEzC,+EAA+E;AAC/E,QAAQ;AACR,+EAA+E;AAExE,KAAK,UAAU,QAAQ,CAC5B,QAA4C,EAC5C,mBAA0D,EAC1D,MAAmB,EACnB,KAAa,EACb,eAAiC,MAAM,EACvC,qBAA8B,KAAK;IAEnC,MAAM,YAAY,GAAG,IAAA,8BAAe,GAAE,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAA;IAE7D,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,MAAM,IAAI,KAAK,CAAC,WAAW,YAAY,EAAE,CAAC,CAAA;IAC5C,CAAC;IAED,IAAI,CAAC;QACH,MAAM,UAAU,GAAG,IAAA,4BAAc,GAAE,CAAC,aAAa,EAAE,CAAA;QACnD,MAAM,cAAc,GAAG,UAAU,EAAE,cAAc,CAAA,CAAE,QAAQ;QAC3D,MAAM,YAAY,GAAG,UAAU,EAAE,MAAM,KAAK,KAAK,CAAA,CAAE,OAAO;QAC1D,MAAM,cAAc,GAAG,UAAU,EAAE,QAAQ,KAAK,KAAK,CAAC,CAAE,OAAO;QAC/D,MAAM,eAAe,GAAG,CAAC,kBAAkB,IAAI,YAAY,KAAK,KAAK,CAAC,CAAE,OAAO;QAE/E,IAAI,cAAc,EAAE,CAAC;YACnB,MAAM,cAAc,GAAG,MAAM,IAAA,4BAAoB,EAC/C,QAAQ,EACR,mBAAmB,EACnB,YAAY,CAAC,SAAS,EACtB,YAAY,EACZ,cAAc,EACd,eAAe,EACf,MAAM,CACP,CAAA;YACD,IAAI,cAAc,EAAE,CAAC;gBACnB,gBAAgB;gBAChB,IAAA,sBAAa,EAAC,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,YAAY,CAAC,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAA;gBACxE,IAAA,uBAAc,EAAC,cAAc,CAAC,CAAA;gBAC9B,OAAO,cAAc,CAAA;YACvB,CAAC;QACH,CAAC;QAED,sCAAsC;QACtC,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,IAAI,IAAA,wBAAc,EAAC,YAAY,CAAC,QAAQ,EAAE,YAAY,CAAC,SAAS,CAAC,CAAA;QACjG,IAAI,MAAwB,CAAA;QAE5B,QAAQ,KAAK,EAAE,CAAC;YACd,KAAK,WAAW;gBACd,MAAM,GAAG,MAAM,IAAA,0BAAc,EAAC,QAAQ,EAAE,mBAAmB,EAAE,KAAK,EAAE,MAAM,EAAE,YAAY,EAAE,cAAc,EAAE,eAAe,CAAC,CAAA;gBAC1H,MAAK;YACP,KAAK,QAAQ,CAAC;YACd;gBACE,MAAM,GAAG,MAAM,IAAA,oBAAW,EAAC,QAAQ,EAAE,mBAAmB,EAAE,KAAK,EAAE,MAAM,EAAE,YAAY,EAAE,cAAc,EAAE,eAAe,CAAC,CAAA;gBACvH,MAAK;QACT,CAAC;QAED,IAAA,uBAAc,EAAC,MAAM,CAAC,CAAA;QAEtB,4BAA4B;QAC5B,IAAI,cAAc,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACtC,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CACrD,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC;gBACvD,KAAK,CAAC,IAAI,KAAK,UAAU,CAC1B,CAAA;YACD,IAAI,UAAU,EAAE,CAAC;gBACf,IAAA,yBAAiB,EAAC,QAAQ,EAAE,mBAAmB,EAAE,YAAY,CAAC,SAAS,EAAE,MAAM,EAAE,cAAc,CAAC,CAAA;gBAChG,IAAA,cAAQ,EAAC,qBAAqB,IAAA,oBAAY,GAAE,EAAE,CAAC,CAAA;YACjD,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAA;IACf,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,sBAAsB,EAAE,CAAC;YACpE,gBAAgB,CAAC,KAAK,CAAC,CAAA;QACzB,CAAC;QACD,MAAM,KAAK,CAAA;IACb,CAAC;AACH,CAAC;AAED,+EAA+E;AAC/E,WAAW;AACX,+EAA+E;AAE/E,SAAS,gBAAgB,CAAC,KAAU,EAAE,OAAiC,WAAW;IAChF,MAAM,QAAQ,GAAG,IAAA,yBAAW,GAAE,CAAA;IAE9B,IAAI,SAAS,GAAG,eAAe,CAAA;IAC/B,IAAI,YAAY,GAAG,MAAM,CAAA;IAEzB,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAC3B,YAAY,GAAG,KAAK,CAAC,OAAO,CAAA;QAE5B,kBAAkB;QAClB,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YAC7E,OAAM,CAAC,mBAAmB;QAC5B,CAAC;QAED,oCAAoC;QACpC,MAAM,aAAa,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,gCAAgC,CAAC,CAAA;QAC3E,IAAI,aAAa,EAAE,CAAC;YAClB,MAAM,UAAU,GAAG,aAAa,CAAC,CAAC,CAAC,CAAA;YACnC,SAAS,GAAG,aAAa,UAAU,EAAE,CAAA;YACrC,YAAY,GAAG,KAAK,CAAC,OAAO,CAAA;YAC5B,IAAI,GAAG,WAAW,CAAA;QACpB,CAAC;aAAM,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1C,SAAS,GAAG,oBAAoB,CAAA;YAChC,YAAY,GAAG,kBAAkB,CAAA;YACjC,IAAI,GAAG,WAAW,CAAA;QACpB,CAAC;aAAM,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YAChF,SAAS,GAAG,eAAe,CAAA;YAC3B,YAAY,GAAG,gBAAgB,CAAA;YAC/B,IAAI,GAAG,WAAW,CAAA;QACpB,CAAC;aAAM,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YAC3E,SAAS,GAAG,YAAY,CAAA;YACxB,YAAY,GAAG,kBAAkB,CAAA;YACjC,IAAI,GAAG,WAAW,CAAA;QACpB,CAAC;aAAM,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;YACjF,SAAS,GAAG,YAAY,CAAA;YACxB,YAAY,GAAG,iBAAiB,CAAA;YAChC,IAAI,GAAG,WAAW,CAAA;QACpB,CAAC;aAAM,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAChF,SAAS,GAAG,kBAAkB,CAAA;YAC9B,YAAY,GAAG,WAAW,CAAA;YAC1B,IAAI,GAAG,yBAAyB,CAAA;QAClC,CAAC;IACH,CAAC;IAED,MAAM,YAAY,GAAqB;QACrC,IAAI;QACJ,KAAK,EAAE;YACL,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,YAAY;YACrB,OAAO,EAAE,KAAK;SACf;KACF,CAAA;IAED,QAAQ,CAAC,IAAI,CAAC,eAAe,EAAE,YAAY,CAAC,CAAA;IAC5C,IAAA,cAAQ,EAAC,SAAS,SAAS,MAAM,YAAY,EAAE,CAAC,CAAA;AAClD,CAAC;AAED,+EAA+E;AAC/E,UAAU;AACV,+EAA+E;AAExE,KAAK,UAAU,UAAU,CAAC,EAC/B,YAAY,GAAG,EAAE,EACjB,UAAU,EACV,MAAM,GAMP;IACC,MAAM,QAAQ,GAAG;QACf;YACE,OAAO,EAAE;gBACP,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;aAC9C;YACD,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,IAAA,mBAAU,GAAE;SACnB;KACoC,CAAA;IAEvC,OAAO,QAAQ,CACb,QAAQ,EACR,YAAY,EACZ,MAAM,IAAI,IAAI,eAAe,EAAE,CAAC,MAAM,EACtC,EAAE,EACF,OAAO,EACP,IAAI,CAAC,SAAS;KACf,CAAA;AACH,CAAC"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 系统命令处理模块
|
|
3
|
+
*/
|
|
4
|
+
/**
|
|
5
|
+
* 处理系统命令(如 /clear, /compact 等)
|
|
6
|
+
* @returns true 表示是系统命令且已处理,false 表示不是系统命令 无需进入后续步骤
|
|
7
|
+
*/
|
|
8
|
+
export declare function handleSystemCommand(input: string): Promise<boolean>;
|
|
9
|
+
/**
|
|
10
|
+
* 尝试处理自定义命令
|
|
11
|
+
* @param input 用户输入
|
|
12
|
+
* @returns { processedInput: string, handled: boolean } 处理后的输入和是否被处理标志
|
|
13
|
+
*/
|
|
14
|
+
export declare function tryHandleCustomCommand(input: string): Promise<{
|
|
15
|
+
processedInput: string;
|
|
16
|
+
handled: boolean;
|
|
17
|
+
}>;
|
|
18
|
+
//# sourceMappingURL=runCommand.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"runCommand.d.ts","sourceRoot":"","sources":["../../../src/services/command/runCommand.ts"],"names":[],"mappings":"AAQA;;GAEG;AAEH;;;GAGG;AACH,wBAAsB,mBAAmB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAazE;AAED;;;;GAIG;AACH,wBAAsB,sBAAsB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC;IACnE,cAAc,EAAE,MAAM,CAAC;IACvB,OAAO,EAAE,OAAO,CAAC;CAClB,CAAC,CA2CD"}
|
|
@@ -0,0 +1,149 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.handleSystemCommand = handleSystemCommand;
|
|
4
|
+
exports.tryHandleCustomCommand = tryHandleCustomCommand;
|
|
5
|
+
const log_1 = require("../../util/log");
|
|
6
|
+
const EventSystem_1 = require("../../events/EventSystem");
|
|
7
|
+
const StateManager_1 = require("../../manager/StateManager");
|
|
8
|
+
const compact_1 = require("../../util/compact");
|
|
9
|
+
const errors_1 = require("../../types/errors");
|
|
10
|
+
const customCommands_1 = require("../plugins/customCommands");
|
|
11
|
+
/**
|
|
12
|
+
* 系统命令处理模块
|
|
13
|
+
*/
|
|
14
|
+
/**
|
|
15
|
+
* 处理系统命令(如 /clear, /compact 等)
|
|
16
|
+
* @returns true 表示是系统命令且已处理,false 表示不是系统命令 无需进入后续步骤
|
|
17
|
+
*/
|
|
18
|
+
async function handleSystemCommand(input) {
|
|
19
|
+
switch (input) {
|
|
20
|
+
case '/clear':
|
|
21
|
+
await handleClearCommand();
|
|
22
|
+
return true;
|
|
23
|
+
case '/compact':
|
|
24
|
+
await handleCompactCommand();
|
|
25
|
+
return true;
|
|
26
|
+
default:
|
|
27
|
+
return false;
|
|
28
|
+
}
|
|
29
|
+
}
|
|
30
|
+
/**
|
|
31
|
+
* 尝试处理自定义命令
|
|
32
|
+
* @param input 用户输入
|
|
33
|
+
* @returns { processedInput: string, handled: boolean } 处理后的输入和是否被处理标志
|
|
34
|
+
*/
|
|
35
|
+
async function tryHandleCustomCommand(input) {
|
|
36
|
+
const eventBus = EventSystem_1.EventBus.getInstance();
|
|
37
|
+
// 支持字母、数字、下划线、连字符和冒号分隔符
|
|
38
|
+
const customCommandMatch = input.match(/^\/([a-zA-Z0-9_-]+(?::[a-zA-Z0-9_-]+)*)(?:\s+(.*))?$/);
|
|
39
|
+
if (!customCommandMatch) {
|
|
40
|
+
return { processedInput: input, handled: false };
|
|
41
|
+
}
|
|
42
|
+
const [, commandName, args = ''] = customCommandMatch;
|
|
43
|
+
try {
|
|
44
|
+
const customCommands = await (0, customCommands_1.getCachedCustomCommands)();
|
|
45
|
+
const command = customCommands.find(cmd => cmd.name === commandName);
|
|
46
|
+
if (command) {
|
|
47
|
+
// 找到自定义命令,替换输入内容
|
|
48
|
+
const processedInput = (0, customCommands_1.resolveArguments)(command.content, args);
|
|
49
|
+
(0, log_1.logInfo)(`Custom command '/${commandName}' resolved, content length: ${processedInput.length}`);
|
|
50
|
+
// 发出自定义命令处理事件
|
|
51
|
+
eventBus.emit('command:custom:resolved', {
|
|
52
|
+
commandName,
|
|
53
|
+
displayName: command.displayName,
|
|
54
|
+
description: command.description,
|
|
55
|
+
scope: command.scope,
|
|
56
|
+
args,
|
|
57
|
+
originalInput: input,
|
|
58
|
+
processedContent: processedInput
|
|
59
|
+
});
|
|
60
|
+
return { processedInput, handled: true };
|
|
61
|
+
}
|
|
62
|
+
else {
|
|
63
|
+
// 命令不存在,保持原输入,让 LLM 处理
|
|
64
|
+
(0, log_1.logDebug)(`Custom command '/${commandName}' not found, treating as normal input`);
|
|
65
|
+
return { processedInput: input, handled: false };
|
|
66
|
+
}
|
|
67
|
+
}
|
|
68
|
+
catch (error) {
|
|
69
|
+
(0, log_1.logWarn)(`Failed to process custom command: ${error}`);
|
|
70
|
+
// 出错时保持原输入
|
|
71
|
+
return { processedInput: input, handled: false };
|
|
72
|
+
}
|
|
73
|
+
}
|
|
74
|
+
/**
|
|
75
|
+
* 处理清空命令
|
|
76
|
+
*/
|
|
77
|
+
async function handleClearCommand() {
|
|
78
|
+
(0, log_1.logInfo)('执行清空命令...');
|
|
79
|
+
const eventBus = EventSystem_1.EventBus.getInstance();
|
|
80
|
+
const stateManager = (0, StateManager_1.getStateManager)();
|
|
81
|
+
stateManager.setMessageHistory([]);
|
|
82
|
+
stateManager.updateState('idle');
|
|
83
|
+
eventBus.emit('session:cleared', {
|
|
84
|
+
sessionId: stateManager.getSessionId()
|
|
85
|
+
});
|
|
86
|
+
// 清空所有状态数据
|
|
87
|
+
stateManager.clearAllState();
|
|
88
|
+
}
|
|
89
|
+
/**
|
|
90
|
+
* 处理压缩命令
|
|
91
|
+
*/
|
|
92
|
+
async function handleCompactCommand() {
|
|
93
|
+
(0, log_1.logInfo)('执行压缩命令...');
|
|
94
|
+
const eventBus = EventSystem_1.EventBus.getInstance();
|
|
95
|
+
const stateManager = (0, StateManager_1.getStateManager)();
|
|
96
|
+
const mainAgentState = stateManager.forAgent(StateManager_1.MAIN_AGENT_ID);
|
|
97
|
+
// 获取当前消息历史
|
|
98
|
+
const messages = mainAgentState.getMessageHistory();
|
|
99
|
+
// 如果消息历史为空,无需压缩
|
|
100
|
+
if (messages.length === 0) {
|
|
101
|
+
const errMsg = 'Empty history, skip compact';
|
|
102
|
+
(0, log_1.logInfo)(errMsg);
|
|
103
|
+
const compactExecData = {
|
|
104
|
+
errMsg: errMsg,
|
|
105
|
+
tokenBefore: 0,
|
|
106
|
+
tokenCompact: 0,
|
|
107
|
+
compactRate: 0
|
|
108
|
+
};
|
|
109
|
+
eventBus.emit('compact:exec', compactExecData);
|
|
110
|
+
stateManager.currentAbortController = null;
|
|
111
|
+
mainAgentState.updateState('idle');
|
|
112
|
+
return;
|
|
113
|
+
}
|
|
114
|
+
// 使用共享的 AbortController,这样中断会话时可以中断压缩
|
|
115
|
+
stateManager.currentAbortController = new AbortController();
|
|
116
|
+
try {
|
|
117
|
+
(0, log_1.logDebug)(`开始手动压缩,当前消息数: ${messages.length}`);
|
|
118
|
+
// 直接执行压缩,不检查阈值
|
|
119
|
+
const compactedMessages = await (0, compact_1.compactMessages)(messages, stateManager.currentAbortController);
|
|
120
|
+
// 更新消息历史
|
|
121
|
+
mainAgentState.setMessageHistory(compactedMessages);
|
|
122
|
+
(0, log_1.logDebug)(`压缩完成,压缩后消息数: ${compactedMessages.length}`);
|
|
123
|
+
}
|
|
124
|
+
catch (error) {
|
|
125
|
+
// 用户中断不作为错误处理
|
|
126
|
+
if ((0, errors_1.isInterruptedException)(error)) {
|
|
127
|
+
(0, log_1.logInfo)('压缩操作被用户中断');
|
|
128
|
+
return;
|
|
129
|
+
}
|
|
130
|
+
const errorMsg = `压缩失败: ${error instanceof Error ? error.message : String(error)}`;
|
|
131
|
+
(0, log_1.logError)(errorMsg);
|
|
132
|
+
eventBus.emit('session:error', {
|
|
133
|
+
type: 'compact_error',
|
|
134
|
+
error: {
|
|
135
|
+
code: 'COMPACT_FAILED',
|
|
136
|
+
message: errorMsg,
|
|
137
|
+
details: { error }
|
|
138
|
+
}
|
|
139
|
+
});
|
|
140
|
+
}
|
|
141
|
+
finally {
|
|
142
|
+
// 清理共享的 AbortController
|
|
143
|
+
stateManager.currentAbortController = null;
|
|
144
|
+
// 确保状态恢复为 idle(无论成功、跳过还是失败)
|
|
145
|
+
mainAgentState.updateState('idle');
|
|
146
|
+
}
|
|
147
|
+
(0, log_1.logDebug)('压缩命令执行完成');
|
|
148
|
+
}
|
|
149
|
+
//# sourceMappingURL=runCommand.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"runCommand.js","sourceRoot":"","sources":["../../../src/services/command/runCommand.ts"],"names":[],"mappings":";;AAgBA,kDAaC;AAOD,wDA8CC;AAlFD,wCAAsE;AACtE,0DAAoD;AACpD,6DAA4E;AAC5E,gDAAqD;AAErD,+CAA4D;AAC5D,8DAAsF;AAEtF;;GAEG;AAEH;;;GAGG;AACI,KAAK,UAAU,mBAAmB,CAAC,KAAa;IACrD,QAAQ,KAAK,EAAE,CAAC;QACd,KAAK,QAAQ;YACX,MAAM,kBAAkB,EAAE,CAAC;YAC3B,OAAO,IAAI,CAAC;QAEd,KAAK,UAAU;YACb,MAAM,oBAAoB,EAAE,CAAC;YAC7B,OAAO,IAAI,CAAC;QAEd;YACE,OAAO,KAAK,CAAC;IACjB,CAAC;AACH,CAAC;AAED;;;;GAIG;AACI,KAAK,UAAU,sBAAsB,CAAC,KAAa;IAIxD,MAAM,QAAQ,GAAG,sBAAQ,CAAC,WAAW,EAAE,CAAC;IAExC,wBAAwB;IACxB,MAAM,kBAAkB,GAAG,KAAK,CAAC,KAAK,CAAC,sDAAsD,CAAC,CAAC;IAE/F,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACxB,OAAO,EAAE,cAAc,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;IACnD,CAAC;IAED,MAAM,CAAC,EAAE,WAAW,EAAE,IAAI,GAAG,EAAE,CAAC,GAAG,kBAAkB,CAAC;IAEtD,IAAI,CAAC;QACH,MAAM,cAAc,GAAG,MAAM,IAAA,wCAAuB,GAAE,CAAC;QACvD,MAAM,OAAO,GAAG,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;QAErE,IAAI,OAAO,EAAE,CAAC;YACZ,iBAAiB;YACjB,MAAM,cAAc,GAAG,IAAA,iCAAgB,EAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YAC/D,IAAA,aAAO,EAAC,oBAAoB,WAAW,+BAA+B,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC;YAE/F,cAAc;YACd,QAAQ,CAAC,IAAI,CAAC,yBAAyB,EAAE;gBACvC,WAAW;gBACX,WAAW,EAAE,OAAO,CAAC,WAAW;gBAChC,WAAW,EAAE,OAAO,CAAC,WAAW;gBAChC,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,IAAI;gBACJ,aAAa,EAAE,KAAK;gBACpB,gBAAgB,EAAE,cAAc;aACjC,CAAC,CAAC;YAEH,OAAO,EAAE,cAAc,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC3C,CAAC;aAAM,CAAC;YACN,uBAAuB;YACvB,IAAA,cAAQ,EAAC,oBAAoB,WAAW,uCAAuC,CAAC,CAAC;YACjF,OAAO,EAAE,cAAc,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;QACnD,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAA,aAAO,EAAC,qCAAqC,KAAK,EAAE,CAAC,CAAC;QACtD,WAAW;QACX,OAAO,EAAE,cAAc,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;IACnD,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,kBAAkB;IAC/B,IAAA,aAAO,EAAC,WAAW,CAAC,CAAC;IAErB,MAAM,QAAQ,GAAG,sBAAQ,CAAC,WAAW,EAAE,CAAC;IACxC,MAAM,YAAY,GAAG,IAAA,8BAAe,GAAE,CAAC;IAEvC,YAAY,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAA;IAElC,YAAY,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IAEjC,QAAQ,CAAC,IAAI,CAAC,iBAAiB,EAAE;QAC/B,SAAS,EAAE,YAAY,CAAC,YAAY,EAAE;KACvC,CAAC,CAAC;IAEH,WAAW;IACX,YAAY,CAAC,aAAa,EAAE,CAAC;AAC/B,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,oBAAoB;IACjC,IAAA,aAAO,EAAC,WAAW,CAAC,CAAC;IAErB,MAAM,QAAQ,GAAG,sBAAQ,CAAC,WAAW,EAAE,CAAC;IACxC,MAAM,YAAY,GAAG,IAAA,8BAAe,GAAE,CAAC;IACvC,MAAM,cAAc,GAAG,YAAY,CAAC,QAAQ,CAAC,4BAAa,CAAC,CAAC;IAE5D,WAAW;IACX,MAAM,QAAQ,GAAG,cAAc,CAAC,iBAAiB,EAAE,CAAC;IAEpD,gBAAgB;IAChB,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1B,MAAM,MAAM,GAAG,6BAA6B,CAAA;QAC5C,IAAA,aAAO,EAAC,MAAM,CAAC,CAAC;QAChB,MAAM,eAAe,GAAoB;YACvC,MAAM,EAAE,MAAM;YACd,WAAW,EAAE,CAAC;YACd,YAAY,EAAE,CAAC;YACf,WAAW,EAAE,CAAC;SACf,CAAC;QACF,QAAQ,CAAC,IAAI,CAAC,cAAc,EAAE,eAAe,CAAC,CAAC;QAE/C,YAAY,CAAC,sBAAsB,GAAG,IAAI,CAAC;QAC3C,cAAc,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QACnC,OAAO;IACT,CAAC;IAED,sCAAsC;IACtC,YAAY,CAAC,sBAAsB,GAAG,IAAI,eAAe,EAAE,CAAC;IAE5D,IAAI,CAAC;QACH,IAAA,cAAQ,EAAC,iBAAiB,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAE7C,eAAe;QACf,MAAM,iBAAiB,GAAG,MAAM,IAAA,yBAAe,EAC7C,QAAQ,EACR,YAAY,CAAC,sBAAsB,CACpC,CAAC;QAEF,SAAS;QACT,cAAc,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;QAEpD,IAAA,cAAQ,EAAC,gBAAgB,iBAAiB,CAAC,MAAM,EAAE,CAAC,CAAC;IAEvD,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,cAAc;QACd,IAAI,IAAA,+BAAsB,EAAC,KAAK,CAAC,EAAE,CAAC;YAClC,IAAA,aAAO,EAAC,WAAW,CAAC,CAAC;YACrB,OAAO;QACT,CAAC;QACD,MAAM,QAAQ,GAAG,SAAS,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;QACnF,IAAA,cAAQ,EAAC,QAAQ,CAAC,CAAC;QACnB,QAAQ,CAAC,IAAI,CAAC,eAAe,EAAE;YAC7B,IAAI,EAAE,eAAe;YACrB,KAAK,EAAE;gBACL,IAAI,EAAE,gBAAgB;gBACtB,OAAO,EAAE,QAAQ;gBACjB,OAAO,EAAE,EAAE,KAAK,EAAE;aACnB;SACF,CAAC,CAAC;IACL,CAAC;YAAS,CAAC;QACT,wBAAwB;QACxB,YAAY,CAAC,sBAAsB,GAAG,IAAI,CAAC;QAC3C,4BAA4B;QAC5B,cAAc,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IACrC,CAAC;IAED,IAAA,cAAQ,EAAC,UAAU,CAAC,CAAC;AACvB,CAAC"}
|
|
@@ -0,0 +1,35 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* MCP 客户端 - 负责与单个 MCP Server 的通信
|
|
3
|
+
*/
|
|
4
|
+
import { MCPServerConfig, MCPServerCapabilities, MCPToolResult, MCPServerStatus } from '../../types/mcp';
|
|
5
|
+
export declare class MCPClient {
|
|
6
|
+
private config;
|
|
7
|
+
private client;
|
|
8
|
+
private transport;
|
|
9
|
+
private _status;
|
|
10
|
+
private _capabilities;
|
|
11
|
+
constructor(config: MCPServerConfig);
|
|
12
|
+
get status(): MCPServerStatus;
|
|
13
|
+
get capabilities(): MCPServerCapabilities | null;
|
|
14
|
+
/** 连接超时时间 (ms) */
|
|
15
|
+
private static readonly CONNECT_TIMEOUT;
|
|
16
|
+
/**
|
|
17
|
+
* 连接到 MCP Server
|
|
18
|
+
*/
|
|
19
|
+
connect(): Promise<void>;
|
|
20
|
+
/**
|
|
21
|
+
* 带超时的 Promise 包装
|
|
22
|
+
*/
|
|
23
|
+
private withTimeout;
|
|
24
|
+
/**
|
|
25
|
+
* 断开连接
|
|
26
|
+
*/
|
|
27
|
+
disconnect(): Promise<void>;
|
|
28
|
+
/**
|
|
29
|
+
* 调用 MCP 工具
|
|
30
|
+
*/
|
|
31
|
+
callTool(toolName: string, args: Record<string, unknown>): Promise<MCPToolResult>;
|
|
32
|
+
private createTransport;
|
|
33
|
+
private fetchCapabilities;
|
|
34
|
+
}
|
|
35
|
+
//# sourceMappingURL=MCPClient.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"MCPClient.d.ts","sourceRoot":"","sources":["../../../src/services/mcp/MCPClient.ts"],"names":[],"mappings":"AAAA;;GAEG;AAMH,OAAO,EACL,eAAe,EACf,qBAAqB,EACrB,aAAa,EACb,eAAe,EAEhB,MAAM,iBAAiB,CAAA;AAKxB,qBAAa,SAAS;IAMR,OAAO,CAAC,MAAM;IAL1B,OAAO,CAAC,MAAM,CAAQ;IACtB,OAAO,CAAC,SAAS,CAAyB;IAC1C,OAAO,CAAC,OAAO,CAAkC;IACjD,OAAO,CAAC,aAAa,CAAqC;gBAEtC,MAAM,EAAE,eAAe;IAY3C,IAAI,MAAM,IAAI,eAAe,CAE5B;IAED,IAAI,YAAY,IAAI,qBAAqB,GAAG,IAAI,CAE/C;IAED,kBAAkB;IAClB,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAQ;IAE/C;;OAEG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IA+B9B;;OAEG;YACW,WAAW;IAYzB;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAsCjC;;OAEG;IACG,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,aAAa,CAAC;IAgBvF,OAAO,CAAC,eAAe;YAgCT,iBAAiB;CAUhC"}
|
|
@@ -0,0 +1,166 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* MCP 客户端 - 负责与单个 MCP Server 的通信
|
|
4
|
+
*/
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.MCPClient = void 0;
|
|
7
|
+
const index_js_1 = require("@modelcontextprotocol/sdk/client/index.js");
|
|
8
|
+
const stdio_js_1 = require("@modelcontextprotocol/sdk/client/stdio.js");
|
|
9
|
+
const sse_js_1 = require("@modelcontextprotocol/sdk/client/sse.js");
|
|
10
|
+
const streamableHttp_js_1 = require("@modelcontextprotocol/sdk/client/streamableHttp.js");
|
|
11
|
+
const log_1 = require("../../util/log");
|
|
12
|
+
class MCPClient {
|
|
13
|
+
constructor(config) {
|
|
14
|
+
this.config = config;
|
|
15
|
+
this.transport = null;
|
|
16
|
+
this._status = 'disconnected';
|
|
17
|
+
this._capabilities = null;
|
|
18
|
+
this.client = new index_js_1.Client({
|
|
19
|
+
name: 'sema-core',
|
|
20
|
+
version: '1.0.0'
|
|
21
|
+
}, {
|
|
22
|
+
capabilities: {}
|
|
23
|
+
});
|
|
24
|
+
}
|
|
25
|
+
get status() {
|
|
26
|
+
return this._status;
|
|
27
|
+
}
|
|
28
|
+
get capabilities() {
|
|
29
|
+
return this._capabilities;
|
|
30
|
+
}
|
|
31
|
+
/**
|
|
32
|
+
* 连接到 MCP Server
|
|
33
|
+
*/
|
|
34
|
+
async connect() {
|
|
35
|
+
try {
|
|
36
|
+
this._status = 'connecting';
|
|
37
|
+
// 根据传输类型创建 transport
|
|
38
|
+
this.transport = this.createTransport();
|
|
39
|
+
// 带超时的连接
|
|
40
|
+
await this.withTimeout(this.client.connect(this.transport), MCPClient.CONNECT_TIMEOUT, `连接超时 (${MCPClient.CONNECT_TIMEOUT / 1000}s)`);
|
|
41
|
+
// 获取能力(也需要超时保护)
|
|
42
|
+
await this.withTimeout(this.fetchCapabilities(), MCPClient.CONNECT_TIMEOUT, `获取能力超时 (${MCPClient.CONNECT_TIMEOUT / 1000}s)`);
|
|
43
|
+
this._status = 'connected';
|
|
44
|
+
(0, log_1.logDebug)(`MCP Server [${this.config.name}] 连接成功`);
|
|
45
|
+
}
|
|
46
|
+
catch (error) {
|
|
47
|
+
this._status = 'error';
|
|
48
|
+
const errorMsg = error instanceof Error ? error.message : String(error);
|
|
49
|
+
(0, log_1.logError)(`MCP Server [${this.config.name}] 连接失败: ${errorMsg}`);
|
|
50
|
+
throw error;
|
|
51
|
+
}
|
|
52
|
+
}
|
|
53
|
+
/**
|
|
54
|
+
* 带超时的 Promise 包装
|
|
55
|
+
*/
|
|
56
|
+
async withTimeout(promise, ms, message) {
|
|
57
|
+
let timeoutId;
|
|
58
|
+
const timeoutPromise = new Promise((_, reject) => {
|
|
59
|
+
timeoutId = setTimeout(() => reject(new Error(message)), ms);
|
|
60
|
+
});
|
|
61
|
+
try {
|
|
62
|
+
return await Promise.race([promise, timeoutPromise]);
|
|
63
|
+
}
|
|
64
|
+
finally {
|
|
65
|
+
clearTimeout(timeoutId);
|
|
66
|
+
}
|
|
67
|
+
}
|
|
68
|
+
/**
|
|
69
|
+
* 断开连接
|
|
70
|
+
*/
|
|
71
|
+
async disconnect() {
|
|
72
|
+
if (this._status === 'disconnected') {
|
|
73
|
+
return;
|
|
74
|
+
}
|
|
75
|
+
this._status = 'disconnected';
|
|
76
|
+
this._capabilities = null;
|
|
77
|
+
try {
|
|
78
|
+
// 先尝试优雅关闭客户端
|
|
79
|
+
if (this.client) {
|
|
80
|
+
await this.withTimeout(this.client.close(), 5000, // 5秒超时
|
|
81
|
+
'关闭客户端超时');
|
|
82
|
+
}
|
|
83
|
+
(0, log_1.logDebug)(`MCP Server [${this.config.name}] 已断开`);
|
|
84
|
+
}
|
|
85
|
+
catch (error) {
|
|
86
|
+
// 记录错误但不抛出,确保清理流程继续
|
|
87
|
+
(0, log_1.logError)(`MCP Server [${this.config.name}] 断开失败: ${error}`);
|
|
88
|
+
}
|
|
89
|
+
finally {
|
|
90
|
+
// 确保 transport 被清理
|
|
91
|
+
try {
|
|
92
|
+
if (this.transport) {
|
|
93
|
+
// 对于 stdio transport,强制关闭可能仍在运行的子进程
|
|
94
|
+
if (this.config.transport === 'stdio' && 'close' in this.transport) {
|
|
95
|
+
;
|
|
96
|
+
this.transport.close?.();
|
|
97
|
+
}
|
|
98
|
+
this.transport = null;
|
|
99
|
+
}
|
|
100
|
+
}
|
|
101
|
+
catch (transportError) {
|
|
102
|
+
// 忽略 transport 清理错误
|
|
103
|
+
(0, log_1.logDebug)(`Transport 清理错误 [${this.config.name}]: ${transportError}`);
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
}
|
|
107
|
+
/**
|
|
108
|
+
* 调用 MCP 工具
|
|
109
|
+
*/
|
|
110
|
+
async callTool(toolName, args) {
|
|
111
|
+
if (this._status !== 'connected') {
|
|
112
|
+
throw new Error(`MCP Server [${this.config.name}] 未连接`);
|
|
113
|
+
}
|
|
114
|
+
const result = await this.client.callTool({
|
|
115
|
+
name: toolName,
|
|
116
|
+
arguments: args
|
|
117
|
+
});
|
|
118
|
+
return {
|
|
119
|
+
content: result.content,
|
|
120
|
+
isError: result.isError
|
|
121
|
+
};
|
|
122
|
+
}
|
|
123
|
+
createTransport() {
|
|
124
|
+
switch (this.config.transport) {
|
|
125
|
+
case 'stdio':
|
|
126
|
+
if (!this.config.command) {
|
|
127
|
+
throw new Error('stdio 传输需要配置 command');
|
|
128
|
+
}
|
|
129
|
+
return new stdio_js_1.StdioClientTransport({
|
|
130
|
+
command: this.config.command,
|
|
131
|
+
args: this.config.args,
|
|
132
|
+
env: this.config.env,
|
|
133
|
+
// cwd: this.config.cwd
|
|
134
|
+
});
|
|
135
|
+
case 'sse':
|
|
136
|
+
if (!this.config.url) {
|
|
137
|
+
throw new Error('sse 传输需要配置 url');
|
|
138
|
+
}
|
|
139
|
+
return new sse_js_1.SSEClientTransport(new URL(this.config.url));
|
|
140
|
+
case 'http':
|
|
141
|
+
if (!this.config.url) {
|
|
142
|
+
throw new Error('http 传输需要配置 url');
|
|
143
|
+
}
|
|
144
|
+
return new streamableHttp_js_1.StreamableHTTPClientTransport(new URL(this.config.url), {
|
|
145
|
+
requestInit: this.config.headers ? { headers: this.config.headers } : undefined
|
|
146
|
+
});
|
|
147
|
+
default:
|
|
148
|
+
throw new Error(`不支持的传输类型: ${this.config.transport}`);
|
|
149
|
+
}
|
|
150
|
+
}
|
|
151
|
+
async fetchCapabilities() {
|
|
152
|
+
try {
|
|
153
|
+
const toolsResult = await this.client.listTools();
|
|
154
|
+
this._capabilities = {
|
|
155
|
+
tools: toolsResult.tools
|
|
156
|
+
};
|
|
157
|
+
}
|
|
158
|
+
catch {
|
|
159
|
+
this._capabilities = { tools: [] };
|
|
160
|
+
}
|
|
161
|
+
}
|
|
162
|
+
}
|
|
163
|
+
exports.MCPClient = MCPClient;
|
|
164
|
+
/** 连接超时时间 (ms) */
|
|
165
|
+
MCPClient.CONNECT_TIMEOUT = 30000;
|
|
166
|
+
//# sourceMappingURL=MCPClient.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"MCPClient.js","sourceRoot":"","sources":["../../../src/services/mcp/MCPClient.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAEH,wEAAkE;AAClE,wEAAgF;AAChF,oEAA4E;AAC5E,0FAAkG;AAQlG,wCAAmD;AAInD,MAAa,SAAS;IAMpB,YAAoB,MAAuB;QAAvB,WAAM,GAAN,MAAM,CAAiB;QAJnC,cAAS,GAAqB,IAAI,CAAA;QAClC,YAAO,GAAoB,cAAc,CAAA;QACzC,kBAAa,GAAiC,IAAI,CAAA;QAGxD,IAAI,CAAC,MAAM,GAAG,IAAI,iBAAM,CACtB;YACE,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,OAAO;SACjB,EACD;YACE,YAAY,EAAE,EAAE;SACjB,CACF,CAAA;IACH,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,OAAO,CAAA;IACrB,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,aAAa,CAAA;IAC3B,CAAC;IAKD;;OAEG;IACH,KAAK,CAAC,OAAO;QACX,IAAI,CAAC;YACH,IAAI,CAAC,OAAO,GAAG,YAAY,CAAA;YAE3B,qBAAqB;YACrB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,eAAe,EAAE,CAAA;YAEvC,SAAS;YACT,MAAM,IAAI,CAAC,WAAW,CACpB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,EACnC,SAAS,CAAC,eAAe,EACzB,SAAS,SAAS,CAAC,eAAe,GAAG,IAAI,IAAI,CAC9C,CAAA;YAED,gBAAgB;YAChB,MAAM,IAAI,CAAC,WAAW,CACpB,IAAI,CAAC,iBAAiB,EAAE,EACxB,SAAS,CAAC,eAAe,EACzB,WAAW,SAAS,CAAC,eAAe,GAAG,IAAI,IAAI,CAChD,CAAA;YAED,IAAI,CAAC,OAAO,GAAG,WAAW,CAAA;YAC1B,IAAA,cAAQ,EAAC,eAAe,IAAI,CAAC,MAAM,CAAC,IAAI,QAAQ,CAAC,CAAA;QACnD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,OAAO,GAAG,OAAO,CAAA;YACtB,MAAM,QAAQ,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;YACvE,IAAA,cAAQ,EAAC,eAAe,IAAI,CAAC,MAAM,CAAC,IAAI,WAAW,QAAQ,EAAE,CAAC,CAAA;YAC9D,MAAM,KAAK,CAAA;QACb,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,WAAW,CAAI,OAAmB,EAAE,EAAU,EAAE,OAAe;QAC3E,IAAI,SAAyB,CAAA;QAC7B,MAAM,cAAc,GAAG,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;YACtD,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;QAC9D,CAAC,CAAC,CAAA;QACF,IAAI,CAAC;YACH,OAAO,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC,CAAA;QACtD,CAAC;gBAAS,CAAC;YACT,YAAY,CAAC,SAAU,CAAC,CAAA;QAC1B,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,OAAO,KAAK,cAAc,EAAE,CAAC;YACpC,OAAM;QACR,CAAC;QAED,IAAI,CAAC,OAAO,GAAG,cAAc,CAAA;QAC7B,IAAI,CAAC,aAAa,GAAG,IAAI,CAAA;QAEzB,IAAI,CAAC;YACH,aAAa;YACb,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBAChB,MAAM,IAAI,CAAC,WAAW,CACpB,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,EACnB,IAAI,EAAE,OAAO;gBACb,SAAS,CACV,CAAA;YACH,CAAC;YACD,IAAA,cAAQ,EAAC,eAAe,IAAI,CAAC,MAAM,CAAC,IAAI,OAAO,CAAC,CAAA;QAClD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,oBAAoB;YACpB,IAAA,cAAQ,EAAC,eAAe,IAAI,CAAC,MAAM,CAAC,IAAI,WAAW,KAAK,EAAE,CAAC,CAAA;QAC7D,CAAC;gBAAS,CAAC;YACT,mBAAmB;YACnB,IAAI,CAAC;gBACH,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;oBACnB,oCAAoC;oBACpC,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,KAAK,OAAO,IAAI,OAAO,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;wBACnE,CAAC;wBAAC,IAAI,CAAC,SAAiB,CAAC,KAAK,EAAE,EAAE,CAAA;oBACpC,CAAC;oBACD,IAAI,CAAC,SAAS,GAAG,IAAI,CAAA;gBACvB,CAAC;YACH,CAAC;YAAC,OAAO,cAAc,EAAE,CAAC;gBACxB,oBAAoB;gBACpB,IAAA,cAAQ,EAAC,mBAAmB,IAAI,CAAC,MAAM,CAAC,IAAI,MAAM,cAAc,EAAE,CAAC,CAAA;YACrE,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CAAC,QAAgB,EAAE,IAA6B;QAC5D,IAAI,IAAI,CAAC,OAAO,KAAK,WAAW,EAAE,CAAC;YACjC,MAAM,IAAI,KAAK,CAAC,eAAe,IAAI,CAAC,MAAM,CAAC,IAAI,OAAO,CAAC,CAAA;QACzD,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;YACxC,IAAI,EAAE,QAAQ;YACd,SAAS,EAAE,IAAI;SAChB,CAAC,CAAA;QAEF,OAAO;YACL,OAAO,EAAE,MAAM,CAAC,OAAmC;YACnD,OAAO,EAAE,MAAM,CAAC,OAA8B;SAC/C,CAAA;IACH,CAAC;IAEO,eAAe;QACrB,QAAQ,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;YAC9B,KAAK,OAAO;gBACV,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;oBACzB,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAA;gBACzC,CAAC;gBACD,OAAO,IAAI,+BAAoB,CAAC;oBAC9B,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;oBAC5B,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;oBACtB,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,GAAyC;oBAC1D,uBAAuB;iBACxB,CAAC,CAAA;YAEJ,KAAK,KAAK;gBACR,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC;oBACrB,MAAM,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAA;gBACnC,CAAC;gBACD,OAAO,IAAI,2BAAkB,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAA;YAEzD,KAAK,MAAM;gBACT,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC;oBACrB,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAA;gBACpC,CAAC;gBACD,OAAO,IAAI,iDAA6B,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE;oBACjE,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,SAAS;iBAChF,CAAC,CAAA;YAEJ;gBACE,MAAM,IAAI,KAAK,CAAC,aAAa,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,CAAA;QACzD,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,iBAAiB;QAC7B,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAA;YACjD,IAAI,CAAC,aAAa,GAAG;gBACnB,KAAK,EAAE,WAAW,CAAC,KAA4B;aAChD,CAAA;QACH,CAAC;QAAC,MAAM,CAAC;YACP,IAAI,CAAC,aAAa,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,CAAA;QACpC,CAAC;IACH,CAAC;;AAnLH,8BAoLC;AA1JC,kBAAkB;AACM,yBAAe,GAAG,KAAK,AAAR,CAAQ"}
|