openmcp-sdk 0.0.1

package/service/hook/db.js

@@ -0,0 +1,179 @@
+"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;
+    };
+})();
+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) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
+    if (kind === "m") throw new TypeError("Private method is not writable");
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
+    return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
+};
+var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
+    return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+var _DiskStorage_storageHome;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.systemPromptDB = exports.ocrDB = exports.settingDB = exports.diskStorage = exports.LocalDB = void 0;
+const nedb_1 = __importDefault(require("@seald-io/nedb"));
+const os = __importStar(require("os"));
+const path = __importStar(require("path"));
+const fs = __importStar(require("fs"));
+const dbConnections = {};
+class LocalDB {
+    constructor(tableName) {
+        this.tableName = tableName;
+        this.init();
+    }
+    init() {
+        return __awaiter(this, void 0, void 0, function* () {
+            const homedir = os.homedir();
+            const dbPath = path.join(homedir, '.openmcp', 'nedb');
+            if (!fs.existsSync(dbPath)) {
+                fs.mkdirSync(dbPath, { recursive: true });
+            }
+            const filename = path.join(dbPath, `${this.tableName}.db`);
+            if (!dbConnections[filename]) {
+                dbConnections[filename] = new nedb_1.default({
+                    filename,
+                    autoload: true,
+                    timestampData: true
+                });
+            }
+            this.db = dbConnections[filename];
+            yield this.db.ensureIndex({ fieldName: 'id', unique: true });
+        });
+    }
+    insert(entity) {
+        return __awaiter(this, void 0, void 0, function* () {
+            yield this.db.update({ id: entity.id }, entity, { upsert: true });
+        });
+    }
+    findById(id) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return yield this.db.findOne({ id });
+        });
+    }
+    findAll() {
+        return __awaiter(this, void 0, void 0, function* () {
+            return yield this.db.find({});
+        });
+    }
+    delete(id) {
+        return __awaiter(this, void 0, void 0, function* () {
+            yield this.db.remove({ id });
+        });
+    }
+    close() {
+        return __awaiter(this, void 0, void 0, function* () {
+            // NeDB 不需要显式关闭
+        });
+    }
+}
+exports.LocalDB = LocalDB;
+class DiskStorage {
+    constructor() {
+        _DiskStorage_storageHome.set(this, void 0);
+        const homedir = os.homedir();
+        const imageStorageFolder = path.join(homedir, '.openmcp', 'storage');
+        // 确保存储目录存在
+        if (!fs.existsSync(imageStorageFolder)) {
+            fs.mkdirSync(imageStorageFolder, { recursive: true });
+        }
+        __classPrivateFieldSet(this, _DiskStorage_storageHome, imageStorageFolder, "f");
+    }
+    get(filename) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const filePath = path.join(__classPrivateFieldGet(this, _DiskStorage_storageHome, "f"), filename);
+            if (fs.existsSync(filePath)) {
+                return fs.promises.readFile(filePath);
+            }
+            return null;
+        });
+    }
+    set(filename, data, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const filePath = path.join(__classPrivateFieldGet(this, _DiskStorage_storageHome, "f"), filename);
+            yield fs.promises.writeFile(filePath, data, options);
+        });
+    }
+    delete(filename) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const filePath = path.join(__classPrivateFieldGet(this, _DiskStorage_storageHome, "f"), filename);
+            if (fs.existsSync(filePath)) {
+                yield fs.promises.unlink(filePath);
+            }
+        });
+    }
+    getSync(filename) {
+        const filePath = path.join(__classPrivateFieldGet(this, _DiskStorage_storageHome, "f"), filename);
+        if (fs.existsSync(filePath)) {
+            return fs.readFileSync(filePath);
+        }
+        return null;
+    }
+    setSync(filename, data, options) {
+        if (!fs.existsSync(__classPrivateFieldGet(this, _DiskStorage_storageHome, "f"))) {
+            fs.mkdirSync(__classPrivateFieldGet(this, _DiskStorage_storageHome, "f"), { recursive: true });
+        }
+        const filePath = path.join(__classPrivateFieldGet(this, _DiskStorage_storageHome, "f"), filename);
+        fs.writeFileSync(filePath, data, options);
+    }
+    getStoragePath(filename) {
+        return path.join(__classPrivateFieldGet(this, _DiskStorage_storageHome, "f"), filename);
+    }
+    deleteSync(filename) {
+        const filePath = path.join(__classPrivateFieldGet(this, _DiskStorage_storageHome, "f"), filename);
+        if (fs.existsSync(filePath)) {
+            fs.unlinkSync(filePath);
+        }
+    }
+}
+_DiskStorage_storageHome = new WeakMap();
+exports.diskStorage = new DiskStorage();
+exports.settingDB = new LocalDB('setting');
+exports.ocrDB = new LocalDB('ocr');
+exports.systemPromptDB = new LocalDB('systemPrompt');

