sema-core 1.0.2

package/dist/services/api/adapt/anthropic.js

@@ -0,0 +1,233 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MAIN_QUERY_TEMPERATURE = void 0;
+exports.queryAnthropic = queryAnthropic;
+const sdk_1 = __importDefault(require("@anthropic-ai/sdk"));
+const crypto_1 = require("crypto");
+const tools_1 = require("../../../tools/base/tools");
+const logLLM_1 = require("../../../util/logLLM");
+const log_1 = require("../../../util/log");
+const util_1 = require("./util");
+Object.defineProperty(exports, "MAIN_QUERY_TEMPERATURE", { enumerable: true, get: function () { return util_1.MAIN_QUERY_TEMPERATURE; } });
+// --- Core ---
+async function streamChat(params, signal, emitChunkEvents = false) {
+    const { url, headers = {}, body } = params;
+    // 从 headers 里提取 apiKey
+    const apiKey = headers["x-api-key"] || headers["Authorization"]?.replace("Bearer ", "") || "sk-placeholder";
+    const client = new sdk_1.default({
+        apiKey,
+        baseURL: url,
+        defaultHeaders: headers,
+    });
+    // --- 累积 stream chunks ---
+    let accumulatedText = '';
+    let accumulatedThinking = '';
+    let thinkingSignature = '';
+    const contentBlocks = [];
+    let messageId = '';
+    let stopReason = null;
+    let usage = {
+        input_tokens: 0,
+        output_tokens: 0,
+        cache_creation_input_tokens: 0,
+        cache_read_input_tokens: 0,
+    };
+    const eventBus = (0, util_1.getChunkEventBus)(emitChunkEvents);
+    // 使用 stream 方法
+    const stream = client.messages.stream({
+        ...body,
+        stream: true,
+    });
+    try {
+        for await (const event of stream) {
+            if (signal?.aborted) {
+                break; // 返回已累积的部分内容，而非抛出异常
+            }
+            switch (event.type) {
+                case 'message_start':
+                    messageId = event.message.id;
+                    break;
+                case 'content_block_start':
+                    if (event.content_block.type === 'tool_use') {
+                        contentBlocks[event.index] = { ...event.content_block, input: '' };
+                    }
+                    break;
+                case 'content_block_delta': {
+                    const delta = event.delta;
+                    if (delta.type === 'thinking_delta') {
+                        accumulatedThinking += delta.thinking;
+                        if (eventBus) {
+                            (0, util_1.emitChunkEvent)(eventBus, 'thinking', accumulatedThinking, delta.thinking);
+                        }
+                    }
+                    else if (delta.type === 'text_delta') {
+                        accumulatedText += delta.text;
+                        if (eventBus) {
+                            (0, util_1.emitChunkEvent)(eventBus, 'text', accumulatedText, delta.text);
+                        }
+                    }
+                    else if (delta.type === 'input_json_delta') {
+                        const block = contentBlocks[event.index];
+                        if (block && block.type === 'tool_use') {
+                            block.input = (block.input || '') + delta.partial_json;
+                        }
+                    }
+                    else if (delta.type === 'signature_delta') {
+                        thinkingSignature = delta.signature;
+                    }
+                    break;
+                }
+                case 'message_delta':
+                    stopReason = event.delta.stop_reason;
+                    if (event.usage) {
+                        const deltaUsage = event.usage;
+                        usage = {
+                            input_tokens: deltaUsage.input_tokens || 0,
+                            output_tokens: deltaUsage.output_tokens || 0,
+                            cache_creation_input_tokens: deltaUsage.cache_creation_input_tokens || 0,
+                            cache_read_input_tokens: deltaUsage.cache_read_input_tokens || 0,
+                        };
+                    }
+                    break;
+            }
+        }
+    }
+    catch (error) {
+        if (signal?.aborted) {
+            // 中断时不抛出，返回已累积的部分内容
+        }
+        else {
+            throw error;
+        }
+    }
+    // --- 构建结果 ---
+    const finalContentBlocks = [];
+    // 添加 thinking block（如果有）
+    if (accumulatedThinking) {
+        finalContentBlocks.push({
+            type: 'thinking',
+            thinking: accumulatedThinking,
+            signature: thinkingSignature,
+        });
+    }
+    // 添加 text block（如果有）
+    if (accumulatedText) {
+        finalContentBlocks.push({
+            type: 'text',
+            text: accumulatedText,
+        });
+    }
+    // 添加 tool_use blocks（解析 JSON input）
+    for (const block of contentBlocks) {
+        if (block && block.type === 'tool_use') {
+            const toolBlock = block;
+            finalContentBlocks.push({
+                type: 'tool_use',
+                id: toolBlock.id,
+                name: toolBlock.name,
+                input: typeof toolBlock.input === 'string'
+                    ? JSON.parse(toolBlock.input || '{}')
+                    : toolBlock.input,
+            });
+        }
+    }
+    return {
+        id: messageId,
+        type: 'message',
+        role: 'assistant',
+        model: body.model,
+        content: finalContentBlocks,
+        stop_reason: stopReason,
+        stop_sequence: null,
+        usage,
+    };
+}
+// ============================================================================
+// Anthropic 实现
+// ============================================================================
+async function queryAnthropic(messages, systemPromptContent, tools, signal, modelProfile, enableThinking, emitChunkEvents) {
+    const start = Date.now();
+    const baseURL = modelProfile.baseURL || 'https://api.anthropic.com';
+    // 构建 header
+    const headers = {
+        'Content-Type': 'application/json',
+    };
+    const apiKey = modelProfile.apiKey;
+    if (apiKey) {
+        headers['x-api-key'] = apiKey;
+    }
+    // 构建消息列表
+    const anthropicMessages = buildAnthropicMessages(messages, enableThinking);
+    // 转换工具定义
+    const anthropicTools = tools.length > 0 ? (0, tools_1.buildTools)(tools) : undefined;
+    // 构建请求参数
+    const requestBody = {
+        model: modelProfile.modelName,
+        messages: anthropicMessages,
+        system: systemPromptContent,
+        max_tokens: modelProfile.maxTokens,
+        temperature: util_1.MAIN_QUERY_TEMPERATURE,
+        stream: true,
+        ...(anthropicTools && { tools: anthropicTools }),
+    };
+    // 如果启用了 thinking，添加相关参数
+    if (enableThinking) {
+        requestBody.thinking = {
+            type: 'enabled',
+            budget_tokens: Math.min(Math.floor(modelProfile.maxTokens / 2) || 8000, 4000)
+        };
+    }
+    (0, logLLM_1.logLLMRequest)(requestBody);
+    // 统一使用 streamChat 处理请求
+    const parsedMessage = await streamChat({
+        url: baseURL,
+        headers,
+        body: requestBody,
+    }, signal, emitChunkEvents);
+    const durationMs = Date.now() - start;
+    // 转换为 AssistantMessage
+    return convertToAssistantMessage(parsedMessage, durationMs);
+}
+function buildAnthropicMessages(messages, enableThinking = false) {
+    const result = [];
+    messages.forEach(message => {
+        const role = message.message.role;
+        const content = message.message.content;
+        if (Array.isArray(content)) {
+            // 启用 thinking 时，必须保留 assistant 消息中的 thinking block，否则 API 会报错
+            // 未启用 thinking 时，需要过滤掉 thinking block
+            const filteredContent = enableThinking
+                ? content
+                : content.filter((block) => block.type !== 'thinking');
+            if (filteredContent.length > 0) {
+                result.push({
+                    role,
+                    content: filteredContent,
+                });
+            }
+        }
+        else {
+            result.push({
+                role,
+                content: content,
+            });
+        }
+    });
+    (0, log_1.logDebug)(`转换后的 Anthropic Messages: ${JSON.stringify(result, null, 2)}`);
+    return result;
+}
+/**
+ * 将 Anthropic.Message 转换为 AssistantMessage
+ */
+function convertToAssistantMessage(message, durationMs) {
+    return {
+        type: 'assistant',
+        uuid: (0, crypto_1.randomUUID)(),
+        durationMs,
+        message,
+    };
+}
+//# sourceMappingURL=anthropic.js.map

