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.
Files changed (291) hide show
  1. package/LICENSE +22 -0
  2. package/README.md +244 -0
  3. package/dist/adapters/adapter-factory.d.ts +21 -0
  4. package/dist/adapters/adapter-factory.d.ts.map +1 -0
  5. package/dist/adapters/adapter-factory.js +157 -0
  6. package/dist/adapters/adapter-factory.js.map +1 -0
  7. package/dist/adapters/http-sse-adapter.d.ts +49 -0
  8. package/dist/adapters/http-sse-adapter.d.ts.map +1 -0
  9. package/dist/adapters/http-sse-adapter.js +485 -0
  10. package/dist/adapters/http-sse-adapter.js.map +1 -0
  11. package/dist/adapters/stdio-adapter.d.ts +69 -0
  12. package/dist/adapters/stdio-adapter.d.ts.map +1 -0
  13. package/dist/adapters/stdio-adapter.js +664 -0
  14. package/dist/adapters/stdio-adapter.js.map +1 -0
  15. package/dist/adapters/streamable-http-adapter.d.ts +38 -0
  16. package/dist/adapters/streamable-http-adapter.d.ts.map +1 -0
  17. package/dist/adapters/streamable-http-adapter.js +321 -0
  18. package/dist/adapters/streamable-http-adapter.js.map +1 -0
  19. package/dist/ai-gateway/anthropic-errors.d.ts +4 -0
  20. package/dist/ai-gateway/anthropic-errors.d.ts.map +1 -0
  21. package/dist/ai-gateway/anthropic-errors.js +17 -0
  22. package/dist/ai-gateway/anthropic-errors.js.map +1 -0
  23. package/dist/ai-gateway/auth.d.ts +3 -0
  24. package/dist/ai-gateway/auth.d.ts.map +1 -0
  25. package/dist/ai-gateway/auth.js +14 -0
  26. package/dist/ai-gateway/auth.js.map +1 -0
  27. package/dist/ai-gateway/auth.test.d.ts +2 -0
  28. package/dist/ai-gateway/auth.test.d.ts.map +1 -0
  29. package/dist/ai-gateway/auth.test.js +57 -0
  30. package/dist/ai-gateway/auth.test.js.map +1 -0
  31. package/dist/ai-gateway/count-tokens.d.ts +5 -0
  32. package/dist/ai-gateway/count-tokens.d.ts.map +1 -0
  33. package/dist/ai-gateway/count-tokens.js +37 -0
  34. package/dist/ai-gateway/count-tokens.js.map +1 -0
  35. package/dist/ai-gateway/count-tokens.test.d.ts +2 -0
  36. package/dist/ai-gateway/count-tokens.test.d.ts.map +1 -0
  37. package/dist/ai-gateway/count-tokens.test.js +19 -0
  38. package/dist/ai-gateway/count-tokens.test.js.map +1 -0
  39. package/dist/ai-gateway/dialects.d.ts +4 -0
  40. package/dist/ai-gateway/dialects.d.ts.map +1 -0
  41. package/dist/ai-gateway/dialects.js +15 -0
  42. package/dist/ai-gateway/dialects.js.map +1 -0
  43. package/dist/ai-gateway/gateway-server.d.ts +15 -0
  44. package/dist/ai-gateway/gateway-server.d.ts.map +1 -0
  45. package/dist/ai-gateway/gateway-server.js +121 -0
  46. package/dist/ai-gateway/gateway-server.js.map +1 -0
  47. package/dist/ai-gateway/integration.test.d.ts +2 -0
  48. package/dist/ai-gateway/integration.test.d.ts.map +1 -0
  49. package/dist/ai-gateway/integration.test.js +205 -0
  50. package/dist/ai-gateway/integration.test.js.map +1 -0
  51. package/dist/ai-gateway/ir/anthropic-in.d.ts +3 -0
  52. package/dist/ai-gateway/ir/anthropic-in.d.ts.map +1 -0
  53. package/dist/ai-gateway/ir/anthropic-in.js +121 -0
  54. package/dist/ai-gateway/ir/anthropic-in.js.map +1 -0
  55. package/dist/ai-gateway/ir/anthropic-in.test.d.ts +2 -0
  56. package/dist/ai-gateway/ir/anthropic-in.test.d.ts.map +1 -0
  57. package/dist/ai-gateway/ir/anthropic-in.test.js +93 -0
  58. package/dist/ai-gateway/ir/anthropic-in.test.js.map +1 -0
  59. package/dist/ai-gateway/ir/anthropic-sse.d.ts +22 -0
  60. package/dist/ai-gateway/ir/anthropic-sse.d.ts.map +1 -0
  61. package/dist/ai-gateway/ir/anthropic-sse.js +200 -0
  62. package/dist/ai-gateway/ir/anthropic-sse.js.map +1 -0
  63. package/dist/ai-gateway/ir/anthropic-sse.test.d.ts +2 -0
  64. package/dist/ai-gateway/ir/anthropic-sse.test.d.ts.map +1 -0
  65. package/dist/ai-gateway/ir/anthropic-sse.test.js +101 -0
  66. package/dist/ai-gateway/ir/anthropic-sse.test.js.map +1 -0
  67. package/dist/ai-gateway/ir/gemini-out.d.ts +20 -0
  68. package/dist/ai-gateway/ir/gemini-out.d.ts.map +1 -0
  69. package/dist/ai-gateway/ir/gemini-out.js +218 -0
  70. package/dist/ai-gateway/ir/gemini-out.js.map +1 -0
  71. package/dist/ai-gateway/ir/gemini-out.test.d.ts +2 -0
  72. package/dist/ai-gateway/ir/gemini-out.test.d.ts.map +1 -0
  73. package/dist/ai-gateway/ir/gemini-out.test.js +125 -0
  74. package/dist/ai-gateway/ir/gemini-out.test.js.map +1 -0
  75. package/dist/ai-gateway/ir/openai-out.d.ts +17 -0
  76. package/dist/ai-gateway/ir/openai-out.d.ts.map +1 -0
  77. package/dist/ai-gateway/ir/openai-out.js +226 -0
  78. package/dist/ai-gateway/ir/openai-out.js.map +1 -0
  79. package/dist/ai-gateway/ir/openai-out.test.d.ts +2 -0
  80. package/dist/ai-gateway/ir/openai-out.test.d.ts.map +1 -0
  81. package/dist/ai-gateway/ir/openai-out.test.js +144 -0
  82. package/dist/ai-gateway/ir/openai-out.test.js.map +1 -0
  83. package/dist/ai-gateway/ir/types.d.ts +77 -0
  84. package/dist/ai-gateway/ir/types.d.ts.map +1 -0
  85. package/dist/ai-gateway/ir/types.js +4 -0
  86. package/dist/ai-gateway/ir/types.js.map +1 -0
  87. package/dist/ai-gateway/ir-channel.d.ts +15 -0
  88. package/dist/ai-gateway/ir-channel.d.ts.map +1 -0
  89. package/dist/ai-gateway/ir-channel.js +120 -0
  90. package/dist/ai-gateway/ir-channel.js.map +1 -0
  91. package/dist/ai-gateway/model-router.d.ts +11 -0
  92. package/dist/ai-gateway/model-router.d.ts.map +1 -0
  93. package/dist/ai-gateway/model-router.js +27 -0
  94. package/dist/ai-gateway/model-router.js.map +1 -0
  95. package/dist/ai-gateway/model-router.test.d.ts +2 -0
  96. package/dist/ai-gateway/model-router.test.d.ts.map +1 -0
  97. package/dist/ai-gateway/model-router.test.js +43 -0
  98. package/dist/ai-gateway/model-router.test.js.map +1 -0
  99. package/dist/ai-gateway/passthrough.d.ts +9 -0
  100. package/dist/ai-gateway/passthrough.d.ts.map +1 -0
  101. package/dist/ai-gateway/passthrough.js +81 -0
  102. package/dist/ai-gateway/passthrough.js.map +1 -0
  103. package/dist/ai-gateway/passthrough.test.d.ts +2 -0
  104. package/dist/ai-gateway/passthrough.test.d.ts.map +1 -0
  105. package/dist/ai-gateway/passthrough.test.js +108 -0
  106. package/dist/ai-gateway/passthrough.test.js.map +1 -0
  107. package/dist/ai-gateway/upstream/model-lister.d.ts +13 -0
  108. package/dist/ai-gateway/upstream/model-lister.d.ts.map +1 -0
  109. package/dist/ai-gateway/upstream/model-lister.js +61 -0
  110. package/dist/ai-gateway/upstream/model-lister.js.map +1 -0
  111. package/dist/ai-gateway/upstream/provider-client.d.ts +20 -0
  112. package/dist/ai-gateway/upstream/provider-client.d.ts.map +1 -0
  113. package/dist/ai-gateway/upstream/provider-client.js +76 -0
  114. package/dist/ai-gateway/upstream/provider-client.js.map +1 -0
  115. package/dist/ai-gateway/upstream/provider-client.test.d.ts +2 -0
  116. package/dist/ai-gateway/upstream/provider-client.test.d.ts.map +1 -0
  117. package/dist/ai-gateway/upstream/provider-client.test.js +69 -0
  118. package/dist/ai-gateway/upstream/provider-client.test.js.map +1 -0
  119. package/dist/cli/cli-main.d.ts +7 -0
  120. package/dist/cli/cli-main.d.ts.map +1 -0
  121. package/dist/cli/cli-main.js +116 -0
  122. package/dist/cli/cli-main.js.map +1 -0
  123. package/dist/cli/cli-router.d.ts +20 -0
  124. package/dist/cli/cli-router.d.ts.map +1 -0
  125. package/dist/cli/cli-router.js +271 -0
  126. package/dist/cli/cli-router.js.map +1 -0
  127. package/dist/cli/cli-utils.d.ts +30 -0
  128. package/dist/cli/cli-utils.d.ts.map +1 -0
  129. package/dist/cli/cli-utils.js +219 -0
  130. package/dist/cli/cli-utils.js.map +1 -0
  131. package/dist/cli/commands/audit-commands.d.ts +30 -0
  132. package/dist/cli/commands/audit-commands.d.ts.map +1 -0
  133. package/dist/cli/commands/audit-commands.js +417 -0
  134. package/dist/cli/commands/audit-commands.js.map +1 -0
  135. package/dist/cli/commands/autostart-command.d.ts +21 -0
  136. package/dist/cli/commands/autostart-command.d.ts.map +1 -0
  137. package/dist/cli/commands/autostart-command.js +243 -0
  138. package/dist/cli/commands/autostart-command.js.map +1 -0
  139. package/dist/cli/commands/autostart-command.test.d.ts +2 -0
  140. package/dist/cli/commands/autostart-command.test.d.ts.map +1 -0
  141. package/dist/cli/commands/autostart-command.test.js +35 -0
  142. package/dist/cli/commands/autostart-command.test.js.map +1 -0
  143. package/dist/cli/commands/config-commands.d.ts +30 -0
  144. package/dist/cli/commands/config-commands.d.ts.map +1 -0
  145. package/dist/cli/commands/config-commands.js +1013 -0
  146. package/dist/cli/commands/config-commands.js.map +1 -0
  147. package/dist/cli/commands/daemon-commands.d.ts +76 -0
  148. package/dist/cli/commands/daemon-commands.d.ts.map +1 -0
  149. package/dist/cli/commands/daemon-commands.js +451 -0
  150. package/dist/cli/commands/daemon-commands.js.map +1 -0
  151. package/dist/cli/commands/detect-commands.d.ts +17 -0
  152. package/dist/cli/commands/detect-commands.d.ts.map +1 -0
  153. package/dist/cli/commands/detect-commands.js +293 -0
  154. package/dist/cli/commands/detect-commands.js.map +1 -0
  155. package/dist/cli/commands/diagnose-commands.d.ts +19 -0
  156. package/dist/cli/commands/diagnose-commands.d.ts.map +1 -0
  157. package/dist/cli/commands/diagnose-commands.js +326 -0
  158. package/dist/cli/commands/diagnose-commands.js.map +1 -0
  159. package/dist/cli/commands/optimize-commands.d.ts +14 -0
  160. package/dist/cli/commands/optimize-commands.d.ts.map +1 -0
  161. package/dist/cli/commands/optimize-commands.js +189 -0
  162. package/dist/cli/commands/optimize-commands.js.map +1 -0
  163. package/dist/cli/commands/proxy-command.d.ts +27 -0
  164. package/dist/cli/commands/proxy-command.d.ts.map +1 -0
  165. package/dist/cli/commands/proxy-command.js +212 -0
  166. package/dist/cli/commands/proxy-command.js.map +1 -0
  167. package/dist/cli/commands/start-command.d.ts +22 -0
  168. package/dist/cli/commands/start-command.d.ts.map +1 -0
  169. package/dist/cli/commands/start-command.js +275 -0
  170. package/dist/cli/commands/start-command.js.map +1 -0
  171. package/dist/config/config-manager.d.ts +125 -0
  172. package/dist/config/config-manager.d.ts.map +1 -0
  173. package/dist/config/config-manager.js +569 -0
  174. package/dist/config/config-manager.js.map +1 -0
  175. package/dist/config/config-manager.test.d.ts +2 -0
  176. package/dist/config/config-manager.test.d.ts.map +1 -0
  177. package/dist/config/config-manager.test.js +209 -0
  178. package/dist/config/config-manager.test.js.map +1 -0
  179. package/dist/core/auto-config-generator.d.ts +55 -0
  180. package/dist/core/auto-config-generator.d.ts.map +1 -0
  181. package/dist/core/auto-config-generator.js +278 -0
  182. package/dist/core/auto-config-generator.js.map +1 -0
  183. package/dist/core/mcpdog-server.d.ts +49 -0
  184. package/dist/core/mcpdog-server.d.ts.map +1 -0
  185. package/dist/core/mcpdog-server.js +552 -0
  186. package/dist/core/mcpdog-server.js.map +1 -0
  187. package/dist/core/protocol-detector.d.ts +56 -0
  188. package/dist/core/protocol-detector.d.ts.map +1 -0
  189. package/dist/core/protocol-detector.js +365 -0
  190. package/dist/core/protocol-detector.js.map +1 -0
  191. package/dist/daemon/daemon-client.d.ts +34 -0
  192. package/dist/daemon/daemon-client.d.ts.map +1 -0
  193. package/dist/daemon/daemon-client.js +130 -0
  194. package/dist/daemon/daemon-client.js.map +1 -0
  195. package/dist/daemon/daemon-client.test.d.ts +2 -0
  196. package/dist/daemon/daemon-client.test.d.ts.map +1 -0
  197. package/dist/daemon/daemon-client.test.js +77 -0
  198. package/dist/daemon/daemon-client.test.js.map +1 -0
  199. package/dist/daemon/daemon-info.d.ts +7 -0
  200. package/dist/daemon/daemon-info.d.ts.map +1 -0
  201. package/dist/daemon/daemon-info.js +34 -0
  202. package/dist/daemon/daemon-info.js.map +1 -0
  203. package/dist/daemon/daemon-info.test.d.ts +2 -0
  204. package/dist/daemon/daemon-info.test.d.ts.map +1 -0
  205. package/dist/daemon/daemon-info.test.js +57 -0
  206. package/dist/daemon/daemon-info.test.js.map +1 -0
  207. package/dist/daemon/daemon-web-server.ai.test.d.ts +2 -0
  208. package/dist/daemon/daemon-web-server.ai.test.d.ts.map +1 -0
  209. package/dist/daemon/daemon-web-server.ai.test.js +119 -0
  210. package/dist/daemon/daemon-web-server.ai.test.js.map +1 -0
  211. package/dist/daemon/daemon-web-server.d.ts +59 -0
  212. package/dist/daemon/daemon-web-server.d.ts.map +1 -0
  213. package/dist/daemon/daemon-web-server.js +1236 -0
  214. package/dist/daemon/daemon-web-server.js.map +1 -0
  215. package/dist/daemon/daemon-web-server.test.d.ts +2 -0
  216. package/dist/daemon/daemon-web-server.test.d.ts.map +1 -0
  217. package/dist/daemon/daemon-web-server.test.js +93 -0
  218. package/dist/daemon/daemon-web-server.test.js.map +1 -0
  219. package/dist/daemon/mcpdog-daemon.d.ts +42 -0
  220. package/dist/daemon/mcpdog-daemon.d.ts.map +1 -0
  221. package/dist/daemon/mcpdog-daemon.js +302 -0
  222. package/dist/daemon/mcpdog-daemon.js.map +1 -0
  223. package/dist/daemon/stdio-proxy.d.ts +17 -0
  224. package/dist/daemon/stdio-proxy.d.ts.map +1 -0
  225. package/dist/daemon/stdio-proxy.js +143 -0
  226. package/dist/daemon/stdio-proxy.js.map +1 -0
  227. package/dist/index.d.ts +14 -0
  228. package/dist/index.d.ts.map +1 -0
  229. package/dist/index.js +198 -0
  230. package/dist/index.js.map +1 -0
  231. package/dist/intent-analyzer.d.ts +27 -0
  232. package/dist/intent-analyzer.d.ts.map +1 -0
  233. package/dist/intent-analyzer.js +140 -0
  234. package/dist/intent-analyzer.js.map +1 -0
  235. package/dist/logging/server-log-manager.d.ts +75 -0
  236. package/dist/logging/server-log-manager.d.ts.map +1 -0
  237. package/dist/logging/server-log-manager.js +185 -0
  238. package/dist/logging/server-log-manager.js.map +1 -0
  239. package/dist/mcp-registry.d.ts +35 -0
  240. package/dist/mcp-registry.d.ts.map +1 -0
  241. package/dist/mcp-registry.js +166 -0
  242. package/dist/mcp-registry.js.map +1 -0
  243. package/dist/middleware/auth.d.ts +5 -0
  244. package/dist/middleware/auth.d.ts.map +1 -0
  245. package/dist/middleware/auth.js +81 -0
  246. package/dist/middleware/auth.js.map +1 -0
  247. package/dist/mock-tool-database.d.ts +41 -0
  248. package/dist/mock-tool-database.d.ts.map +1 -0
  249. package/dist/mock-tool-database.js +461 -0
  250. package/dist/mock-tool-database.js.map +1 -0
  251. package/dist/proxy-server-manager.d.ts +11 -0
  252. package/dist/proxy-server-manager.d.ts.map +1 -0
  253. package/dist/proxy-server-manager.js +144 -0
  254. package/dist/proxy-server-manager.js.map +1 -0
  255. package/dist/router/tool-router.d.ts +61 -0
  256. package/dist/router/tool-router.d.ts.map +1 -0
  257. package/dist/router/tool-router.js +539 -0
  258. package/dist/router/tool-router.js.map +1 -0
  259. package/dist/router/tool-router.test.d.ts +2 -0
  260. package/dist/router/tool-router.test.d.ts.map +1 -0
  261. package/dist/router/tool-router.test.js +113 -0
  262. package/dist/router/tool-router.test.js.map +1 -0
  263. package/dist/streamable-http-server.d.ts +27 -0
  264. package/dist/streamable-http-server.d.ts.map +1 -0
  265. package/dist/streamable-http-server.js +302 -0
  266. package/dist/streamable-http-server.js.map +1 -0
  267. package/dist/tool-finder.d.ts +21 -0
  268. package/dist/tool-finder.d.ts.map +1 -0
  269. package/dist/tool-finder.js +86 -0
  270. package/dist/tool-finder.js.map +1 -0
  271. package/dist/tool-recommender.d.ts +23 -0
  272. package/dist/tool-recommender.d.ts.map +1 -0
  273. package/dist/tool-recommender.js +200 -0
  274. package/dist/tool-recommender.js.map +1 -0
  275. package/dist/types/index.d.ts +145 -0
  276. package/dist/types/index.d.ts.map +1 -0
  277. package/dist/types/index.js +2 -0
  278. package/dist/types/index.js.map +1 -0
  279. package/dist/utils/server-name-validator.d.ts +47 -0
  280. package/dist/utils/server-name-validator.d.ts.map +1 -0
  281. package/dist/utils/server-name-validator.js +156 -0
  282. package/dist/utils/server-name-validator.js.map +1 -0
  283. package/dist/web/web-server.d.ts +44 -0
  284. package/dist/web/web-server.d.ts.map +1 -0
  285. package/dist/web/web-server.js +500 -0
  286. package/dist/web/web-server.js.map +1 -0
  287. package/package.json +68 -0
  288. package/web/dist/assets/index-DMVhVtNU.css +1 -0
  289. package/web/dist/assets/index-VMXIgcOH.js +92 -0
  290. package/web/dist/assets/index-VMXIgcOH.js.map +1 -0
  291. 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,2 @@
1
+ export {};
2
+ //# sourceMappingURL=gemini-out.test.d.ts.map
@@ -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