agentdog 1.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +22 -0
- package/README.md +244 -0
- package/dist/adapters/adapter-factory.d.ts +21 -0
- package/dist/adapters/adapter-factory.d.ts.map +1 -0
- package/dist/adapters/adapter-factory.js +157 -0
- package/dist/adapters/adapter-factory.js.map +1 -0
- package/dist/adapters/http-sse-adapter.d.ts +49 -0
- package/dist/adapters/http-sse-adapter.d.ts.map +1 -0
- package/dist/adapters/http-sse-adapter.js +485 -0
- package/dist/adapters/http-sse-adapter.js.map +1 -0
- package/dist/adapters/stdio-adapter.d.ts +69 -0
- package/dist/adapters/stdio-adapter.d.ts.map +1 -0
- package/dist/adapters/stdio-adapter.js +664 -0
- package/dist/adapters/stdio-adapter.js.map +1 -0
- package/dist/adapters/streamable-http-adapter.d.ts +38 -0
- package/dist/adapters/streamable-http-adapter.d.ts.map +1 -0
- package/dist/adapters/streamable-http-adapter.js +321 -0
- package/dist/adapters/streamable-http-adapter.js.map +1 -0
- package/dist/ai-gateway/anthropic-errors.d.ts +4 -0
- package/dist/ai-gateway/anthropic-errors.d.ts.map +1 -0
- package/dist/ai-gateway/anthropic-errors.js +17 -0
- package/dist/ai-gateway/anthropic-errors.js.map +1 -0
- package/dist/ai-gateway/auth.d.ts +3 -0
- package/dist/ai-gateway/auth.d.ts.map +1 -0
- package/dist/ai-gateway/auth.js +14 -0
- package/dist/ai-gateway/auth.js.map +1 -0
- package/dist/ai-gateway/auth.test.d.ts +2 -0
- package/dist/ai-gateway/auth.test.d.ts.map +1 -0
- package/dist/ai-gateway/auth.test.js +57 -0
- package/dist/ai-gateway/auth.test.js.map +1 -0
- package/dist/ai-gateway/count-tokens.d.ts +5 -0
- package/dist/ai-gateway/count-tokens.d.ts.map +1 -0
- package/dist/ai-gateway/count-tokens.js +37 -0
- package/dist/ai-gateway/count-tokens.js.map +1 -0
- package/dist/ai-gateway/count-tokens.test.d.ts +2 -0
- package/dist/ai-gateway/count-tokens.test.d.ts.map +1 -0
- package/dist/ai-gateway/count-tokens.test.js +19 -0
- package/dist/ai-gateway/count-tokens.test.js.map +1 -0
- package/dist/ai-gateway/dialects.d.ts +4 -0
- package/dist/ai-gateway/dialects.d.ts.map +1 -0
- package/dist/ai-gateway/dialects.js +15 -0
- package/dist/ai-gateway/dialects.js.map +1 -0
- package/dist/ai-gateway/gateway-server.d.ts +15 -0
- package/dist/ai-gateway/gateway-server.d.ts.map +1 -0
- package/dist/ai-gateway/gateway-server.js +121 -0
- package/dist/ai-gateway/gateway-server.js.map +1 -0
- package/dist/ai-gateway/integration.test.d.ts +2 -0
- package/dist/ai-gateway/integration.test.d.ts.map +1 -0
- package/dist/ai-gateway/integration.test.js +205 -0
- package/dist/ai-gateway/integration.test.js.map +1 -0
- package/dist/ai-gateway/ir/anthropic-in.d.ts +3 -0
- package/dist/ai-gateway/ir/anthropic-in.d.ts.map +1 -0
- package/dist/ai-gateway/ir/anthropic-in.js +121 -0
- package/dist/ai-gateway/ir/anthropic-in.js.map +1 -0
- package/dist/ai-gateway/ir/anthropic-in.test.d.ts +2 -0
- package/dist/ai-gateway/ir/anthropic-in.test.d.ts.map +1 -0
- package/dist/ai-gateway/ir/anthropic-in.test.js +93 -0
- package/dist/ai-gateway/ir/anthropic-in.test.js.map +1 -0
- package/dist/ai-gateway/ir/anthropic-sse.d.ts +22 -0
- package/dist/ai-gateway/ir/anthropic-sse.d.ts.map +1 -0
- package/dist/ai-gateway/ir/anthropic-sse.js +200 -0
- package/dist/ai-gateway/ir/anthropic-sse.js.map +1 -0
- package/dist/ai-gateway/ir/anthropic-sse.test.d.ts +2 -0
- package/dist/ai-gateway/ir/anthropic-sse.test.d.ts.map +1 -0
- package/dist/ai-gateway/ir/anthropic-sse.test.js +101 -0
- package/dist/ai-gateway/ir/anthropic-sse.test.js.map +1 -0
- package/dist/ai-gateway/ir/gemini-out.d.ts +20 -0
- package/dist/ai-gateway/ir/gemini-out.d.ts.map +1 -0
- package/dist/ai-gateway/ir/gemini-out.js +218 -0
- package/dist/ai-gateway/ir/gemini-out.js.map +1 -0
- package/dist/ai-gateway/ir/gemini-out.test.d.ts +2 -0
- package/dist/ai-gateway/ir/gemini-out.test.d.ts.map +1 -0
- package/dist/ai-gateway/ir/gemini-out.test.js +125 -0
- package/dist/ai-gateway/ir/gemini-out.test.js.map +1 -0
- package/dist/ai-gateway/ir/openai-out.d.ts +17 -0
- package/dist/ai-gateway/ir/openai-out.d.ts.map +1 -0
- package/dist/ai-gateway/ir/openai-out.js +226 -0
- package/dist/ai-gateway/ir/openai-out.js.map +1 -0
- package/dist/ai-gateway/ir/openai-out.test.d.ts +2 -0
- package/dist/ai-gateway/ir/openai-out.test.d.ts.map +1 -0
- package/dist/ai-gateway/ir/openai-out.test.js +144 -0
- package/dist/ai-gateway/ir/openai-out.test.js.map +1 -0
- package/dist/ai-gateway/ir/types.d.ts +77 -0
- package/dist/ai-gateway/ir/types.d.ts.map +1 -0
- package/dist/ai-gateway/ir/types.js +4 -0
- package/dist/ai-gateway/ir/types.js.map +1 -0
- package/dist/ai-gateway/ir-channel.d.ts +15 -0
- package/dist/ai-gateway/ir-channel.d.ts.map +1 -0
- package/dist/ai-gateway/ir-channel.js +120 -0
- package/dist/ai-gateway/ir-channel.js.map +1 -0
- package/dist/ai-gateway/model-router.d.ts +11 -0
- package/dist/ai-gateway/model-router.d.ts.map +1 -0
- package/dist/ai-gateway/model-router.js +27 -0
- package/dist/ai-gateway/model-router.js.map +1 -0
- package/dist/ai-gateway/model-router.test.d.ts +2 -0
- package/dist/ai-gateway/model-router.test.d.ts.map +1 -0
- package/dist/ai-gateway/model-router.test.js +43 -0
- package/dist/ai-gateway/model-router.test.js.map +1 -0
- package/dist/ai-gateway/passthrough.d.ts +9 -0
- package/dist/ai-gateway/passthrough.d.ts.map +1 -0
- package/dist/ai-gateway/passthrough.js +81 -0
- package/dist/ai-gateway/passthrough.js.map +1 -0
- package/dist/ai-gateway/passthrough.test.d.ts +2 -0
- package/dist/ai-gateway/passthrough.test.d.ts.map +1 -0
- package/dist/ai-gateway/passthrough.test.js +108 -0
- package/dist/ai-gateway/passthrough.test.js.map +1 -0
- package/dist/ai-gateway/upstream/model-lister.d.ts +13 -0
- package/dist/ai-gateway/upstream/model-lister.d.ts.map +1 -0
- package/dist/ai-gateway/upstream/model-lister.js +61 -0
- package/dist/ai-gateway/upstream/model-lister.js.map +1 -0
- package/dist/ai-gateway/upstream/provider-client.d.ts +20 -0
- package/dist/ai-gateway/upstream/provider-client.d.ts.map +1 -0
- package/dist/ai-gateway/upstream/provider-client.js +76 -0
- package/dist/ai-gateway/upstream/provider-client.js.map +1 -0
- package/dist/ai-gateway/upstream/provider-client.test.d.ts +2 -0
- package/dist/ai-gateway/upstream/provider-client.test.d.ts.map +1 -0
- package/dist/ai-gateway/upstream/provider-client.test.js +69 -0
- package/dist/ai-gateway/upstream/provider-client.test.js.map +1 -0
- package/dist/cli/cli-main.d.ts +7 -0
- package/dist/cli/cli-main.d.ts.map +1 -0
- package/dist/cli/cli-main.js +116 -0
- package/dist/cli/cli-main.js.map +1 -0
- package/dist/cli/cli-router.d.ts +20 -0
- package/dist/cli/cli-router.d.ts.map +1 -0
- package/dist/cli/cli-router.js +271 -0
- package/dist/cli/cli-router.js.map +1 -0
- package/dist/cli/cli-utils.d.ts +30 -0
- package/dist/cli/cli-utils.d.ts.map +1 -0
- package/dist/cli/cli-utils.js +219 -0
- package/dist/cli/cli-utils.js.map +1 -0
- package/dist/cli/commands/audit-commands.d.ts +30 -0
- package/dist/cli/commands/audit-commands.d.ts.map +1 -0
- package/dist/cli/commands/audit-commands.js +417 -0
- package/dist/cli/commands/audit-commands.js.map +1 -0
- package/dist/cli/commands/autostart-command.d.ts +21 -0
- package/dist/cli/commands/autostart-command.d.ts.map +1 -0
- package/dist/cli/commands/autostart-command.js +243 -0
- package/dist/cli/commands/autostart-command.js.map +1 -0
- package/dist/cli/commands/autostart-command.test.d.ts +2 -0
- package/dist/cli/commands/autostart-command.test.d.ts.map +1 -0
- package/dist/cli/commands/autostart-command.test.js +35 -0
- package/dist/cli/commands/autostart-command.test.js.map +1 -0
- package/dist/cli/commands/config-commands.d.ts +30 -0
- package/dist/cli/commands/config-commands.d.ts.map +1 -0
- package/dist/cli/commands/config-commands.js +1013 -0
- package/dist/cli/commands/config-commands.js.map +1 -0
- package/dist/cli/commands/daemon-commands.d.ts +76 -0
- package/dist/cli/commands/daemon-commands.d.ts.map +1 -0
- package/dist/cli/commands/daemon-commands.js +451 -0
- package/dist/cli/commands/daemon-commands.js.map +1 -0
- package/dist/cli/commands/detect-commands.d.ts +17 -0
- package/dist/cli/commands/detect-commands.d.ts.map +1 -0
- package/dist/cli/commands/detect-commands.js +293 -0
- package/dist/cli/commands/detect-commands.js.map +1 -0
- package/dist/cli/commands/diagnose-commands.d.ts +19 -0
- package/dist/cli/commands/diagnose-commands.d.ts.map +1 -0
- package/dist/cli/commands/diagnose-commands.js +326 -0
- package/dist/cli/commands/diagnose-commands.js.map +1 -0
- package/dist/cli/commands/optimize-commands.d.ts +14 -0
- package/dist/cli/commands/optimize-commands.d.ts.map +1 -0
- package/dist/cli/commands/optimize-commands.js +189 -0
- package/dist/cli/commands/optimize-commands.js.map +1 -0
- package/dist/cli/commands/proxy-command.d.ts +27 -0
- package/dist/cli/commands/proxy-command.d.ts.map +1 -0
- package/dist/cli/commands/proxy-command.js +212 -0
- package/dist/cli/commands/proxy-command.js.map +1 -0
- package/dist/cli/commands/start-command.d.ts +22 -0
- package/dist/cli/commands/start-command.d.ts.map +1 -0
- package/dist/cli/commands/start-command.js +275 -0
- package/dist/cli/commands/start-command.js.map +1 -0
- package/dist/config/config-manager.d.ts +125 -0
- package/dist/config/config-manager.d.ts.map +1 -0
- package/dist/config/config-manager.js +569 -0
- package/dist/config/config-manager.js.map +1 -0
- package/dist/config/config-manager.test.d.ts +2 -0
- package/dist/config/config-manager.test.d.ts.map +1 -0
- package/dist/config/config-manager.test.js +209 -0
- package/dist/config/config-manager.test.js.map +1 -0
- package/dist/core/auto-config-generator.d.ts +55 -0
- package/dist/core/auto-config-generator.d.ts.map +1 -0
- package/dist/core/auto-config-generator.js +278 -0
- package/dist/core/auto-config-generator.js.map +1 -0
- package/dist/core/mcpdog-server.d.ts +49 -0
- package/dist/core/mcpdog-server.d.ts.map +1 -0
- package/dist/core/mcpdog-server.js +552 -0
- package/dist/core/mcpdog-server.js.map +1 -0
- package/dist/core/protocol-detector.d.ts +56 -0
- package/dist/core/protocol-detector.d.ts.map +1 -0
- package/dist/core/protocol-detector.js +365 -0
- package/dist/core/protocol-detector.js.map +1 -0
- package/dist/daemon/daemon-client.d.ts +34 -0
- package/dist/daemon/daemon-client.d.ts.map +1 -0
- package/dist/daemon/daemon-client.js +130 -0
- package/dist/daemon/daemon-client.js.map +1 -0
- package/dist/daemon/daemon-client.test.d.ts +2 -0
- package/dist/daemon/daemon-client.test.d.ts.map +1 -0
- package/dist/daemon/daemon-client.test.js +77 -0
- package/dist/daemon/daemon-client.test.js.map +1 -0
- package/dist/daemon/daemon-info.d.ts +7 -0
- package/dist/daemon/daemon-info.d.ts.map +1 -0
- package/dist/daemon/daemon-info.js +34 -0
- package/dist/daemon/daemon-info.js.map +1 -0
- package/dist/daemon/daemon-info.test.d.ts +2 -0
- package/dist/daemon/daemon-info.test.d.ts.map +1 -0
- package/dist/daemon/daemon-info.test.js +57 -0
- package/dist/daemon/daemon-info.test.js.map +1 -0
- package/dist/daemon/daemon-web-server.ai.test.d.ts +2 -0
- package/dist/daemon/daemon-web-server.ai.test.d.ts.map +1 -0
- package/dist/daemon/daemon-web-server.ai.test.js +119 -0
- package/dist/daemon/daemon-web-server.ai.test.js.map +1 -0
- package/dist/daemon/daemon-web-server.d.ts +59 -0
- package/dist/daemon/daemon-web-server.d.ts.map +1 -0
- package/dist/daemon/daemon-web-server.js +1236 -0
- package/dist/daemon/daemon-web-server.js.map +1 -0
- package/dist/daemon/daemon-web-server.test.d.ts +2 -0
- package/dist/daemon/daemon-web-server.test.d.ts.map +1 -0
- package/dist/daemon/daemon-web-server.test.js +93 -0
- package/dist/daemon/daemon-web-server.test.js.map +1 -0
- package/dist/daemon/mcpdog-daemon.d.ts +42 -0
- package/dist/daemon/mcpdog-daemon.d.ts.map +1 -0
- package/dist/daemon/mcpdog-daemon.js +302 -0
- package/dist/daemon/mcpdog-daemon.js.map +1 -0
- package/dist/daemon/stdio-proxy.d.ts +17 -0
- package/dist/daemon/stdio-proxy.d.ts.map +1 -0
- package/dist/daemon/stdio-proxy.js +143 -0
- package/dist/daemon/stdio-proxy.js.map +1 -0
- package/dist/index.d.ts +14 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +198 -0
- package/dist/index.js.map +1 -0
- package/dist/intent-analyzer.d.ts +27 -0
- package/dist/intent-analyzer.d.ts.map +1 -0
- package/dist/intent-analyzer.js +140 -0
- package/dist/intent-analyzer.js.map +1 -0
- package/dist/logging/server-log-manager.d.ts +75 -0
- package/dist/logging/server-log-manager.d.ts.map +1 -0
- package/dist/logging/server-log-manager.js +185 -0
- package/dist/logging/server-log-manager.js.map +1 -0
- package/dist/mcp-registry.d.ts +35 -0
- package/dist/mcp-registry.d.ts.map +1 -0
- package/dist/mcp-registry.js +166 -0
- package/dist/mcp-registry.js.map +1 -0
- package/dist/middleware/auth.d.ts +5 -0
- package/dist/middleware/auth.d.ts.map +1 -0
- package/dist/middleware/auth.js +81 -0
- package/dist/middleware/auth.js.map +1 -0
- package/dist/mock-tool-database.d.ts +41 -0
- package/dist/mock-tool-database.d.ts.map +1 -0
- package/dist/mock-tool-database.js +461 -0
- package/dist/mock-tool-database.js.map +1 -0
- package/dist/proxy-server-manager.d.ts +11 -0
- package/dist/proxy-server-manager.d.ts.map +1 -0
- package/dist/proxy-server-manager.js +144 -0
- package/dist/proxy-server-manager.js.map +1 -0
- package/dist/router/tool-router.d.ts +61 -0
- package/dist/router/tool-router.d.ts.map +1 -0
- package/dist/router/tool-router.js +539 -0
- package/dist/router/tool-router.js.map +1 -0
- package/dist/router/tool-router.test.d.ts +2 -0
- package/dist/router/tool-router.test.d.ts.map +1 -0
- package/dist/router/tool-router.test.js +113 -0
- package/dist/router/tool-router.test.js.map +1 -0
- package/dist/streamable-http-server.d.ts +27 -0
- package/dist/streamable-http-server.d.ts.map +1 -0
- package/dist/streamable-http-server.js +302 -0
- package/dist/streamable-http-server.js.map +1 -0
- package/dist/tool-finder.d.ts +21 -0
- package/dist/tool-finder.d.ts.map +1 -0
- package/dist/tool-finder.js +86 -0
- package/dist/tool-finder.js.map +1 -0
- package/dist/tool-recommender.d.ts +23 -0
- package/dist/tool-recommender.d.ts.map +1 -0
- package/dist/tool-recommender.js +200 -0
- package/dist/tool-recommender.js.map +1 -0
- package/dist/types/index.d.ts +145 -0
- package/dist/types/index.d.ts.map +1 -0
- package/dist/types/index.js +2 -0
- package/dist/types/index.js.map +1 -0
- package/dist/utils/server-name-validator.d.ts +47 -0
- package/dist/utils/server-name-validator.d.ts.map +1 -0
- package/dist/utils/server-name-validator.js +156 -0
- package/dist/utils/server-name-validator.js.map +1 -0
- package/dist/web/web-server.d.ts +44 -0
- package/dist/web/web-server.d.ts.map +1 -0
- package/dist/web/web-server.js +500 -0
- package/dist/web/web-server.js.map +1 -0
- package/package.json +68 -0
- package/web/dist/assets/index-DMVhVtNU.css +1 -0
- package/web/dist/assets/index-VMXIgcOH.js +92 -0
- package/web/dist/assets/index-VMXIgcOH.js.map +1 -0
- package/web/dist/index.html +14 -0
|
@@ -0,0 +1,218 @@
|
|
|
1
|
+
// IR → Gemini generateContent 请求体;Gemini SSE/JSON 响应 → IR 事件流
|
|
2
|
+
// 约束(设计文档 §6.2):
|
|
3
|
+
// - functionResponse.response 必须为 object(text→{content},isError→{error})
|
|
4
|
+
// - tool_result 的函数名需从历史 assistant 消息按 tool_use_id 回查
|
|
5
|
+
// - tools 的 JSON Schema 需净化(剥离 $schema 等 Gemini 不支持的 draft 元数据键;additionalProperties 保留)
|
|
6
|
+
import { randomUUID } from 'crypto';
|
|
7
|
+
import { geminiGenerateUrl } from '../upstream/model-lister.js';
|
|
8
|
+
// ===== 请求方向 =====
|
|
9
|
+
export function irToGemini(ir, provider, model) {
|
|
10
|
+
// tool_use_id → 函数名回查表(从历史 assistant 消息)
|
|
11
|
+
const nameByToolUseId = new Map();
|
|
12
|
+
for (const msg of ir.messages) {
|
|
13
|
+
if (msg.role === 'assistant') {
|
|
14
|
+
for (const part of msg.content) {
|
|
15
|
+
if (part.kind === 'toolUse') {
|
|
16
|
+
nameByToolUseId.set(part.id, part.name);
|
|
17
|
+
}
|
|
18
|
+
}
|
|
19
|
+
}
|
|
20
|
+
}
|
|
21
|
+
const contents = [];
|
|
22
|
+
for (const msg of ir.messages) {
|
|
23
|
+
const role = msg.role === 'assistant' ? 'model' : 'user';
|
|
24
|
+
const parts = [];
|
|
25
|
+
const functionResponses = [];
|
|
26
|
+
for (const part of msg.content) {
|
|
27
|
+
if (part.kind === 'text') {
|
|
28
|
+
parts.push({ text: part.text });
|
|
29
|
+
}
|
|
30
|
+
else if (part.kind === 'image') {
|
|
31
|
+
parts.push({ inlineData: { mimeType: part.mime, data: part.data } }); // base64 无前缀
|
|
32
|
+
}
|
|
33
|
+
else if (part.kind === 'toolUse') {
|
|
34
|
+
parts.push({ functionCall: { name: part.name, args: part.args ?? {} } });
|
|
35
|
+
}
|
|
36
|
+
else if (part.kind === 'toolResult') {
|
|
37
|
+
const fname = nameByToolUseId.get(part.toolUseId) || 'unknown';
|
|
38
|
+
functionResponses.push({
|
|
39
|
+
functionResponse: {
|
|
40
|
+
name: fname,
|
|
41
|
+
// response 必须为 object
|
|
42
|
+
response: part.isError ? { error: part.content } : { content: part.content },
|
|
43
|
+
},
|
|
44
|
+
});
|
|
45
|
+
}
|
|
46
|
+
}
|
|
47
|
+
// tool_result 提为独立 user 轮的 functionResponse parts(紧随 model functionCall 之后)
|
|
48
|
+
if (functionResponses.length > 0) {
|
|
49
|
+
if (parts.length > 0) {
|
|
50
|
+
contents.push({ role, parts });
|
|
51
|
+
}
|
|
52
|
+
contents.push({ role: 'user', parts: functionResponses });
|
|
53
|
+
}
|
|
54
|
+
else {
|
|
55
|
+
if (parts.length > 0) {
|
|
56
|
+
contents.push({ role, parts });
|
|
57
|
+
}
|
|
58
|
+
}
|
|
59
|
+
}
|
|
60
|
+
const body = { contents };
|
|
61
|
+
if (ir.system.length > 0) {
|
|
62
|
+
body.systemInstruction = { parts: ir.system.map(p => ({ text: p.text })) };
|
|
63
|
+
}
|
|
64
|
+
if (ir.tools.length > 0) {
|
|
65
|
+
body.tools = [{
|
|
66
|
+
functionDeclarations: ir.tools.map(t => ({
|
|
67
|
+
name: t.name,
|
|
68
|
+
description: t.description,
|
|
69
|
+
parameters: sanitizeSchema(t.inputSchema),
|
|
70
|
+
})),
|
|
71
|
+
}];
|
|
72
|
+
}
|
|
73
|
+
if (ir.toolChoice) {
|
|
74
|
+
const toolConfig = { functionCallingConfig: {} };
|
|
75
|
+
switch (ir.toolChoice.mode) {
|
|
76
|
+
case 'auto':
|
|
77
|
+
toolConfig.functionCallingConfig.mode = 'AUTO';
|
|
78
|
+
break;
|
|
79
|
+
case 'required':
|
|
80
|
+
toolConfig.functionCallingConfig.mode = 'ANY';
|
|
81
|
+
break;
|
|
82
|
+
case 'none':
|
|
83
|
+
toolConfig.functionCallingConfig.mode = 'NONE';
|
|
84
|
+
break;
|
|
85
|
+
case 'specific':
|
|
86
|
+
toolConfig.functionCallingConfig.mode = 'ANY';
|
|
87
|
+
toolConfig.functionCallingConfig.allowedFunctionNames = [ir.toolChoice.name];
|
|
88
|
+
break;
|
|
89
|
+
}
|
|
90
|
+
body.toolConfig = toolConfig;
|
|
91
|
+
}
|
|
92
|
+
const gen = { maxOutputTokens: ir.sampling.maxTokens };
|
|
93
|
+
if (ir.sampling.temperature !== undefined) {
|
|
94
|
+
gen.temperature = ir.sampling.temperature;
|
|
95
|
+
}
|
|
96
|
+
if (ir.sampling.topP !== undefined) {
|
|
97
|
+
gen.topP = ir.sampling.topP;
|
|
98
|
+
}
|
|
99
|
+
if (ir.sampling.topK !== undefined) {
|
|
100
|
+
gen.topK = ir.sampling.topK; // gemini 支持,可保留
|
|
101
|
+
}
|
|
102
|
+
if (ir.sampling.stopSequences?.length) {
|
|
103
|
+
gen.stopSequences = ir.sampling.stopSequences;
|
|
104
|
+
}
|
|
105
|
+
body.generationConfig = gen;
|
|
106
|
+
return {
|
|
107
|
+
url: geminiGenerateUrl(provider.baseUrl, model, ir.stream),
|
|
108
|
+
headers: { 'content-type': 'application/json', 'x-goog-api-key': provider.apiKey, ...(provider.headers || {}) },
|
|
109
|
+
body,
|
|
110
|
+
};
|
|
111
|
+
}
|
|
112
|
+
// 剥离 Gemini 不支持的 JSON Schema draft 元数据键($schema 等);additionalProperties 官方已支持、保留
|
|
113
|
+
const GEMINI_UNSUPPORTED_KEYS = new Set(['$schema', '$id', '$defs', '$ref', '$comment', 'definitions']);
|
|
114
|
+
export function sanitizeSchema(schema) {
|
|
115
|
+
if (Array.isArray(schema)) {
|
|
116
|
+
return schema.map(sanitizeSchema);
|
|
117
|
+
}
|
|
118
|
+
if (schema && typeof schema === 'object') {
|
|
119
|
+
const out = {};
|
|
120
|
+
for (const [key, value] of Object.entries(schema)) {
|
|
121
|
+
if (GEMINI_UNSUPPORTED_KEYS.has(key)) {
|
|
122
|
+
continue;
|
|
123
|
+
}
|
|
124
|
+
out[key] = sanitizeSchema(value);
|
|
125
|
+
}
|
|
126
|
+
return out;
|
|
127
|
+
}
|
|
128
|
+
return schema;
|
|
129
|
+
}
|
|
130
|
+
// ===== 响应方向 =====
|
|
131
|
+
// gemini SSE(alt=sse)每帧为完整 GenerateContentResponse JSON;
|
|
132
|
+
// functionCall 为非增量全量 args;id 可选——上游给了就透传,缺省合成 call_<uuid>
|
|
133
|
+
// 并记 name↔id 映射(Claude Code 回传 tool_result 按 tool_use_id 回查函数名)
|
|
134
|
+
export class GeminiStreamState {
|
|
135
|
+
nameById = new Map();
|
|
136
|
+
finishEmitted = false;
|
|
137
|
+
lookupName(id) {
|
|
138
|
+
return this.nameById.get(id);
|
|
139
|
+
}
|
|
140
|
+
frameToEvents(frame) {
|
|
141
|
+
const events = [];
|
|
142
|
+
const candidate = frame.candidates?.[0];
|
|
143
|
+
for (const part of candidate?.content?.parts || []) {
|
|
144
|
+
if (typeof part.text === 'string' && part.text && part.thought !== true) {
|
|
145
|
+
events.push({ type: 'textDelta', text: part.text });
|
|
146
|
+
}
|
|
147
|
+
if (part.functionCall && part.functionCall.name) {
|
|
148
|
+
const id = part.functionCall.id || `call_${randomUUID()}`;
|
|
149
|
+
this.nameById.set(id, part.functionCall.name);
|
|
150
|
+
events.push({ type: 'toolCallStart', id, name: part.functionCall.name });
|
|
151
|
+
events.push({ type: 'toolCallDelta', argsDelta: JSON.stringify(part.functionCall.args ?? {}) });
|
|
152
|
+
events.push({ type: 'toolCallEnd' });
|
|
153
|
+
}
|
|
154
|
+
// thought: true 的 part 丢弃
|
|
155
|
+
}
|
|
156
|
+
if (candidate?.finishReason && !this.finishEmitted) {
|
|
157
|
+
this.finishEmitted = true;
|
|
158
|
+
events.push({ type: 'finish', reason: mapGeminiFinish(candidate.finishReason) });
|
|
159
|
+
}
|
|
160
|
+
const usage = frame.usageMetadata;
|
|
161
|
+
if (usage) {
|
|
162
|
+
events.push({
|
|
163
|
+
type: 'usage',
|
|
164
|
+
input: usage.promptTokenCount ?? 0,
|
|
165
|
+
output: usage.candidatesTokenCount ?? 0,
|
|
166
|
+
});
|
|
167
|
+
}
|
|
168
|
+
return events;
|
|
169
|
+
}
|
|
170
|
+
}
|
|
171
|
+
export function mapGeminiFinish(reason) {
|
|
172
|
+
// gemini 不显式区分工具调用结束:functionCall 出现时由调用方把 stop 覆盖为 tool_calls(见 frameToEvents 调用方)
|
|
173
|
+
if (reason === 'MAX_TOKENS') {
|
|
174
|
+
return 'length';
|
|
175
|
+
}
|
|
176
|
+
return 'stop';
|
|
177
|
+
}
|
|
178
|
+
// 非流式响应 → IR 事件
|
|
179
|
+
export function geminiFinalToIREvents(json) {
|
|
180
|
+
const state = new GeminiStreamState();
|
|
181
|
+
const events = state.frameToEvents(json);
|
|
182
|
+
// 非流式含 functionCall 时 finish 映射为 tool_calls
|
|
183
|
+
if (events.some(e => e.type === 'toolCallStart')) {
|
|
184
|
+
const finish = events.find(e => e.type === 'finish');
|
|
185
|
+
if (finish && finish.reason === 'stop') {
|
|
186
|
+
finish.reason = 'tool_calls';
|
|
187
|
+
}
|
|
188
|
+
}
|
|
189
|
+
return [{ type: 'messageStart' }, ...events];
|
|
190
|
+
}
|
|
191
|
+
// SSE 行 → IR 事件(data: 前缀,JSON 全帧)
|
|
192
|
+
export function geminiSSELineToEvents(line, state) {
|
|
193
|
+
if (!line.startsWith('data:')) {
|
|
194
|
+
return null;
|
|
195
|
+
}
|
|
196
|
+
const data = line.slice(5).trim();
|
|
197
|
+
if (!data) {
|
|
198
|
+
return [];
|
|
199
|
+
}
|
|
200
|
+
try {
|
|
201
|
+
const events = state.frameToEvents(JSON.parse(data));
|
|
202
|
+
// 流式 functionCall 后接 STOP:覆盖为 tool_calls
|
|
203
|
+
if (events.some(e => e.type === 'toolCallStart')) {
|
|
204
|
+
// 记录本帧有工具调用,供后续 STOP 帧修正
|
|
205
|
+
state.sawToolCall = true;
|
|
206
|
+
}
|
|
207
|
+
const finish = events.find(e => e.type === 'finish');
|
|
208
|
+
if (finish && finish.reason === 'stop' && state.sawToolCall) {
|
|
209
|
+
finish.reason = 'tool_calls';
|
|
210
|
+
}
|
|
211
|
+
return events;
|
|
212
|
+
}
|
|
213
|
+
catch {
|
|
214
|
+
return null;
|
|
215
|
+
}
|
|
216
|
+
}
|
|
217
|
+
export { randomUUID };
|
|
218
|
+
//# sourceMappingURL=gemini-out.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gemini-out.js","sourceRoot":"","sources":["../../../src/ai-gateway/ir/gemini-out.ts"],"names":[],"mappings":"AAAA,8DAA8D;AAC9D,iBAAiB;AACjB,yEAAyE;AACzE,sDAAsD;AACtD,yFAAyF;AACzF,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAEpC,OAAO,EAAE,iBAAiB,EAAE,MAAM,6BAA6B,CAAC;AAGhE,mBAAmB;AAEnB,MAAM,UAAU,UAAU,CAAC,EAAa,EAAE,QAA0B,EAAE,KAAa;IAGjF,yCAAyC;IACzC,MAAM,eAAe,GAAG,IAAI,GAAG,EAAkB,CAAC;IAClD,KAAK,MAAM,GAAG,IAAI,EAAE,CAAC,QAAQ,EAAE,CAAC;QAC9B,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC7B,KAAK,MAAM,IAAI,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;gBAC/B,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;oBAC5B,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC1C,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,QAAQ,GAAU,EAAE,CAAC;IAC3B,KAAK,MAAM,GAAG,IAAI,EAAE,CAAC,QAAQ,EAAE,CAAC;QAC9B,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC;QACzD,MAAM,KAAK,GAAU,EAAE,CAAC;QACxB,MAAM,iBAAiB,GAAU,EAAE,CAAC;QAEpC,KAAK,MAAM,IAAI,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;YAC/B,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACzB,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YAClC,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBACjC,KAAK,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa;YACrF,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;gBACnC,KAAK,CAAC,IAAI,CAAC,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;YAC3E,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBACtC,MAAM,KAAK,GAAG,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,SAAS,CAAC;gBAC/D,iBAAiB,CAAC,IAAI,CAAC;oBACrB,gBAAgB,EAAE;wBAChB,IAAI,EAAE,KAAK;wBACX,sBAAsB;wBACtB,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE;qBAC7E;iBACF,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,4EAA4E;QAC5E,IAAI,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACjC,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACrB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;YACjC,CAAC;YACD,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,iBAAiB,EAAE,CAAC,CAAC;QAC5D,CAAC;aAAM,CAAC;YACN,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACrB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;YACjC,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,IAAI,GAAQ,EAAE,QAAQ,EAAE,CAAC;IAC/B,IAAI,EAAE,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzB,IAAI,CAAC,iBAAiB,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;IAC7E,CAAC;IACD,IAAI,EAAE,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxB,IAAI,CAAC,KAAK,GAAG,CAAC;gBACZ,oBAAoB,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;oBACvC,IAAI,EAAE,CAAC,CAAC,IAAI;oBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;oBAC1B,UAAU,EAAE,cAAc,CAAC,CAAC,CAAC,WAAW,CAAC;iBAC1C,CAAC,CAAC;aACJ,CAAC,CAAC;IACL,CAAC;IACD,IAAI,EAAE,CAAC,UAAU,EAAE,CAAC;QAClB,MAAM,UAAU,GAAQ,EAAE,qBAAqB,EAAE,EAAE,EAAE,CAAC;QACtD,QAAQ,EAAE,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;YAC3B,KAAK,MAAM;gBAAE,UAAU,CAAC,qBAAqB,CAAC,IAAI,GAAG,MAAM,CAAC;gBAAC,MAAM;YACnE,KAAK,UAAU;gBAAE,UAAU,CAAC,qBAAqB,CAAC,IAAI,GAAG,KAAK,CAAC;gBAAC,MAAM;YACtE,KAAK,MAAM;gBAAE,UAAU,CAAC,qBAAqB,CAAC,IAAI,GAAG,MAAM,CAAC;gBAAC,MAAM;YACnE,KAAK,UAAU;gBACb,UAAU,CAAC,qBAAqB,CAAC,IAAI,GAAG,KAAK,CAAC;gBAC9C,UAAU,CAAC,qBAAqB,CAAC,oBAAoB,GAAG,CAAC,EAAE,CAAC,UAAU,CAAC,IAAK,CAAC,CAAC;gBAC9E,MAAM;QACV,CAAC;QACD,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IAC/B,CAAC;IAED,MAAM,GAAG,GAAQ,EAAE,eAAe,EAAE,EAAE,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC;IAC5D,IAAI,EAAE,CAAC,QAAQ,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;QAC1C,GAAG,CAAC,WAAW,GAAG,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC;IAC5C,CAAC;IACD,IAAI,EAAE,CAAC,QAAQ,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;QACnC,GAAG,CAAC,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC;IAC9B,CAAC;IACD,IAAI,EAAE,CAAC,QAAQ,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;QACnC,GAAG,CAAC,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,gBAAgB;IAC/C,CAAC;IACD,IAAI,EAAE,CAAC,QAAQ,CAAC,aAAa,EAAE,MAAM,EAAE,CAAC;QACtC,GAAG,CAAC,aAAa,GAAG,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC;IAChD,CAAC;IACD,IAAI,CAAC,gBAAgB,GAAG,GAAG,CAAC;IAE5B,OAAO;QACL,GAAG,EAAE,iBAAiB,CAAC,QAAQ,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC;QAC1D,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,QAAQ,CAAC,MAAM,EAAE,GAAG,CAAC,QAAQ,CAAC,OAAO,IAAI,EAAE,CAAC,EAAE;QAC/G,IAAI;KACL,CAAC;AACJ,CAAC;AAED,iFAAiF;AACjF,MAAM,uBAAuB,GAAG,IAAI,GAAG,CAAC,CAAC,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,aAAa,CAAC,CAAC,CAAC;AACxG,MAAM,UAAU,cAAc,CAAC,MAAW;IACxC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1B,OAAO,MAAM,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;IACpC,CAAC;IACD,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QACzC,MAAM,GAAG,GAAQ,EAAE,CAAC;QACpB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YAClD,IAAI,uBAAuB,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBACrC,SAAS;YACX,CAAC;YACD,GAAG,CAAC,GAAG,CAAC,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;QACnC,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,mBAAmB;AAEnB,yDAAyD;AACzD,2DAA2D;AAC3D,gEAAgE;AAChE,MAAM,OAAO,iBAAiB;IACpB,QAAQ,GAAG,IAAI,GAAG,EAAkB,CAAC;IACrC,aAAa,GAAG,KAAK,CAAC;IAE9B,UAAU,CAAC,EAAU;QACnB,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC/B,CAAC;IAED,aAAa,CAAC,KAAU;QACtB,MAAM,MAAM,GAAc,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;QAExC,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,OAAO,EAAE,KAAK,IAAI,EAAE,EAAE,CAAC;YACnD,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,OAAO,KAAK,IAAI,EAAE,CAAC;gBACxE,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YACtD,CAAC;YACD,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC;gBAChD,MAAM,EAAE,GAAG,IAAI,CAAC,YAAY,CAAC,EAAE,IAAI,QAAQ,UAAU,EAAE,EAAE,CAAC;gBAC1D,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;gBAC9C,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC;gBACzE,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;gBAChG,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,CAAC;YACvC,CAAC;YACD,0BAA0B;QAC5B,CAAC;QAED,IAAI,SAAS,EAAE,YAAY,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;YACnD,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;YAC1B,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,eAAe,CAAC,SAAS,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;QACnF,CAAC;QAED,MAAM,KAAK,GAAG,KAAK,CAAC,aAAa,CAAC;QAClC,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,KAAK,CAAC,gBAAgB,IAAI,CAAC;gBAClC,MAAM,EAAE,KAAK,CAAC,oBAAoB,IAAI,CAAC;aACxC,CAAC,CAAC;QACL,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AAED,MAAM,UAAU,eAAe,CAAC,MAAc;IAC5C,oFAAoF;IACpF,IAAI,MAAM,KAAK,YAAY,EAAE,CAAC;QAC5B,OAAO,QAAQ,CAAC;IAClB,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,gBAAgB;AAChB,MAAM,UAAU,qBAAqB,CAAC,IAAS;IAC7C,MAAM,KAAK,GAAG,IAAI,iBAAiB,EAAE,CAAC;IACtC,MAAM,MAAM,GAAG,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;IACzC,4CAA4C;IAC5C,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,eAAe,CAAC,EAAE,CAAC;QACjD,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAgD,CAAC;QACpG,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;YACvC,MAAM,CAAC,MAAM,GAAG,YAAY,CAAC;QAC/B,CAAC;IACH,CAAC;IACD,OAAO,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,EAAE,GAAG,MAAM,CAAC,CAAC;AAC/C,CAAC;AAED,kCAAkC;AAClC,MAAM,UAAU,qBAAqB,CAAC,IAAY,EAAE,KAAwB;IAC1E,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;QAC9B,OAAO,IAAI,CAAC;IACd,CAAC;IACD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAClC,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;QACrD,yCAAyC;QACzC,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,eAAe,CAAC,EAAE,CAAC;YACjD,yBAAyB;YACxB,KAAa,CAAC,WAAW,GAAG,IAAI,CAAC;QACpC,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAgD,CAAC;QACpG,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,MAAM,IAAK,KAAa,CAAC,WAAW,EAAE,CAAC;YACrE,MAAM,CAAC,MAAM,GAAG,YAAY,CAAC;QAC/B,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,OAAO,EAAE,UAAU,EAAE,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gemini-out.test.d.ts","sourceRoot":"","sources":["../../../src/ai-gateway/ir/gemini-out.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,125 @@
|
|
|
1
|
+
import { describe, it, expect } from 'vitest';
|
|
2
|
+
import { irToGemini, sanitizeSchema, GeminiStreamState, geminiSSELineToEvents, geminiFinalToIREvents } from './gemini-out.js';
|
|
3
|
+
const provider = {
|
|
4
|
+
id: '1', slug: 'gemini', dialect: 'gemini',
|
|
5
|
+
baseUrl: 'https://generativelanguage.googleapis.com', apiKey: 'gk', enabled: true,
|
|
6
|
+
};
|
|
7
|
+
function baseIR(over = {}) {
|
|
8
|
+
return { system: [], messages: [], tools: [], toolChoice: null, sampling: { maxTokens: 100 }, stream: false, ...over };
|
|
9
|
+
}
|
|
10
|
+
describe('irToGemini(请求方向)', () => {
|
|
11
|
+
it('system → systemInstruction.parts、采样参数映射(topK 保留)', () => {
|
|
12
|
+
const ir = baseIR({
|
|
13
|
+
system: [{ kind: 'text', text: 'S' }],
|
|
14
|
+
sampling: { maxTokens: 9, temperature: 0.7, topP: 0.8, topK: 40, stopSequences: ['X'] },
|
|
15
|
+
});
|
|
16
|
+
const { body, url, headers } = irToGemini(ir, provider, 'gemini-2.0-flash');
|
|
17
|
+
expect(body.systemInstruction).toEqual({ parts: [{ text: 'S' }] });
|
|
18
|
+
expect(body.generationConfig).toEqual({ maxOutputTokens: 9, temperature: 0.7, topP: 0.8, topK: 40, stopSequences: ['X'] });
|
|
19
|
+
expect(url).toBe('https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent');
|
|
20
|
+
expect(headers['x-goog-api-key']).toBe('gk');
|
|
21
|
+
// 流式 URL
|
|
22
|
+
expect(irToGemini({ ...ir, stream: true }, provider, 'm').url).toContain(':streamGenerateContent?alt=sse');
|
|
23
|
+
});
|
|
24
|
+
it('图片 → inlineData(base64 无前缀)', () => {
|
|
25
|
+
const { body } = irToGemini(baseIR({
|
|
26
|
+
messages: [{ role: 'user', content: [{ kind: 'image', mime: 'image/png', data: 'AA==' }] }],
|
|
27
|
+
}), provider, 'm');
|
|
28
|
+
expect(body.contents[0].parts[0]).toEqual({ inlineData: { mimeType: 'image/png', data: 'AA==' } });
|
|
29
|
+
});
|
|
30
|
+
it('tool_use → functionCall;tool_result 按函数名配对且 response 为 object', () => {
|
|
31
|
+
const { body } = irToGemini(baseIR({
|
|
32
|
+
messages: [
|
|
33
|
+
{ role: 'user', content: [{ kind: 'text', text: 'ls' }] },
|
|
34
|
+
{ role: 'assistant', content: [{ kind: 'toolUse', id: 'tu_9', name: 'bash', args: { cmd: 'ls' } }] },
|
|
35
|
+
{ role: 'user', content: [{ kind: 'toolResult', toolUseId: 'tu_9', content: 'a.txt', isError: false }] },
|
|
36
|
+
],
|
|
37
|
+
}), provider, 'm');
|
|
38
|
+
const modelTurn = body.contents.find((c) => c.role === 'model');
|
|
39
|
+
expect(modelTurn.parts[0].functionCall).toEqual({ name: 'bash', args: { cmd: 'ls' } });
|
|
40
|
+
const respTurn = body.contents.find((c) => c.parts?.[0]?.functionResponse);
|
|
41
|
+
expect(respTurn.parts[0].functionResponse).toEqual({ name: 'bash', response: { content: 'a.txt' } });
|
|
42
|
+
// functionResponse 紧随 model 轮
|
|
43
|
+
expect(body.contents.indexOf(respTurn)).toBe(body.contents.indexOf(modelTurn) + 1);
|
|
44
|
+
});
|
|
45
|
+
it('isError 的 tool_result 包 {error}', () => {
|
|
46
|
+
const { body } = irToGemini(baseIR({
|
|
47
|
+
messages: [
|
|
48
|
+
{ role: 'assistant', content: [{ kind: 'toolUse', id: 't1', name: 'f', args: {} }] },
|
|
49
|
+
{ role: 'user', content: [{ kind: 'toolResult', toolUseId: 't1', content: 'boom', isError: true }] },
|
|
50
|
+
],
|
|
51
|
+
}), provider, 'm');
|
|
52
|
+
const respTurn = body.contents.find((c) => c.parts?.[0]?.functionResponse);
|
|
53
|
+
expect(respTurn.parts[0].functionResponse.response).toEqual({ error: 'boom' });
|
|
54
|
+
});
|
|
55
|
+
it('tools 净化 JSON Schema(剥 $schema,保留 additionalProperties)', () => {
|
|
56
|
+
const { body } = irToGemini(baseIR({
|
|
57
|
+
tools: [{
|
|
58
|
+
name: 'f', description: 'd',
|
|
59
|
+
inputSchema: { $schema: 'http://json-schema.org/draft-07/schema#', type: 'object', additionalProperties: false, properties: { a: { type: 'string' } } },
|
|
60
|
+
}],
|
|
61
|
+
}), provider, 'm');
|
|
62
|
+
const decl = body.tools[0].functionDeclarations[0];
|
|
63
|
+
expect(decl.parameters).toEqual({ type: 'object', additionalProperties: false, properties: { a: { type: 'string' } } });
|
|
64
|
+
});
|
|
65
|
+
it('tool_choice 四态映射', () => {
|
|
66
|
+
let r = irToGemini(baseIR({ toolChoice: { mode: 'auto' } }), provider, 'm');
|
|
67
|
+
expect(r.body.toolConfig.functionCallingConfig.mode).toBe('AUTO');
|
|
68
|
+
r = irToGemini(baseIR({ toolChoice: { mode: 'required' } }), provider, 'm');
|
|
69
|
+
expect(r.body.toolConfig.functionCallingConfig.mode).toBe('ANY');
|
|
70
|
+
r = irToGemini(baseIR({ toolChoice: { mode: 'specific', name: 'f' } }), provider, 'm');
|
|
71
|
+
expect(r.body.toolConfig.functionCallingConfig).toEqual({ mode: 'ANY', allowedFunctionNames: ['f'] });
|
|
72
|
+
expect(irToGemini(baseIR(), provider, 'm').body).not.toHaveProperty('toolConfig');
|
|
73
|
+
});
|
|
74
|
+
});
|
|
75
|
+
describe('Gemini SSE → IR 事件', () => {
|
|
76
|
+
it('文本帧 + STOP finish + usageMetadata', () => {
|
|
77
|
+
const state = new GeminiStreamState();
|
|
78
|
+
const ev = geminiSSELineToEvents('data: {"candidates":[{"content":{"parts":[{"text":"Hi"}]},"finishReason":"STOP","index":0}],"usageMetadata":{"promptTokenCount":7,"candidatesTokenCount":2}}', state);
|
|
79
|
+
expect(ev).toEqual([
|
|
80
|
+
{ type: 'textDelta', text: 'Hi' },
|
|
81
|
+
{ type: 'finish', reason: 'stop' },
|
|
82
|
+
{ type: 'usage', input: 7, output: 2 },
|
|
83
|
+
]);
|
|
84
|
+
});
|
|
85
|
+
it('functionCall 全量 args 一帧三事件,缺 id 合成、STOP 修正为 tool_calls', () => {
|
|
86
|
+
const state = new GeminiStreamState();
|
|
87
|
+
const ev1 = geminiSSELineToEvents('data: {"candidates":[{"content":{"parts":[{"functionCall":{"name":"bash","args":{"cmd":"ls"}}}]}}]}', state);
|
|
88
|
+
expect(ev1).toHaveLength(3);
|
|
89
|
+
expect(ev1[0].type).toBe('toolCallStart');
|
|
90
|
+
expect(ev1[0].id).toMatch(/^call_/);
|
|
91
|
+
expect(ev1[1]).toEqual({ type: 'toolCallDelta', argsDelta: '{"cmd":"ls"}' });
|
|
92
|
+
expect(ev1[2]).toEqual({ type: 'toolCallEnd' });
|
|
93
|
+
const ev2 = geminiSSELineToEvents('data: {"candidates":[{"content":{"parts":[{"text":"done"}]},"finishReason":"STOP"}]}', state);
|
|
94
|
+
expect(ev2).toContainEqual({ type: 'finish', reason: 'tool_calls' });
|
|
95
|
+
});
|
|
96
|
+
it('thought part 丢弃、MAX_TOKENS → length', () => {
|
|
97
|
+
const state = new GeminiStreamState();
|
|
98
|
+
const ev = geminiSSELineToEvents('data: {"candidates":[{"content":{"parts":[{"text":"think","thought":true},{"text":"ans"}]},"finishReason":"MAX_TOKENS"}]}', state);
|
|
99
|
+
expect(ev).toEqual([
|
|
100
|
+
{ type: 'textDelta', text: 'ans' },
|
|
101
|
+
{ type: 'finish', reason: 'length' },
|
|
102
|
+
]);
|
|
103
|
+
});
|
|
104
|
+
it('非流式 geminiFinalToIREvents', () => {
|
|
105
|
+
const ev = geminiFinalToIREvents({
|
|
106
|
+
candidates: [{ content: { parts: [{ functionCall: { name: 'f', args: { a: 1 } } }] }, finishReason: 'STOP' }],
|
|
107
|
+
usageMetadata: { promptTokenCount: 3, candidatesTokenCount: 1 },
|
|
108
|
+
});
|
|
109
|
+
expect(ev[0]).toEqual({ type: 'messageStart' });
|
|
110
|
+
expect(ev.find(e => e.type === 'finish')).toEqual({ type: 'finish', reason: 'tool_calls' });
|
|
111
|
+
expect(ev.find(e => e.type === 'usage')).toEqual({ type: 'usage', input: 3, output: 1 });
|
|
112
|
+
});
|
|
113
|
+
});
|
|
114
|
+
describe('sanitizeSchema', () => {
|
|
115
|
+
it('递归剥离不支持键', () => {
|
|
116
|
+
expect(sanitizeSchema({
|
|
117
|
+
$schema: 'x', type: 'object',
|
|
118
|
+
properties: { a: { type: 'string', $comment: 'c' }, b: { type: 'array', items: { $ref: '#/x', type: 'number' } } },
|
|
119
|
+
})).toEqual({
|
|
120
|
+
type: 'object',
|
|
121
|
+
properties: { a: { type: 'string' }, b: { type: 'array', items: { type: 'number' } } },
|
|
122
|
+
});
|
|
123
|
+
});
|
|
124
|
+
});
|
|
125
|
+
//# sourceMappingURL=gemini-out.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gemini-out.test.js","sourceRoot":"","sources":["../../../src/ai-gateway/ir/gemini-out.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,iBAAiB,EAAE,qBAAqB,EAAE,qBAAqB,EAAE,MAAM,iBAAiB,CAAC;AAI9H,MAAM,QAAQ,GAAqB;IACjC,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,QAAQ;IAC1C,OAAO,EAAE,2CAA2C,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI;CAClF,CAAC;AAEF,SAAS,MAAM,CAAC,OAA2B,EAAE;IAC3C,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,SAAS,EAAE,GAAG,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,EAAE,CAAC;AACzH,CAAC;AAED,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,EAAE,GAAG,MAAM,CAAC;YAChB,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC;YACrC,QAAQ,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC,GAAG,CAAC,EAAE;SACxF,CAAC,CAAC;QACH,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,UAAU,CAAC,EAAE,EAAE,QAAQ,EAAE,kBAAkB,CAAC,CAAC;QAC5E,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;QACnE,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,OAAO,CAAC,EAAE,eAAe,EAAE,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC3H,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,0FAA0F,CAAC,CAAC;QAC7G,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7C,SAAS;QACT,MAAM,CAAC,UAAU,CAAC,EAAE,GAAG,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,gCAAgC,CAAC,CAAC;IAC7G,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,EAAE,IAAI,EAAE,GAAG,UAAU,CAAC,MAAM,CAAC;YACjC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;SAC5F,CAAC,EAAE,QAAQ,EAAE,GAAG,CAAC,CAAC;QACnB,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;IACrG,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,MAAM,EAAE,IAAI,EAAE,GAAG,UAAU,CAAC,MAAM,CAAC;YACjC,QAAQ,EAAE;gBACR,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE;gBACzD,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE;gBACpG,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE;aACzG;SACF,CAAC,EAAE,QAAQ,EAAE,GAAG,CAAC,CAAC;QACnB,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC;QACrE,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;QACvF,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAC;QAChF,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;QACrG,8BAA8B;QAC9B,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;IACrF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,EAAE,IAAI,EAAE,GAAG,UAAU,CAAC,MAAM,CAAC;YACjC,QAAQ,EAAE;gBACR,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,EAAE;gBACpF,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,SAAS,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE;aACrG;SACF,CAAC,EAAE,QAAQ,EAAE,GAAG,CAAC,CAAC;QACnB,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAC;QAChF,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;IACjF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,EAAE,IAAI,EAAE,GAAG,UAAU,CAAC,MAAM,CAAC;YACjC,KAAK,EAAE,CAAC;oBACN,IAAI,EAAE,GAAG,EAAE,WAAW,EAAE,GAAG;oBAC3B,WAAW,EAAE,EAAE,OAAO,EAAE,yCAAyC,EAAE,IAAI,EAAE,QAAQ,EAAE,oBAAoB,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE;iBACxJ,CAAC;SACH,CAAC,EAAE,QAAQ,EAAE,GAAG,CAAC,CAAC;QACnB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC;QACnD,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,oBAAoB,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;IAC1H,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAC1B,IAAI,CAAC,GAAG,UAAU,CAAC,MAAM,CAAC,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,CAAC,EAAE,QAAQ,EAAE,GAAG,CAAC,CAAC;QAC5E,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAClE,CAAC,GAAG,UAAU,CAAC,MAAM,CAAC,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,CAAC,EAAE,QAAQ,EAAE,GAAG,CAAC,CAAC;QAC5E,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACjE,CAAC,GAAG,UAAU,CAAC,MAAM,CAAC,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC,EAAE,QAAQ,EAAE,GAAG,CAAC,CAAC;QACvF,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,oBAAoB,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACtG,MAAM,CAAC,UAAU,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;IACpF,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,KAAK,GAAG,IAAI,iBAAiB,EAAE,CAAC;QACtC,MAAM,EAAE,GAAG,qBAAqB,CAAC,8JAA8J,EAAE,KAAK,CAAE,CAAC;QACzM,MAAM,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC;YACjB,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE;YACjC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE;YAClC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE;SACvC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,MAAM,KAAK,GAAG,IAAI,iBAAiB,EAAE,CAAC;QACtC,MAAM,GAAG,GAAG,qBAAqB,CAAC,qGAAqG,EAAE,KAAK,CAAE,CAAC;QACjJ,MAAM,CAAC,GAAG,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC5B,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC1C,MAAM,CAAE,GAAG,CAAC,CAAC,CAAS,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC7C,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,SAAS,EAAE,cAAc,EAAE,CAAC,CAAC;QAC7E,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,CAAC;QAChD,MAAM,GAAG,GAAG,qBAAqB,CAAC,sFAAsF,EAAE,KAAK,CAAE,CAAC;QAClI,MAAM,CAAC,GAAG,CAAC,CAAC,cAAc,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,KAAK,GAAG,IAAI,iBAAiB,EAAE,CAAC;QACtC,MAAM,EAAE,GAAG,qBAAqB,CAAC,2HAA2H,EAAE,KAAK,CAAE,CAAC;QACtK,MAAM,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC;YACjB,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,KAAK,EAAE;YAClC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE;SACrC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACnC,MAAM,EAAE,GAAG,qBAAqB,CAAC;YAC/B,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE,CAAC;YAC7G,aAAa,EAAE,EAAE,gBAAgB,EAAE,CAAC,EAAE,oBAAoB,EAAE,CAAC,EAAE;SAChE,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC;QAChD,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,CAAC;QAC5F,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC;IAC3F,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,EAAE,CAAC,UAAU,EAAE,GAAG,EAAE;QAClB,MAAM,CAAC,cAAc,CAAC;YACpB,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,QAAQ;YAC5B,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE;SACnH,CAAC,CAAC,CAAC,OAAO,CAAC;YACV,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE;SACvF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
import { IRRequest, IREvent, IRMessage } from './types.js';
|
|
2
|
+
import { AIProviderConfig } from '../../types/index.js';
|
|
3
|
+
export declare function irToOpenAI(ir: IRRequest, provider: AIProviderConfig, model: string): {
|
|
4
|
+
url: string;
|
|
5
|
+
headers: Record<string, string>;
|
|
6
|
+
body: any;
|
|
7
|
+
};
|
|
8
|
+
export declare class OpenAIStreamState {
|
|
9
|
+
private tools;
|
|
10
|
+
private finishEmitted;
|
|
11
|
+
chunkToEvents(chunk: any): IREvent[];
|
|
12
|
+
}
|
|
13
|
+
export declare function mapFinishReason(reason: string): 'stop' | 'tool_calls' | 'length';
|
|
14
|
+
export declare function openaiFinalToIREvents(json: any): IREvent[];
|
|
15
|
+
export declare function openaiSSELineToEvents(line: string, state: OpenAIStreamState): IREvent[] | null;
|
|
16
|
+
export type { IRMessage };
|
|
17
|
+
//# sourceMappingURL=openai-out.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai-out.d.ts","sourceRoot":"","sources":["../../../src/ai-gateway/ir/openai-out.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAE3D,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAIxD,wBAAgB,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,QAAQ,EAAE,gBAAgB,EAAE,KAAK,EAAE,MAAM,GAAG;IACpF,GAAG,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAAC,IAAI,EAAE,GAAG,CAAC;CACzD,CAkGA;AAkBD,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,KAAK,CAAqE;IAClF,OAAO,CAAC,aAAa,CAAS;IAE9B,aAAa,CAAC,KAAK,EAAE,GAAG,GAAG,OAAO,EAAE;CAuDrC;AAED,wBAAgB,eAAe,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,GAAG,YAAY,GAAG,QAAQ,CAQhF;AAID,wBAAgB,qBAAqB,CAAC,IAAI,EAAE,GAAG,GAAG,OAAO,EAAE,CAsB1D;AAGD,wBAAgB,qBAAqB,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,iBAAiB,GAAG,OAAO,EAAE,GAAG,IAAI,CAa9F;AAED,YAAY,EAAE,SAAS,EAAE,CAAC"}
|
|
@@ -0,0 +1,226 @@
|
|
|
1
|
+
import { openaiChatUrl } from '../upstream/model-lister.js';
|
|
2
|
+
// ===== 请求方向 =====
|
|
3
|
+
export function irToOpenAI(ir, provider, model) {
|
|
4
|
+
const messages = [];
|
|
5
|
+
// system 并入首条 system 消息(分块合并为单字符串)
|
|
6
|
+
if (ir.system.length > 0) {
|
|
7
|
+
messages.push({ role: 'system', content: ir.system.map(p => p.text).join('\n\n') });
|
|
8
|
+
}
|
|
9
|
+
// openai 历史约束:assistant tool_calls 之后每个调用必须紧跟一条 role:tool 消息。
|
|
10
|
+
// 状态机:assistant 轮吐出后,收集下一个 user 轮的 toolResult 按 id 配对输出;缺失补空 content。
|
|
11
|
+
const pendingToolCalls = [];
|
|
12
|
+
const toolResultById = new Map();
|
|
13
|
+
for (const msg of ir.messages) {
|
|
14
|
+
if (msg.role === 'assistant') {
|
|
15
|
+
const toolCalls = [];
|
|
16
|
+
let text = '';
|
|
17
|
+
for (const part of msg.content) {
|
|
18
|
+
if (part.kind === 'text') {
|
|
19
|
+
text += part.text;
|
|
20
|
+
}
|
|
21
|
+
else if (part.kind === 'toolUse') {
|
|
22
|
+
toolCalls.push({
|
|
23
|
+
id: part.id,
|
|
24
|
+
type: 'function',
|
|
25
|
+
function: { name: part.name, arguments: JSON.stringify(part.args ?? {}) },
|
|
26
|
+
});
|
|
27
|
+
pendingToolCalls.push({ id: part.id });
|
|
28
|
+
}
|
|
29
|
+
// 图片块在 assistant 历史中无意义,丢弃
|
|
30
|
+
}
|
|
31
|
+
messages.push({
|
|
32
|
+
role: 'assistant',
|
|
33
|
+
content: text || (toolCalls.length ? null : ''),
|
|
34
|
+
...(toolCalls.length ? { tool_calls: toolCalls } : {}),
|
|
35
|
+
});
|
|
36
|
+
}
|
|
37
|
+
else {
|
|
38
|
+
// user 轮:先收集本轮 toolResult 与普通内容,再冲刷 pending tool 调用
|
|
39
|
+
//(openai 约束:assistant tool_calls 之后每个调用必须紧跟一条 role:tool 消息,缺失补空 content)
|
|
40
|
+
const normalParts = [];
|
|
41
|
+
for (const part of msg.content) {
|
|
42
|
+
if (part.kind === 'toolResult') {
|
|
43
|
+
toolResultById.set(part.toolUseId, { content: part.content });
|
|
44
|
+
}
|
|
45
|
+
else if (part.kind === 'text') {
|
|
46
|
+
normalParts.push({ type: 'text', text: part.text });
|
|
47
|
+
}
|
|
48
|
+
else if (part.kind === 'image') {
|
|
49
|
+
normalParts.push({ type: 'image_url', image_url: { url: `data:${part.mime};base64,${part.data}` } });
|
|
50
|
+
}
|
|
51
|
+
}
|
|
52
|
+
if (pendingToolCalls.length > 0) {
|
|
53
|
+
flushToolResults(messages, pendingToolCalls, toolResultById);
|
|
54
|
+
}
|
|
55
|
+
if (normalParts.length > 0) {
|
|
56
|
+
messages.push({ role: 'user', content: normalParts.length === 1 && normalParts[0].type === 'text' ? normalParts[0].text : normalParts });
|
|
57
|
+
}
|
|
58
|
+
}
|
|
59
|
+
}
|
|
60
|
+
// 末轮遗留未配对调用补空
|
|
61
|
+
flushToolResults(messages, pendingToolCalls, toolResultById);
|
|
62
|
+
const body = {
|
|
63
|
+
model,
|
|
64
|
+
messages,
|
|
65
|
+
max_tokens: ir.sampling.maxTokens,
|
|
66
|
+
};
|
|
67
|
+
if (ir.sampling.temperature !== undefined) {
|
|
68
|
+
body.temperature = ir.sampling.temperature;
|
|
69
|
+
}
|
|
70
|
+
if (ir.sampling.topP !== undefined) {
|
|
71
|
+
body.top_p = ir.sampling.topP;
|
|
72
|
+
}
|
|
73
|
+
// top_k:openai 不支持,丢弃
|
|
74
|
+
if (ir.sampling.stopSequences?.length) {
|
|
75
|
+
body.stop = ir.sampling.stopSequences;
|
|
76
|
+
}
|
|
77
|
+
if (ir.tools.length > 0) {
|
|
78
|
+
body.tools = ir.tools.map(t => ({
|
|
79
|
+
type: 'function',
|
|
80
|
+
function: { name: t.name, description: t.description, parameters: t.inputSchema },
|
|
81
|
+
}));
|
|
82
|
+
}
|
|
83
|
+
if (ir.toolChoice) {
|
|
84
|
+
switch (ir.toolChoice.mode) {
|
|
85
|
+
case 'auto':
|
|
86
|
+
body.tool_choice = 'auto';
|
|
87
|
+
break;
|
|
88
|
+
case 'required':
|
|
89
|
+
body.tool_choice = 'required';
|
|
90
|
+
break;
|
|
91
|
+
case 'specific':
|
|
92
|
+
body.tool_choice = { type: 'function', function: { name: ir.toolChoice.name } };
|
|
93
|
+
break;
|
|
94
|
+
case 'none':
|
|
95
|
+
body.tool_choice = 'none';
|
|
96
|
+
break;
|
|
97
|
+
}
|
|
98
|
+
}
|
|
99
|
+
if (ir.stream) {
|
|
100
|
+
body.stream = true;
|
|
101
|
+
body.stream_options = { include_usage: true };
|
|
102
|
+
}
|
|
103
|
+
return {
|
|
104
|
+
url: openaiChatUrl(provider.baseUrl),
|
|
105
|
+
headers: { 'content-type': 'application/json', authorization: `Bearer ${provider.apiKey}`, ...(provider.headers || {}) },
|
|
106
|
+
body,
|
|
107
|
+
};
|
|
108
|
+
}
|
|
109
|
+
// 输出 pendingToolCalls 对应的 role:tool 消息;缺失的补空 content 容错(避免上游 400)
|
|
110
|
+
function flushToolResults(messages, pending, results) {
|
|
111
|
+
for (const call of pending.splice(0)) {
|
|
112
|
+
const result = results.get(call.id);
|
|
113
|
+
messages.push({
|
|
114
|
+
role: 'tool',
|
|
115
|
+
tool_call_id: call.id,
|
|
116
|
+
content: result ? result.content : '',
|
|
117
|
+
});
|
|
118
|
+
results.delete(call.id);
|
|
119
|
+
}
|
|
120
|
+
}
|
|
121
|
+
// ===== 响应方向(流式) =====
|
|
122
|
+
// delta.tool_calls 按 index 有状态聚合:首帧 id/name → toolCallStart,args 增量 → toolCallDelta
|
|
123
|
+
export class OpenAIStreamState {
|
|
124
|
+
tools = new Map();
|
|
125
|
+
finishEmitted = false;
|
|
126
|
+
chunkToEvents(chunk) {
|
|
127
|
+
const events = [];
|
|
128
|
+
const delta = chunk.choices?.[0]?.delta ?? {};
|
|
129
|
+
const choice = chunk.choices?.[0];
|
|
130
|
+
if (typeof delta.content === 'string' && delta.content) {
|
|
131
|
+
events.push({ type: 'textDelta', text: delta.content });
|
|
132
|
+
}
|
|
133
|
+
// DeepSeek reasoner 类 reasoning_content:一期丢弃
|
|
134
|
+
//(调用方可记录 debug 日志)
|
|
135
|
+
if (Array.isArray(delta.tool_calls)) {
|
|
136
|
+
for (const tc of delta.tool_calls) {
|
|
137
|
+
const index = tc.index ?? 0;
|
|
138
|
+
let entry = this.tools.get(index);
|
|
139
|
+
if (!entry) {
|
|
140
|
+
entry = { id: tc.id || `call_${index}`, name: tc.function?.name || '', emitted: false };
|
|
141
|
+
this.tools.set(index, entry);
|
|
142
|
+
}
|
|
143
|
+
if (tc.id && tc.id !== entry.id) {
|
|
144
|
+
entry.id = tc.id;
|
|
145
|
+
}
|
|
146
|
+
if (tc.function?.name && tc.function.name !== entry.name) {
|
|
147
|
+
entry.name = tc.function.name;
|
|
148
|
+
}
|
|
149
|
+
if (!entry.emitted && entry.name) {
|
|
150
|
+
entry.emitted = true;
|
|
151
|
+
events.push({ type: 'toolCallStart', id: entry.id, name: entry.name });
|
|
152
|
+
}
|
|
153
|
+
if (tc.function?.arguments) {
|
|
154
|
+
events.push({ type: 'toolCallDelta', argsDelta: tc.function.arguments });
|
|
155
|
+
}
|
|
156
|
+
}
|
|
157
|
+
}
|
|
158
|
+
if (choice?.finish_reason && !this.finishEmitted) {
|
|
159
|
+
this.finishEmitted = true;
|
|
160
|
+
// 每个 toolCallStart 对应一个 toolCallEnd(openai 增量到 finish 时参数已完整)
|
|
161
|
+
let toolEnds = 0;
|
|
162
|
+
for (const entry of this.tools.values()) {
|
|
163
|
+
if (entry.emitted) {
|
|
164
|
+
toolEnds++;
|
|
165
|
+
}
|
|
166
|
+
}
|
|
167
|
+
events.push({ type: 'finish', reason: mapFinishReason(choice.finish_reason) });
|
|
168
|
+
for (let i = 0; i < toolEnds; i++) {
|
|
169
|
+
events.push({ type: 'toolCallEnd' });
|
|
170
|
+
}
|
|
171
|
+
}
|
|
172
|
+
if (chunk.usage) {
|
|
173
|
+
events.push({ type: 'usage', input: chunk.usage.prompt_tokens ?? 0, output: chunk.usage.completion_tokens ?? 0 });
|
|
174
|
+
}
|
|
175
|
+
return events;
|
|
176
|
+
}
|
|
177
|
+
}
|
|
178
|
+
export function mapFinishReason(reason) {
|
|
179
|
+
if (reason === 'tool_calls' || reason === 'function_call') {
|
|
180
|
+
return 'tool_calls';
|
|
181
|
+
}
|
|
182
|
+
if (reason === 'length') {
|
|
183
|
+
return 'length';
|
|
184
|
+
}
|
|
185
|
+
return 'stop';
|
|
186
|
+
}
|
|
187
|
+
// ===== 响应方向(非流式) =====
|
|
188
|
+
export function openaiFinalToIREvents(json) {
|
|
189
|
+
const events = [{ type: 'messageStart' }];
|
|
190
|
+
const choice = json.choices?.[0];
|
|
191
|
+
const message = choice?.message ?? {};
|
|
192
|
+
if (typeof message.content === 'string' && message.content) {
|
|
193
|
+
events.push({ type: 'textDelta', text: message.content });
|
|
194
|
+
}
|
|
195
|
+
if (Array.isArray(message.tool_calls)) {
|
|
196
|
+
for (const tc of message.tool_calls) {
|
|
197
|
+
events.push({ type: 'toolCallStart', id: tc.id, name: tc.function?.name || '' });
|
|
198
|
+
events.push({ type: 'toolCallDelta', argsDelta: tc.function?.arguments ?? '{}' });
|
|
199
|
+
events.push({ type: 'toolCallEnd' });
|
|
200
|
+
}
|
|
201
|
+
}
|
|
202
|
+
if (json.usage) {
|
|
203
|
+
events.push({ type: 'usage', input: json.usage.prompt_tokens ?? 0, output: json.usage.completion_tokens ?? 0 });
|
|
204
|
+
}
|
|
205
|
+
if (choice?.finish_reason) {
|
|
206
|
+
events.push({ type: 'finish', reason: mapFinishReason(choice.finish_reason) });
|
|
207
|
+
}
|
|
208
|
+
return events;
|
|
209
|
+
}
|
|
210
|
+
// SSE 行(不含 "data: " 前缀的 JSON 文本或 "[DONE]")→ IR 事件
|
|
211
|
+
export function openaiSSELineToEvents(line, state) {
|
|
212
|
+
if (!line.startsWith('data:')) {
|
|
213
|
+
return null;
|
|
214
|
+
}
|
|
215
|
+
const data = line.slice(5).trim();
|
|
216
|
+
if (!data || data === '[DONE]') {
|
|
217
|
+
return [];
|
|
218
|
+
}
|
|
219
|
+
try {
|
|
220
|
+
return state.chunkToEvents(JSON.parse(data));
|
|
221
|
+
}
|
|
222
|
+
catch {
|
|
223
|
+
return null;
|
|
224
|
+
}
|
|
225
|
+
}
|
|
226
|
+
//# sourceMappingURL=openai-out.js.map
|