package/dist/services/api/adapt/anthropic.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../../../src/services/api/adapt/anthropic.ts"],"names":[],"mappings":";;;;;;AA2KA,wCA2DC;AAtOD,4DAAyC;AACzC,mCAAmC;AAInC,qDAAsD;AACtD,iDAAoD;AACpD,2CAA4C;AAC5C,iCAAiF;AAExE,uGAFA,6BAAsB,OAEA;AAU/B,eAAe;AAEf,KAAK,UAAU,UAAU,CACvB,MAAoB,EACpB,MAAoB,EACpB,kBAA2B,KAAK;IAEhC,MAAM,EAAE,GAAG,EAAE,OAAO,GAAG,EAAE,EAAE,IAAI,EAAE,GAAG,MAAM,CAAC;IAE3C,uBAAuB;IACvB,MAAM,MAAM,GACV,OAAO,CAAC,WAAW,CAAC,IAAI,OAAO,CAAC,eAAe,CAAC,EAAE,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,IAAI,gBAAgB,CAAC;IAE/F,MAAM,MAAM,GAAG,IAAI,aAAS,CAAC;QAC3B,MAAM;QACN,OAAO,EAAE,GAAG;QACZ,cAAc,EAAE,OAAO;KACxB,CAAC,CAAC;IAEH,2BAA2B;IAC3B,IAAI,eAAe,GAAG,EAAE,CAAC;IACzB,IAAI,mBAAmB,GAAG,EAAE,CAAC;IAC7B,IAAI,iBAAiB,GAAG,EAAE,CAAC;IAC3B,MAAM,aAAa,GAA6B,EAAE,CAAC;IACnD,IAAI,SAAS,GAAG,EAAE,CAAC;IACnB,IAAI,UAAU,GAAqC,IAAI,CAAC;IACxD,IAAI,KAAK,GAA+B;QACtC,YAAY,EAAE,CAAC;QACf,aAAa,EAAE,CAAC;QAChB,2BAA2B,EAAE,CAAC;QAC9B,uBAAuB,EAAE,CAAC;KAC3B,CAAC;IAEF,MAAM,QAAQ,GAAG,IAAA,uBAAgB,EAAC,eAAe,CAAC,CAAC;IAEnD,eAAe;IACf,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;QACpC,GAAG,IAAI;QACP,MAAM,EAAE,IAAI;KACb,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACjC,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;gBACpB,MAAM,CAAC,oBAAoB;YAC7B,CAAC;YAED,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;gBACnB,KAAK,eAAe;oBAClB,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC7B,MAAM;gBACR,KAAK,qBAAqB;oBACxB,IAAI,KAAK,CAAC,aAAa,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBAC5C,aAAa,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,aAAa,EAAE,KAAK,EAAE,EAAE,EAA4B,CAAC;oBAC/F,CAAC;oBACD,MAAM;gBACR,KAAK,qBAAqB,CAAC,CAAC,CAAC;oBAC3B,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;oBAC1B,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;wBACpC,mBAAmB,IAAI,KAAK,CAAC,QAAQ,CAAC;wBACtC,IAAI,QAAQ,EAAE,CAAC;4BACb,IAAA,qBAAc,EAAC,QAAQ,EAAE,UAAU,EAAE,mBAAmB,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;wBAC5E,CAAC;oBACH,CAAC;yBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;wBACvC,eAAe,IAAI,KAAK,CAAC,IAAI,CAAC;wBAC9B,IAAI,QAAQ,EAAE,CAAC;4BACb,IAAA,qBAAc,EAAC,QAAQ,EAAE,MAAM,EAAE,eAAe,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;wBAChE,CAAC;oBACH,CAAC;yBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;wBAC7C,MAAM,KAAK,GAAG,aAAa,CAAC,KAAK,CAAC,KAAK,CAAQ,CAAC;wBAChD,IAAI,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;4BACvC,KAAK,CAAC,KAAK,GAAG,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC,GAAG,KAAK,CAAC,YAAY,CAAC;wBACzD,CAAC;oBACH,CAAC;yBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;wBAC5C,iBAAiB,GAAG,KAAK,CAAC,SAAS,CAAC;oBACtC,CAAC;oBACD,MAAM;gBACR,CAAC;gBACD,KAAK,eAAe;oBAClB,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC;oBACrC,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;wBAChB,MAAM,UAAU,GAAG,KAAK,CAAC,KAAY,CAAC;wBACtC,KAAK,GAAG;4BACN,YAAY,EAAE,UAAU,CAAC,YAAY,IAAI,CAAC;4BAC1C,aAAa,EAAE,UAAU,CAAC,aAAa,IAAI,CAAC;4BAC5C,2BAA2B,EAAE,UAAU,CAAC,2BAA2B,IAAI,CAAC;4BACxE,uBAAuB,EAAE,UAAU,CAAC,uBAAuB,IAAI,CAAC;yBACjE,CAAC;oBACJ,CAAC;oBACD,MAAM;YACV,CAAC;QACH,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;YACpB,oBAAoB;QACtB,CAAC;aAAM,CAAC;YACN,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,eAAe;IACf,MAAM,kBAAkB,GAA6B,EAAE,CAAC;IAExD,yBAAyB;IACzB,IAAI,mBAAmB,EAAE,CAAC;QACxB,kBAAkB,CAAC,IAAI,CAAC;YACtB,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE,mBAAmB;YAC7B,SAAS,EAAE,iBAAiB;SACH,CAAC,CAAC;IAC/B,CAAC;IAED,qBAAqB;IACrB,IAAI,eAAe,EAAE,CAAC;QACpB,kBAAkB,CAAC,IAAI,CAAC;YACtB,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,eAAe;SACI,CAAC,CAAC;IAC/B,CAAC;IAED,oCAAoC;IACpC,KAAK,MAAM,KAAK,IAAI,aAAa,EAAE,CAAC;QAClC,IAAI,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YACvC,MAAM,SAAS,GAAG,KAAY,CAAC;YAC/B,kBAAkB,CAAC,IAAI,CAAC;gBACtB,IAAI,EAAE,UAAU;gBAChB,EAAE,EAAE,SAAS,CAAC,EAAE;gBAChB,IAAI,EAAE,SAAS,CAAC,IAAI;gBACpB,KAAK,EAAE,OAAO,SAAS,CAAC,KAAK,KAAK,QAAQ;oBACxC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,IAAI,IAAI,CAAC;oBACrC,CAAC,CAAC,SAAS,CAAC,KAAK;aACM,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAED,OAAO;QACL,EAAE,EAAE,SAAS;QACb,IAAI,EAAE,SAAS;QACf,IAAI,EAAE,WAAW;QACjB,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,OAAO,EAAE,kBAAkB;QAC3B,WAAW,EAAE,UAAU;QACvB,aAAa,EAAE,IAAI;QACnB,KAAK;KACN,CAAC;AACJ,CAAC;AAED,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAExE,KAAK,UAAU,cAAc,CAClC,QAA4C,EAC5C,mBAA0D,EAC1D,KAAa,EACb,MAAmB,EACnB,YAA0B,EAC1B,cAAuB,EACvB,eAAwB;IAExB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;IACxB,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,IAAI,2BAA2B,CAAA;IAEnE,YAAY;IACZ,MAAM,OAAO,GAA2B;QACtC,cAAc,EAAE,kBAAkB;KACnC,CAAA;IACD,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,CAAA;IAClC,IAAI,MAAM,EAAE,CAAC;QACX,OAAO,CAAC,WAAW,CAAC,GAAG,MAAM,CAAA;IAC/B,CAAC;IAED,SAAS;IACT,MAAM,iBAAiB,GAAG,sBAAsB,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAA;IAE1E,SAAS;IACT,MAAM,cAAc,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAA,kBAAU,EAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAA;IAEvE,SAAS;IACT,MAAM,WAAW,GAAyB;QACxC,KAAK,EAAE,YAAY,CAAC,SAAS;QAC7B,QAAQ,EAAE,iBAAiB;QAC3B,MAAM,EAAE,mBAAmB;QAC3B,UAAU,EAAE,YAAY,CAAC,SAAS;QAClC,WAAW,EAAE,6BAAsB;QACnC,MAAM,EAAE,IAAI;QACZ,GAAG,CAAC,cAAc,IAAI,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC;KACjD,CAAA;IAED,wBAAwB;IACxB,IAAI,cAAc,EAAE,CAAC;QACnB,WAAW,CAAC,QAAQ,GAAG;YACrB,IAAI,EAAE,SAAS;YACf,aAAa,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,SAAS,GAAG,CAAC,CAAC,IAAI,IAAI,EAAE,IAAI,CAAC;SAC9E,CAAA;IACH,CAAC;IAED,IAAA,sBAAa,EAAC,WAAW,CAAC,CAAA;IAE1B,uBAAuB;IACvB,MAAM,aAAa,GAAG,MAAM,UAAU,CAAC;QACrC,GAAG,EAAE,OAAO;QACZ,OAAO;QACP,IAAI,EAAE,WAAW;KAClB,EAAE,MAAM,EAAE,eAAe,CAAC,CAAA;IAE3B,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAA;IAErC,uBAAuB;IACvB,OAAO,yBAAyB,CAAC,aAAa,EAAE,UAAU,CAAC,CAAA;AAC7D,CAAC;AAED,SAAS,sBAAsB,CAC7B,QAA4C,EAC5C,iBAA0B,KAAK;IAE/B,MAAM,MAAM,GAA6B,EAAE,CAAA;IAE3C,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;QACzB,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC,IAA4B,CAAA;QACzD,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,OAAO,CAAA;QAEvC,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC3B,8DAA8D;YAC9D,sCAAsC;YACtC,MAAM,eAAe,GAAG,cAAc;gBACpC,CAAC,CAAC,OAAO;gBACT,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,KAAU,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,CAAC,CAAA;YAC7D,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC/B,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI;oBACJ,OAAO,EAAE,eAAe;iBACzB,CAAC,CAAA;YACJ,CAAC;QACH,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,IAAI,CAAC;gBACV,IAAI;gBACJ,OAAO,EAAE,OAAO;aACjB,CAAC,CAAA;QACJ,CAAC;IACH,CAAC,CAAC,CAAA;IAEF,IAAA,cAAQ,EAAC,4BAA4B,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAA;IACvE,OAAO,MAAM,CAAA;AACf,CAAC;AAED;;GAEG;AACH,SAAS,yBAAyB,CAChC,OAA0B,EAC1B,UAAkB;IAElB,OAAO;QACL,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,IAAA,mBAAU,GAAE;QAClB,UAAU;QACV,OAAO;KACR,CAAA;AACH,CAAC"}

