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,93 @@
|
|
|
1
|
+
import { describe, it, expect } from 'vitest';
|
|
2
|
+
import { anthropicToIR } from './anthropic-in.js';
|
|
3
|
+
describe('anthropicToIR', () => {
|
|
4
|
+
it('string content 与 blocks 混合解析', () => {
|
|
5
|
+
const ir = anthropicToIR({
|
|
6
|
+
messages: [
|
|
7
|
+
{ role: 'user', content: '你好' },
|
|
8
|
+
{ role: 'assistant', content: [{ type: 'text', text: '在' }] },
|
|
9
|
+
{ role: 'user', content: [{ type: 'text', text: '继续' }] },
|
|
10
|
+
],
|
|
11
|
+
max_tokens: 100,
|
|
12
|
+
});
|
|
13
|
+
expect(ir.messages).toHaveLength(3);
|
|
14
|
+
expect(ir.messages[0].content).toEqual([{ kind: 'text', text: '你好' }]);
|
|
15
|
+
expect(ir.messages[1].role).toBe('assistant');
|
|
16
|
+
expect(ir.sampling.maxTokens).toBe(100);
|
|
17
|
+
});
|
|
18
|
+
it('tool_use 与 tool_result 按 id 配对保留', () => {
|
|
19
|
+
const ir = anthropicToIR({
|
|
20
|
+
messages: [
|
|
21
|
+
{
|
|
22
|
+
role: 'assistant',
|
|
23
|
+
content: [{ type: 'tool_use', id: 'tu_1', name: 'bash', input: { cmd: 'ls' } }],
|
|
24
|
+
},
|
|
25
|
+
{
|
|
26
|
+
role: 'user',
|
|
27
|
+
content: [{ type: 'tool_result', tool_use_id: 'tu_1', content: [{ type: 'text', text: 'a.txt' }], is_error: false }],
|
|
28
|
+
},
|
|
29
|
+
],
|
|
30
|
+
max_tokens: 10,
|
|
31
|
+
});
|
|
32
|
+
expect(ir.messages[0].content[0]).toEqual({ kind: 'toolUse', id: 'tu_1', name: 'bash', args: { cmd: 'ls' } });
|
|
33
|
+
expect(ir.messages[1].content[0]).toEqual({
|
|
34
|
+
kind: 'toolResult', toolUseId: 'tu_1', content: 'a.txt', isError: false,
|
|
35
|
+
});
|
|
36
|
+
});
|
|
37
|
+
it('system 分块解析并保留 cache 标记', () => {
|
|
38
|
+
const ir = anthropicToIR({
|
|
39
|
+
system: [
|
|
40
|
+
{ type: 'text', text: 'A', cache_control: { type: 'ephemeral' } },
|
|
41
|
+
{ type: 'text', text: 'B' },
|
|
42
|
+
],
|
|
43
|
+
messages: [],
|
|
44
|
+
max_tokens: 10,
|
|
45
|
+
});
|
|
46
|
+
expect(ir.system).toEqual([
|
|
47
|
+
{ kind: 'text', text: 'A', cache: true },
|
|
48
|
+
{ kind: 'text', text: 'B' },
|
|
49
|
+
]);
|
|
50
|
+
});
|
|
51
|
+
it('system 字符串形态', () => {
|
|
52
|
+
expect(anthropicToIR({ system: 'sys', messages: [], max_tokens: 1 }).system)
|
|
53
|
+
.toEqual([{ kind: 'text', text: 'sys' }]);
|
|
54
|
+
});
|
|
55
|
+
it('图片 base64 块解析', () => {
|
|
56
|
+
const ir = anthropicToIR({
|
|
57
|
+
messages: [{ role: 'user', content: [{ type: 'image', source: { type: 'base64', media_type: 'image/png', data: 'AAAA' } }] }],
|
|
58
|
+
max_tokens: 1,
|
|
59
|
+
});
|
|
60
|
+
expect(ir.messages[0].content[0]).toEqual({ kind: 'image', mime: 'image/png', data: 'AAAA' });
|
|
61
|
+
});
|
|
62
|
+
it('thinking 历史块丢弃', () => {
|
|
63
|
+
const ir = anthropicToIR({
|
|
64
|
+
messages: [{ role: 'assistant', content: [
|
|
65
|
+
{ type: 'thinking', thinking: '...', signature: 'x' },
|
|
66
|
+
{ type: 'text', text: '答案' },
|
|
67
|
+
] }],
|
|
68
|
+
max_tokens: 1,
|
|
69
|
+
});
|
|
70
|
+
expect(ir.messages[0].content).toEqual([{ kind: 'text', text: '答案' }]);
|
|
71
|
+
});
|
|
72
|
+
it('tools 与 tool_choice 四态映射', () => {
|
|
73
|
+
const tools = [{ name: 'bash', description: 'run', input_schema: { type: 'object' } }];
|
|
74
|
+
expect(anthropicToIR({ messages: [], tools, tool_choice: { type: 'auto' }, max_tokens: 1 }).toolChoice)
|
|
75
|
+
.toEqual({ mode: 'auto' });
|
|
76
|
+
expect(anthropicToIR({ messages: [], tools, tool_choice: { type: 'any' }, max_tokens: 1 }).toolChoice)
|
|
77
|
+
.toEqual({ mode: 'required' });
|
|
78
|
+
expect(anthropicToIR({ messages: [], tools, tool_choice: { type: 'tool', name: 'bash' }, max_tokens: 1 }).toolChoice)
|
|
79
|
+
.toEqual({ mode: 'specific', name: 'bash' });
|
|
80
|
+
expect(anthropicToIR({ messages: [], tools, max_tokens: 1 }).toolChoice).toBeNull();
|
|
81
|
+
});
|
|
82
|
+
it('sampling 全参', () => {
|
|
83
|
+
const ir = anthropicToIR({
|
|
84
|
+
messages: [], max_tokens: 7, temperature: 0.5, top_p: 0.9, top_k: 40, stop_sequences: ['\n\n'],
|
|
85
|
+
});
|
|
86
|
+
expect(ir.sampling).toEqual({ maxTokens: 7, temperature: 0.5, topP: 0.9, topK: 40, stopSequences: ['\n\n'] });
|
|
87
|
+
});
|
|
88
|
+
it('stream 标志', () => {
|
|
89
|
+
expect(anthropicToIR({ messages: [], max_tokens: 1, stream: true }).stream).toBe(true);
|
|
90
|
+
expect(anthropicToIR({ messages: [], max_tokens: 1 }).stream).toBe(false);
|
|
91
|
+
});
|
|
92
|
+
});
|
|
93
|
+
//# sourceMappingURL=anthropic-in.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic-in.test.js","sourceRoot":"","sources":["../../../src/ai-gateway/ir/anthropic-in.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAElD,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,EAAE,GAAG,aAAa,CAAC;YACvB,QAAQ,EAAE;gBACR,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE;gBAC/B,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,EAAE;gBAC7D,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE;aAC1D;YACD,UAAU,EAAE,GAAG;SAChB,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QACvE,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC9C,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,EAAE,GAAG,aAAa,CAAC;YACvB,QAAQ,EAAE;gBACR;oBACE,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC;iBAChF;gBACD;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;iBACrH;aACF;YACD,UAAU,EAAE,EAAE;SACf,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;QAC9G,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACxC,IAAI,EAAE,YAAY,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK;SACxE,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACjC,MAAM,EAAE,GAAG,aAAa,CAAC;YACvB,MAAM,EAAE;gBACN,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,aAAa,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE;gBACjE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE;aAC5B;YACD,QAAQ,EAAE,EAAE;YACZ,UAAU,EAAE,EAAE;SACf,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACxB,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE;YACxC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE;SAC5B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,cAAc,EAAE,GAAG,EAAE;QACtB,MAAM,CAAC,aAAa,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC;aACzE,OAAO,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,eAAe,EAAE,GAAG,EAAE;QACvB,MAAM,EAAE,GAAG,aAAa,CAAC;YACvB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;YAC7H,UAAU,EAAE,CAAC;SACd,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;IAChG,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gBAAgB,EAAE,GAAG,EAAE;QACxB,MAAM,EAAE,GAAG,aAAa,CAAC;YACvB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE;wBACvC,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,EAAE;wBACrD,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE;qBAC7B,EAAE,CAAC;YACJ,UAAU,EAAE,CAAC;SACd,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;QAClC,MAAM,KAAK,GAAG,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;QACvF,MAAM,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC;aACpG,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;QAC7B,MAAM,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC;aACnG,OAAO,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;QACjC,MAAM,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC;aAClH,OAAO,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;QAC/C,MAAM,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,CAAC,QAAQ,EAAE,CAAC;IACtF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,aAAa,EAAE,GAAG,EAAE;QACrB,MAAM,EAAE,GAAG,aAAa,CAAC;YACvB,QAAQ,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,cAAc,EAAE,CAAC,MAAM,CAAC;SAC/F,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAChH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,WAAW,EAAE,GAAG,EAAE;QACnB,MAAM,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvF,MAAM,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import { Response } from 'express';
|
|
2
|
+
import { IREvent } from './types.js';
|
|
3
|
+
export declare class AnthropicSSEWriter {
|
|
4
|
+
private res;
|
|
5
|
+
private model;
|
|
6
|
+
private nextIndex;
|
|
7
|
+
private openBlock;
|
|
8
|
+
private started;
|
|
9
|
+
private finished;
|
|
10
|
+
private inputTokens;
|
|
11
|
+
private outputTokens;
|
|
12
|
+
private stopReason;
|
|
13
|
+
private pingTimer?;
|
|
14
|
+
constructor(res: Response, model: string);
|
|
15
|
+
private ensureStart;
|
|
16
|
+
write(ev: IREvent): void;
|
|
17
|
+
private closeBlock;
|
|
18
|
+
finish(): void;
|
|
19
|
+
isCommitted(): boolean;
|
|
20
|
+
}
|
|
21
|
+
export declare function aggregateIR(events: IREvent[], model: string): any;
|
|
22
|
+
//# sourceMappingURL=anthropic-sse.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic-sse.d.ts","sourceRoot":"","sources":["../../../src/ai-gateway/ir/anthropic-sse.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAC;AACnC,OAAO,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAgBrC,qBAAa,kBAAkB;IAUjB,OAAO,CAAC,GAAG;IAAY,OAAO,CAAC,KAAK;IAThD,OAAO,CAAC,SAAS,CAAK;IACtB,OAAO,CAAC,SAAS,CAAmC;IACpD,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,WAAW,CAAK;IACxB,OAAO,CAAC,YAAY,CAAK;IACzB,OAAO,CAAC,UAAU,CAAc;IAChC,OAAO,CAAC,SAAS,CAAC,CAAiB;gBAEf,GAAG,EAAE,QAAQ,EAAU,KAAK,EAAE,MAAM;IAGxD,OAAO,CAAC,WAAW;IAyBnB,KAAK,CAAC,EAAE,EAAE,OAAO,GAAG,IAAI;IAyDxB,OAAO,CAAC,UAAU;IAQlB,MAAM,IAAI,IAAI;IAmBd,WAAW,IAAI,OAAO;CAGvB;AAGD,wBAAgB,WAAW,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,MAAM,GAAG,GAAG,CA6DjE"}
|
|
@@ -0,0 +1,200 @@
|
|
|
1
|
+
function sse(res, event, data) {
|
|
2
|
+
res.write(`event: ${event}\ndata: ${JSON.stringify(data)}\n\n`);
|
|
3
|
+
}
|
|
4
|
+
function mapStopReason(reason) {
|
|
5
|
+
if (reason === 'tool_calls') {
|
|
6
|
+
return 'tool_use';
|
|
7
|
+
}
|
|
8
|
+
if (reason === 'length') {
|
|
9
|
+
return 'max_tokens';
|
|
10
|
+
}
|
|
11
|
+
return 'end_turn';
|
|
12
|
+
}
|
|
13
|
+
export class AnthropicSSEWriter {
|
|
14
|
+
res;
|
|
15
|
+
model;
|
|
16
|
+
nextIndex = 0;
|
|
17
|
+
openBlock = null;
|
|
18
|
+
started = false;
|
|
19
|
+
finished = false;
|
|
20
|
+
inputTokens = 0;
|
|
21
|
+
outputTokens = 0;
|
|
22
|
+
stopReason = 'end_turn';
|
|
23
|
+
pingTimer;
|
|
24
|
+
constructor(res, model) {
|
|
25
|
+
this.res = res;
|
|
26
|
+
this.model = model;
|
|
27
|
+
}
|
|
28
|
+
// 延迟提交后调用:写响应头 + message_start
|
|
29
|
+
ensureStart() {
|
|
30
|
+
if (this.started) {
|
|
31
|
+
return;
|
|
32
|
+
}
|
|
33
|
+
this.started = true;
|
|
34
|
+
this.res.writeHead(200, {
|
|
35
|
+
'content-type': 'text/event-stream',
|
|
36
|
+
'cache-control': 'no-cache',
|
|
37
|
+
connection: 'keep-alive',
|
|
38
|
+
});
|
|
39
|
+
sse(this.res, 'message_start', {
|
|
40
|
+
type: 'message_start',
|
|
41
|
+
message: {
|
|
42
|
+
id: `msg_${Date.now().toString(36)}`,
|
|
43
|
+
type: 'message', role: 'assistant', model: this.model,
|
|
44
|
+
content: [], stop_reason: null, stop_sequence: null,
|
|
45
|
+
// 占位 usage:权威 input/output 在 message_delta 汇总(openai/gemini 流末才可得)
|
|
46
|
+
usage: { input_tokens: 0, output_tokens: 0 },
|
|
47
|
+
},
|
|
48
|
+
});
|
|
49
|
+
this.pingTimer = setInterval(() => {
|
|
50
|
+
sse(this.res, 'ping', { type: 'ping' });
|
|
51
|
+
}, 30_000);
|
|
52
|
+
}
|
|
53
|
+
write(ev) {
|
|
54
|
+
if (this.finished) {
|
|
55
|
+
return;
|
|
56
|
+
}
|
|
57
|
+
switch (ev.type) {
|
|
58
|
+
case 'messageStart':
|
|
59
|
+
this.ensureStart();
|
|
60
|
+
break;
|
|
61
|
+
case 'textDelta':
|
|
62
|
+
this.ensureStart();
|
|
63
|
+
if (this.openBlock !== 'text') {
|
|
64
|
+
this.closeBlock();
|
|
65
|
+
this.openBlock = 'text';
|
|
66
|
+
sse(this.res, 'content_block_start', {
|
|
67
|
+
type: 'content_block_start', index: this.nextIndex++,
|
|
68
|
+
content_block: { type: 'text', text: '' },
|
|
69
|
+
});
|
|
70
|
+
}
|
|
71
|
+
sse(this.res, 'content_block_delta', {
|
|
72
|
+
type: 'content_block_delta', index: this.nextIndex - 1,
|
|
73
|
+
delta: { type: 'text_delta', text: ev.text },
|
|
74
|
+
});
|
|
75
|
+
break;
|
|
76
|
+
case 'toolCallStart':
|
|
77
|
+
this.ensureStart();
|
|
78
|
+
this.closeBlock();
|
|
79
|
+
this.openBlock = 'toolUse';
|
|
80
|
+
sse(this.res, 'content_block_start', {
|
|
81
|
+
type: 'content_block_start', index: this.nextIndex++,
|
|
82
|
+
content_block: { type: 'tool_use', id: ev.id, name: ev.name, input: {} },
|
|
83
|
+
});
|
|
84
|
+
break;
|
|
85
|
+
case 'toolCallDelta':
|
|
86
|
+
sse(this.res, 'content_block_delta', {
|
|
87
|
+
type: 'content_block_delta', index: this.nextIndex - 1,
|
|
88
|
+
delta: { type: 'input_json_delta', partial_json: ev.argsDelta },
|
|
89
|
+
});
|
|
90
|
+
break;
|
|
91
|
+
case 'toolCallEnd':
|
|
92
|
+
this.closeBlock();
|
|
93
|
+
break;
|
|
94
|
+
case 'usage':
|
|
95
|
+
this.inputTokens = ev.input;
|
|
96
|
+
this.outputTokens = ev.output;
|
|
97
|
+
break;
|
|
98
|
+
case 'finish':
|
|
99
|
+
this.stopReason = mapStopReason(ev.reason);
|
|
100
|
+
break;
|
|
101
|
+
case 'error':
|
|
102
|
+
// 流内错误:若尚未提交响应头,由调用方兜底为 HTTP 错误;已提交则发 error 事件
|
|
103
|
+
if (this.started) {
|
|
104
|
+
sse(this.res, 'error', { type: 'error', error: { type: 'api_error', message: ev.message } });
|
|
105
|
+
}
|
|
106
|
+
break;
|
|
107
|
+
}
|
|
108
|
+
}
|
|
109
|
+
closeBlock() {
|
|
110
|
+
if (this.openBlock) {
|
|
111
|
+
sse(this.res, 'content_block_stop', { type: 'content_block_stop', index: this.nextIndex - 1 });
|
|
112
|
+
this.openBlock = null;
|
|
113
|
+
}
|
|
114
|
+
}
|
|
115
|
+
// 事件流结束:汇总 message_delta + message_stop
|
|
116
|
+
finish() {
|
|
117
|
+
if (this.finished) {
|
|
118
|
+
return;
|
|
119
|
+
}
|
|
120
|
+
this.ensureStart();
|
|
121
|
+
this.closeBlock();
|
|
122
|
+
this.finished = true;
|
|
123
|
+
if (this.pingTimer) {
|
|
124
|
+
clearInterval(this.pingTimer);
|
|
125
|
+
}
|
|
126
|
+
sse(this.res, 'message_delta', {
|
|
127
|
+
type: 'message_delta',
|
|
128
|
+
delta: { stop_reason: this.stopReason, stop_sequence: null },
|
|
129
|
+
usage: { input_tokens: this.inputTokens, output_tokens: this.outputTokens },
|
|
130
|
+
});
|
|
131
|
+
sse(this.res, 'message_stop', { type: 'message_stop' });
|
|
132
|
+
this.res.end();
|
|
133
|
+
}
|
|
134
|
+
isCommitted() {
|
|
135
|
+
return this.started;
|
|
136
|
+
}
|
|
137
|
+
}
|
|
138
|
+
// 非流式:IR 事件序列聚合为单个 Anthropic Message JSON(Claude Code 也发非流式请求,如标题生成)
|
|
139
|
+
export function aggregateIR(events, model) {
|
|
140
|
+
const content = [];
|
|
141
|
+
let text = '';
|
|
142
|
+
let currentTool = null;
|
|
143
|
+
let input = 0, output = 0;
|
|
144
|
+
let stopReason = null;
|
|
145
|
+
for (const ev of events) {
|
|
146
|
+
switch (ev.type) {
|
|
147
|
+
case 'textDelta':
|
|
148
|
+
text += ev.text;
|
|
149
|
+
break;
|
|
150
|
+
case 'toolCallStart': {
|
|
151
|
+
// 文本块在首个工具前落盘
|
|
152
|
+
if (text) {
|
|
153
|
+
content.push({ type: 'text', text });
|
|
154
|
+
text = '';
|
|
155
|
+
}
|
|
156
|
+
currentTool = { id: ev.id, name: ev.name, args: '' };
|
|
157
|
+
break;
|
|
158
|
+
}
|
|
159
|
+
case 'toolCallDelta':
|
|
160
|
+
if (currentTool) {
|
|
161
|
+
currentTool.args += ev.argsDelta;
|
|
162
|
+
}
|
|
163
|
+
break;
|
|
164
|
+
case 'toolCallEnd':
|
|
165
|
+
if (currentTool) {
|
|
166
|
+
let parsed = {};
|
|
167
|
+
try {
|
|
168
|
+
parsed = currentTool.args ? JSON.parse(currentTool.args) : {};
|
|
169
|
+
}
|
|
170
|
+
catch {
|
|
171
|
+
parsed = {}; // 截断的参数无法解析时给空对象,不作为成功静默——stop_reason 仍如实
|
|
172
|
+
}
|
|
173
|
+
content.push({ type: 'tool_use', id: currentTool.id, name: currentTool.name, input: parsed });
|
|
174
|
+
currentTool = null;
|
|
175
|
+
}
|
|
176
|
+
break;
|
|
177
|
+
case 'usage':
|
|
178
|
+
input = ev.input;
|
|
179
|
+
output = ev.output;
|
|
180
|
+
break;
|
|
181
|
+
case 'finish':
|
|
182
|
+
stopReason = mapStopReason(ev.reason);
|
|
183
|
+
break;
|
|
184
|
+
case 'error':
|
|
185
|
+
throw new Error(ev.message);
|
|
186
|
+
default:
|
|
187
|
+
break;
|
|
188
|
+
}
|
|
189
|
+
}
|
|
190
|
+
if (text) {
|
|
191
|
+
content.push({ type: 'text', text });
|
|
192
|
+
}
|
|
193
|
+
return {
|
|
194
|
+
id: `msg_${Date.now().toString(36)}`,
|
|
195
|
+
type: 'message', role: 'assistant', model,
|
|
196
|
+
content, stop_reason: stopReason, stop_sequence: null,
|
|
197
|
+
usage: { input_tokens: input, output_tokens: output },
|
|
198
|
+
};
|
|
199
|
+
}
|
|
200
|
+
//# sourceMappingURL=anthropic-sse.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic-sse.js","sourceRoot":"","sources":["../../../src/ai-gateway/ir/anthropic-sse.ts"],"names":[],"mappings":"AAQA,SAAS,GAAG,CAAC,GAAa,EAAE,KAAa,EAAE,IAAS;IAClD,GAAG,CAAC,KAAK,CAAC,UAAU,KAAK,WAAW,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAClE,CAAC;AAED,SAAS,aAAa,CAAC,MAAwC;IAC7D,IAAI,MAAM,KAAK,YAAY,EAAE,CAAC;QAC5B,OAAO,UAAU,CAAC;IACpB,CAAC;IACD,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;QACxB,OAAO,YAAY,CAAC;IACtB,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,MAAM,OAAO,kBAAkB;IAUT;IAAuB;IATnC,SAAS,GAAG,CAAC,CAAC;IACd,SAAS,GAA8B,IAAI,CAAC;IAC5C,OAAO,GAAG,KAAK,CAAC;IAChB,QAAQ,GAAG,KAAK,CAAC;IACjB,WAAW,GAAG,CAAC,CAAC;IAChB,YAAY,GAAG,CAAC,CAAC;IACjB,UAAU,GAAG,UAAU,CAAC;IACxB,SAAS,CAAkB;IAEnC,YAAoB,GAAa,EAAU,KAAa;QAApC,QAAG,GAAH,GAAG,CAAU;QAAU,UAAK,GAAL,KAAK,CAAQ;IAAG,CAAC;IAE5D,+BAA+B;IACvB,WAAW;QACjB,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,OAAO;QACT,CAAC;QACD,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE;YACtB,cAAc,EAAE,mBAAmB;YACnC,eAAe,EAAE,UAAU;YAC3B,UAAU,EAAE,YAAY;SACzB,CAAC,CAAC;QACH,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,eAAe,EAAE;YAC7B,IAAI,EAAE,eAAe;YACrB,OAAO,EAAE;gBACP,EAAE,EAAE,OAAO,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE;gBACpC,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK;gBACrD,OAAO,EAAE,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI;gBACnD,mEAAmE;gBACnE,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE;aAC7C;SACF,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE;YAChC,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;QAC1C,CAAC,EAAE,MAAM,CAAC,CAAC;IACb,CAAC;IAED,KAAK,CAAC,EAAW;QACf,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,OAAO;QACT,CAAC;QACD,QAAQ,EAAE,CAAC,IAAI,EAAE,CAAC;YAChB,KAAK,cAAc;gBACjB,IAAI,CAAC,WAAW,EAAE,CAAC;gBACnB,MAAM;YACR,KAAK,WAAW;gBACd,IAAI,CAAC,WAAW,EAAE,CAAC;gBACnB,IAAI,IAAI,CAAC,SAAS,KAAK,MAAM,EAAE,CAAC;oBAC9B,IAAI,CAAC,UAAU,EAAE,CAAC;oBAClB,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC;oBACxB,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,qBAAqB,EAAE;wBACnC,IAAI,EAAE,qBAAqB,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,EAAE;wBACpD,aAAa,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE;qBAC1C,CAAC,CAAC;gBACL,CAAC;gBACD,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,qBAAqB,EAAE;oBACnC,IAAI,EAAE,qBAAqB,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,GAAG,CAAC;oBACtD,KAAK,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE;iBAC7C,CAAC,CAAC;gBACH,MAAM;YACR,KAAK,eAAe;gBAClB,IAAI,CAAC,WAAW,EAAE,CAAC;gBACnB,IAAI,CAAC,UAAU,EAAE,CAAC;gBAClB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;gBAC3B,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,qBAAqB,EAAE;oBACnC,IAAI,EAAE,qBAAqB,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,EAAE;oBACpD,aAAa,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE;iBACzE,CAAC,CAAC;gBACH,MAAM;YACR,KAAK,eAAe;gBAClB,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,qBAAqB,EAAE;oBACnC,IAAI,EAAE,qBAAqB,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,GAAG,CAAC;oBACtD,KAAK,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE,YAAY,EAAE,EAAE,CAAC,SAAS,EAAE;iBAChE,CAAC,CAAC;gBACH,MAAM;YACR,KAAK,aAAa;gBAChB,IAAI,CAAC,UAAU,EAAE,CAAC;gBAClB,MAAM;YACR,KAAK,OAAO;gBACV,IAAI,CAAC,WAAW,GAAG,EAAE,CAAC,KAAK,CAAC;gBAC5B,IAAI,CAAC,YAAY,GAAG,EAAE,CAAC,MAAM,CAAC;gBAC9B,MAAM;YACR,KAAK,QAAQ;gBACX,IAAI,CAAC,UAAU,GAAG,aAAa,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC;gBAC3C,MAAM;YACR,KAAK,OAAO;gBACV,+CAA+C;gBAC/C,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;oBACjB,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;gBAC/F,CAAC;gBACD,MAAM;QACV,CAAC;IACH,CAAC;IAEO,UAAU;QAChB,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,oBAAoB,EAAE,EAAE,IAAI,EAAE,oBAAoB,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,GAAG,CAAC,EAAE,CAAC,CAAC;YAC/F,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACxB,CAAC;IACH,CAAC;IAED,wCAAwC;IACxC,MAAM;QACJ,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,OAAO;QACT,CAAC;QACD,IAAI,CAAC,WAAW,EAAE,CAAC;QACnB,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAChC,CAAC;QACD,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,eAAe,EAAE;YAC7B,IAAI,EAAE,eAAe;YACrB,KAAK,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,UAAU,EAAE,aAAa,EAAE,IAAI,EAAE;YAC5D,KAAK,EAAE,EAAE,YAAY,EAAE,IAAI,CAAC,WAAW,EAAE,aAAa,EAAE,IAAI,CAAC,YAAY,EAAE;SAC5E,CAAC,CAAC;QACH,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,cAAc,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC;QACxD,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC;IACjB,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;CACF;AAED,qEAAqE;AACrE,MAAM,UAAU,WAAW,CAAC,MAAiB,EAAE,KAAa;IAC1D,MAAM,OAAO,GAAU,EAAE,CAAC;IAC1B,IAAI,IAAI,GAAG,EAAE,CAAC;IACd,IAAI,WAAW,GAAsD,IAAI,CAAC;IAC1E,IAAI,KAAK,GAAG,CAAC,EAAE,MAAM,GAAG,CAAC,CAAC;IAC1B,IAAI,UAAU,GAAkB,IAAI,CAAC;IAErC,KAAK,MAAM,EAAE,IAAI,MAAM,EAAE,CAAC;QACxB,QAAQ,EAAE,CAAC,IAAI,EAAE,CAAC;YAChB,KAAK,WAAW;gBACd,IAAI,IAAI,EAAE,CAAC,IAAI,CAAC;gBAChB,MAAM;YACR,KAAK,eAAe,CAAC,CAAC,CAAC;gBACrB,cAAc;gBACd,IAAI,IAAI,EAAE,CAAC;oBACT,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;oBACrC,IAAI,GAAG,EAAE,CAAC;gBACZ,CAAC;gBACD,WAAW,GAAG,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;gBACrD,MAAM;YACR,CAAC;YACD,KAAK,eAAe;gBAClB,IAAI,WAAW,EAAE,CAAC;oBAChB,WAAW,CAAC,IAAI,IAAI,EAAE,CAAC,SAAS,CAAC;gBACnC,CAAC;gBACD,MAAM;YACR,KAAK,aAAa;gBAChB,IAAI,WAAW,EAAE,CAAC;oBAChB,IAAI,MAAM,GAAQ,EAAE,CAAC;oBACrB,IAAI,CAAC;wBACH,MAAM,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;oBAChE,CAAC;oBAAC,MAAM,CAAC;wBACP,MAAM,GAAG,EAAE,CAAC,CAAC,0CAA0C;oBACzD,CAAC;oBACD,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,WAAW,CAAC,EAAE,EAAE,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;oBAC9F,WAAW,GAAG,IAAI,CAAC;gBACrB,CAAC;gBACD,MAAM;YACR,KAAK,OAAO;gBACV,KAAK,GAAG,EAAE,CAAC,KAAK,CAAC;gBACjB,MAAM,GAAG,EAAE,CAAC,MAAM,CAAC;gBACnB,MAAM;YACR,KAAK,QAAQ;gBACX,UAAU,GAAG,aAAa,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC;gBACtC,MAAM;YACR,KAAK,OAAO;gBACV,MAAM,IAAI,KAAK,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC;YAC9B;gBACE,MAAM;QACV,CAAC;IACH,CAAC;IACD,IAAI,IAAI,EAAE,CAAC;QACT,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IACvC,CAAC;IAED,OAAO;QACL,EAAE,EAAE,OAAO,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE;QACpC,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK;QACzC,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,aAAa,EAAE,IAAI;QACrD,KAAK,EAAE,EAAE,YAAY,EAAE,KAAK,EAAE,aAAa,EAAE,MAAM,EAAE;KACtD,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic-sse.test.d.ts","sourceRoot":"","sources":["../../../src/ai-gateway/ir/anthropic-sse.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,101 @@
|
|
|
1
|
+
import { describe, it, expect } from 'vitest';
|
|
2
|
+
import { AnthropicSSEWriter, aggregateIR } from './anthropic-sse.js';
|
|
3
|
+
function collectWriter() {
|
|
4
|
+
const frames = [];
|
|
5
|
+
const res = {
|
|
6
|
+
headersSent: true,
|
|
7
|
+
writeHead: () => { },
|
|
8
|
+
write: (chunk) => {
|
|
9
|
+
const eventMatch = chunk.match(/^event: (.+)\ndata: (.+)\n\n/s);
|
|
10
|
+
if (eventMatch) {
|
|
11
|
+
frames.push({ event: eventMatch[1], data: JSON.parse(eventMatch[2]) });
|
|
12
|
+
}
|
|
13
|
+
},
|
|
14
|
+
end: () => { },
|
|
15
|
+
};
|
|
16
|
+
return { res, frames };
|
|
17
|
+
}
|
|
18
|
+
describe('AnthropicSSEWriter', () => {
|
|
19
|
+
it('完整事件序列:message_start → blocks → message_delta → message_stop,index 递增', () => {
|
|
20
|
+
const { res, frames } = collectWriter();
|
|
21
|
+
const writer = new AnthropicSSEWriter(res, 'm');
|
|
22
|
+
const events = [
|
|
23
|
+
{ type: 'messageStart' },
|
|
24
|
+
{ type: 'textDelta', text: 'Hi' },
|
|
25
|
+
{ type: 'toolCallStart', id: 'c1', name: 'bash' },
|
|
26
|
+
{ type: 'toolCallDelta', argsDelta: '{"cmd"' },
|
|
27
|
+
{ type: 'toolCallEnd' },
|
|
28
|
+
{ type: 'usage', input: 10, output: 5 },
|
|
29
|
+
{ type: 'finish', reason: 'tool_calls' },
|
|
30
|
+
];
|
|
31
|
+
events.forEach(e => writer.write(e));
|
|
32
|
+
writer.finish();
|
|
33
|
+
expect(frames.map(f => f.event)).toEqual([
|
|
34
|
+
'message_start',
|
|
35
|
+
'content_block_start', 'content_block_delta', 'content_block_stop',
|
|
36
|
+
'content_block_start', 'content_block_delta', 'content_block_stop',
|
|
37
|
+
'message_delta', 'message_stop',
|
|
38
|
+
]);
|
|
39
|
+
// text block index 0
|
|
40
|
+
expect(frames[1].data.index).toBe(0);
|
|
41
|
+
expect(frames[1].data.content_block.type).toBe('text');
|
|
42
|
+
expect(frames[2].data.delta).toEqual({ type: 'text_delta', text: 'Hi' });
|
|
43
|
+
// tool_use block index 1,带 id/name
|
|
44
|
+
expect(frames[4].data.index).toBe(1);
|
|
45
|
+
expect(frames[4].data.content_block).toEqual({ type: 'tool_use', id: 'c1', name: 'bash', input: {} });
|
|
46
|
+
expect(frames[5].data.delta).toEqual({ type: 'input_json_delta', partial_json: '{"cmd"' });
|
|
47
|
+
// message_delta 汇总 usage + stop_reason
|
|
48
|
+
expect(frames[7].data.delta.stop_reason).toBe('tool_use');
|
|
49
|
+
expect(frames[7].data.usage).toEqual({ input_tokens: 10, output_tokens: 5 });
|
|
50
|
+
expect(frames[0].data.message.usage).toEqual({ input_tokens: 0, output_tokens: 0 }); // 占位
|
|
51
|
+
});
|
|
52
|
+
it('stop_reason 映射:stop→end_turn、length→max_tokens', () => {
|
|
53
|
+
for (const [reason, expected] of [['stop', 'end_turn'], ['length', 'max_tokens']]) {
|
|
54
|
+
const { res, frames } = collectWriter();
|
|
55
|
+
const writer = new AnthropicSSEWriter(res, 'm');
|
|
56
|
+
writer.write({ type: 'messageStart' });
|
|
57
|
+
writer.write({ type: 'textDelta', text: 'x' });
|
|
58
|
+
writer.write({ type: 'finish', reason });
|
|
59
|
+
writer.finish();
|
|
60
|
+
expect(frames.at(-2).data.delta.stop_reason).toBe(expected);
|
|
61
|
+
}
|
|
62
|
+
});
|
|
63
|
+
it('连续 textDelta 复用同一 block,不重复 start', () => {
|
|
64
|
+
const { res, frames } = collectWriter();
|
|
65
|
+
const writer = new AnthropicSSEWriter(res, 'm');
|
|
66
|
+
writer.write({ type: 'messageStart' });
|
|
67
|
+
writer.write({ type: 'textDelta', text: 'a' });
|
|
68
|
+
writer.write({ type: 'textDelta', text: 'b' });
|
|
69
|
+
writer.finish();
|
|
70
|
+
expect(frames.filter(f => f.event === 'content_block_start')).toHaveLength(1);
|
|
71
|
+
});
|
|
72
|
+
});
|
|
73
|
+
describe('aggregateIR(非流式聚合)', () => {
|
|
74
|
+
it('文本 + 工具调用聚合为 Message JSON', () => {
|
|
75
|
+
const message = aggregateIR([
|
|
76
|
+
{ type: 'messageStart' },
|
|
77
|
+
{ type: 'textDelta', text: '查一下' },
|
|
78
|
+
{ type: 'toolCallStart', id: 'c1', name: 'bash' },
|
|
79
|
+
{ type: 'toolCallDelta', argsDelta: '{"cmd":' },
|
|
80
|
+
{ type: 'toolCallDelta', argsDelta: '"ls"}' },
|
|
81
|
+
{ type: 'toolCallEnd' },
|
|
82
|
+
{ type: 'usage', input: 3, output: 4 },
|
|
83
|
+
{ type: 'finish', reason: 'tool_calls' },
|
|
84
|
+
], 'm');
|
|
85
|
+
expect(message.content).toEqual([
|
|
86
|
+
{ type: 'text', text: '查一下' },
|
|
87
|
+
{ type: 'tool_use', id: 'c1', name: 'bash', input: { cmd: 'ls' } },
|
|
88
|
+
]);
|
|
89
|
+
expect(message.stop_reason).toBe('tool_use');
|
|
90
|
+
expect(message.usage).toEqual({ input_tokens: 3, output_tokens: 4 });
|
|
91
|
+
});
|
|
92
|
+
it('纯文本回复', () => {
|
|
93
|
+
const message = aggregateIR([
|
|
94
|
+
{ type: 'textDelta', text: '好的' },
|
|
95
|
+
{ type: 'finish', reason: 'stop' },
|
|
96
|
+
], 'm');
|
|
97
|
+
expect(message.content).toEqual([{ type: 'text', text: '好的' }]);
|
|
98
|
+
expect(message.stop_reason).toBe('end_turn');
|
|
99
|
+
});
|
|
100
|
+
});
|
|
101
|
+
//# sourceMappingURL=anthropic-sse.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic-sse.test.js","sourceRoot":"","sources":["../../../src/ai-gateway/ir/anthropic-sse.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,kBAAkB,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAGrE,SAAS,aAAa;IACpB,MAAM,MAAM,GAAmC,EAAE,CAAC;IAClD,MAAM,GAAG,GAAQ;QACf,WAAW,EAAE,IAAI;QACjB,SAAS,EAAE,GAAG,EAAE,GAAE,CAAC;QACnB,KAAK,EAAE,CAAC,KAAa,EAAE,EAAE;YACvB,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,+BAA+B,CAAC,CAAC;YAChE,IAAI,UAAU,EAAE,CAAC;gBACf,MAAM,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACzE,CAAC;QACH,CAAC;QACD,GAAG,EAAE,GAAG,EAAE,GAAE,CAAC;KACd,CAAC;IACF,OAAO,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC;AACzB,CAAC;AAED,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;QAC/E,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,aAAa,EAAE,CAAC;QACxC,MAAM,MAAM,GAAG,IAAI,kBAAkB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAChD,MAAM,MAAM,GAAc;YACxB,EAAE,IAAI,EAAE,cAAc,EAAE;YACxB,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE;YACjC,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE;YACjD,EAAE,IAAI,EAAE,eAAe,EAAE,SAAS,EAAE,QAAQ,EAAE;YAC9C,EAAE,IAAI,EAAE,aAAa,EAAE;YACvB,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE;YACvC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,YAAY,EAAE;SACzC,CAAC;QACF,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,EAAE,CAAC;QAEhB,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC;YACvC,eAAe;YACf,qBAAqB,EAAE,qBAAqB,EAAE,oBAAoB;YAClE,qBAAqB,EAAE,qBAAqB,EAAE,oBAAoB;YAClE,eAAe,EAAE,cAAc;SAChC,CAAC,CAAC;QACH,qBAAqB;QACrB,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACzE,mCAAmC;QACnC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QACtG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,YAAY,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC3F,uCAAuC;QACvC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC;QAC7E,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK;IAC5F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,KAAK,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,UAAU,CAAC,EAAE,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAU,EAAE,CAAC;YAC3F,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,aAAa,EAAE,CAAC;YACxC,MAAM,MAAM,GAAG,IAAI,kBAAkB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YAChD,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC;YACvC,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;YAC/C,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,aAAa,EAAE,CAAC;QACxC,MAAM,MAAM,GAAG,IAAI,kBAAkB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAChD,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC;QACvC,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;QAC/C,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,EAAE,CAAC;QAChB,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,qBAAqB,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAChF,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACnC,MAAM,OAAO,GAAG,WAAW,CAAC;YAC1B,EAAE,IAAI,EAAE,cAAc,EAAE;YACxB,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,KAAK,EAAE;YAClC,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE;YACjD,EAAE,IAAI,EAAE,eAAe,EAAE,SAAS,EAAE,SAAS,EAAE;YAC/C,EAAE,IAAI,EAAE,eAAe,EAAE,SAAS,EAAE,OAAO,EAAE;YAC7C,EAAE,IAAI,EAAE,aAAa,EAAE;YACvB,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE;YACtC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,YAAY,EAAE;SACzC,EAAE,GAAG,CAAC,CAAC;QACR,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;YAC9B,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE;YAC7B,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;SACnE,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC7C,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;QACf,MAAM,OAAO,GAAG,WAAW,CAAC;YAC1B,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE;YACjC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE;SACnC,EAAE,GAAG,CAAC,CAAC;QACR,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAChE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import { randomUUID } from 'crypto';
|
|
2
|
+
import { IRRequest, IREvent } from './types.js';
|
|
3
|
+
import { AIProviderConfig } from '../../types/index.js';
|
|
4
|
+
export declare function irToGemini(ir: IRRequest, provider: AIProviderConfig, model: string): {
|
|
5
|
+
url: string;
|
|
6
|
+
headers: Record<string, string>;
|
|
7
|
+
body: any;
|
|
8
|
+
};
|
|
9
|
+
export declare function sanitizeSchema(schema: any): any;
|
|
10
|
+
export declare class GeminiStreamState {
|
|
11
|
+
private nameById;
|
|
12
|
+
private finishEmitted;
|
|
13
|
+
lookupName(id: string): string | undefined;
|
|
14
|
+
frameToEvents(frame: any): IREvent[];
|
|
15
|
+
}
|
|
16
|
+
export declare function mapGeminiFinish(reason: string): 'stop' | 'tool_calls' | 'length';
|
|
17
|
+
export declare function geminiFinalToIREvents(json: any): IREvent[];
|
|
18
|
+
export declare function geminiSSELineToEvents(line: string, state: GeminiStreamState): IREvent[] | null;
|
|
19
|
+
export { randomUUID };
|
|
20
|
+
//# sourceMappingURL=gemini-out.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gemini-out.d.ts","sourceRoot":"","sources":["../../../src/ai-gateway/ir/gemini-out.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpC,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAEhD,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,CAiGA;AAID,wBAAgB,cAAc,CAAC,MAAM,EAAE,GAAG,GAAG,GAAG,CAe/C;AAOD,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,QAAQ,CAA6B;IAC7C,OAAO,CAAC,aAAa,CAAS;IAE9B,UAAU,CAAC,EAAE,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IAI1C,aAAa,CAAC,KAAK,EAAE,GAAG,GAAG,OAAO,EAAE;CAiCrC;AAED,wBAAgB,eAAe,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,GAAG,YAAY,GAAG,QAAQ,CAMhF;AAGD,wBAAgB,qBAAqB,CAAC,IAAI,EAAE,GAAG,GAAG,OAAO,EAAE,CAW1D;AAGD,wBAAgB,qBAAqB,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,iBAAiB,GAAG,OAAO,EAAE,GAAG,IAAI,CAuB9F;AAED,OAAO,EAAE,UAAU,EAAE,CAAC"}
|