openmcp-sdk 0.0.2 → 0.0.4

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "openmcp-sdk",
-    "version": "0.0.2",
+    "version": "0.0.4",
     "description": "openmcp-sdk",
     "scripts": {
         "test": "echo \"Error: no test specified\" && exit 1"

package/service/hook/adapter.d.ts

@@ -59,9 +59,15 @@ export declare class TaskLoopAdapter {
      * @returns
      */
     listTools(): Promise<{
-        name: string;
-        description: string | undefined;
         enabled: boolean;
+        name: string;
+        inputSchema: {
+            type: "object";
+            properties?: import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough"> | undefined;
+        } & {
+            [k: string]: unknown;
+        };
+        description?: string | undefined;
     }[]>;
 }
 //# sourceMappingURL=adapter.d.ts.map

package/service/hook/adapter.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"adapter.d.ts","sourceRoot":"","sources":["../../src/hook/adapter.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAC/B,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AAEtC,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAI/C,MAAM,WAAW,gBAAgB;IAC7B,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,GAAG,CAAC;CACb;AAGD,MAAM,WAAW,iBAAiB;IAC9B,MAAM,CAAC,EAAE,GAAG,CAAC;IACb,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,cAAc;IAC3B,WAAW,CAAC,OAAO,EAAE,GAAG,GAAG,IAAI,CAAC;CACnC;AAGD,MAAM,MAAM,cAAc,GAAG,CAAC,OAAO,EAAE,GAAG,KAAK,IAAI,CAAC;AAEpD,qBAAa,iBAAiB;IAC1B,OAAO,CAAC,EAAE,CAAY;IACtB,OAAO,CAAC,eAAe,CAAsB;gBAEjC,EAAE,EAAE,SAAS;IAezB;;;OAGG;IACH,WAAW,CAAC,OAAO,EAAE,gBAAgB,GAAG,IAAI;IAQ5C;;;;OAIG;IACH,mBAAmB,CAAC,QAAQ,EAAE,cAAc,GAAG;QAAE,OAAO,EAAE,MAAM,IAAI,CAAA;KAAE;CAMzE;AAGD,qBAAa,eAAe;IACjB,OAAO,EAAE,YAAY,CAAC;IAC7B,OAAO,CAAC,eAAe,CAAsB;gBAEjC,MAAM,CAAC,EAAE,GAAG;IAgBxB;;;OAGG;IACI,WAAW,CAAC,OAAO,EAAE,gBAAgB,GAAG,IAAI;IAInD;;;;OAIG;IACI,mBAAmB,CAAC,QAAQ,EAAE,cAAc,GAAG;QAAE,OAAO,EAAE,MAAM,IAAI,CAAA;KAAE;IAO7E;;;OAGG;IACU,gBAAgB,CAAC,SAAS,EAAE,UAAU;IAWnD;;;OAGG;IACU,SAAS;;;;;CAWzB"}
+{"version":3,"file":"adapter.d.ts","sourceRoot":"","sources":["../../src/hook/adapter.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAC/B,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AAEtC,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAI/C,MAAM,WAAW,gBAAgB;IAC7B,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,GAAG,CAAC;CACb;AAGD,MAAM,WAAW,iBAAiB;IAC9B,MAAM,CAAC,EAAE,GAAG,CAAC;IACb,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,cAAc;IAC3B,WAAW,CAAC,OAAO,EAAE,GAAG,GAAG,IAAI,CAAC;CACnC;AAGD,MAAM,MAAM,cAAc,GAAG,CAAC,OAAO,EAAE,GAAG,KAAK,IAAI,CAAC;AAEpD,qBAAa,iBAAiB;IAC1B,OAAO,CAAC,EAAE,CAAY;IACtB,OAAO,CAAC,eAAe,CAAsB;gBAEjC,EAAE,EAAE,SAAS;IAezB;;;OAGG;IACH,WAAW,CAAC,OAAO,EAAE,gBAAgB,GAAG,IAAI;IAQ5C;;;;OAIG;IACH,mBAAmB,CAAC,QAAQ,EAAE,cAAc,GAAG;QAAE,OAAO,EAAE,MAAM,IAAI,CAAA;KAAE;CAMzE;AAGD,qBAAa,eAAe;IACjB,OAAO,EAAE,YAAY,CAAC;IAC7B,OAAO,CAAC,eAAe,CAAsB;gBAEjC,MAAM,CAAC,EAAE,GAAG;IAgBxB;;;OAGG;IACI,WAAW,CAAC,OAAO,EAAE,gBAAgB,GAAG,IAAI;IAInD;;;;OAIG;IACI,mBAAmB,CAAC,QAAQ,EAAE,cAAc,GAAG;QAAE,OAAO,EAAE,MAAM,IAAI,CAAA;KAAE;IAO7E;;;OAGG;IACU,gBAAgB,CAAC,SAAS,EAAE,UAAU;IAWnD;;;OAGG;IACU,SAAS;;;;;;;;;;;CAUzB"}

package/service/hook/adapter.js

@@ -110,11 +110,10 @@ class TaskLoopAdapter {
         return __awaiter(this, void 0, void 0, function* () {
             const tools = yield (connect_service_1.client === null || connect_service_1.client === void 0 ? void 0 : connect_service_1.client.listTools());
             if (tools === null || tools === void 0 ? void 0 : tools.tools) {
-                return tools.tools.map((tool) => ({
-                    name: tool.name,
-                    description: tool.description,
-                    enabled: true
-                }));
+                return tools.tools.map((tool) => {
+                    const enabledTools = Object.assign(Object.assign({}, tool), { enabled: true });
+                    return enabledTools;
+                });
             }
             return [];
         });

package/task-loop.d.ts

@@ -24,6 +24,17 @@ export interface ToolCall {
     }
 }
 
+export interface ToolCallContent {
+    type: string;
+    text: string;
+	[key: string]: any;
+}
+
+export interface ToolCallResult {
+    state: MessageState;
+    content: ToolCallContent[];
+}
+
 export enum MessageState {
     ServerError = 'server internal error',
     ReceiveChunkError = 'receive chunk error',
@@ -58,6 +69,7 @@ export class TaskLoop {
     private onError;
     private onChunk;
     private onDone;
+    private onToolCalled;
     private onEpoch;
     private completionUsage;
     private llmConfig;
@@ -68,14 +80,46 @@ export class TaskLoop {
     private doConversation;
     makeChatData(tabStorage: any): ChatCompletionCreateParamsBase | undefined;
     abort(): void;
+
+    /**
+     * @description 注册 error 发生时触发的回调函数
+     * @param handler 
+     */
     registerOnError(handler: (msg: IErrorMssage) => void): void;
     registerOnChunk(handler: (chunk: ChatCompletionChunk) => void): void;
+
+    /**
+     * @description 注册 chat.completion 完成时触发的回调函数
+     * @param handler 
+     */
     registerOnDone(handler: () => void): void;
+
+    /**
+     * @description 注册每一个 epoch 开始时触发的回调函数
+     * @param handler 
+     */
     registerOnEpoch(handler: () => void): void;
-    setMaxEpochs(maxEpochs: number): void;
+
+    /**
+     * @description 注册当工具调用完成时的回调函数，会调用这个方法，可以拦截并修改 toolcall 的输出
+     * @param handler 
+     */
+    registerOnToolCalled(handler: (toolCallResult: ToolCallResult) => ToolCallResult): void;
+
+    /**
+     * @description 注册当工具调用前的回调函数，可以拦截并修改 toolcall 的输入
+     * @param handler 
+     */
+    registerOnToolCall(handler: (toolCall: ToolCall) => ToolCall): void;
+
+    /**
+     * @description 获取当前的 LLM 配置
+     */
+    getLlmConfig(): any;
+
     /**
      * @description 设置当前的 LLM 配置，用于 nodejs 环境运行
-     * @param config
+     * @param config 
      * @example
      * setLlmConfig({
      *     id: 'openai',
@@ -85,7 +129,12 @@ export class TaskLoop {
      * })
      */
     setLlmConfig(config: any): void;
-    getLlmConfig(): any;
+
+    /**
+     * @description 设置最大 epoch 次数
+     * @param maxEpochs 
+     */
+    setMaxEpochs(maxEpochs: number): void;
     bindStreaming(content: Ref<string>, toolCalls: Ref<ToolCall[]>): void;
     connectToService(): Promise<void>;
     /**