From 4a8385d7ad857fe22bf74fe431202b2a473f1cd5 Mon Sep 17 00:00:00 2001 From: Kirigaya <1193466151@qq.com> Date: Fri, 20 Jun 2025 21:06:01 +0800 Subject: [PATCH] task loop --- renderer/src/api/message-bridge.ts | 8 +- .../main-panel/chat/core/task-loop.ts | 27 +- renderer/src/views/connect/core.ts | 6 +- resources/openmcp-sdk-release/task-loop.d.ts | 13 +- service/src/hook/adapter.ts | 319 ----------------- service/src/hook/sdk.ts | 327 ++++++++++++++++++ service/src/index.ts | 2 +- service/src/mcp/client.service.ts | 2 +- service/src/mcp/connect.service.ts | 32 +- service/src/sdk.ts | 1 + service/task-loop.d.ts | 39 ++- service/tsconfig.json | 2 +- 12 files changed, 416 insertions(+), 362 deletions(-) create mode 100644 service/src/hook/sdk.ts create mode 100644 service/src/sdk.ts diff --git a/renderer/src/api/message-bridge.ts b/renderer/src/api/message-bridge.ts index ae888b1..231973d 100644 --- a/renderer/src/api/message-bridge.ts +++ b/renderer/src/api/message-bridge.ts @@ -39,22 +39,22 @@ export class MessageBridge { switch (platform) { case 'vscode': this.setupVsCodeListener(); - pinkLog('当前模式: vscode'); + pinkLog('current platform: vscode'); break; case 'electron': this.setupElectronListener(); - pinkLog('当前模式: electron'); + pinkLog('current platform: electron'); break; case 'nodejs': this.setupNodejsListener(); - pinkLog('当前模式: nodejs'); + pinkLog('current platform: nodejs'); break; case 'web': this.setupWebSocket(); - pinkLog('当前模式: web'); + pinkLog('current platform: web'); break; } } diff --git a/renderer/src/components/main-panel/chat/core/task-loop.ts b/renderer/src/components/main-panel/chat/core/task-loop.ts index a2048dc..7269522 100644 --- a/renderer/src/components/main-panel/chat/core/task-loop.ts +++ b/renderer/src/components/main-panel/chat/core/task-loop.ts @@ -65,7 +65,7 @@ export class TaskLoop { }; constructor( - private readonly taskOptions: TaskLoopOptions = { + private taskOptions: TaskLoopOptions = { maxEpochs: 50, maxJsonParseRetry: 3, adapter: undefined, @@ -99,6 +99,25 @@ export class TaskLoop { mcpClientAdapter.addConnectRefreshListener(); } + public async waitConnection() { + await this.nodejsStatus.connectionFut; + } + + public setTaskLoopOptions(taskOptions: TaskLoopOptions) { + const { + maxEpochs = 50, + maxJsonParseRetry = 3, + verbose = 1, + } = taskOptions; + + this.taskOptions = { + maxEpochs, + maxJsonParseRetry, + verbose, + ...this.taskOptions + }; + } + /** * @description 处理 streaming 输出的每一个分块的 content 部分 * @param chunk @@ -784,9 +803,11 @@ export class TaskLoop { } } - public async getPrompt(promptId: string, args: Record) { + public async getPrompt(promptId: string, args?: Record) { const prompt = await mcpClientAdapter.readPromptTemplate(promptId, args); - return prompt; + // transform prompt to string + const promptString = prompt.messages.map(m => m.content.text).join('\n'); + return promptString; } public async getResource(resourceUri: string) { diff --git a/renderer/src/views/connect/core.ts b/renderer/src/views/connect/core.ts index 31d44ee..f36a77c 100644 --- a/renderer/src/views/connect/core.ts +++ b/renderer/src/views/connect/core.ts @@ -610,7 +610,7 @@ class McpClientAdapter { }, { once: false }); } - + const launchSignature = await this.getLaunchSignature(); let allOk = true; @@ -677,7 +677,7 @@ class McpClientAdapter { return msg; } - public async readPromptTemplate(promptId: string, args: Record) { + public async readPromptTemplate(promptId: string, args?: Record) { // TODO: 如果遇到不同服务器的同名 tool，请拓展解决方案 // 目前只找到第一个匹配 toolName 的工具进行调用 let clientId = this.clients[0].clientId; @@ -691,6 +691,8 @@ class McpClientAdapter { } } + console.log(args); + const bridge = useMessageBridge(); const { code, msg } = await bridge.commandRequest('prompts/get', { clientId, promptId, args }); return msg; diff --git a/resources/openmcp-sdk-release/task-loop.d.ts b/resources/openmcp-sdk-release/task-loop.d.ts index e66729c..1240c5f 100644 --- a/resources/openmcp-sdk-release/task-loop.d.ts +++ b/resources/openmcp-sdk-release/task-loop.d.ts @@ -189,6 +189,17 @@ interface ChatSetting { export class TaskLoop { constructor(taskOptions?: TaskLoopOptions); + /** + * @description wait for connection + */ + waitConnection(): Promise; + + /** + * @description Set the task loop options + * @param taskOptions + */ + setTaskLoopOptions(taskOptions: TaskLoopOptions): void; + /** * @description make chat data * @param tabStorage @@ -294,7 +305,7 @@ export class TaskLoop { /** * @description Get prompt template from mcp server */ - getPrompt(promptId: string, args: Record): Promise; + getPrompt(promptId: string, args?: Record): Promise; /** * @description Get resource template from mcp server diff --git a/service/src/hook/adapter.ts b/service/src/hook/adapter.ts index 234645a..0e65837 100644 --- a/service/src/hook/adapter.ts +++ b/service/src/hook/adapter.ts @@ -1,8 +1,4 @@ -import * as fs from 'fs'; - import { WebSocket } from 'ws'; -import { EventEmitter } from 'events'; -import { routeMessage } from '../common/router.js'; import { ConnectionType } from '../mcp/client.dto.js'; // WebSocket 消息格式 @@ -80,318 +76,3 @@ export class VSCodeWebViewLike { }; } } - - -export class TaskLoopAdapter { - public emitter: EventEmitter; - private messageHandlers: Set; - private connectionOptions: IConnectionArgs[] = []; - - constructor(option?: any) { - this.emitter = new EventEmitter(option); - this.messageHandlers = new Set(); - - this.emitter.on('message/renderer', (message: WebSocketMessage) => { - this.messageHandlers.forEach((handler) => handler(message)); - }); - - // 默认需要将监听的消息导入到 routeMessage 中 - this.onDidReceiveMessage((message) => { - const { command, data } = message; - - switch (command) { - case 'nodejs/launch-signature': - this.postMessage({ - command: 'nodejs/launch-signature', - data: { - code: 200, - msg: this.connectionOptions - } - }) - break; - - case 'nodejs/update-connection-signature': - // sdk 模式下不需要自动保存连接参数 - break; - - default: - routeMessage(command, data, this); - break; - } - }); - - } - - /** - * @description 发送消息 - * @param message - 包含 command 和 args 的消息 - */ - public postMessage(message: WebSocketMessage): void { - this.emitter.emit('message/service', message); - } - - /** - * @description 注册接受消息的句柄 - * @param callback - 消息回调 - * @returns {{ dispose: () => void }} - 可销毁的监听器 - */ - public onDidReceiveMessage(callback: MessageHandler): { dispose: () => void } { - this.messageHandlers.add(callback); - return { - dispose: () => this.messageHandlers.delete(callback), - }; - } - - /** - * @description 连接到 mcp 服务端 - * @param mcpOption - */ - public addMcp(mcpOption: IConnectionArgs) { - - // 0.1.4 新版本开始，此处修改为懒加载连接 - // 实际的连接移交给前端 mcpAdapter 中进行统一的调度 - // 调度步骤如下： - // getLaunchSignature 先获取访问签名，访问签名通过当前函数 push 到 class 中 - - this.connectionOptions.push(mcpOption); - } -} - -interface StdioMCPConfig { - command: string; - args: string[]; - env?: { - [key: string]: string; - }; - description?: string; - prompts?: string[]; - resources?: string[]; -} - -interface HttpMCPConfig { - url: string; - type?: string; - env?: { - [key: string]: string; - }; - description?: string; - prompts?: string[]; - resources?: string[]; -} - -export interface OmAgentConfiguration { - version: string; - mcpServers: { - [key: string]: StdioMCPConfig | HttpMCPConfig; - }; - defaultLLM: { - baseURL: string; - apiToken: string; - model: string; - } -} - -export interface DefaultLLM { - baseURL: string; - apiToken?: string; - model: string; -} - -import { - MessageState, - type TaskLoopOptions, - type ChatMessage, - type ChatSetting, - type TaskLoop, - type TextMessage -} from '../../task-loop.js'; - -export function UserMessage(content: string): TextMessage { - return { - role: 'user', - content, - extraInfo: { - created: Date.now(), - state: MessageState.None, - serverName: '', - enableXmlWrapper: false - } - } -} - -export function AssistantMessage(content: string): TextMessage { - return { - role: 'assistant', - content, - extraInfo: { - created: Date.now(), - state: MessageState.None, - serverName: '', - enableXmlWrapper: false - } - } -} - -export class OmAgent { - private _adapter: TaskLoopAdapter; - private _loop?: TaskLoop; - private _defaultLLM?: DefaultLLM; - - constructor() { - this._adapter = new TaskLoopAdapter(); - } - - /** - * @description Load MCP configuration from file. - * Supports multiple MCP backends and a default LLM model configuration. - * - * @example - * Example configuration: - * { - * "version": "1.0.0", - * "mcpServers": { - * "openmemory": { - * "command": "npx", - * "args": ["-y", "openmemory"], - * "env": { - * "OPENMEMORY_API_KEY": "YOUR_API_KEY", - * "CLIENT_NAME": "openmemory" - * }, - * "description": "A MCP for long-term memory support" - * } - * }, - * "defaultLLM": { - * "baseURL": "https://api.openmemory.ai", - * "apiToken": "YOUR_API_KEY", - * "model": "deepseek-chat" - * } - * } - * - * @param configPath - Path to the configuration file - */ - public loadMcpConfig(configPath: string) { - const config = JSON.parse(fs.readFileSync(configPath, 'utf-8')) as OmAgentConfiguration; - const { mcpServers, defaultLLM } = config; - - // set default llm - this.setDefaultLLM(defaultLLM); - - for (const key in mcpServers) { - const mcpConfig = mcpServers[key]; - if ('command' in mcpConfig) { - const commandString = ( - mcpConfig.command + ' ' + mcpConfig.args.join(' ') - ).trim(); - - this._adapter.addMcp({ - commandString, - connectionType: 'STDIO', - env: mcpConfig.env, - description: mcpConfig.description, - }); - } else { - const connectionType: ConnectionType = mcpConfig.type === 'http' ? 'STREAMABLE_HTTP' : 'SSE'; - this._adapter.addMcp({ - url: mcpConfig.url, - env: mcpConfig.env, - connectionType, - description: mcpConfig.description, - }); - } - } - } - - /** - * @description Add MCP server - */ - public addMcpServer(connectionArgs: IConnectionArgs) { - this._adapter.addMcp(connectionArgs); - } - - private async getLoop(loopOption?: TaskLoopOptions) { - if (this._loop) { - return this._loop; - } - - const { - verbose = 1, - maxEpochs = 50, - maxJsonParseRetry = 3, - } = loopOption || {} - - const adapter = this._adapter; - const { TaskLoop } = await import('../../task-loop.js'); - this._loop = new TaskLoop({ adapter, verbose, maxEpochs, maxJsonParseRetry }); - - return this._loop; - } - - public setDefaultLLM(option: DefaultLLM) { - this._defaultLLM = option; - } - - public async getPrompt(promptId: string, args: Record) { - const loop = await this.getLoop(); - const prompt = await loop.getPrompt(promptId, args); - return prompt; - } - - /** - * @description Asynchronous invoking agent by string or messages - * @param messages Chat message or string - * @param settings Chat setting and task loop options - * @returns - */ - public async ainvoke( - { messages, settings }: { messages: ChatMessage[] | string; settings?: ChatSetting & Partial; } - ) { - if (messages.length === 0) { - throw new Error('messages is empty'); - } - - // detach taskloop option from settings and set default value - const { - maxEpochs = 50, - maxJsonParseRetry = 3, - verbose = 1 - } = settings || {}; - - const loop = await this.getLoop({ maxEpochs, maxJsonParseRetry, verbose }); - const storage = await loop.createStorage(settings); - - // set input message - // user can invoke [UserMessage("CONTENT")] to make messages quickly - // or use string directly - let userMessage: string; - if (typeof messages === 'string') { - userMessage = messages; - } else { - const lastMessageContent = messages.at(-1)?.content; - if (typeof lastMessageContent === 'string') { - userMessage = lastMessageContent; - } else { - throw new Error('last message content is undefined'); - } - } - - // select correct llm config - // user can set llm config via omagent.setDefaultLLM() - // or write "defaultLLM" in mcpconfig.json to specify - if (this._defaultLLM) { - loop.setLlmConfig({ - baseUrl: this._defaultLLM.baseURL, - userToken: this._defaultLLM.apiToken, - userModel: this._defaultLLM.model, - }); - } else { - // throw error to user and give the suggestion - throw new Error('default LLM is not set, please set it via omagent.setDefaultLLM() or write "defaultLLM" in mcpconfig.json'); - } - - await loop.start(storage, userMessage); - - // get response from last message in message list - const lastMessage = storage.messages.at(-1)?.content; - return lastMessage; - } -} diff --git a/service/src/hook/sdk.ts b/service/src/hook/sdk.ts new file mode 100644 index 0000000..d0e17c4 --- /dev/null +++ b/service/src/hook/sdk.ts @@ -0,0 +1,327 @@ +import { EventEmitter } from 'events'; +import { routeMessage } from '../common/router.js'; +import * as fs from 'fs'; + +export class TaskLoopAdapter { + public emitter: EventEmitter; + private messageHandlers: Set; + private connectionOptions: IConnectionArgs[] = []; + + constructor(option?: any) { + this.emitter = new EventEmitter(option); + this.messageHandlers = new Set(); + + this.emitter.on('message/renderer', (message: WebSocketMessage) => { + this.messageHandlers.forEach((handler) => handler(message)); + }); + + // 默认需要将监听的消息导入到 routeMessage 中 + this.onDidReceiveMessage((message) => { + const { command, data } = message; + + switch (command) { + case 'nodejs/launch-signature': + this.postMessage({ + command: 'nodejs/launch-signature', + data: { + code: 200, + msg: this.connectionOptions + } + }) + break; + + case 'nodejs/update-connection-signature': + // sdk 模式下不需要自动保存连接参数 + break; + + default: + routeMessage(command, data, this); + break; + } + }); + + } + + /** + * @description 发送消息 + * @param message - 包含 command 和 args 的消息 + */ + public postMessage(message: WebSocketMessage): void { + this.emitter.emit('message/service', message); + } + + /** + * @description 注册接受消息的句柄 + * @param callback - 消息回调 + * @returns {{ dispose: () => void }} - 可销毁的监听器 + */ + public onDidReceiveMessage(callback: MessageHandler): { dispose: () => void } { + this.messageHandlers.add(callback); + return { + dispose: () => this.messageHandlers.delete(callback), + }; + } + + /** + * @description 连接到 mcp 服务端 + * @param mcpOption + */ + public addMcp(mcpOption: IConnectionArgs) { + + // 0.1.4 新版本开始，此处修改为懒加载连接 + // 实际的连接移交给前端 mcpAdapter 中进行统一的调度 + // 调度步骤如下： + // getLaunchSignature 先获取访问签名，访问签名通过当前函数 push 到 class 中 + + this.connectionOptions.push(mcpOption); + } +} + +interface StdioMCPConfig { + command: string; + args: string[]; + env?: { + [key: string]: string; + }; + description?: string; + prompts?: string[]; + resources?: string[]; +} + +interface HttpMCPConfig { + url: string; + type?: string; + env?: { + [key: string]: string; + }; + description?: string; + prompts?: string[]; + resources?: string[]; +} + +export interface OmAgentConfiguration { + version: string; + mcpServers: { + [key: string]: StdioMCPConfig | HttpMCPConfig; + }; + defaultLLM: { + baseURL: string; + apiToken: string; + model: string; + } +} + +export interface DefaultLLM { + baseURL: string; + apiToken?: string; + model: string; +} + +import { + MessageState, + type TaskLoopOptions, + type ChatMessage, + type ChatSetting, + type TaskLoop, + type TextMessage +} from '../../task-loop.js'; +import { IConnectionArgs, MessageHandler, WebSocketMessage } from './adapter.js'; +import { ConnectionType } from 'src/mcp/client.dto.js'; + +export function UserMessage(content: string): TextMessage { + return { + role: 'user', + content, + extraInfo: { + created: Date.now(), + state: MessageState.None, + serverName: '', + enableXmlWrapper: false + } + } +} + +export function AssistantMessage(content: string): TextMessage { + return { + role: 'assistant', + content, + extraInfo: { + created: Date.now(), + state: MessageState.None, + serverName: '', + enableXmlWrapper: false + } + } +} + +export class OmAgent { + private _adapter: TaskLoopAdapter; + private _loop?: TaskLoop; + private _defaultLLM?: DefaultLLM; + + constructor() { + this._adapter = new TaskLoopAdapter(); + } + + /** + * @description Load MCP configuration from file. + * Supports multiple MCP backends and a default LLM model configuration. + * + * @example + * Example configuration: + * { + * "version": "1.0.0", + * "mcpServers": { + * "openmemory": { + * "command": "npx", + * "args": ["-y", "openmemory"], + * "env": { + * "OPENMEMORY_API_KEY": "YOUR_API_KEY", + * "CLIENT_NAME": "openmemory" + * }, + * "description": "A MCP for long-term memory support" + * } + * }, + * "defaultLLM": { + * "baseURL": "https://api.openmemory.ai", + * "apiToken": "YOUR_API_KEY", + * "model": "deepseek-chat" + * } + * } + * + * @param configPath - Path to the configuration file + */ + public loadMcpConfig(configPath: string) { + const config = JSON.parse(fs.readFileSync(configPath, 'utf-8')) as OmAgentConfiguration; + const { mcpServers, defaultLLM } = config; + + // set default llm + this.setDefaultLLM(defaultLLM); + + for (const key in mcpServers) { + const mcpConfig = mcpServers[key]; + if ('command' in mcpConfig) { + const commandString = ( + mcpConfig.command + ' ' + mcpConfig.args.join(' ') + ).trim(); + + this._adapter.addMcp({ + commandString, + connectionType: 'STDIO', + env: mcpConfig.env, + description: mcpConfig.description, + }); + } else { + const connectionType: ConnectionType = mcpConfig.type === 'http' ? 'STREAMABLE_HTTP' : 'SSE'; + this._adapter.addMcp({ + url: mcpConfig.url, + env: mcpConfig.env, + connectionType, + description: mcpConfig.description, + }); + } + } + } + + /** + * @description Add MCP server + */ + public addMcpServer(connectionArgs: IConnectionArgs) { + this._adapter.addMcp(connectionArgs); + } + + private async getLoop(loopOption?: TaskLoopOptions) { + + if (this._loop) { + if (loopOption) { + this._loop.setTaskLoopOptions(loopOption); + } + return this._loop; + } + + const { + verbose = 1, + maxEpochs = 50, + maxJsonParseRetry = 3, + } = loopOption || {} + + const adapter = this._adapter; + const { TaskLoop } = await import('../../task-loop.js'); + + this._loop = new TaskLoop({ adapter, verbose, maxEpochs, maxJsonParseRetry }); + await this._loop.waitConnection(); + + return this._loop; + } + + public setDefaultLLM(option: DefaultLLM) { + this._defaultLLM = option; + } + + public async getPrompt(promptId: string, args: Record) { + const loop = await this.getLoop(); + + const prompt = await loop.getPrompt(promptId, args); + + return prompt; + } + + /** + * @description Asynchronous invoking agent by string or messages + * @param messages Chat message or string + * @param settings Chat setting and task loop options + * @returns + */ + public async ainvoke( + { messages, settings }: { messages: ChatMessage[] | string; settings?: ChatSetting & Partial; } + ) { + if (messages.length === 0) { + throw new Error('messages is empty'); + } + + // detach taskloop option from settings and set default value + const { + maxEpochs = 50, + maxJsonParseRetry = 3, + verbose = 1 + } = settings || {}; + + const loop = await this.getLoop({ maxEpochs, maxJsonParseRetry, verbose }); + const storage = await loop.createStorage(settings); + + // set input message + // user can invoke [UserMessage("CONTENT")] to make messages quickly + // or use string directly + let userMessage: string; + if (typeof messages === 'string') { + userMessage = messages; + } else { + const lastMessageContent = messages.at(-1)?.content; + if (typeof lastMessageContent === 'string') { + userMessage = lastMessageContent; + } else { + throw new Error('last message content is undefined'); + } + } + + // select correct llm config + // user can set llm config via omagent.setDefaultLLM() + // or write "defaultLLM" in mcpconfig.json to specify + if (this._defaultLLM) { + loop.setLlmConfig({ + baseUrl: this._defaultLLM.baseURL, + userToken: this._defaultLLM.apiToken, + userModel: this._defaultLLM.model, + }); + } else { + // throw error to user and give the suggestion + throw new Error('default LLM is not set, please set it via omagent.setDefaultLLM() or write "defaultLLM" in mcpconfig.json'); + } + + await loop.start(storage, userMessage); + + // get response from last message in message list + const lastMessage = storage.messages.at(-1)?.content; + return lastMessage; + } +} diff --git a/service/src/index.ts b/service/src/index.ts index 600ef9e..ae85e0c 100644 --- a/service/src/index.ts +++ b/service/src/index.ts @@ -1,4 +1,4 @@ export { routeMessage } from './common/router.js'; -export { VSCodeWebViewLike, TaskLoopAdapter, OmAgent, OmAgentConfiguration, UserMessage, AssistantMessage } from './hook/adapter.js'; +export { VSCodeWebViewLike } from './hook/adapter.js'; export { setVscodeWorkspace, setRunningCWD } from './hook/setting.js'; export { clientMap } from './mcp/connect.service.js'; \ No newline at end of file diff --git a/service/src/mcp/client.service.ts b/service/src/mcp/client.service.ts index 9d8b875..6869bce 100644 --- a/service/src/mcp/client.service.ts +++ b/service/src/mcp/client.service.ts @@ -137,7 +137,7 @@ export class McpClient { } // 获取提示 - public async getPrompt(name: string, args: Record = {}) { + public async getPrompt(name: string, args: Record = {}) { return await this.client.getPrompt({ name, arguments: args }); diff --git a/service/src/mcp/connect.service.ts b/service/src/mcp/connect.service.ts index ef6d5cd..efc26e0 100644 --- a/service/src/mcp/connect.service.ts +++ b/service/src/mcp/connect.service.ts @@ -268,24 +268,24 @@ export async function connectService( ): Promise { try { // 使用cli-table3创建美观的表格 - const table = new Table({ - head: ['Property', 'Value'], - colWidths: [20, 60], - style: { - head: ['green'], - border: ['grey'] - } - }); + // const table = new Table({ + // head: ['Property', 'Value'], + // colWidths: [20, 60], + // style: { + // head: ['green'], + // border: ['grey'] + // } + // }); - table.push( - ['Connection Type', option.connectionType], - ['Command', option.command || 'N/A'], - ['Arguments', option.args?.join(' ') || 'N/A'], - ['Working Directory', option.cwd || 'N/A'], - ['URL', option.url || 'N/A'] - ); + // table.push( + // ['Connection Type', option.connectionType], + // ['Command', option.command || 'N/A'], + // ['Arguments', option.args?.join(' ') || 'N/A'], + // ['Working Directory', option.cwd || 'N/A'], + // ['URL', option.url || 'N/A'] + // ); - console.log(table.toString()); + // console.log(table.toString()); // 预处理字符串 await preprocessCommand(option, webview); diff --git a/service/src/sdk.ts b/service/src/sdk.ts new file mode 100644 index 0000000..fa0086f --- /dev/null +++ b/service/src/sdk.ts @@ -0,0 +1 @@ +export { TaskLoopAdapter, OmAgent, OmAgentConfiguration, UserMessage, AssistantMessage } from './hook/sdk.js'; \ No newline at end of file diff --git a/service/task-loop.d.ts b/service/task-loop.d.ts index 06af785..1240c5f 100644 --- a/service/task-loop.d.ts +++ b/service/task-loop.d.ts @@ -5,25 +5,25 @@ export type ChatCompletionChunk = OpenAI.Chat.Completions.ChatCompletionChunk; export type ChatCompletionCreateParamsBase = OpenAI.Chat.Completions.ChatCompletionCreateParams & { id?: string }; interface SchemaProperty { - title: string; - type: string; - description?: string; + title: string; + type: string; + description?: string; } interface InputSchema { - type: string; - properties: Record; - required?: string[]; - title?: string; - $defs?: any; + type: string; + properties: Record; + required?: string[]; + title?: string; + $defs?: any; } interface ToolItem { - name: string; - description: string; - inputSchema: InputSchema; + name: string; + description: string; + inputSchema: InputSchema; enabled: boolean; - anyOf?: any; + anyOf?: any; } interface IExtraInfo { @@ -87,7 +87,7 @@ export interface ToolCall { export interface ToolCallContent { type: string; text: string; - [key: string]: any; + [key: string]: any; } export interface ToolCallResult { @@ -189,6 +189,17 @@ interface ChatSetting { export class TaskLoop { constructor(taskOptions?: TaskLoopOptions); + /** + * @description wait for connection + */ + waitConnection(): Promise; + + /** + * @description Set the task loop options + * @param taskOptions + */ + setTaskLoopOptions(taskOptions: TaskLoopOptions): void; + /** * @description make chat data * @param tabStorage @@ -294,7 +305,7 @@ export class TaskLoop { /** * @description Get prompt template from mcp server */ - getPrompt(promptId: string, args: Record): Promise; + getPrompt(promptId: string, args?: Record): Promise; /** * @description Get resource template from mcp server diff --git a/service/tsconfig.json b/service/tsconfig.json index b372d45..d10762d 100644 --- a/service/tsconfig.json +++ b/service/tsconfig.json @@ -10,7 +10,7 @@ "forceConsistentCasingInFileNames": true, "outDir": "./dist", "declaration": true, - "declarationMap": true, + "declarationMap": false, "experimentalDecorators": true, }, "paths": {