package/dist/services/api/adapt/openai.d.ts

@@ -0,0 +1,8 @@
+import { UserMessage, AssistantMessage } from '../../../types/message';
+import { ModelProfile } from '../../../types/model';
+import { Tool } from '../../../tools/base/Tool';
+export declare function queryOpenAI(messages: (UserMessage | AssistantMessage)[], systemPromptContent: Array<{
+    type: 'text';
+    text: string;
+}>, tools: Tool[], signal: AbortSignal, modelProfile: ModelProfile, enableThinking: boolean, emitChunkEvents: boolean): Promise<AssistantMessage>;
+//# sourceMappingURL=openai.d.ts.map

package/dist/services/api/adapt/openai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../../../src/services/api/adapt/openai.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,WAAW,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAA;AACtE,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAA;AACnD,OAAO,EAAE,IAAI,EAAE,MAAM,0BAA0B,CAAA;AAmK/C,wBAAsB,WAAW,CAC/B,QAAQ,EAAE,CAAC,WAAW,GAAG,gBAAgB,CAAC,EAAE,EAC5C,mBAAmB,EAAE,KAAK,CAAC;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,CAAC,EAC1D,KAAK,EAAE,IAAI,EAAE,EACb,MAAM,EAAE,WAAW,EACnB,YAAY,EAAE,YAAY,EAC1B,cAAc,EAAE,OAAO,EACvB,eAAe,EAAE,OAAO,GACvB,OAAO,CAAC,gBAAgB,CAAC,CAwD3B"}