package/service/hook/llm.d.ts

@@ -0,0 +1,13 @@
+export declare const llms: {
+    id: string;
+    name: string;
+    baseUrl: string;
+    models: string[];
+    provider: string;
+    isOpenAICompatible: boolean;
+    description: string;
+    website: string;
+    userToken: string;
+    userModel: string;
+}[];
+//# sourceMappingURL=llm.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"llm.d.ts","sourceRoot":"","sources":["../../src/hook/llm.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,IAAI;;;;;;;;;;;GAqFhB,CAAC"}

package/service/hook/llm.js

@@ -0,0 +1,89 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.llms = void 0;
+exports.llms = [
+    {
+        id: 'deepseek',
+        name: 'DeepSeek',
+        baseUrl: 'https://api.deepseek.com/v1',
+        models: ['deepseek-chat', 'deepseek-reasoner'],
+        provider: 'DeepSeek',
+        isOpenAICompatible: true,
+        description: '深度求索推出的大模型，擅长中文和代码',
+        website: 'https://www.deepseek.com',
+        userToken: '',
+        userModel: 'deepseek-chat'
+    },
+    {
+        id: 'openai',
+        name: 'OpenAI',
+        baseUrl: 'https://api.openai.com/v1',
+        models: ['gpt-4-turbo', 'gpt-4', 'gpt-3.5-turbo'],
+        provider: 'OpenAI',
+        isOpenAICompatible: true,
+        description: 'OpenAI官方API',
+        website: 'https://openai.com',
+        userToken: '',
+        userModel: 'gpt-4-turbo'
+    },
+    {
+        id: 'mistral',
+        name: 'Mistral',
+        baseUrl: 'https://api.mistral.ai/v1',
+        models: ['mistral-tiny', 'mistral-small', 'mistral-medium'],
+        provider: 'Mistral AI',
+        isOpenAICompatible: true,
+        description: '欧洲开源大模型代表',
+        website: 'https://mistral.ai',
+        userToken: '',
+        userModel: 'mistral-tiny'
+    },
+    {
+        id: 'ollama',
+        name: 'Ollama (Local)',
+        baseUrl: 'http://localhost:11434/v1',
+        models: ['llama2', 'mistral', 'codellama'],
+        provider: 'Ollama',
+        isOpenAICompatible: true,
+        description: '本地运行的大模型',
+        website: 'https://ollama.com',
+        userToken: '',
+        userModel: 'llama2'
+    },
+    {
+        id: 'groq',
+        name: 'Groq',
+        baseUrl: 'https://api.groq.com/openai/v1',
+        models: ['mixtral-8x7b-32768', 'llama2-70b-4096'],
+        provider: 'Groq',
+        isOpenAICompatible: true,
+        description: '超高速推理API',
+        website: 'https://groq.com',
+        userToken: '',
+        userModel: 'mixtral-8x7b-32768'
+    },
+    {
+        id: 'perplexity',
+        name: 'Perplexity',
+        baseUrl: 'https://api.perplexity.ai/v1',
+        models: ['pplx-7b-online', 'pplx-70b-online'],
+        provider: 'Perplexity AI',
+        isOpenAICompatible: true,
+        description: '联网搜索增强的大模型',
+        website: 'https://www.perplexity.ai',
+        userToken: '',
+        userModel: 'pplx-7b-online'
+    },
+    {
+        id: 'kimi',
+        name: 'Kimi Chat',
+        baseUrl: 'https://api.moonshot.cn/v1',
+        models: ['moonshot-v1-8k', 'moonshot-v1-32k', 'moonshot-v1-128k'],
+        provider: '月之暗面 (Moonshot AI)',
+        isOpenAICompatible: true,
+        description: '支持超长上下文的中文大模型，上下文窗口高达128K',
+        website: 'https://kimi.moonshot.cn',
+        userToken: '',
+        userModel: 'moonshot-v1-8k'
+    }
+];

