openmcp-sdk 0.0.3 → 0.0.5

package/service/setting/setting.controller.js

@@ -1,4 +1,3 @@
-"use strict";
 var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
     var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
     if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
@@ -14,14 +13,12 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.SettingController = void 0;
-const common_1 = require("../common");
-const setting_service_1 = require("./setting.service");
-class SettingController {
-    saveSetting(client, data, webview) {
+import { Controller } from "../common/index.js";
+import { getTour, loadSetting, saveSetting, setTour } from "./setting.service.js";
+export class SettingController {
+    saveSetting(data, webview) {
         return __awaiter(this, void 0, void 0, function* () {
-            (0, setting_service_1.saveSetting)(data);
+            saveSetting(data);
             console.log('Settings saved successfully');
             return {
                 code: 200,
@@ -29,30 +26,28 @@ class SettingController {
             };
         });
     }
-    loadSetting(client, data, webview) {
+    loadSetting(data, webview) {
         return __awaiter(this, void 0, void 0, function* () {
-            console.log('enter loading');
-            const config = (0, setting_service_1.loadSetting)();
+            const config = loadSetting();
             return {
                 code: 200,
                 msg: config
             };
         });
     }
-    setTourController(client, data, webview) {
+    setTourController(data, webview) {
         return __awaiter(this, void 0, void 0, function* () {
             const { userHasReadGuide } = data;
-            (0, setting_service_1.setTour)(userHasReadGuide);
+            setTour(userHasReadGuide);
             return {
                 code: 200,
                 msg: 'setTour success'
             };
         });
     }
-    getTourController(client, data, webview) {
+    getTourController(data, webview) {
         return __awaiter(this, void 0, void 0, function* () {
-            console.log('enter');
-            const { userHasReadGuide } = (0, setting_service_1.getTour)();
+            const { userHasReadGuide } = getTour();
             return {
                 code: 200,
                 msg: {
@@ -62,16 +57,15 @@ class SettingController {
         });
     }
 }
-exports.SettingController = SettingController;
 __decorate([
-    (0, common_1.Controller)('setting/save')
+    Controller('setting/save')
 ], SettingController.prototype, "saveSetting", null);
 __decorate([
-    (0, common_1.Controller)('setting/load')
+    Controller('setting/load')
 ], SettingController.prototype, "loadSetting", null);
 __decorate([
-    (0, common_1.Controller)('setting/set-tour')
+    Controller('setting/set-tour')
 ], SettingController.prototype, "setTourController", null);
 __decorate([
-    (0, common_1.Controller)('setting/get-tour')
+    Controller('setting/get-tour')
 ], SettingController.prototype, "getTourController", null);

package/service/setting/setting.dto.js

@@ -1,2 +1 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
+export {};

package/service/setting/setting.service.d.ts

@@ -1,4 +1,4 @@
-import { IConfig } from './setting.dto';
+import { IConfig } from './setting.dto.js';
 export declare function loadSetting(): IConfig;
 export declare function saveSetting(config: Partial<IConfig>): void;
 export declare function getTour(): {

package/service/setting/setting.service.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"setting.service.d.ts","sourceRoot":"","sources":["../../src/setting/setting.service.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAyCxC,wBAAgB,WAAW,IAAI,OAAO,CAcrC;AAED,wBAAgB,WAAW,CAAC,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,GAAG,IAAI,CAU1D;AAED,wBAAgB,OAAO;;EAatB;AAED,wBAAgB,OAAO,CAAC,gBAAgB,EAAE,OAAO,GAAG,IAAI,CASvD"}
+{"version":3,"file":"setting.service.d.ts","sourceRoot":"","sources":["../../src/setting/setting.service.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAyC3C,wBAAgB,WAAW,IAAI,OAAO,CAkBrC;AAED,wBAAgB,WAAW,CAAC,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,GAAG,IAAI,CAU1D;AAED,wBAAgB,OAAO;;EAatB;AAED,wBAAgB,OAAO,CAAC,gBAAgB,EAAE,OAAO,GAAG,IAAI,CASvD"}

package/service/setting/setting.service.js

@@ -1,46 +1,7 @@
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || (function () {
-    var ownKeys = function(o) {
-        ownKeys = Object.getOwnPropertyNames || function (o) {
-            var ar = [];
-            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
-            return ar;
-        };
-        return ownKeys(o);
-    };
-    return function (mod) {
-        if (mod && mod.__esModule) return mod;
-        var result = {};
-        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
-        __setModuleDefault(result, mod);
-        return result;
-    };
-})();
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.loadSetting = loadSetting;
-exports.saveSetting = saveSetting;
-exports.getTour = getTour;
-exports.setTour = setTour;
-const fs = __importStar(require("fs"));
-const os = __importStar(require("os"));
-const path = __importStar(require("path"));
-const llm_1 = require("../hook/llm");
+import * as fs from 'fs';
+import * as os from 'os';
+import * as path from 'path';
+import { llms } from '../hook/llm.js';
 function getConfigurationPath() {
     const homeDir = os.homedir();
     const configDir = path.join(homeDir, '.openmcp');
@@ -57,7 +18,7 @@ function getDefaultLanguage() {
 }
 const DEFAULT_CONFIG = {
     MODEL_INDEX: 0,
-    LLM_INFO: llm_1.llms,
+    LLM_INFO: llms,
     LANG: getDefaultLanguage(), MCP_TIMEOUT_SEC: 60
 };
 function createConfig() {
@@ -71,21 +32,25 @@ function createConfig() {
     fs.writeFileSync(configPath, JSON.stringify(DEFAULT_CONFIG, null, 2), 'utf-8');
     return DEFAULT_CONFIG;
 }
-function loadSetting() {
+export function loadSetting() {
     const configPath = getConfigurationPath();
     if (!fs.existsSync(configPath)) {
         return createConfig();
     }
     try {
         const configData = fs.readFileSync(configPath, 'utf-8');
-        return JSON.parse(configData);
+        const config = JSON.parse(configData);
+        if (!config.LLM_INFO || (Array.isArray(config.LLM_INFO) && config.LLM_INFO.length === 0)) {
+            config.LLM_INFO = llms;
+        }
+        return config;
     }
     catch (error) {
         console.error('Error loading config file, creating new one:', error);
         return createConfig();
     }
 }
-function saveSetting(config) {
+export function saveSetting(config) {
     const configPath = getConfigurationPath();
     console.log('save to ' + configPath);
     try {
@@ -96,7 +61,7 @@ function saveSetting(config) {
         throw error;
     }
 }
-function getTour() {
+export function getTour() {
     const configPath = getConfigurationPath();
     const KEY = path.join(path.dirname(configPath), 'KEY');
     console.log(KEY);
@@ -109,7 +74,7 @@ function getTour() {
         userHasReadGuide: true
     };
 }
-function setTour(userHasReadGuide) {
+export function setTour(userHasReadGuide) {
     const configPath = getConfigurationPath();
     const KEY = path.join(path.dirname(configPath), 'KEY');
     if (userHasReadGuide) {

package/task-loop.d.ts

@@ -10,6 +10,29 @@ export interface TaskLoopOptions {
     adapter?: any;
 }
 
+
+export interface SchemaProperty {
+	title: string;
+	type: string;
+	description?: string;
+}
+
+export interface InputSchema {
+	type: string;
+	properties: Record<string, SchemaProperty>;
+	required?: string[];
+	title?: string;
+	$defs?: any;
+}
+
+export interface ToolItem {
+	name: string;
+	description: string;
+	inputSchema: InputSchema;
+    enabled: boolean;
+	anyOf?: any;
+}
+
 export type Ref<T> = {
     value: T;
 };
@@ -24,6 +47,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 +92,7 @@ export class TaskLoop {
     private onError;
     private onChunk;
     private onDone;
+    private onToolCalled;
     private onEpoch;
     private completionUsage;
     private llmConfig;
@@ -68,14 +103,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,9 +152,26 @@ 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>;
+
+    /**
+     * @description 设置代理服务器
+     * @param proxyServer 
+     */
+    setProxyServer(proxyServer: string): void;
+
+    /**
+     * @description 获取所有可用的工具列表
+     */
+    listTools(): Promise<ToolItem[]>;
+
     /**
      * @description 开启循环，异步更新 DOM
      */