package/dist/services/api/adapt/openai.js

@@ -0,0 +1,412 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.queryOpenAI = queryOpenAI;
+const openai_1 = __importDefault(require("openai"));
+const crypto_1 = require("crypto");
+const nanoid_1 = require("nanoid");
+const tools_1 = require("../../../tools/base/tools");
+const logLLM_1 = require("../../../util/logLLM");
+const log_1 = require("../../../util/log");
+const adapter_1 = require("../../../util/adapter");
+const util_1 = require("./util");
+// --- Core ---
+async function streamChat(params, signal, emitChunkEvents = false) {
+    const { url, headers = {}, body } = params;
+    // 从 headers 里提取 apiKey，或直接用 Authorization
+    const apiKey = headers["Authorization"]?.replace("Bearer ", "") || "sk-placeholder";
+    const client = new openai_1.default({
+        apiKey,
+        baseURL: url,
+        defaultHeaders: headers,
+    });
+    // --- 累积 stream chunks ---
+    let accumulatedContent = '';
+    let accumulatedReasoning = '';
+    const toolCallsMap = new Map();
+    const eventBus = (0, util_1.getChunkEventBus)(emitChunkEvents);
+    // 使用 stream 方法，添加 stream_options 以获取 usage 信息
+    const stream = client.chat.completions.stream({
+        ...body,
+        stream: true,
+        stream_options: { include_usage: true },
+    });
+    try {
+        for await (const chunk of stream) {
+            if (signal?.aborted) {
+                break; // 返回已累积的部分内容，而非抛出异常
+            }
+            const delta = chunk.choices[0]?.delta;
+            if (!delta)
+                continue;
+            // content
+            if (delta.content) {
+                // 过滤 openrouter 的处理消息
+                if (delta.content.includes(':OPENROUTER PROCESSING')) {
+                    continue;
+                }
+                accumulatedContent += delta.content;
+                if (eventBus) {
+                    (0, util_1.emitChunkEvent)(eventBus, 'text', accumulatedContent, delta.content);
+                }
+            }
+            // reasoning_content（部分模型支持，如 o1/o3/deepseek）
+            if ("reasoning_content" in delta && delta.reasoning_content) {
+                accumulatedReasoning += delta.reasoning_content;
+                if (eventBus) {
+                    (0, util_1.emitChunkEvent)(eventBus, 'thinking', accumulatedReasoning, delta.reasoning_content);
+                }
+            }
+            // tool_calls
+            if (delta.tool_calls) {
+                for (const tc of delta.tool_calls) {
+                    const idx = tc.index;
+                    if (!toolCallsMap.has(idx)) {
+                        toolCallsMap.set(idx, { id: "", name: "", arguments: "" });
+                    }
+                    const entry = toolCallsMap.get(idx);
+                    if (tc.id)
+                        entry.id = tc.id;
+                    if (tc.function?.name)
+                        entry.name += tc.function.name;
+                    if (tc.function?.arguments)
+                        entry.arguments += tc.function.arguments;
+                }
+            }
+        }
+    }
+    catch (error) {
+        if (signal?.aborted) {
+            // 中断时不抛出，返回已累积的部分内容
+        }
+        else {
+            throw error;
+        }
+    }
+    // --- 获取最终响应信息（中断时跳过，避免挂起）---
+    let finalResponse = null;
+    if (!signal?.aborted) {
+        try {
+            finalResponse = await stream.finalChatCompletion();
+        }
+        catch {
+            // stream 已中断，忽略
+        }
+    }
+    // --- 构建结果，添加累积的 reasoning_content ---
+    const tool_calls = toolCallsMap.size
+        ? Array.from(toolCallsMap.entries())
+            .sort(([a], [b]) => a - b)
+            .map(([_index, tc]) => ({
+            id: tc.id,
+            type: "function",
+            function: { name: tc.name, arguments: tc.arguments },
+        }))
+        : undefined;
+    // 构建符合 OpenAI.ChatCompletion 格式的返回值
+    const message = {
+        role: 'assistant',
+        content: accumulatedContent || null,
+        refusal: null,
+    };
+    if (accumulatedReasoning) {
+        message.reasoning_content = accumulatedReasoning;
+    }
+    if (tool_calls && tool_calls.length > 0) {
+        message.tool_calls = tool_calls;
+    }
+    return {
+        id: finalResponse?.id || `chatcmpl-${(0, crypto_1.randomUUID)()}`,
+        object: 'chat.completion',
+        created: finalResponse?.created || Math.floor(Date.now() / 1000),
+        model: body.model,
+        choices: [{
+                index: 0,
+                message,
+                finish_reason: finalResponse?.choices[0]?.finish_reason || 'stop',
+                logprobs: null,
+            }],
+        usage: {
+            prompt_tokens: finalResponse?.usage?.prompt_tokens || 0,
+            completion_tokens: finalResponse?.usage?.completion_tokens || 0,
+            total_tokens: (finalResponse?.usage?.prompt_tokens || 0) + (finalResponse?.usage?.completion_tokens || 0),
+        },
+    };
+}
+// ============================================================================
+// OpenAI 实现
+// ============================================================================
+async function queryOpenAI(messages, systemPromptContent, tools, signal, modelProfile, enableThinking, emitChunkEvents) {
+    const start = Date.now();
+    let baseURL = modelProfile.baseURL || 'https://api.openai.com/v1';
+    if (modelProfile.provider !== 'glm' && !baseURL.endsWith('/v1')) {
+        baseURL = baseURL.replace(/\/$/, '') + '/v1';
+    }
+    // 构建 header
+    const headers = {
+        'Content-Type': 'application/json',
+    };
+    const apiKey = modelProfile.apiKey;
+    if (apiKey) {
+        headers['Authorization'] = `Bearer ${apiKey}`;
+    }
+    // 构建消息列表（Anthropic 转 OpenAI）
+    const openaiMessages = buildOpenAIMessages(messages, systemPromptContent, enableThinking);
+    // 转换工具定义
+    const openaiTools = tools.length > 0 ? convertToolsToOpenAI(tools) : undefined;
+    // 构建请求参数
+    const requestBody = {
+        model: modelProfile.modelName,
+        messages: openaiMessages,
+        stream: true,
+        ...(openaiTools && { tools: openaiTools }),
+        ...((0, adapter_1.useMaxCompletionTokens)(modelProfile.modelName)
+            ? { max_completion_tokens: modelProfile.maxTokens || 8000 }
+            : { max_tokens: modelProfile.maxTokens || 8000 }),
+        // 根据 provider 决定 reasoning 启用方式：
+        // - openai 官方：使用 reasoning_effort
+        // - 其他 provider（如 DeepSeek）：使用 thinking 参数
+        ...(modelProfile.provider === 'openai'
+            ? (enableThinking ? { reasoning_effort: 'medium' } : {})
+            : (enableThinking
+                ? { thinking: { type: "enabled" } }
+                : { thinking: { type: "disabled" } })),
+    };
+    (0, logLLM_1.logLLMRequest)(requestBody);
+    // 统一使用 streamChat 处理请求
+    const chatCompletion = await streamChat({
+        url: baseURL,
+        headers,
+        body: requestBody,
+    }, signal, emitChunkEvents);
+    const durationMs = Date.now() - start;
+    // 转换为 AssistantMessage
+    return convertToAssistantMessage(chatCompletion, durationMs);
+}
+// ============================================================================
+// 消息格式转换函数
+// ============================================================================
+function buildOpenAIMessages(messages, systemPromptContent, enableThinking = false) {
+    // 转换消息
+    const openaiMessages = convertAnthropicMessagesToOpenAI(messages, enableThinking);
+    // 加入系统提示
+    if (systemPromptContent.length > 0) {
+        const systemPrompt = systemPromptContent.map(item => item.text).filter(Boolean).join('\n\n');
+        if (systemPrompt) {
+            openaiMessages.unshift({
+                role: 'system',
+                content: systemPrompt
+            });
+        }
+    }
+    (0, log_1.logDebug)(`转换后的 OpenAI Messages: ${JSON.stringify(openaiMessages, null, 2)}`);
+    return openaiMessages;
+}
+function convertAnthropicMessagesToOpenAI(messages, enableThinking = false) {
+    const openaiMessages = [];
+    const toolResults = {};
+    function normalizeContentBlocks(content) {
+        if (typeof content === 'string') {
+            return [{ type: 'text', text: content }];
+        }
+        const blocks = Array.isArray(content) ? content : [content];
+        // 保留 thinking blocks，某些模型（如 DeepSeek）需要在历史消息中包含 reasoning_content
+        return blocks;
+    }
+    messages.forEach(message => {
+        const contentBlocks = normalizeContentBlocks(message.message.content);
+        if (message.message.role === 'assistant') {
+            processAssistantMessage(contentBlocks, openaiMessages, toolResults, enableThinking);
+        }
+        else {
+            processUserMessage(message, contentBlocks, openaiMessages, toolResults);
+        }
+    });
+    return buildFinalMessages(openaiMessages, toolResults);
+}
+function processAssistantMessage(contentBlocks, messages, toolResults, enableThinking = false) {
+    let textContent = '';
+    let reasoningContent = '';
+    const toolCalls = [];
+    contentBlocks.forEach(block => {
+        if (block.type === 'text') {
+            textContent += block.text;
+        }
+        else if (block.type === 'thinking' && enableThinking) {
+            // 启用 thinking 时，保留 reasoning_content，某些模型（如 DeepSeek）需要在历史消息中包含
+            reasoningContent += block.thinking;
+        }
+        else if (block.type === 'tool_use') {
+            toolCalls.push({
+                type: 'function',
+                function: {
+                    name: block.name,
+                    arguments: JSON.stringify(block.input),
+                },
+                id: block.id,
+            });
+        }
+        else if (block.type === 'tool_result') {
+            toolResults[block.tool_use_id] = createToolResultMessage(block);
+        }
+    });
+    if (textContent || toolCalls.length > 0 || reasoningContent) {
+        const message = { role: 'assistant' };
+        if (textContent) {
+            message.content = textContent;
+        }
+        // 某些模型（如 DeepSeek）要求历史消息中包含 reasoning_content
+        if (reasoningContent) {
+            message.reasoning_content = reasoningContent;
+        }
+        if (toolCalls.length > 0) {
+            message.tool_calls = toolCalls;
+            if (!textContent) {
+                message.content = null;
+            }
+        }
+        messages.push(message);
+    }
+}
+function processUserMessage(message, contentBlocks, messages, toolResults) {
+    const textBlocks = [];
+    contentBlocks.forEach(block => {
+        if (block.type === 'text') {
+            textBlocks.push(block.text);
+        }
+        else if (block.type === 'tool_result') {
+            toolResults[block.tool_use_id] = createToolResultMessage(block);
+        }
+    });
+    if (textBlocks.length > 0) {
+        messages.push({
+            role: message.message.role,
+            content: textBlocks.join('\n'),
+        });
+    }
+}
+function createToolResultMessage(block) {
+    const content = typeof block.content === 'string'
+        ? block.content
+        : JSON.stringify(block.content);
+    return {
+        role: 'tool',
+        content,
+        tool_call_id: block.tool_use_id,
+    };
+}
+function buildFinalMessages(messages, toolResults) {
+    const finalMessages = [];
+    messages.forEach(message => {
+        finalMessages.push(message);
+        if ('tool_calls' in message && message.tool_calls) {
+            message.tool_calls.forEach((toolCall) => {
+                if (toolResults[toolCall.id]) {
+                    finalMessages.push(toolResults[toolCall.id]);
+                }
+            });
+        }
+    });
+    return finalMessages;
+}
+// ============================================================================
+// 工具转换函数
+// ============================================================================
+function convertToolsToOpenAI(tools) {
+    const anthropicTools = (0, tools_1.buildTools)(tools);
+    return anthropicTools.map(tool => ({
+        type: 'function',
+        function: {
+            name: tool.name,
+            description: tool.description || '',
+            parameters: tool.input_schema
+        }
+    }));
+}
+// ============================================================================
+// 响应转换函数
+// ============================================================================
+/**
+ * 将 OpenAI 的 finish_reason 映射为 Anthropic 的 stop_reason
+ */
+function mapFinishReasonToStopReason(finishReason) {
+    switch (finishReason) {
+        case 'stop':
+            return 'end_turn';
+        case 'length':
+            return 'max_tokens';
+        case 'tool_calls':
+            return 'tool_use';
+        case 'content_filter':
+            return 'end_turn';
+        default:
+            return finishReason ? 'end_turn' : null;
+    }
+}
+/**
+ * 将 OpenAI.ChatCompletion 转换为 AssistantMessage
+ */
+function convertToAssistantMessage(chatCompletion, durationMs) {
+    const contentBlocks = [];
+    const choice = chatCompletion.choices[0];
+    const message = choice?.message;
+    // 处理 thinking/reasoning 内容
+    if (message?.reasoning_content) {
+        contentBlocks.push({
+            type: 'thinking',
+            thinking: message.reasoning_content,
+            signature: '',
+        });
+    }
+    // 处理工具调用
+    if (message?.tool_calls) {
+        message.tool_calls.forEach((toolCall) => {
+            if (toolCall.type === 'function') {
+                contentBlocks.push({
+                    type: 'tool_use',
+                    input: safeParseJSON(toolCall.function.arguments),
+                    name: toolCall.function.name,
+                    id: toolCall.id || (0, nanoid_1.nanoid)(),
+                });
+            }
+        });
+    }
+    // 处理文本内容
+    if (message?.content) {
+        contentBlocks.push({
+            type: 'text',
+            text: message.content,
+            citations: null,
+        });
+    }
+    return {
+        type: 'assistant',
+        uuid: (0, crypto_1.randomUUID)(),
+        durationMs,
+        message: {
+            id: chatCompletion.id,
+            type: 'message',
+            role: 'assistant',
+            content: contentBlocks,
+            stop_reason: mapFinishReasonToStopReason(choice?.finish_reason),
+            stop_sequence: null,
+            usage: {
+                input_tokens: chatCompletion.usage?.prompt_tokens || 0,
+                output_tokens: chatCompletion.usage?.completion_tokens || 0,
+            },
+        },
+    };
+}
+function safeParseJSON(jsonString) {
+    if (!jsonString)
+        return {};
+    try {
+        return JSON.parse(jsonString);
+    }
+    catch {
+        return {};
+    }
+}
+//# sourceMappingURL=openai.js.map