package/service/hook/setting.d.ts

@@ -0,0 +1,5 @@
+export declare let VSCODE_WORKSPACE: string;
+export declare let RUNNING_CWD: string;
+export declare function setVscodeWorkspace(workspace: string): void;
+export declare function setRunningCWD(path: string): void;
+//# sourceMappingURL=setting.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"setting.d.ts","sourceRoot":"","sources":["../../src/hook/setting.ts"],"names":[],"mappings":"AAAA,eAAO,IAAI,gBAAgB,QAAK,CAAC;AACjC,eAAO,IAAI,WAAW,QAAK,CAAC;AAE5B,wBAAgB,kBAAkB,CAAC,SAAS,EAAE,MAAM,QAEnD;AAED,wBAAgB,aAAa,CAAC,IAAI,EAAE,MAAM,QAEzC"}

package/service/hook/setting.js

@@ -0,0 +1,13 @@
+"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;
+}
+function setRunningCWD(path) {
+    exports.RUNNING_CWD = path;
+}

package/service/hook/types.d.ts

@@ -0,0 +1,90 @@
+export interface IMessage {
+    type: string;
+    data: Record<string, unknown>;
+    timestamp?: number;
+}
+export type MessageHandler = (message: IMessage) => void;
+export interface SchemaProperty {
+    title: string;
+    type: string;
+}
+export interface InputSchema {
+    type: string;
+    properties: Record<string, SchemaProperty>;
+    required?: string[];
+    title?: string;
+}
+export interface Argument {
+    name: string;
+    required: boolean;
+}
+export interface Content {
+    uri: string;
+    mimeType: string;
+    text: string;
+}
+export interface MessageContent {
+    type: string;
+    text: string;
+}
+export interface ToolsListResponse {
+    tools: Array<{
+        name: string;
+        description: string;
+        inputSchema: InputSchema;
+    }>;
+}
+export interface PromptsListResponse {
+    prompts: Array<{
+        name: string;
+        description: string;
+        arguments: Argument[];
+    }>;
+}
+export interface ResourceTemplatesListResponse {
+    resourceTemplates: Array<{
+        uriTemplate: string;
+        name: string;
+        description: string;
+    }>;
+}
+export interface ResourcesListResponse {
+    resources: any[];
+}
+export interface ResourcesReadResponse {
+    contents: Content[];
+}
+export interface PromptsGetResponse {
+    messages: Array<{
+        role: string;
+        content: MessageContent;
+    }>;
+}
+export interface ToolListItem {
+    name: string;
+    description: string;
+    inputSchema: InputSchema;
+}
+export interface ToolsListResponse {
+    tools: ToolListItem[];
+}
+export interface BaseRequest {
+    method: string;
+    params: Record<string, any>;
+}
+export interface ResourcesReadRequest extends BaseRequest {
+    method: 'resources/read';
+    params: {
+        uri: string;
+    };
+}
+export interface PromptsGetRequest extends BaseRequest {
+    method: 'prompts/get';
+    params: {
+        name: string;
+        arguments: Record<string, any>;
+    };
+}
+export type APIResponse = ToolsListResponse | PromptsListResponse | ResourceTemplatesListResponse | ResourcesListResponse | ResourcesReadResponse | PromptsGetResponse;
+export type APIRequest = BaseRequest | ResourcesReadRequest | PromptsGetRequest;
+//# sourceMappingURL=types.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/hook/types.ts"],"names":[],"mappings":"AACA,MAAM,WAAW,QAAQ;IACrB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC9B,SAAS,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,MAAM,cAAc,GAAG,CAAC,OAAO,EAAE,QAAQ,KAAK,IAAI,CAAC;AAGzD,MAAM,WAAW,cAAc;IAC9B,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;CACb;AAED,MAAM,WAAW,WAAW;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;IAC3C,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;IACpB,KAAK,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,QAAQ;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,OAAO,CAAC;CAClB;AAED,MAAM,WAAW,OAAO;IACvB,GAAG,EAAE,MAAM,CAAC;IACZ,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;CACb;AAED,MAAM,WAAW,cAAc;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;CACb;AAGD,MAAM,WAAW,iBAAiB;IACjC,KAAK,EAAE,KAAK,CAAC;QACZ,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;QACpB,WAAW,EAAE,WAAW,CAAC;KACzB,CAAC,CAAC;CACH;AAED,MAAM,WAAW,mBAAmB;IACnC,OAAO,EAAE,KAAK,CAAC;QACd,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;QACpB,SAAS,EAAE,QAAQ,EAAE,CAAC;KACtB,CAAC,CAAC;CACH;AAED,MAAM,WAAW,6BAA6B;IAC7C,iBAAiB,EAAE,KAAK,CAAC;QACxB,WAAW,EAAE,MAAM,CAAC;QACpB,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;KACpB,CAAC,CAAC;CACH;AAED,MAAM,WAAW,qBAAqB;IACrC,SAAS,EAAE,GAAG,EAAE,CAAC;CACjB;AAED,MAAM,WAAW,qBAAqB;IACrC,QAAQ,EAAE,OAAO,EAAE,CAAC;CACpB;AAED,MAAM,WAAW,kBAAkB;IAClC,QAAQ,EAAE,KAAK,CAAC;QACf,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,cAAc,CAAC;KACxB,CAAC,CAAC;CACH;AAED,MAAM,WAAW,YAAY;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,WAAW,CAAC;CAC5B;AAED,MAAM,WAAW,iBAAiB;IAC9B,KAAK,EAAE,YAAY,EAAE,CAAC;CACzB;AAGD,MAAM,WAAW,WAAW;IAC3B,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC5B;AAED,MAAM,WAAW,oBAAqB,SAAQ,WAAW;IACxD,MAAM,EAAE,gBAAgB,CAAC;IACzB,MAAM,EAAE;QACP,GAAG,EAAE,MAAM,CAAC;KACZ,CAAC;CACF;AAED,MAAM,WAAW,iBAAkB,SAAQ,WAAW;IACrD,MAAM,EAAE,aAAa,CAAC;IACtB,MAAM,EAAE;QACP,IAAI,EAAE,MAAM,CAAC;QACb,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAC/B,CAAC;CACF;AAGD,MAAM,MAAM,WAAW,GACpB,iBAAiB,GACjB,mBAAmB,GACnB,6BAA6B,GAC7B,qBAAqB,GACrB,qBAAqB,GACrB,kBAAkB,CAAC;AAEtB,MAAM,MAAM,UAAU,GACnB,WAAW,GACX,oBAAoB,GACpB,iBAAiB,CAAC"}

