openmcp-sdk 0.0.3 → 0.0.5

package/service/hook/llm.js

@@ -1,7 +1,4 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.llms = void 0;
-exports.llms = [
+export const llms = [
     {
         id: 'deepseek',
         name: 'DeepSeek',
@@ -26,6 +23,54 @@ exports.llms = [
         userToken: '',
         userModel: 'gpt-4-turbo'
     },
+    {
+        id: 'qwen',
+        name: '通义千问 Qwen',
+        baseUrl: 'https://dashscope.aliyuncs.com/compatible-mode/v1',
+        models: ['qwen-max', 'qwen-plus', 'qwen-turbo', 'qwen-long', 'qwen-omni-turbo', 'qwen-omni-turbo-realtime'],
+        provider: 'Alibaba',
+        isOpenAICompatible: true,
+        description: '阿里巴巴通义千问',
+        website: 'https://help.aliyun.com/zh/model-studio/models#cfc131abafghw',
+        userToken: '',
+        userModel: 'qwen-plus'
+    },
+    {
+        id: 'doubao',
+        name: '豆包 Seed',
+        baseUrl: 'https://ark.cn-beijing.volces.com/api/v3',
+        models: ['doubao-1.5-pro-32k', 'doubao-1.5-pro-256k', 'doubao-1.5-lite', 'deepseek-v3'],
+        provider: 'bytedance',
+        isOpenAICompatible: true,
+        description: '字节跳动豆包 Seed',
+        website: 'https://help.aliyun.com/zh/model-studio/models#cfc131abafghw',
+        userToken: '',
+        userModel: 'doubao-1.5-pro-32k'
+    },
+    {
+        id: 'gemini',
+        name: 'Gemini',
+        baseUrl: 'https://generativelanguage.googleapis.com/v1beta/openai/',
+        models: ['gemini-2.0-flash', 'gemini-2.5-flash-preview-05-20', 'gemini-2.5-pro-preview-05-06'],
+        provider: 'google',
+        isOpenAICompatible: true,
+        description: 'Google Gemini',
+        website: 'https://ai.google.dev/gemini-api/docs/models?hl=zh-cn%2F%2Fgemini-2.5-pro-preview-05-06#gemini-2.5-pro-preview-05-06',
+        userToken: '',
+        userModel: 'gemini-2.0-flash'
+    },
+    {
+        id: 'grok',
+        name: 'Grok',
+        baseUrl: 'https://api.x.ai/v1',
+        models: ['grok-3', 'grok-3-fast', 'grok-3-mini', 'grok-3-mini-fast'],
+        provider: 'xai',
+        isOpenAICompatible: true,
+        description: 'xAI Grok',
+        website: 'https://docs.x.ai/docs/models',
+        userToken: '',
+        userModel: 'grok-3-mini'
+    },
     {
         id: 'mistral',
         name: 'Mistral',

package/service/hook/setting.js

@@ -1,13 +1,8 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.RUNNING_CWD = exports.VSCODE_WORKSPACE = void 0;
-exports.setVscodeWorkspace = setVscodeWorkspace;
-exports.setRunningCWD = setRunningCWD;
-exports.VSCODE_WORKSPACE = '';
-exports.RUNNING_CWD = '';
-function setVscodeWorkspace(workspace) {
-    exports.VSCODE_WORKSPACE = workspace;
+export let VSCODE_WORKSPACE = '';
+export let RUNNING_CWD = '';
+export function setVscodeWorkspace(workspace) {
+    VSCODE_WORKSPACE = workspace;
 }
-function setRunningCWD(path) {
-    exports.RUNNING_CWD = path;
+export function setRunningCWD(path) {
+    RUNNING_CWD = path;
 }

package/service/hook/types.js

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

package/service/hook/util.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=util.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"util.d.ts","sourceRoot":"","sources":["../../src/hook/util.ts"],"names":[],"mappings":""}

package/service/hook/util.js

@@ -0,0 +1 @@
+export {};

package/service/index.d.ts

@@ -1,5 +1,5 @@
-export { routeMessage } from './common/router';
-export { VSCodeWebViewLike, TaskLoopAdapter } from './hook/adapter';
-export { setVscodeWorkspace, setRunningCWD } from './hook/setting';
-export { client } from './mcp/connect.service';
+export { routeMessage } from './common/router.js';
+export { VSCodeWebViewLike, TaskLoopAdapter } from './hook/adapter.js';
+export { setVscodeWorkspace, setRunningCWD } from './hook/setting.js';
+export { clientMap } from './mcp/connect.service.js';
 //# sourceMappingURL=index.d.ts.map

package/service/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAC/C,OAAO,EAAE,iBAAiB,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AACpE,OAAO,EAAE,kBAAkB,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAEnE,OAAO,EAAE,MAAM,EAAE,MAAM,uBAAuB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,iBAAiB,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACvE,OAAO,EAAE,kBAAkB,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AACtE,OAAO,EAAE,SAAS,EAAE,MAAM,0BAA0B,CAAC"}

package/service/index.js

@@ -1,14 +1,4 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.client = exports.setRunningCWD = exports.setVscodeWorkspace = exports.TaskLoopAdapter = exports.VSCodeWebViewLike = exports.routeMessage = void 0;
-var router_1 = require("./common/router");
-Object.defineProperty(exports, "routeMessage", { enumerable: true, get: function () { return router_1.routeMessage; } });
-var adapter_1 = require("./hook/adapter");
-Object.defineProperty(exports, "VSCodeWebViewLike", { enumerable: true, get: function () { return adapter_1.VSCodeWebViewLike; } });
-Object.defineProperty(exports, "TaskLoopAdapter", { enumerable: true, get: function () { return adapter_1.TaskLoopAdapter; } });
-var setting_1 = require("./hook/setting");
-Object.defineProperty(exports, "setVscodeWorkspace", { enumerable: true, get: function () { return setting_1.setVscodeWorkspace; } });
-Object.defineProperty(exports, "setRunningCWD", { enumerable: true, get: function () { return setting_1.setRunningCWD; } });
-// TODO: 更加规范
-var connect_service_1 = require("./mcp/connect.service");
-Object.defineProperty(exports, "client", { enumerable: true, get: function () { return connect_service_1.client; } });
+export { routeMessage } from './common/router.js';
+export { VSCodeWebViewLike, TaskLoopAdapter } from './hook/adapter.js';
+export { setVscodeWorkspace, setRunningCWD } from './hook/setting.js';
+export { clientMap } from './mcp/connect.service.js';

package/service/llm/llm.controller.d.ts

@@ -1,10 +1,15 @@
-import { RequestClientType } from "../common";
-import { PostMessageble } from "../hook/adapter";
+import { RequestData } from "../common/index.dto.js";
+import { PostMessageble } from "../hook/adapter.js";
+import { OpenAI } from "openai";
 export declare class LlmController {
-    chatCompletion(client: RequestClientType, data: any, webview: PostMessageble): Promise<{
+    chatCompletion(data: RequestData, webview: PostMessageble): Promise<{
         code: number;
         msg: string;
     }>;
-    abortChatCompletion(client: RequestClientType, data: any, webview: PostMessageble): Promise<import("../common/index.dto").RestfulResponse>;
+    abortChatCompletion(data: RequestData, webview: PostMessageble): Promise<import("../common/index.dto.js").RestfulResponse>;
+    getModels(data: RequestData, webview: PostMessageble): Promise<{
+        code: number;
+        msg: OpenAI.Models.Model[];
+    }>;
 }
 //# sourceMappingURL=llm.controller.d.ts.map

package/service/llm/llm.controller.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"llm.controller.d.ts","sourceRoot":"","sources":["../../src/llm/llm.controller.ts"],"names":[],"mappings":"AAAA,OAAO,EAAc,iBAAiB,EAAE,MAAM,WAAW,CAAC;AAC1D,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAGjD,qBAAa,aAAa;IAGhB,cAAc,CAAC,MAAM,EAAE,iBAAiB,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE,cAAc;;;;IAgC5E,mBAAmB,CAAC,MAAM,EAAE,iBAAiB,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE,cAAc;CAI1F"}
+{"version":3,"file":"llm.controller.d.ts","sourceRoot":"","sources":["../../src/llm/llm.controller.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAGpD,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAEhC,qBAAa,aAAa;IAGhB,cAAc,CAAC,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,cAAc;;;;IAuBzD,mBAAmB,CAAC,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,cAAc;IAM9D,SAAS,CAAC,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,cAAc;;;;CAmB7D"}

package/service/llm/llm.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,22 +13,14 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.LlmController = void 0;
-const common_1 = require("../common");
-const llm_service_1 = require("./llm.service");
-class LlmController {
-    chatCompletion(client, data, webview) {
+import { Controller } from "../common/index.js";
+import { abortMessageService, streamingChatCompletion } from "./llm.service.js";
+import { OpenAI } from "openai";
+export class LlmController {
+    chatCompletion(data, webview) {
         return __awaiter(this, void 0, void 0, function* () {
-            let { tools = [] } = data;
-            if (tools.length > 0 && !client) {
-                return {
-                    code: 501,
-                    msg: 'mcp client 尚未连接'
-                };
-            }
             try {
-                yield (0, llm_service_1.streamingChatCompletion)(data, webview);
+                yield streamingChatCompletion(data, webview);
             }
             catch (error) {
                 console.log('error' + error);
@@ -46,16 +37,32 @@ class LlmController {
             };
         });
     }
-    abortChatCompletion(client, data, webview) {
+    abortChatCompletion(data, webview) {
         return __awaiter(this, void 0, void 0, function* () {
-            return (0, llm_service_1.abortMessageService)(data, webview);
+            return abortMessageService(data, webview);
+        });
+    }
+    getModels(data, webview) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const { baseURL, apiKey, proxyServer } = data;
+            const client = new OpenAI({
+                apiKey,
+                baseURL,
+            });
+            const models = yield client.models.list();
+            return {
+                code: 200,
+                msg: models.data
+            };
         });
     }
 }
-exports.LlmController = LlmController;
 __decorate([
-    (0, common_1.Controller)('llm/chat/completions')
+    Controller('llm/chat/completions')
 ], LlmController.prototype, "chatCompletion", null);
 __decorate([
-    (0, common_1.Controller)('llm/chat/completions/abort')
+    Controller('llm/chat/completions/abort')
 ], LlmController.prototype, "abortChatCompletion", null);
+__decorate([
+    Controller('llm/models')
+], LlmController.prototype, "getModels", null);

package/service/llm/llm.dto.d.ts

@@ -5,4 +5,19 @@ export type MyMessageType = OpenAI.Chat.ChatCompletionMessageParam & {
 export type MyToolMessageType = OpenAI.Chat.ChatCompletionToolMessageParam & {
     extraInfo?: any;
 };
+export interface OpenMcpChatOption {
+    baseURL: string;
+    apiKey: string;
+    model: string;
+    messages: any[];
+    temperature?: number;
+    tools?: any[];
+    parallelToolCalls?: boolean;
+}
+export interface MyStream<T> extends AsyncIterable<T> {
+    [Symbol.asyncIterator](): AsyncIterator<T>;
+    controller: {
+        abort(): void;
+    };
+}
 //# sourceMappingURL=llm.dto.d.ts.map

package/service/llm/llm.dto.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"llm.dto.d.ts","sourceRoot":"","sources":["../../src/llm/llm.dto.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAEhC,MAAM,MAAM,aAAa,GAAG,MAAM,CAAC,IAAI,CAAC,0BAA0B,GAAG;IACpE,SAAS,CAAC,EAAE,GAAG,CAAC;CAChB,CAAA;AAED,MAAM,MAAM,iBAAiB,GAAG,MAAM,CAAC,IAAI,CAAC,8BAA8B,GAAG;IAC5E,SAAS,CAAC,EAAE,GAAG,CAAC;CAChB,CAAA"}
+{"version":3,"file":"llm.dto.d.ts","sourceRoot":"","sources":["../../src/llm/llm.dto.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAEhC,MAAM,MAAM,aAAa,GAAG,MAAM,CAAC,IAAI,CAAC,0BAA0B,GAAG;IACpE,SAAS,CAAC,EAAE,GAAG,CAAC;CAChB,CAAA;AAED,MAAM,MAAM,iBAAiB,GAAG,MAAM,CAAC,IAAI,CAAC,8BAA8B,GAAG;IAC5E,SAAS,CAAC,EAAE,GAAG,CAAC;CAChB,CAAA;AAED,MAAM,WAAW,iBAAiB;IACjC,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,GAAG,EAAE,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,GAAG,EAAE,CAAC;IACd,iBAAiB,CAAC,EAAE,OAAO,CAAC;CAC5B;AAED,MAAM,WAAW,QAAQ,CAAC,CAAC,CAAE,SAAQ,aAAa,CAAC,CAAC,CAAC;IACjD,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,aAAa,CAAC,CAAC,CAAC,CAAC;IAC3C,UAAU,EAAE;QACR,KAAK,IAAI,IAAI,CAAC;KACjB,CAAC;CACL"}

package/service/llm/llm.dto.js

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

package/service/llm/llm.service.d.ts

@@ -1,6 +1,6 @@
-import { PostMessageble } from "../hook/adapter";
-import { MyMessageType } from "./llm.dto";
-import { RestfulResponse } from "../common/index.dto";
+import { PostMessageble } from "../hook/adapter.js";
+import { MyMessageType } from "./llm.dto.js";
+import { RestfulResponse } from "../common/index.dto.js";
 export declare let currentStream: AsyncIterable<any> | null;
 export declare function streamingChatCompletion(data: any, webview: PostMessageble): Promise<void>;
 export declare function abortMessageService(data: any, webview: PostMessageble): RestfulResponse;

package/service/llm/llm.service.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"llm.service.d.ts","sourceRoot":"","sources":["../../src/llm/llm.service.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAEjD,OAAO,EAAE,aAAa,EAAqB,MAAM,WAAW,CAAC;AAC7D,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAKtD,eAAO,IAAI,aAAa,EAAE,aAAa,CAAC,GAAG,CAAC,GAAG,IAAW,CAAC;AAE3D,wBAAsB,uBAAuB,CACzC,IAAI,EAAE,GAAG,EACT,OAAO,EAAE,cAAc,iBA0E1B;AAID,wBAAgB,mBAAmB,CAAC,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE,cAAc,GAAG,eAAe,CAYvF;AA0CD,wBAAsB,mBAAmB,CAAC,QAAQ,EAAE,aAAa,EAAE,iBAyBlE"}
+{"version":3,"file":"llm.service.d.ts","sourceRoot":"","sources":["../../src/llm/llm.service.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAEpD,OAAO,EAAE,aAAa,EAAqB,MAAM,cAAc,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAOzD,eAAO,IAAI,aAAa,EAAE,aAAa,CAAC,GAAG,CAAC,GAAG,IAAW,CAAC;AAE3D,wBAAsB,uBAAuB,CACzC,IAAI,EAAE,GAAG,EACT,OAAO,EAAE,cAAc,iBAwH1B;AAID,wBAAgB,mBAAmB,CAAC,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE,cAAc,GAAG,eAAe,CAYvF;AA0CD,wBAAsB,mBAAmB,CAAC,QAAQ,EAAE,aAAa,EAAE,iBAyBlE"}

package/service/llm/llm.service.js

@@ -1,4 +1,3 @@
-"use strict";
 var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
     function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
     return new (P || (P = Promise))(function (resolve, reject) {
@@ -15,45 +14,66 @@ var __asyncValues = (this && this.__asyncValues) || function (o) {
     function verb(n) { i[n] = o[n] && function (v) { return new Promise(function (resolve, reject) { v = o[n](v), settle(resolve, reject, v.done, v.value); }); }; }
     function settle(resolve, reject, d, v) { Promise.resolve(v).then(function(v) { resolve({ value: v, done: d }); }, reject); }
 };
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.currentStream = void 0;
-exports.streamingChatCompletion = streamingChatCompletion;
-exports.abortMessageService = abortMessageService;
-exports.postProcessMessages = postProcessMessages;
-const openai_1 = require("openai");
-const db_1 = require("../hook/db");
-const ocr_service_1 = require("../mcp/ocr.service");
-exports.currentStream = null;
-function streamingChatCompletion(data, webview) {
+import { OpenAI } from "openai";
+import { ocrDB } from "../hook/db.js";
+import { ocrWorkerStorage } from "../mcp/ocr.service.js";
+import { axiosFetch } from "../hook/axios-fetch.js";
+import Table from 'cli-table3';
+export let currentStream = null;
+export function streamingChatCompletion(data, webview) {
     return __awaiter(this, void 0, void 0, function* () {
         var _a, e_1, _b, _c;
-        let { baseURL, apiKey, model, messages, temperature, tools = [] } = data;
-        const client = new openai_1.OpenAI({
+        const { baseURL, apiKey, model, messages, temperature, tools = [], parallelToolCalls = true, proxyServer = '' } = data;
+        // 创建请求参数表格
+        const requestTable = new Table({
+            head: ['Parameter', 'Value'],
+            colWidths: [20, 40],
+            style: {
+                head: ['cyan'],
+                border: ['grey']
+            }
+        });
+        requestTable.push(['Model', model], ['Base URL', baseURL || 'Default'], ['Temperature', temperature], ['Tools Count', tools.length], ['Parallel Tool Calls', parallelToolCalls], ['Proxy Server', proxyServer || 'No Proxy']);
+        console.log('\nOpenAI Request Parameters:');
+        console.log(requestTable.toString());
+        const client = new OpenAI({
             baseURL,
-            apiKey
+            apiKey,
+            fetch: (input, init) => __awaiter(this, void 0, void 0, function* () {
+                if (model.startsWith('gemini') && init) {
+                    // 该死的 google
+                    init.headers = {
+                        'Content-Type': 'application/json',
+                        'Authorization': `Bearer ${apiKey}`
+                    };
+                }
+                return yield axiosFetch(input, init, { proxyServer });
+            })
         });
-        if (tools.length === 0) {
-            tools = undefined;
-        }
+        const seriableTools = (tools.length === 0) ? undefined : tools;
+        const seriableParallelToolCalls = (tools.length === 0) ?
+            undefined : model.startsWith('gemini') ? undefined : parallelToolCalls;
         yield postProcessMessages(messages);
+        // 使用表格渲染请求参数
+        console.log('seriableTools', seriableTools);
+        console.log('seriableParallelToolCalls', seriableParallelToolCalls);
         const stream = yield client.chat.completions.create({
             model,
             messages,
             temperature,
-            tools,
-            tool_choice: 'auto',
-            web_search_options: {},
+            tools: seriableTools,
+            parallel_tool_calls: seriableParallelToolCalls,
             stream: true
         });
         // 存储当前的流式传输对象
-        exports.currentStream = stream;
+        currentStream = stream;
         try {
             // 流式传输结果
             for (var _d = true, stream_1 = __asyncValues(stream), stream_1_1; stream_1_1 = yield stream_1.next(), _a = stream_1_1.done, !_a; _d = true) {
                 _c = stream_1_1.value;
                 _d = false;
                 const chunk = _c;
-                if (!exports.currentStream) {
+                if (!currentStream) {
                     // 如果流被中止，则停止循环
                     // TODO: 为每一个标签页设置不同的 currentStream 管理器
                     stream.controller.abort();
@@ -105,10 +125,10 @@ function streamingChatCompletion(data, webview) {
     });
 }
 // 处理中止消息的函数
-function abortMessageService(data, webview) {
-    if (exports.currentStream) {
+export function abortMessageService(data, webview) {
+    if (currentStream) {
         // 标记流已中止
-        exports.currentStream = null;
+        currentStream = null;
     }
     return {
         code: 200,
@@ -132,13 +152,13 @@ function postProcessToolMessages(message) {
                 // 2. 图片的 OCR 仍然在进行中
                 // 3. 图片已被删除
                 // rawContent.data 就是 filename
-                const result = yield db_1.ocrDB.findById(rawContent.data);
+                const result = yield ocrDB.findById(rawContent.data);
                 if (result) {
                     rawContent.text = result.text || '';
                 }
                 else if (rawContent._meta) {
                     const workerId = rawContent._meta.workerId;
-                    const worker = ocr_service_1.ocrWorkerStorage.get(workerId);
+                    const worker = ocrWorkerStorage.get(workerId);
                     if (worker) {
                         const text = yield worker.fut;
                         rawContent.text = text;
@@ -153,7 +173,7 @@ function postProcessToolMessages(message) {
         message.content = JSON.stringify(message.content);
     });
 }
-function postProcessMessages(messages) {
+export function postProcessMessages(messages) {
     return __awaiter(this, void 0, void 0, function* () {
         for (const message of messages) {
             // 去除 extraInfo 属性

package/service/main.d.ts

@@ -0,0 +1,7 @@
+export interface VSCodeMessage {
+    command: string;
+    data?: unknown;
+    callbackId?: string;
+}
+export type MessageHandler = (message: VSCodeMessage) => void;
+//# sourceMappingURL=main.d.ts.map

package/service/main.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"main.d.ts","sourceRoot":"","sources":["../src/main.ts"],"names":[],"mappings":"AAWA,MAAM,WAAW,aAAa;IAC1B,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,OAAO,CAAC;IACf,UAAU,CAAC,EAAE,MAAM,CAAC;CACvB;AAcD,MAAM,MAAM,cAAc,GAAG,CAAC,OAAO,EAAE,aAAa,KAAK,IAAI,CAAC"}

package/service/main.js

@@ -0,0 +1,107 @@
+import { WebSocketServer } from 'ws';
+import { pino } from 'pino';
+import { fileURLToPath } from 'url';
+import { dirname } from 'path';
+import { routeMessage } from './common/router.js';
+import { VSCodeWebViewLike } from './hook/adapter.js';
+import path from 'node:path';
+import * as fs from 'node:fs';
+import { setRunningCWD } from './hook/setting.js';
+const logger = pino({
+    transport: {
+        target: 'pino-pretty', // 启用 pino-pretty
+        options: {
+            colorize: true, // 开启颜色
+            levelFirst: true, // 先打印日志级别
+            translateTime: 'SYS:yyyy-mm-dd HH:MM:ss', // 格式化时间
+            ignore: 'pid,hostname', // 忽略部分字段
+        }
+    }
+});
+function refreshConnectionOption(envPath) {
+    const serverPath = path.join(__dirname, '..', '..', 'servers');
+    const defaultOption = {
+        connectionType: 'STDIO',
+        commandString: 'mcp run main.py',
+        cwd: serverPath
+    };
+    fs.writeFileSync(envPath, JSON.stringify(defaultOption, null, 4));
+    return { items: [defaultOption] };
+}
+function acquireConnectionOption() {
+    const envPath = path.join(__dirname, '..', '.env');
+    if (!fs.existsSync(envPath)) {
+        return refreshConnectionOption(envPath);
+    }
+    try {
+        const option = JSON.parse(fs.readFileSync(envPath, 'utf-8'));
+        if (!option.items) {
+            return refreshConnectionOption(envPath);
+        }
+        if (option.items && option.items.length === 0) {
+            return refreshConnectionOption(envPath);
+        }
+        // 按照前端的规范，整理成 commandString 样式
+        option.items = option.items.map((item) => {
+            var _a;
+            if (item.connectionType === 'STDIO') {
+                item.commandString = (_a = [item.command, ...item.args]) === null || _a === void 0 ? void 0 : _a.join(' ');
+            }
+            else {
+                item.url = item.url;
+            }
+            return item;
+        });
+        return option;
+    }
+    catch (error) {
+        logger.error('读取 .env 配置文件');
+        return refreshConnectionOption(envPath);
+    }
+}
+function updateConnectionOption(data) {
+    const envPath = path.join(__dirname, '..', '.env');
+    const connection = { items: data };
+    fs.writeFileSync(envPath, JSON.stringify(connection, null, 4));
+}
+const __dirname = dirname(fileURLToPath(import.meta.url));
+const devHome = path.join(__dirname, '..', '..');
+setRunningCWD(devHome);
+const wss = new WebSocketServer({ port: 8282 });
+console.log('listen on ws://localhost:8282');
+wss.on('connection', (ws) => {
+    // 仿造 webview 进行统一接口访问
+    const webview = new VSCodeWebViewLike(ws);
+    // 先发送成功建立的消息
+    webview.postMessage({
+        command: 'hello',
+        data: {
+            version: '0.0.1',
+            name: '消息桥连接完成'
+        }
+    });
+    const option = acquireConnectionOption();
+    // 注册消息接受的管线
+    webview.onDidReceiveMessage(message => {
+        logger.info(`command: [${message.command || 'No Command'}]`);
+        const { command, data } = message;
+        switch (command) {
+            case 'web/launch-signature':
+                const launchResult = {
+                    code: 200,
+                    msg: option.items
+                };
+                webview.postMessage({
+                    command: 'web/launch-signature',
+                    data: launchResult
+                });
+                break;
+            case 'web/update-connection-signature':
+                updateConnectionOption(data);
+                break;
+            default:
+                routeMessage(command, data, webview);
+                break;
+        }
+    });
+});

package/service/mcp/auth.service.d.ts

@@ -0,0 +1,23 @@
+import { OAuthClientProvider } from '@modelcontextprotocol/sdk/client/auth.js';
+export declare class OAuthClient {
+    port: number;
+    callbackUrl: string;
+    constructor();
+    /**
+     * @description 开启本地服务器上，并监听OAuth回调请求，并解析授权码或错误信息
+     * @returns {Promise<string>} 返回授权码
+     * @throws {Error} 如果没有收到授权码或发生错误
+     */
+    waitForOAuthCallback(): Promise<string>;
+    /**
+     * @description 获取Oauth认证provider
+     * @return {Promise<OAuthClientProvider>} 返回一个OAuthClientProvider实例
+     */
+    getOAuthProvider(): Promise<OAuthClientProvider>;
+    /**
+     * @description 打开浏览器
+     * @param url 授权URL
+     */
+    openBrowser(url: string): Promise<void>;
+}
+//# sourceMappingURL=auth.service.d.ts.map

package/service/mcp/auth.service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth.service.d.ts","sourceRoot":"","sources":["../../src/mcp/auth.service.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,mBAAmB,EAAE,MAAM,0CAA0C,CAAC;AAmE/E,qBAAa,WAAW;IACtB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;;IAUpB;;;;OAIG;IAEU,oBAAoB,IAAI,OAAO,CAAC,MAAM,CAAC;IA0DpD;;;OAGG;IACU,gBAAgB,IAAI,OAAO,CAAC,mBAAmB,CAAC;IAwB7D;;;OAGG;IAEU,WAAW,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAIrD"}