package/service/hook/types.js

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

package/service/index.d.ts

@@ -0,0 +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';
+//# sourceMappingURL=index.d.ts.map

package/service/index.d.ts.map

@@ -0,0 +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"}

package/service/index.js

@@ -0,0 +1,14 @@
+"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; } });

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

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

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

@@ -0,0 +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"}

package/service/llm/llm.controller.js

@@ -0,0 +1,61 @@
+"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);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+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) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        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) {
+        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);
+            }
+            catch (error) {
+                console.log('error' + error);
+                webview.postMessage({
+                    command: 'llm/chat/completions/error',
+                    data: {
+                        msg: error
+                    }
+                });
+            }
+            return {
+                code: -1,
+                msg: 'terminate'
+            };
+        });
+    }
+    abortChatCompletion(client, data, webview) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return (0, llm_service_1.abortMessageService)(data, webview);
+        });
+    }
+}
+exports.LlmController = LlmController;
+__decorate([
+    (0, common_1.Controller)('llm/chat/completions')
+], LlmController.prototype, "chatCompletion", null);
+__decorate([
+    (0, common_1.Controller)('llm/chat/completions/abort')
+], LlmController.prototype, "abortChatCompletion", null);

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

@@ -0,0 +1,8 @@
+import { OpenAI } from "openai";
+export type MyMessageType = OpenAI.Chat.ChatCompletionMessageParam & {
+    extraInfo?: any;
+};
+export type MyToolMessageType = OpenAI.Chat.ChatCompletionToolMessageParam & {
+    extraInfo?: any;
+};
+//# sourceMappingURL=llm.dto.d.ts.map

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

@@ -0,0 +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"}

package/service/llm/llm.dto.js

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

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

@@ -0,0 +1,8 @@
+import { PostMessageble } from "../hook/adapter";
+import { MyMessageType } from "./llm.dto";
+import { RestfulResponse } from "../common/index.dto";
+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;
+export declare function postProcessMessages(messages: MyMessageType[]): Promise<void>;
+//# sourceMappingURL=llm.service.d.ts.map

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

@@ -0,0 +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"}