@amodalai/runtime 0.1.26 → 0.2.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/dist/src/__fixtures__/README.md +84 -0
- package/dist/src/__fixtures__/smoke-agent/amodal.json +11 -0
- package/dist/src/__fixtures__/smoke-agent/automations/test-auto.md +5 -0
- package/dist/src/__fixtures__/smoke-agent/connections/mock-api/access.json +11 -0
- package/dist/src/__fixtures__/smoke-agent/connections/mock-api/spec.json +4 -0
- package/dist/src/__fixtures__/smoke-agent/connections/mock-api/surface.md +9 -0
- package/dist/src/__fixtures__/smoke-agent/connections/mock-mcp/access.json +3 -0
- package/dist/src/__fixtures__/smoke-agent/connections/mock-mcp/spec.json +8 -0
- package/dist/src/__fixtures__/smoke-agent/evals/basic-eval.md +12 -0
- package/dist/src/__fixtures__/smoke-agent/knowledge/test-knowledge.md +3 -0
- package/dist/src/__fixtures__/smoke-agent/skills/test-skill/SKILL.md +11 -0
- package/dist/src/__fixtures__/smoke-agent/stores/test-items.json +11 -0
- package/dist/src/__fixtures__/smoke-agent/tools/echo_tool/handler.d.ts +18 -0
- package/dist/src/__fixtures__/smoke-agent/tools/echo_tool/handler.js +22 -0
- package/dist/src/__fixtures__/smoke-agent/tools/echo_tool/handler.js.map +1 -0
- package/dist/src/__fixtures__/smoke-agent/tools/echo_tool/tool.json +17 -0
- package/dist/src/__fixtures__/smoke.test.js +718 -0
- package/dist/src/__fixtures__/smoke.test.js.map +1 -0
- package/dist/src/__tests__/test-providers.d.ts +40 -0
- package/dist/src/__tests__/test-providers.js +61 -0
- package/dist/src/__tests__/test-providers.js.map +1 -0
- package/dist/src/agent/local-server.d.ts +3 -3
- package/dist/src/agent/local-server.js +213 -122
- package/dist/src/agent/local-server.js.map +1 -1
- package/dist/src/agent/loop-types.d.ts +175 -0
- package/dist/src/agent/loop-types.js +20 -0
- package/dist/src/agent/loop-types.js.map +1 -0
- package/dist/src/agent/loop.d.ts +31 -0
- package/dist/src/agent/loop.js +139 -0
- package/dist/src/agent/loop.js.map +1 -0
- package/dist/src/agent/loop.test.js +1030 -0
- package/dist/src/agent/loop.test.js.map +1 -0
- package/dist/src/agent/mcp-config.d.ts +28 -0
- package/dist/src/agent/mcp-config.js +57 -0
- package/dist/src/agent/mcp-config.js.map +1 -0
- package/dist/src/agent/page-builder.js +6 -1
- package/dist/src/agent/page-builder.js.map +1 -1
- package/dist/src/agent/proactive/proactive-runner.d.ts +24 -8
- package/dist/src/agent/proactive/proactive-runner.js +30 -32
- package/dist/src/agent/proactive/proactive-runner.js.map +1 -1
- package/dist/src/agent/proactive/proactive-runner.test.d.ts +1 -1
- package/dist/src/agent/proactive/proactive-runner.test.js +75 -87
- package/dist/src/agent/proactive/proactive-runner.test.js.map +1 -1
- package/dist/src/agent/routes/admin-chat.d.ts +15 -3
- package/dist/src/agent/routes/admin-chat.js +63 -18
- package/dist/src/agent/routes/admin-chat.js.map +1 -1
- package/dist/src/agent/routes/automations.js +5 -6
- package/dist/src/agent/routes/automations.js.map +1 -1
- package/dist/src/agent/routes/evals.d.ts +3 -2
- package/dist/src/agent/routes/evals.js +25 -12
- package/dist/src/agent/routes/evals.js.map +1 -1
- package/dist/src/agent/routes/files.js +7 -9
- package/dist/src/agent/routes/files.js.map +1 -1
- package/dist/src/agent/routes/inspect.d.ts +6 -2
- package/dist/src/agent/routes/inspect.js +31 -17
- package/dist/src/agent/routes/inspect.js.map +1 -1
- package/dist/src/agent/routes/inspect.test.js +18 -42
- package/dist/src/agent/routes/inspect.test.js.map +1 -1
- package/dist/src/agent/routes/stores.js +9 -12
- package/dist/src/agent/routes/stores.js.map +1 -1
- package/dist/src/agent/routes/task.d.ts +15 -3
- package/dist/src/agent/routes/task.js +16 -7
- package/dist/src/agent/routes/task.js.map +1 -1
- package/dist/src/agent/routes/task.test.d.ts +1 -1
- package/dist/src/agent/routes/task.test.js +70 -53
- package/dist/src/agent/routes/task.test.js.map +1 -1
- package/dist/src/agent/routes/webhooks.js +12 -3
- package/dist/src/agent/routes/webhooks.js.map +1 -1
- package/dist/src/agent/session-store.d.ts +11 -2
- package/dist/src/agent/session-store.js +1 -1
- package/dist/src/agent/session-store.js.map +1 -1
- package/dist/src/agent/snapshot-server.d.ts +2 -22
- package/dist/src/agent/snapshot-server.js +50 -27
- package/dist/src/agent/snapshot-server.js.map +1 -1
- package/dist/src/agent/states/compacting.d.ts +14 -0
- package/dist/src/agent/states/compacting.js +258 -0
- package/dist/src/agent/states/compacting.js.map +1 -0
- package/dist/src/agent/states/confirming.d.ts +10 -0
- package/dist/src/agent/states/confirming.js +76 -0
- package/dist/src/agent/states/confirming.js.map +1 -0
- package/dist/src/agent/states/dispatching.d.ts +18 -0
- package/dist/src/agent/states/dispatching.js +241 -0
- package/dist/src/agent/states/dispatching.js.map +1 -0
- package/dist/src/agent/states/executing.d.ts +21 -0
- package/dist/src/agent/states/executing.js +308 -0
- package/dist/src/agent/states/executing.js.map +1 -0
- package/dist/src/agent/states/streaming.d.ts +10 -0
- package/dist/src/agent/states/streaming.js +155 -0
- package/dist/src/agent/states/streaming.js.map +1 -0
- package/dist/src/agent/states/thinking.d.ts +13 -0
- package/dist/src/agent/states/thinking.js +233 -0
- package/dist/src/agent/states/thinking.js.map +1 -0
- package/dist/src/agent/token-estimate.d.ts +17 -0
- package/dist/src/agent/token-estimate.js +13 -0
- package/dist/src/agent/token-estimate.js.map +1 -0
- package/dist/src/agent/tool-executor-local.js +9 -18
- package/dist/src/agent/tool-executor-local.js.map +1 -1
- package/dist/src/agent/tool-executor-local.test.js +3 -5
- package/dist/src/agent/tool-executor-local.test.js.map +1 -1
- package/dist/src/api/create-agent.d.ts +15 -0
- package/dist/src/api/create-agent.js +137 -0
- package/dist/src/api/create-agent.js.map +1 -0
- package/dist/src/api/types.d.ts +68 -0
- package/dist/src/api/types.js +7 -0
- package/dist/src/api/types.js.map +1 -0
- package/dist/src/context/compiler.d.ts +13 -0
- package/dist/src/context/compiler.js +358 -0
- package/dist/src/context/compiler.js.map +1 -0
- package/dist/src/context/compiler.test.js +532 -0
- package/dist/src/context/compiler.test.js.map +1 -0
- package/dist/src/context/types.d.ts +110 -0
- package/dist/src/context/types.js +7 -0
- package/dist/src/context/types.js.map +1 -0
- package/dist/src/index.d.ts +33 -6
- package/dist/src/index.js +35 -21
- package/dist/src/index.js.map +1 -1
- package/dist/src/providers/create-provider.d.ts +23 -0
- package/dist/src/providers/create-provider.js +185 -0
- package/dist/src/providers/create-provider.js.map +1 -0
- package/dist/src/{agent/stores-e2e.test.d.ts → providers/create-provider.test.d.ts} +1 -1
- package/dist/src/providers/create-provider.test.js +95 -0
- package/dist/src/providers/create-provider.test.js.map +1 -0
- package/dist/src/providers/failover.d.ts +38 -0
- package/dist/src/providers/failover.js +147 -0
- package/dist/src/providers/failover.js.map +1 -0
- package/dist/src/providers/failover.test.d.ts +6 -0
- package/dist/src/providers/failover.test.js +169 -0
- package/dist/src/providers/failover.test.js.map +1 -0
- package/dist/src/providers/types.d.ts +110 -0
- package/dist/src/providers/types.js +7 -0
- package/dist/src/providers/types.js.map +1 -0
- package/dist/src/routes/ai-stream.d.ts +13 -10
- package/dist/src/routes/ai-stream.js +76 -41
- package/dist/src/routes/ai-stream.js.map +1 -1
- package/dist/src/routes/chat-new.test.d.ts +6 -0
- package/dist/src/routes/chat-new.test.js +107 -0
- package/dist/src/routes/chat-new.test.js.map +1 -0
- package/dist/src/routes/chat-stream-new.test.d.ts +6 -0
- package/dist/src/routes/chat-stream-new.test.js +135 -0
- package/dist/src/routes/chat-stream-new.test.js.map +1 -0
- package/dist/src/routes/chat-stream.d.ts +14 -4
- package/dist/src/routes/chat-stream.js +47 -29
- package/dist/src/routes/chat-stream.js.map +1 -1
- package/dist/src/routes/chat.d.ts +13 -4
- package/dist/src/routes/chat.js +60 -23
- package/dist/src/routes/chat.js.map +1 -1
- package/dist/src/routes/health.d.ts +3 -2
- package/dist/src/routes/health.js.map +1 -1
- package/dist/src/routes/route-helpers.d.ts +50 -0
- package/dist/src/routes/route-helpers.js +80 -0
- package/dist/src/routes/route-helpers.js.map +1 -0
- package/dist/src/routes/session-resolver.d.ts +72 -0
- package/dist/src/routes/session-resolver.js +123 -0
- package/dist/src/routes/session-resolver.js.map +1 -0
- package/dist/src/routes/session-resolver.test.d.ts +6 -0
- package/dist/src/routes/session-resolver.test.js +206 -0
- package/dist/src/routes/session-resolver.test.js.map +1 -0
- package/dist/src/routes/webhooks.d.ts +3 -1
- package/dist/src/routes/webhooks.js +12 -4
- package/dist/src/routes/webhooks.js.map +1 -1
- package/dist/src/security/permission-checker.d.ts +80 -0
- package/dist/src/security/permission-checker.js +75 -0
- package/dist/src/security/permission-checker.js.map +1 -0
- package/dist/src/security/permission-checker.test.d.ts +6 -0
- package/dist/src/security/permission-checker.test.js +208 -0
- package/dist/src/security/permission-checker.test.js.map +1 -0
- package/dist/src/server.d.ts +12 -11
- package/dist/src/server.js +44 -46
- package/dist/src/server.js.map +1 -1
- package/dist/src/server.test.d.ts +1 -1
- package/dist/src/server.test.js +6 -144
- package/dist/src/server.test.js.map +1 -1
- package/dist/src/session/manager.d.ts +98 -0
- package/dist/src/session/manager.js +364 -0
- package/dist/src/session/manager.js.map +1 -0
- package/dist/src/session/manager.test.d.ts +6 -0
- package/dist/src/session/manager.test.js +315 -0
- package/dist/src/session/manager.test.js.map +1 -0
- package/dist/src/session/session-builder.d.ts +71 -0
- package/dist/src/session/session-builder.js +364 -0
- package/dist/src/session/session-builder.js.map +1 -0
- package/dist/src/session/session-builder.test.d.ts +6 -0
- package/dist/src/session/session-builder.test.js +352 -0
- package/dist/src/session/session-builder.test.js.map +1 -0
- package/dist/src/session/store.d.ts +57 -0
- package/dist/src/session/store.js +167 -0
- package/dist/src/session/store.js.map +1 -0
- package/dist/src/session/store.test.d.ts +6 -0
- package/dist/src/session/store.test.js +145 -0
- package/dist/src/session/store.test.js.map +1 -0
- package/dist/src/session/stream-hooks.d.ts +39 -0
- package/dist/src/session/stream-hooks.js +7 -0
- package/dist/src/session/stream-hooks.js.map +1 -0
- package/dist/src/session/tool-context-factory.d.ts +60 -0
- package/dist/src/session/tool-context-factory.js +190 -0
- package/dist/src/session/tool-context-factory.js.map +1 -0
- package/dist/src/session/tool-context-factory.test.d.ts +6 -0
- package/dist/src/session/tool-context-factory.test.js +287 -0
- package/dist/src/session/tool-context-factory.test.js.map +1 -0
- package/dist/src/session/types.d.ts +188 -0
- package/dist/src/session/types.js +7 -0
- package/dist/src/session/types.js.map +1 -0
- package/dist/src/stores/drizzle-store-backend.d.ts +49 -0
- package/dist/src/stores/drizzle-store-backend.js +306 -0
- package/dist/src/stores/drizzle-store-backend.js.map +1 -0
- package/dist/src/stores/drizzle-store-backend.test.d.ts +6 -0
- package/dist/src/stores/drizzle-store-backend.test.js +215 -0
- package/dist/src/stores/drizzle-store-backend.test.js.map +1 -0
- package/dist/src/stores/index.d.ts +4 -0
- package/dist/src/stores/index.js +2 -0
- package/dist/src/stores/index.js.map +1 -1
- package/dist/src/stores/pglite-store-backend.d.ts +16 -19
- package/dist/src/stores/pglite-store-backend.js +85 -239
- package/dist/src/stores/pglite-store-backend.js.map +1 -1
- package/dist/src/stores/postgres-store-backend.d.ts +30 -0
- package/dist/src/stores/postgres-store-backend.js +100 -0
- package/dist/src/stores/postgres-store-backend.js.map +1 -0
- package/dist/src/stores/schema.d.ts +491 -0
- package/dist/src/stores/schema.js +57 -0
- package/dist/src/stores/schema.js.map +1 -0
- package/dist/src/tools/admin-file-tools.d.ts +13 -0
- package/dist/src/tools/admin-file-tools.js +200 -0
- package/dist/src/tools/admin-file-tools.js.map +1 -0
- package/dist/src/tools/admin-file-tools.test.d.ts +6 -0
- package/dist/src/tools/admin-file-tools.test.js +152 -0
- package/dist/src/tools/admin-file-tools.test.js.map +1 -0
- package/dist/src/tools/custom-tool-adapter.d.ts +41 -0
- package/dist/src/tools/custom-tool-adapter.js +190 -0
- package/dist/src/tools/custom-tool-adapter.js.map +1 -0
- package/dist/src/tools/custom-tool-adapter.test.d.ts +6 -0
- package/dist/src/tools/custom-tool-adapter.test.js +244 -0
- package/dist/src/tools/custom-tool-adapter.test.js.map +1 -0
- package/dist/src/tools/dispatch-tool.d.ts +52 -0
- package/dist/src/tools/dispatch-tool.js +71 -0
- package/dist/src/tools/dispatch-tool.js.map +1 -0
- package/dist/src/tools/dispatch-tool.test.d.ts +6 -0
- package/dist/src/tools/dispatch-tool.test.js +75 -0
- package/dist/src/tools/dispatch-tool.test.js.map +1 -0
- package/dist/src/tools/mcp-tool-adapter.d.ts +18 -0
- package/dist/src/tools/mcp-tool-adapter.js +135 -0
- package/dist/src/tools/mcp-tool-adapter.js.map +1 -0
- package/dist/src/tools/mcp-tool-adapter.test.d.ts +6 -0
- package/dist/src/tools/mcp-tool-adapter.test.js +227 -0
- package/dist/src/tools/mcp-tool-adapter.test.js.map +1 -0
- package/dist/src/tools/registry.d.ts +25 -0
- package/dist/src/tools/registry.js +72 -0
- package/dist/src/tools/registry.js.map +1 -0
- package/dist/src/tools/registry.test.d.ts +6 -0
- package/dist/src/tools/registry.test.js +121 -0
- package/dist/src/tools/registry.test.js.map +1 -0
- package/dist/src/tools/request-tool.d.ts +42 -0
- package/dist/src/tools/request-tool.js +190 -0
- package/dist/src/tools/request-tool.js.map +1 -0
- package/dist/src/tools/request-tool.test.d.ts +6 -0
- package/dist/src/tools/request-tool.test.js +254 -0
- package/dist/src/tools/request-tool.test.js.map +1 -0
- package/dist/src/tools/store-tools.d.ts +29 -0
- package/dist/src/tools/store-tools.js +224 -0
- package/dist/src/tools/store-tools.js.map +1 -0
- package/dist/src/tools/store-tools.test.d.ts +6 -0
- package/dist/src/tools/store-tools.test.js +216 -0
- package/dist/src/tools/store-tools.test.js.map +1 -0
- package/dist/src/tools/types.d.ts +111 -0
- package/dist/src/tools/types.js +7 -0
- package/dist/src/tools/types.js.map +1 -0
- package/dist/src/types.d.ts +20 -12
- package/dist/src/types.js +3 -2
- package/dist/src/types.js.map +1 -1
- package/dist/tsconfig.tsbuildinfo +1 -1
- package/package.json +13 -4
- package/dist/src/__tests__/sse-contract.test.js +0 -464
- package/dist/src/__tests__/sse-contract.test.js.map +0 -1
- package/dist/src/__tests__/tools.test.js +0 -583
- package/dist/src/__tests__/tools.test.js.map +0 -1
- package/dist/src/agent/agent-runner.d.ts +0 -33
- package/dist/src/agent/agent-runner.js +0 -1040
- package/dist/src/agent/agent-runner.js.map +0 -1
- package/dist/src/agent/custom-tools-e2e.test.d.ts +0 -6
- package/dist/src/agent/custom-tools-e2e.test.js +0 -566
- package/dist/src/agent/custom-tools-e2e.test.js.map +0 -1
- package/dist/src/agent/request-helper.d.ts +0 -16
- package/dist/src/agent/request-helper.js +0 -96
- package/dist/src/agent/request-helper.js.map +0 -1
- package/dist/src/agent/stores-e2e.test.js +0 -433
- package/dist/src/agent/stores-e2e.test.js.map +0 -1
- package/dist/src/agent/tool-context-builder.d.ts +0 -11
- package/dist/src/agent/tool-context-builder.js +0 -102
- package/dist/src/agent/tool-context-builder.js.map +0 -1
- package/dist/src/agent/tool-context-builder.test.d.ts +0 -6
- package/dist/src/agent/tool-context-builder.test.js +0 -152
- package/dist/src/agent/tool-context-builder.test.js.map +0 -1
- package/dist/src/agent/write-repo-file.test.js +0 -270
- package/dist/src/agent/write-repo-file.test.js.map +0 -1
- package/dist/src/cron/heartbeat-runner.d.ts +0 -21
- package/dist/src/cron/heartbeat-runner.js +0 -79
- package/dist/src/cron/heartbeat-runner.js.map +0 -1
- package/dist/src/cron/heartbeat-runner.test.d.ts +0 -6
- package/dist/src/cron/heartbeat-runner.test.js +0 -120
- package/dist/src/cron/heartbeat-runner.test.js.map +0 -1
- package/dist/src/cron/heartbeat-scheduler.d.ts +0 -26
- package/dist/src/cron/heartbeat-scheduler.js +0 -55
- package/dist/src/cron/heartbeat-scheduler.js.map +0 -1
- package/dist/src/cron/heartbeat-scheduler.test.d.ts +0 -6
- package/dist/src/cron/heartbeat-scheduler.test.js +0 -61
- package/dist/src/cron/heartbeat-scheduler.test.js.map +0 -1
- package/dist/src/routes/ai-stream.test.d.ts +0 -6
- package/dist/src/routes/ai-stream.test.js +0 -586
- package/dist/src/routes/ai-stream.test.js.map +0 -1
- package/dist/src/routes/ask-user-response.d.ts +0 -30
- package/dist/src/routes/ask-user-response.js +0 -61
- package/dist/src/routes/ask-user-response.js.map +0 -1
- package/dist/src/routes/ask-user-response.test.d.ts +0 -6
- package/dist/src/routes/ask-user-response.test.js +0 -88
- package/dist/src/routes/ask-user-response.test.js.map +0 -1
- package/dist/src/routes/chat-stream.test.d.ts +0 -6
- package/dist/src/routes/chat-stream.test.js +0 -155
- package/dist/src/routes/chat-stream.test.js.map +0 -1
- package/dist/src/routes/chat.test.d.ts +0 -6
- package/dist/src/routes/chat.test.js +0 -99
- package/dist/src/routes/chat.test.js.map +0 -1
- package/dist/src/routes/widget-actions.d.ts +0 -49
- package/dist/src/routes/widget-actions.js +0 -78
- package/dist/src/routes/widget-actions.js.map +0 -1
- package/dist/src/session/custom-tool-adapter.d.ts +0 -74
- package/dist/src/session/custom-tool-adapter.js +0 -180
- package/dist/src/session/custom-tool-adapter.js.map +0 -1
- package/dist/src/session/history-converter.d.ts +0 -21
- package/dist/src/session/history-converter.js +0 -59
- package/dist/src/session/history-converter.js.map +0 -1
- package/dist/src/session/history-converter.test.d.ts +0 -6
- package/dist/src/session/history-converter.test.js +0 -130
- package/dist/src/session/history-converter.test.js.map +0 -1
- package/dist/src/session/session-manager.d.ts +0 -219
- package/dist/src/session/session-manager.js +0 -915
- package/dist/src/session/session-manager.js.map +0 -1
- package/dist/src/session/session-manager.test.d.ts +0 -6
- package/dist/src/session/session-manager.test.js +0 -455
- package/dist/src/session/session-manager.test.js.map +0 -1
- package/dist/src/session/session-runner.d.ts +0 -45
- package/dist/src/session/session-runner.js +0 -719
- package/dist/src/session/session-runner.js.map +0 -1
- package/dist/src/session/session-runner.test.d.ts +0 -6
- package/dist/src/session/session-runner.test.js +0 -834
- package/dist/src/session/session-runner.test.js.map +0 -1
- /package/dist/src/{__tests__/sse-contract.test.d.ts → __fixtures__/smoke.test.d.ts} +0 -0
- /package/dist/src/{__tests__/tools.test.d.ts → agent/loop.test.d.ts} +0 -0
- /package/dist/src/{agent/write-repo-file.test.d.ts → context/compiler.test.d.ts} +0 -0
|
@@ -1,14 +1,14 @@
|
|
|
1
1
|
/**
|
|
2
2
|
* @license
|
|
3
|
-
* Copyright
|
|
3
|
+
* Copyright 2026 Amodal Labs, Inc.
|
|
4
4
|
* SPDX-License-Identifier: MIT
|
|
5
5
|
*/
|
|
6
6
|
/**
|
|
7
7
|
* Vercel AI SDK UI Message Stream Protocol adapter.
|
|
8
8
|
*
|
|
9
9
|
* Accepts requests in the Vercel AI SDK format, feeds the user message into
|
|
10
|
-
* the
|
|
11
|
-
* `SSEEvent` into the UI Message Stream Protocol that `@ai-sdk/react`'s
|
|
10
|
+
* the standalone session manager's `runMessage()` generator, and translates
|
|
11
|
+
* every `SSEEvent` into the UI Message Stream Protocol that `@ai-sdk/react`'s
|
|
12
12
|
* `useChat` hook expects.
|
|
13
13
|
*
|
|
14
14
|
* Protocol spec: events are newline-delimited JSON objects prefixed with
|
|
@@ -19,8 +19,9 @@ import { Router } from 'express';
|
|
|
19
19
|
import { z } from 'zod';
|
|
20
20
|
import { validate } from '../middleware/request-validation.js';
|
|
21
21
|
import { getAuthContext } from '../middleware/auth.js';
|
|
22
|
-
import { streamMessage } from '../session/session-runner.js';
|
|
23
22
|
import { SSEEventType } from '../types.js';
|
|
23
|
+
import { resolveSession } from './session-resolver.js';
|
|
24
|
+
import { adaptOnUsage, asyncHandler, fireDrainHooks } from './route-helpers.js';
|
|
24
25
|
// ---------------------------------------------------------------------------
|
|
25
26
|
// Request schema (Vercel AI SDK message format)
|
|
26
27
|
// ---------------------------------------------------------------------------
|
|
@@ -61,7 +62,6 @@ function closeTextBlock(state) {
|
|
|
61
62
|
*/
|
|
62
63
|
export function translateEvent(event, state) {
|
|
63
64
|
const out = [];
|
|
64
|
-
// eslint-disable-next-line @typescript-eslint/switch-exhaustiveness-check -- TODO: handle all cases
|
|
65
65
|
switch (event.type) {
|
|
66
66
|
case SSEEventType.Init: {
|
|
67
67
|
out.push({ type: 'message-start', messageId: state.messageId });
|
|
@@ -158,15 +158,6 @@ export function translateEvent(event, state) {
|
|
|
158
158
|
});
|
|
159
159
|
break;
|
|
160
160
|
}
|
|
161
|
-
case SSEEventType.AskUser: {
|
|
162
|
-
// Question[] needs to go into a generic Record<string, unknown> data part
|
|
163
|
-
const askData = {
|
|
164
|
-
ask_id: event.ask_id,
|
|
165
|
-
questions: event.questions,
|
|
166
|
-
};
|
|
167
|
-
out.push({ type: 'data-ask-user', data: askData });
|
|
168
|
-
break;
|
|
169
|
-
}
|
|
170
161
|
case SSEEventType.CredentialSaved: {
|
|
171
162
|
out.push({
|
|
172
163
|
type: 'data-credential-saved',
|
|
@@ -196,8 +187,58 @@ export function translateEvent(event, state) {
|
|
|
196
187
|
out.push({ type: 'finish', finishReason: 'stop' });
|
|
197
188
|
break;
|
|
198
189
|
}
|
|
199
|
-
|
|
190
|
+
case SSEEventType.ConfirmationRequired: {
|
|
191
|
+
out.push({
|
|
192
|
+
type: 'data-confirmation-required',
|
|
193
|
+
data: {
|
|
194
|
+
endpoint: event.endpoint,
|
|
195
|
+
method: event.method,
|
|
196
|
+
reason: event.reason,
|
|
197
|
+
escalated: event.escalated,
|
|
198
|
+
},
|
|
199
|
+
});
|
|
200
|
+
break;
|
|
201
|
+
}
|
|
202
|
+
case SSEEventType.CompactionStart: {
|
|
203
|
+
out.push({
|
|
204
|
+
type: 'data-compaction-start',
|
|
205
|
+
data: {
|
|
206
|
+
estimated_tokens: event.estimated_tokens,
|
|
207
|
+
threshold: event.threshold,
|
|
208
|
+
},
|
|
209
|
+
});
|
|
210
|
+
break;
|
|
211
|
+
}
|
|
212
|
+
case SSEEventType.CompactionEnd: {
|
|
213
|
+
out.push({
|
|
214
|
+
type: 'data-compaction-end',
|
|
215
|
+
data: {
|
|
216
|
+
tokens_before: event.tokens_before,
|
|
217
|
+
tokens_after: event.tokens_after,
|
|
218
|
+
compaction_tokens: event.compaction_tokens,
|
|
219
|
+
},
|
|
220
|
+
});
|
|
221
|
+
break;
|
|
222
|
+
}
|
|
223
|
+
case SSEEventType.ToolLog: {
|
|
224
|
+
out.push({
|
|
225
|
+
type: 'data-tool-log',
|
|
226
|
+
data: { tool_name: event.tool_name, message: event.message },
|
|
227
|
+
});
|
|
228
|
+
break;
|
|
229
|
+
}
|
|
230
|
+
// Audit/telemetry events — not forwarded to the chat UI. They're
|
|
231
|
+
// surfaced via other channels (audit log, field-scrub notifications,
|
|
232
|
+
// explore subagent stream in SubagentEvent).
|
|
233
|
+
case SSEEventType.ExploreStart:
|
|
234
|
+
case SSEEventType.ExploreEnd:
|
|
235
|
+
case SSEEventType.PlanMode:
|
|
236
|
+
case SSEEventType.FieldScrub:
|
|
200
237
|
break;
|
|
238
|
+
default: {
|
|
239
|
+
const _exhaustive = event;
|
|
240
|
+
throw new Error(`Unhandled SSEEvent type: ${_exhaustive.type}`);
|
|
241
|
+
}
|
|
201
242
|
}
|
|
202
243
|
return out;
|
|
203
244
|
}
|
|
@@ -221,9 +262,7 @@ export function extractUserMessage(messages) {
|
|
|
221
262
|
}
|
|
222
263
|
export function createAIStreamRouter(options) {
|
|
223
264
|
const router = Router();
|
|
224
|
-
router.post('/chat/ai-stream', validate(AIStreamRequestSchema),
|
|
225
|
-
// eslint-disable-next-line @typescript-eslint/no-misused-promises -- TODO: wrap async route handler
|
|
226
|
-
async (req, res, next) => {
|
|
265
|
+
router.post('/chat/ai-stream', validate(AIStreamRequestSchema), asyncHandler(async (req, res, next) => {
|
|
227
266
|
try {
|
|
228
267
|
// eslint-disable-next-line @typescript-eslint/no-unsafe-type-assertion -- validated by Zod middleware
|
|
229
268
|
const body = req.body;
|
|
@@ -234,26 +273,16 @@ export function createAIStreamRouter(options) {
|
|
|
234
273
|
});
|
|
235
274
|
return;
|
|
236
275
|
}
|
|
237
|
-
const
|
|
238
|
-
|
|
239
|
-
const
|
|
240
|
-
|
|
241
|
-
|
|
242
|
-
|
|
243
|
-
|
|
244
|
-
|
|
245
|
-
|
|
246
|
-
|
|
247
|
-
}
|
|
248
|
-
if (!session) {
|
|
249
|
-
const auth = getAuthContext(res);
|
|
250
|
-
session = await options.sessionManager.create(role, auth, undefined, undefined, deployId);
|
|
251
|
-
}
|
|
252
|
-
}
|
|
253
|
-
else {
|
|
254
|
-
const auth = getAuthContext(res);
|
|
255
|
-
session = await options.sessionManager.create(role, auth, undefined, undefined, deployId);
|
|
256
|
-
}
|
|
276
|
+
const auth = getAuthContext(res);
|
|
277
|
+
// Resolve session (lookup in memory, resume from store, or create new)
|
|
278
|
+
const { session, toolContextFactory } = await resolveSession(body.session_id, {
|
|
279
|
+
sessionManager: options.sessionManager,
|
|
280
|
+
bundleResolver: options.bundleResolver,
|
|
281
|
+
shared: options.shared,
|
|
282
|
+
role: body.role,
|
|
283
|
+
deployId: body.deploy_id,
|
|
284
|
+
auth,
|
|
285
|
+
});
|
|
257
286
|
// Set up SSE headers with Vercel AI SDK protocol marker
|
|
258
287
|
res.setHeader('Content-Type', 'text/event-stream');
|
|
259
288
|
res.setHeader('Cache-Control', 'no-cache');
|
|
@@ -263,8 +292,13 @@ export function createAIStreamRouter(options) {
|
|
|
263
292
|
const controller = new AbortController();
|
|
264
293
|
res.on('close', () => controller.abort());
|
|
265
294
|
// Build per-request hooks with auth context
|
|
266
|
-
const hooks = options.createStreamHooks?.(
|
|
267
|
-
|
|
295
|
+
const hooks = options.createStreamHooks?.(auth);
|
|
296
|
+
// Run message through the agent loop
|
|
297
|
+
const stream = options.sessionManager.runMessage(session.id, message, {
|
|
298
|
+
signal: controller.signal,
|
|
299
|
+
buildToolContext: toolContextFactory,
|
|
300
|
+
onUsage: adaptOnUsage(hooks, session),
|
|
301
|
+
});
|
|
268
302
|
const state = createAdapterState();
|
|
269
303
|
for await (const event of stream) {
|
|
270
304
|
if (controller.signal.aborted)
|
|
@@ -274,6 +308,7 @@ export function createAIStreamRouter(options) {
|
|
|
274
308
|
res.write(`data: ${JSON.stringify(uiEvent)}\n\n`);
|
|
275
309
|
}
|
|
276
310
|
}
|
|
311
|
+
await fireDrainHooks(options.sessionManager, hooks, { session, toolCalls: [] });
|
|
277
312
|
// Terminal sentinel
|
|
278
313
|
res.write('data: [DONE]\n\n');
|
|
279
314
|
res.end();
|
|
@@ -292,7 +327,7 @@ export function createAIStreamRouter(options) {
|
|
|
292
327
|
next(err);
|
|
293
328
|
}
|
|
294
329
|
}
|
|
295
|
-
});
|
|
330
|
+
}));
|
|
296
331
|
return router;
|
|
297
332
|
}
|
|
298
333
|
//# sourceMappingURL=ai-stream.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ai-stream.js","sourceRoot":"","sources":["../../../src/routes/ai-stream.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;;;;;;;GAWG;AAEH,OAAO,
|
|
1
|
+
{"version":3,"file":"ai-stream.js","sourceRoot":"","sources":["../../../src/routes/ai-stream.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;;;;;;;GAWG;AAEH,OAAO,EAAC,MAAM,EAAC,MAAM,SAAS,CAAC;AAC/B,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AACtB,OAAO,EAAC,QAAQ,EAAC,MAAM,qCAAqC,CAAC;AAC7D,OAAO,EAAC,cAAc,EAAC,MAAM,uBAAuB,CAAC;AAIrD,OAAO,EAAC,YAAY,EAAgB,MAAM,aAAa,CAAC;AACxD,OAAO,EAAC,cAAc,EAAC,MAAM,uBAAuB,CAAC;AAErD,OAAO,EAAC,YAAY,EAAE,YAAY,EAAE,cAAc,EAAC,MAAM,oBAAoB,CAAC;AAE9E,8EAA8E;AAC9E,gDAAgD;AAChD,8EAA8E;AAE9E,MAAM,mBAAmB,GAAG,CAAC,CAAC,MAAM,CAAC;IACnC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;IAChB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CAC5B,CAAC,CAAC;AAEH,MAAM,eAAe,GAAG,CAAC,CAAC,MAAM,CAAC;IAC/B,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,WAAW,EAAE,QAAQ,CAAC,CAAC;IAC7C,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC,QAAQ,EAAE;IAC9C,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CAC/B,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,qBAAqB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC5C,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;IACzC,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACjC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC3B,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CACjC,CAAC,CAAC;AAwGH,SAAS,kBAAkB;IACzB,OAAO;QACL,SAAS,EAAE,OAAO,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE;QACxE,aAAa,EAAE,KAAK;QACpB,WAAW,EAAE,EAAE;QACf,gBAAgB,EAAE,CAAC;KACpB,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,KAAmB;IACzC,IAAI,CAAC,KAAK,CAAC,aAAa;QAAE,OAAO,IAAI,CAAC;IACtC,KAAK,CAAC,aAAa,GAAG,KAAK,CAAC;IAC5B,OAAO,EAAC,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,KAAK,CAAC,WAAW,EAAC,CAAC;AACnD,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,cAAc,CAC5B,KAAe,EACf,KAAmB;IAEnB,MAAM,GAAG,GAAoB,EAAE,CAAC;IAEhC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;QACnB,KAAK,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC;YACvB,GAAG,CAAC,IAAI,CAAC,EAAC,IAAI,EAAE,eAAe,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAC,CAAC,CAAC;YAC9D,GAAG,CAAC,IAAI,CAAC,EAAC,IAAI,EAAE,YAAY,EAAC,CAAC,CAAC;YAC/B,MAAM;QACR,CAAC;QAED,KAAK,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC;YAC5B,IAAI,CAAC,KAAK,CAAC,aAAa,EAAE,CAAC;gBACzB,KAAK,CAAC,gBAAgB,EAAE,CAAC;gBACzB,KAAK,CAAC,WAAW,GAAG,QAAQ,KAAK,CAAC,gBAAgB,EAAE,CAAC;gBACrD,KAAK,CAAC,aAAa,GAAG,IAAI,CAAC;gBAC3B,GAAG,CAAC,IAAI,CAAC,EAAC,IAAI,EAAE,YAAY,EAAE,EAAE,EAAE,KAAK,CAAC,WAAW,EAAC,CAAC,CAAC;YACxD,CAAC;YACD,GAAG,CAAC,IAAI,CAAC;gBACP,IAAI,EAAE,YAAY;gBAClB,EAAE,EAAE,KAAK,CAAC,WAAW;gBACrB,KAAK,EAAE,KAAK,CAAC,OAAO;aACrB,CAAC,CAAC;YACH,MAAM;QACR,CAAC;QAED,KAAK,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC;YAChC,MAAM,MAAM,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;YACrC,IAAI,MAAM;gBAAE,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC7B,GAAG,CAAC,IAAI,CAAC;gBACP,IAAI,EAAE,kBAAkB;gBACxB,UAAU,EAAE,KAAK,CAAC,OAAO;gBACzB,QAAQ,EAAE,KAAK,CAAC,SAAS;aAC1B,CAAC,CAAC;YACH,GAAG,CAAC,IAAI,CAAC;gBACP,IAAI,EAAE,sBAAsB;gBAC5B,UAAU,EAAE,KAAK,CAAC,OAAO;gBACzB,QAAQ,EAAE,KAAK,CAAC,SAAS;gBACzB,KAAK,EAAE,KAAK,CAAC,UAAU;aACxB,CAAC,CAAC;YACH,MAAM;QACR,CAAC;QAED,KAAK,YAAY,CAAC,cAAc,CAAC,CAAC,CAAC;YACjC,IAAI,KAAK,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;gBAC7B,GAAG,CAAC,IAAI,CAAC;oBACP,IAAI,EAAE,mBAAmB;oBACzB,UAAU,EAAE,KAAK,CAAC,OAAO;oBACzB,SAAS,EAAE,KAAK,CAAC,KAAK,IAAI,kBAAkB;iBAC7C,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,IAAI,CAAC;oBACP,IAAI,EAAE,uBAAuB;oBAC7B,UAAU,EAAE,KAAK,CAAC,OAAO;oBACzB,MAAM,EAAE,EAAC,MAAM,EAAE,KAAK,CAAC,MAAM,IAAI,EAAE,EAAC;iBACrC,CAAC,CAAC;YACL,CAAC;YACD,MAAM;QACR,CAAC;QAED,KAAK,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC;YAChC,GAAG,CAAC,IAAI,CAAC;gBACP,IAAI,EAAE,eAAe;gBACrB,IAAI,EAAE;oBACJ,cAAc,EAAE,KAAK,CAAC,cAAc;oBACpC,UAAU,EAAE,KAAK,CAAC,UAAU;oBAC5B,UAAU,EAAE,KAAK,CAAC,UAAU;oBAC5B,SAAS,EAAE,KAAK,CAAC,SAAS;oBAC1B,MAAM,EAAE,KAAK,CAAC,MAAM;oBACpB,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,KAAK,EAAE,KAAK,CAAC,KAAK;iBACnB;aACF,CAAC,CAAC;YACH,MAAM;QACR,CAAC;QAED,KAAK,YAAY,CAAC,cAAc,CAAC,CAAC,CAAC;YACjC,GAAG,CAAC,IAAI,CAAC;gBACP,IAAI,EAAE,sBAAsB;gBAC5B,IAAI,EAAE,EAAC,UAAU,EAAE,KAAK,CAAC,UAAU,EAAC;aACrC,CAAC,CAAC;YACH,MAAM;QACR,CAAC;QAED,KAAK,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC;YACzB,GAAG,CAAC,IAAI,CAAC;gBACP,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE,EAAC,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAC;aACzD,CAAC,CAAC;YACH,MAAM;QACR,CAAC;QAED,KAAK,YAAY,CAAC,UAAU,CAAC,CAAC,CAAC;YAC7B,GAAG,CAAC,IAAI,CAAC;gBACP,IAAI,EAAE,kBAAkB;gBACxB,IAAI,EAAE;oBACJ,WAAW,EAAE,KAAK,CAAC,WAAW;oBAC9B,KAAK,EAAE,KAAK,CAAC,KAAK;oBAClB,KAAK,EAAE,KAAK,CAAC,KAAK;oBAClB,SAAS,EAAE,KAAK,CAAC,SAAS;oBAC1B,MAAM,EAAE,KAAK,CAAC,MAAM;iBACrB;aACF,CAAC,CAAC;YACH,MAAM;QACR,CAAC;QAED,KAAK,YAAY,CAAC,eAAe,CAAC,CAAC,CAAC;YAClC,GAAG,CAAC,IAAI,CAAC;gBACP,IAAI,EAAE,uBAAuB;gBAC7B,IAAI,EAAE,EAAC,eAAe,EAAE,KAAK,CAAC,eAAe,EAAC;aAC/C,CAAC,CAAC;YACH,MAAM;QACR,CAAC;QAED,KAAK,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC;YAC3B,GAAG,CAAC,IAAI,CAAC;gBACP,IAAI,EAAE,eAAe;gBACrB,IAAI,EAAE;oBACJ,aAAa,EAAE,KAAK,CAAC,aAAa;oBAClC,UAAU,EAAE,KAAK,CAAC,UAAU;iBAC7B;aACF,CAAC,CAAC;YACH,MAAM;QACR,CAAC;QAED,KAAK,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC;YACxB,GAAG,CAAC,IAAI,CAAC,EAAC,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,KAAK,CAAC,OAAO,EAAC,CAAC,CAAC;YACpD,MAAM;QACR,CAAC;QAED,KAAK,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC;YACvB,MAAM,MAAM,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;YACrC,IAAI,MAAM;gBAAE,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC7B,GAAG,CAAC,IAAI,CAAC,EAAC,IAAI,EAAE,aAAa,EAAC,CAAC,CAAC;YAChC,GAAG,CAAC,IAAI,CAAC,EAAC,IAAI,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,EAAC,CAAC,CAAC;YACjD,MAAM;QACR,CAAC;QAED,KAAK,YAAY,CAAC,oBAAoB,CAAC,CAAC,CAAC;YACvC,GAAG,CAAC,IAAI,CAAC;gBACP,IAAI,EAAE,4BAA4B;gBAClC,IAAI,EAAE;oBACJ,QAAQ,EAAE,KAAK,CAAC,QAAQ;oBACxB,MAAM,EAAE,KAAK,CAAC,MAAM;oBACpB,MAAM,EAAE,KAAK,CAAC,MAAM;oBACpB,SAAS,EAAE,KAAK,CAAC,SAAS;iBAC3B;aACF,CAAC,CAAC;YACH,MAAM;QACR,CAAC;QAED,KAAK,YAAY,CAAC,eAAe,CAAC,CAAC,CAAC;YAClC,GAAG,CAAC,IAAI,CAAC;gBACP,IAAI,EAAE,uBAAuB;gBAC7B,IAAI,EAAE;oBACJ,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;oBACxC,SAAS,EAAE,KAAK,CAAC,SAAS;iBAC3B;aACF,CAAC,CAAC;YACH,MAAM;QACR,CAAC;QAED,KAAK,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC;YAChC,GAAG,CAAC,IAAI,CAAC;gBACP,IAAI,EAAE,qBAAqB;gBAC3B,IAAI,EAAE;oBACJ,aAAa,EAAE,KAAK,CAAC,aAAa;oBAClC,YAAY,EAAE,KAAK,CAAC,YAAY;oBAChC,iBAAiB,EAAE,KAAK,CAAC,iBAAiB;iBAC3C;aACF,CAAC,CAAC;YACH,MAAM;QACR,CAAC;QAED,KAAK,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC;YAC1B,GAAG,CAAC,IAAI,CAAC;gBACP,IAAI,EAAE,eAAe;gBACrB,IAAI,EAAE,EAAC,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAC;aAC3D,CAAC,CAAC;YACH,MAAM;QACR,CAAC;QAED,iEAAiE;QACjE,qEAAqE;QACrE,6CAA6C;QAC7C,KAAK,YAAY,CAAC,YAAY,CAAC;QAC/B,KAAK,YAAY,CAAC,UAAU,CAAC;QAC7B,KAAK,YAAY,CAAC,QAAQ,CAAC;QAC3B,KAAK,YAAY,CAAC,UAAU;YAC1B,MAAM;QAER,OAAO,CAAC,CAAC,CAAC;YACR,MAAM,WAAW,GAAU,KAAK,CAAC;YACjC,MAAM,IAAI,KAAK,CAAC,4BAA6B,WAAwB,CAAC,IAAI,EAAE,CAAC,CAAC;QAChF,CAAC;IACH,CAAC;IAED,OAAO,GAAG,CAAC;AACb,CAAC;AAED,8EAA8E;AAC9E,wDAAwD;AACxD,8EAA8E;AAE9E,MAAM,UAAU,kBAAkB,CAAC,QAAqC;IACtE,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAClD,IAAI,CAAC,WAAW;QAAE,OAAO,EAAE,CAAC;IAE5B,4CAA4C;IAC5C,IAAI,WAAW,CAAC,KAAK,EAAE,CAAC;QACtB,MAAM,QAAQ,GAAG,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;QAClE,IAAI,QAAQ,EAAE,IAAI;YAAE,OAAO,QAAQ,CAAC,IAAI,CAAC;IAC3C,CAAC;IAED,yDAAyD;IACzD,IAAI,WAAW,CAAC,OAAO;QAAE,OAAO,WAAW,CAAC,OAAO,CAAC;IAEpD,OAAO,EAAE,CAAC;AACZ,CAAC;AAcD,MAAM,UAAU,oBAAoB,CAAC,OAA8B;IACjE,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC;IAExB,MAAM,CAAC,IAAI,CACT,iBAAiB,EACjB,QAAQ,CAAC,qBAAqB,CAAC,EAC/B,YAAY,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QACpC,IAAI,CAAC;YACH,sGAAsG;YACtG,MAAM,IAAI,GAAG,GAAG,CAAC,IAAuB,CAAC;YACzC,MAAM,OAAO,GAAG,kBAAkB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAElD,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;oBACnB,KAAK,EAAE,EAAC,IAAI,EAAE,iBAAiB,EAAE,OAAO,EAAE,yCAAyC,EAAC;iBACrF,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAED,MAAM,IAAI,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;YAEjC,uEAAuE;YACvE,MAAM,EAAC,OAAO,EAAE,kBAAkB,EAAC,GAAG,MAAM,cAAc,CAAC,IAAI,CAAC,UAAU,EAAE;gBAC1E,cAAc,EAAE,OAAO,CAAC,cAAc;gBACtC,cAAc,EAAE,OAAO,CAAC,cAAc;gBACtC,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,QAAQ,EAAE,IAAI,CAAC,SAAS;gBACxB,IAAI;aACL,CAAC,CAAC;YAEH,wDAAwD;YACxD,GAAG,CAAC,SAAS,CAAC,cAAc,EAAE,mBAAmB,CAAC,CAAC;YACnD,GAAG,CAAC,SAAS,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;YAC3C,GAAG,CAAC,SAAS,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;YAC1C,GAAG,CAAC,SAAS,CAAC,+BAA+B,EAAE,IAAI,CAAC,CAAC;YACrD,GAAG,CAAC,YAAY,EAAE,CAAC;YAEnB,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;YACzC,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC,CAAC;YAE1C,4CAA4C;YAC5C,MAAM,KAAK,GAAG,OAAO,CAAC,iBAAiB,EAAE,CAAC,IAAI,CAAC,CAAC;YAEhD,qCAAqC;YACrC,MAAM,MAAM,GAAG,OAAO,CAAC,cAAc,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,EAAE;gBACpE,MAAM,EAAE,UAAU,CAAC,MAAM;gBACzB,gBAAgB,EAAE,kBAAkB;gBACpC,OAAO,EAAE,YAAY,CAAC,KAAK,EAAE,OAAO,CAAC;aACtC,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,kBAAkB,EAAE,CAAC;YAEnC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBACjC,IAAI,UAAU,CAAC,MAAM,CAAC,OAAO;oBAAE,MAAM;gBAErC,MAAM,QAAQ,GAAG,cAAc,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;gBAC9C,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;oBAC/B,GAAG,CAAC,KAAK,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;gBACpD,CAAC;YACH,CAAC;YAED,MAAM,cAAc,CAAC,OAAO,CAAC,cAAc,EAAE,KAAK,EAAE,EAAC,OAAO,EAAE,SAAS,EAAE,EAAE,EAAC,CAAC,CAAC;YAE9E,oBAAoB;YACpB,GAAG,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;YAC9B,GAAG,CAAC,GAAG,EAAE,CAAC;QACZ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,GAAG,CAAC,WAAW,EAAE,CAAC;gBACpB,MAAM,UAAU,GAAY;oBAC1B,IAAI,EAAE,OAAO;oBACb,SAAS,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;iBAChE,CAAC;gBACF,GAAG,CAAC,KAAK,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;gBACrD,GAAG,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;gBAC9B,GAAG,CAAC,GAAG,EAAE,CAAC;YACZ,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,GAAG,CAAC,CAAC;YACZ,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CACH,CAAC;IAEF,OAAO,MAAM,CAAC;AAChB,CAAC"}
|
|
@@ -0,0 +1,107 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2026 Amodal Labs, Inc.
|
|
4
|
+
* SPDX-License-Identifier: MIT
|
|
5
|
+
*/
|
|
6
|
+
/**
|
|
7
|
+
* Tests for the new non-streaming chat route.
|
|
8
|
+
*/
|
|
9
|
+
import { describe, it, expect, vi, beforeEach } from 'vitest';
|
|
10
|
+
import express from 'express';
|
|
11
|
+
import request from 'supertest';
|
|
12
|
+
import { createChatRouter } from './chat.js';
|
|
13
|
+
import { errorHandler } from '../middleware/error-handler.js';
|
|
14
|
+
import { SSEEventType } from '../types.js';
|
|
15
|
+
// ---------------------------------------------------------------------------
|
|
16
|
+
// Mocks
|
|
17
|
+
// ---------------------------------------------------------------------------
|
|
18
|
+
const mockResolveSession = vi.fn();
|
|
19
|
+
vi.mock('./session-resolver.js', () => ({
|
|
20
|
+
resolveSession: (...args) => mockResolveSession(...args),
|
|
21
|
+
}));
|
|
22
|
+
function stubSession(id = 'sess-1') {
|
|
23
|
+
return {
|
|
24
|
+
id,
|
|
25
|
+
model: 'test-model',
|
|
26
|
+
providerName: 'test-provider',
|
|
27
|
+
};
|
|
28
|
+
}
|
|
29
|
+
async function* makeEvents(events) {
|
|
30
|
+
for (const event of events) {
|
|
31
|
+
yield event;
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
function createApp() {
|
|
35
|
+
const mockRunMessage = vi.fn();
|
|
36
|
+
const sessionManager = { runMessage: mockRunMessage, persist: vi.fn().mockResolvedValue(undefined) };
|
|
37
|
+
const app = express();
|
|
38
|
+
app.use(express.json());
|
|
39
|
+
app.use(createChatRouter({
|
|
40
|
+
sessionManager: sessionManager,
|
|
41
|
+
bundleResolver: {},
|
|
42
|
+
shared: { storeBackend: null, mcpManager: null, logger: {} },
|
|
43
|
+
}));
|
|
44
|
+
app.use(errorHandler);
|
|
45
|
+
return { app, mockRunMessage };
|
|
46
|
+
}
|
|
47
|
+
// ---------------------------------------------------------------------------
|
|
48
|
+
// Tests
|
|
49
|
+
// ---------------------------------------------------------------------------
|
|
50
|
+
describe('POST /chat/sync (new route)', () => {
|
|
51
|
+
beforeEach(() => {
|
|
52
|
+
vi.clearAllMocks();
|
|
53
|
+
});
|
|
54
|
+
it('returns JSON with session_id, response, and tool_calls', async () => {
|
|
55
|
+
mockResolveSession.mockResolvedValue({ session: stubSession(), toolContextFactory: vi.fn() });
|
|
56
|
+
const { app, mockRunMessage } = createApp();
|
|
57
|
+
mockRunMessage.mockReturnValue(makeEvents([
|
|
58
|
+
{ type: SSEEventType.Init, session_id: 'sess-1', timestamp: '' },
|
|
59
|
+
{ type: SSEEventType.TextDelta, content: 'Hello ', timestamp: '' },
|
|
60
|
+
{ type: SSEEventType.TextDelta, content: 'world!', timestamp: '' },
|
|
61
|
+
{ type: SSEEventType.Done, timestamp: '' },
|
|
62
|
+
]));
|
|
63
|
+
const res = await request(app)
|
|
64
|
+
.post('/chat/sync')
|
|
65
|
+
.send({ message: 'hello' });
|
|
66
|
+
expect(res.status).toBe(200);
|
|
67
|
+
expect(res.body).toEqual({
|
|
68
|
+
session_id: 'sess-1',
|
|
69
|
+
response: 'Hello world!',
|
|
70
|
+
tool_calls: [],
|
|
71
|
+
});
|
|
72
|
+
});
|
|
73
|
+
it('collects tool calls from SSE events', async () => {
|
|
74
|
+
mockResolveSession.mockResolvedValue({ session: stubSession(), toolContextFactory: vi.fn() });
|
|
75
|
+
const { app, mockRunMessage } = createApp();
|
|
76
|
+
mockRunMessage.mockReturnValue(makeEvents([
|
|
77
|
+
{ type: SSEEventType.ToolCallStart, tool_name: 'request', tool_id: 'tc-1', parameters: {}, timestamp: '' },
|
|
78
|
+
{ type: SSEEventType.ToolCallResult, tool_id: 'tc-1', status: 'success', timestamp: '' },
|
|
79
|
+
{ type: SSEEventType.ToolCallStart, tool_name: 'store_write', tool_id: 'tc-2', parameters: {}, timestamp: '' },
|
|
80
|
+
{ type: SSEEventType.ToolCallResult, tool_id: 'tc-2', status: 'error', error: 'Store unavailable', timestamp: '' },
|
|
81
|
+
{ type: SSEEventType.Done, timestamp: '' },
|
|
82
|
+
]));
|
|
83
|
+
const res = await request(app)
|
|
84
|
+
.post('/chat/sync')
|
|
85
|
+
.send({ message: 'do things' });
|
|
86
|
+
expect(res.body.tool_calls).toEqual([
|
|
87
|
+
{ tool_name: 'request', tool_id: 'tc-1', status: 'success', error: undefined },
|
|
88
|
+
{ tool_name: 'store_write', tool_id: 'tc-2', status: 'error', error: 'Store unavailable' },
|
|
89
|
+
]);
|
|
90
|
+
});
|
|
91
|
+
it('returns 400 for missing message', async () => {
|
|
92
|
+
const { app } = createApp();
|
|
93
|
+
const res = await request(app)
|
|
94
|
+
.post('/chat/sync')
|
|
95
|
+
.send({});
|
|
96
|
+
expect(res.status).toBe(400);
|
|
97
|
+
});
|
|
98
|
+
it('returns 500 on resolveSession error', async () => {
|
|
99
|
+
mockResolveSession.mockRejectedValue(new Error('Bundle not found'));
|
|
100
|
+
const { app } = createApp();
|
|
101
|
+
const res = await request(app)
|
|
102
|
+
.post('/chat/sync')
|
|
103
|
+
.send({ message: 'hello' });
|
|
104
|
+
expect(res.status).toBe(500);
|
|
105
|
+
});
|
|
106
|
+
});
|
|
107
|
+
//# sourceMappingURL=chat-new.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"chat-new.test.js","sourceRoot":"","sources":["../../../src/routes/chat-new.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;GAEG;AAEH,OAAO,EAAC,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAC,MAAM,QAAQ,CAAC;AAC5D,OAAO,OAAO,MAAM,SAAS,CAAC;AAC9B,OAAO,OAAO,MAAM,WAAW,CAAC;AAChC,OAAO,EAAC,gBAAgB,EAAC,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAC,YAAY,EAAC,MAAM,gCAAgC,CAAC;AAC5D,OAAO,EAAC,YAAY,EAAC,MAAM,aAAa,CAAC;AAGzC,8EAA8E;AAC9E,QAAQ;AACR,8EAA8E;AAE9E,MAAM,kBAAkB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AACnC,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,cAAc,EAAE,CAAC,GAAG,IAAe,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,IAAI,CAAC;CACpE,CAAC,CAAC,CAAC;AAEJ,SAAS,WAAW,CAAC,EAAE,GAAG,QAAQ;IAChC,OAAO;QACL,EAAE;QACF,KAAK,EAAE,YAAY;QACnB,YAAY,EAAE,eAAe;KAC9B,CAAC;AACJ,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,UAAU,CAAC,MAAkB;IAC3C,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,SAAS,SAAS;IAChB,MAAM,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAC/B,MAAM,cAAc,GAAG,EAAC,UAAU,EAAE,cAAc,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,EAAC,CAAC;IAEnG,MAAM,GAAG,GAAG,OAAO,EAAE,CAAC;IACtB,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IACxB,GAAG,CAAC,GAAG,CAAC,gBAAgB,CAAC;QACvB,cAAc,EAAE,cAAuB;QACvC,cAAc,EAAE,EAAE;QAClB,MAAM,EAAE,EAAC,YAAY,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,EAAE,EAAW,EAAC;KACpE,CAAC,CAAC,CAAC;IACJ,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;IAEtB,OAAO,EAAC,GAAG,EAAE,cAAc,EAAC,CAAC;AAC/B,CAAC;AAED,8EAA8E;AAC9E,QAAQ;AACR,8EAA8E;AAE9E,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC3C,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,kBAAkB,CAAC,iBAAiB,CAAC,EAAC,OAAO,EAAE,WAAW,EAAE,EAAE,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE,EAAC,CAAC,CAAC;QAE5F,MAAM,EAAC,GAAG,EAAE,cAAc,EAAC,GAAG,SAAS,EAAE,CAAC;QAC1C,cAAc,CAAC,eAAe,CAC5B,UAAU,CAAC;YACT,EAAC,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,EAAa;YAC1E,EAAC,IAAI,EAAE,YAAY,CAAC,SAAS,EAAE,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,EAAa;YAC5E,EAAC,IAAI,EAAE,YAAY,CAAC,SAAS,EAAE,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,EAAa;YAC5E,EAAC,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,EAAa;SACrD,CAAC,CACH,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;aAC3B,IAAI,CAAC,YAAY,CAAC;aAClB,IAAI,CAAC,EAAC,OAAO,EAAE,OAAO,EAAC,CAAC,CAAC;QAE5B,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC;YACvB,UAAU,EAAE,QAAQ;YACpB,QAAQ,EAAE,cAAc;YACxB,UAAU,EAAE,EAAE;SACf,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,kBAAkB,CAAC,iBAAiB,CAAC,EAAC,OAAO,EAAE,WAAW,EAAE,EAAE,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE,EAAC,CAAC,CAAC;QAE5F,MAAM,EAAC,GAAG,EAAE,cAAc,EAAC,GAAG,SAAS,EAAE,CAAC;QAC1C,cAAc,CAAC,eAAe,CAC5B,UAAU,CAAC;YACT,EAAC,IAAI,EAAE,YAAY,CAAC,aAAa,EAAE,SAAS,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAa;YACpH,EAAC,IAAI,EAAE,YAAY,CAAC,cAAc,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,EAAE,EAAa;YAClG,EAAC,IAAI,EAAE,YAAY,CAAC,aAAa,EAAE,SAAS,EAAE,aAAa,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAa;YACxH,EAAC,IAAI,EAAE,YAAY,CAAC,cAAc,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,mBAAmB,EAAE,SAAS,EAAE,EAAE,EAAa;YAC5H,EAAC,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,EAAa;SACrD,CAAC,CACH,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;aAC3B,IAAI,CAAC,YAAY,CAAC;aAClB,IAAI,CAAC,EAAC,OAAO,EAAE,WAAW,EAAC,CAAC,CAAC;QAEhC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC;YAClC,EAAC,SAAS,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAC;YAC5E,EAAC,SAAS,EAAE,aAAa,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,mBAAmB,EAAC;SACzF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,EAAC,GAAG,EAAC,GAAG,SAAS,EAAE,CAAC;QAC1B,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;aAC3B,IAAI,CAAC,YAAY,CAAC;aAClB,IAAI,CAAC,EAAE,CAAC,CAAC;QAEZ,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,kBAAkB,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC,CAAC;QAEpE,MAAM,EAAC,GAAG,EAAC,GAAG,SAAS,EAAE,CAAC;QAC1B,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;aAC3B,IAAI,CAAC,YAAY,CAAC;aAClB,IAAI,CAAC,EAAC,OAAO,EAAE,OAAO,EAAC,CAAC,CAAC;QAE5B,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,135 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2026 Amodal Labs, Inc.
|
|
4
|
+
* SPDX-License-Identifier: MIT
|
|
5
|
+
*/
|
|
6
|
+
/**
|
|
7
|
+
* Tests for the new chat-stream route.
|
|
8
|
+
*
|
|
9
|
+
* Mocks resolveSession and StandaloneSessionManager.runMessage to verify
|
|
10
|
+
* the route wiring: SSE format, hooks adaptation, error handling.
|
|
11
|
+
*/
|
|
12
|
+
import { describe, it, expect, vi, beforeEach } from 'vitest';
|
|
13
|
+
import express from 'express';
|
|
14
|
+
import request from 'supertest';
|
|
15
|
+
import { createChatStreamRouter } from './chat-stream.js';
|
|
16
|
+
import { errorHandler } from '../middleware/error-handler.js';
|
|
17
|
+
import { SSEEventType } from '../types.js';
|
|
18
|
+
// ---------------------------------------------------------------------------
|
|
19
|
+
// Mocks
|
|
20
|
+
// ---------------------------------------------------------------------------
|
|
21
|
+
const mockResolveSession = vi.fn();
|
|
22
|
+
vi.mock('./session-resolver.js', () => ({
|
|
23
|
+
resolveSession: (...args) => mockResolveSession(...args),
|
|
24
|
+
}));
|
|
25
|
+
function stubSession(id = 'sess-1') {
|
|
26
|
+
return {
|
|
27
|
+
id,
|
|
28
|
+
model: 'test-model',
|
|
29
|
+
providerName: 'test-provider',
|
|
30
|
+
};
|
|
31
|
+
}
|
|
32
|
+
function stubFactory() {
|
|
33
|
+
return vi.fn();
|
|
34
|
+
}
|
|
35
|
+
async function* makeEvents(events) {
|
|
36
|
+
for (const event of events) {
|
|
37
|
+
yield event;
|
|
38
|
+
}
|
|
39
|
+
}
|
|
40
|
+
function createApp(overrides = {}) {
|
|
41
|
+
const mockRunMessage = vi.fn();
|
|
42
|
+
const sessionManager = { runMessage: mockRunMessage, persist: vi.fn().mockResolvedValue(undefined), ...overrides };
|
|
43
|
+
const app = express();
|
|
44
|
+
app.use(express.json());
|
|
45
|
+
app.use(createChatStreamRouter({
|
|
46
|
+
sessionManager: sessionManager,
|
|
47
|
+
bundleResolver: {},
|
|
48
|
+
shared: { storeBackend: null, mcpManager: null, logger: {} },
|
|
49
|
+
}));
|
|
50
|
+
app.use(errorHandler);
|
|
51
|
+
return { app, mockRunMessage };
|
|
52
|
+
}
|
|
53
|
+
// ---------------------------------------------------------------------------
|
|
54
|
+
// Tests
|
|
55
|
+
// ---------------------------------------------------------------------------
|
|
56
|
+
describe('POST /chat/stream (new route)', () => {
|
|
57
|
+
beforeEach(() => {
|
|
58
|
+
vi.clearAllMocks();
|
|
59
|
+
});
|
|
60
|
+
it('returns SSE content type and streams events', async () => {
|
|
61
|
+
const factory = stubFactory();
|
|
62
|
+
mockResolveSession.mockResolvedValue({ session: stubSession(), toolContextFactory: factory });
|
|
63
|
+
const { app, mockRunMessage } = createApp();
|
|
64
|
+
mockRunMessage.mockReturnValue(makeEvents([
|
|
65
|
+
{ type: SSEEventType.Init, session_id: 'sess-1', timestamp: '2024-01-01T00:00:00Z' },
|
|
66
|
+
{ type: SSEEventType.TextDelta, content: 'Hello!', timestamp: '2024-01-01T00:00:01Z' },
|
|
67
|
+
{ type: SSEEventType.Done, timestamp: '2024-01-01T00:00:02Z' },
|
|
68
|
+
]));
|
|
69
|
+
const res = await request(app)
|
|
70
|
+
.post('/chat/stream')
|
|
71
|
+
.send({ message: 'hello' });
|
|
72
|
+
expect(res.headers['content-type']).toContain('text/event-stream');
|
|
73
|
+
const lines = res.text.split('\n\n').filter(Boolean);
|
|
74
|
+
expect(lines.length).toBeGreaterThanOrEqual(3);
|
|
75
|
+
for (const line of lines) {
|
|
76
|
+
expect(line.startsWith('data: ')).toBe(true);
|
|
77
|
+
}
|
|
78
|
+
});
|
|
79
|
+
it('passes toolContextFactory as buildToolContext to runMessage', async () => {
|
|
80
|
+
const factory = stubFactory();
|
|
81
|
+
mockResolveSession.mockResolvedValue({ session: stubSession(), toolContextFactory: factory });
|
|
82
|
+
const { app, mockRunMessage } = createApp();
|
|
83
|
+
mockRunMessage.mockReturnValue(makeEvents([
|
|
84
|
+
{ type: SSEEventType.Done, timestamp: '2024-01-01T00:00:00Z' },
|
|
85
|
+
]));
|
|
86
|
+
await request(app)
|
|
87
|
+
.post('/chat/stream')
|
|
88
|
+
.send({ message: 'hello' });
|
|
89
|
+
expect(mockRunMessage).toHaveBeenCalledWith('sess-1', 'hello', expect.objectContaining({ buildToolContext: factory }));
|
|
90
|
+
});
|
|
91
|
+
it('fires onAuditLog hook after stream drains', async () => {
|
|
92
|
+
const factory = stubFactory();
|
|
93
|
+
mockResolveSession.mockResolvedValue({ session: stubSession(), toolContextFactory: factory });
|
|
94
|
+
const onAuditLog = vi.fn();
|
|
95
|
+
const mockRunMessage = vi.fn().mockReturnValue(makeEvents([
|
|
96
|
+
{ type: SSEEventType.ToolCallStart, tool_name: 'request', tool_id: 'tc-1', parameters: {}, timestamp: '' },
|
|
97
|
+
{ type: SSEEventType.ToolCallResult, tool_id: 'tc-1', status: 'success', timestamp: '' },
|
|
98
|
+
{ type: SSEEventType.Done, timestamp: '' },
|
|
99
|
+
]));
|
|
100
|
+
const app = express();
|
|
101
|
+
app.use(express.json());
|
|
102
|
+
app.use(createChatStreamRouter({
|
|
103
|
+
sessionManager: { runMessage: mockRunMessage, persist: vi.fn().mockResolvedValue(undefined) },
|
|
104
|
+
bundleResolver: {},
|
|
105
|
+
shared: { storeBackend: null, mcpManager: null, logger: {} },
|
|
106
|
+
createStreamHooks: () => ({ onAuditLog }),
|
|
107
|
+
}));
|
|
108
|
+
await request(app).post('/chat/stream').send({ message: 'test' });
|
|
109
|
+
expect(onAuditLog).toHaveBeenCalledWith(expect.objectContaining({
|
|
110
|
+
event: 'session_completed',
|
|
111
|
+
details: expect.objectContaining({
|
|
112
|
+
tool_calls: [{ tool_name: 'request', tool_id: 'tc-1', status: 'success' }],
|
|
113
|
+
}),
|
|
114
|
+
}));
|
|
115
|
+
});
|
|
116
|
+
it('writes SSE error event when handler throws after headers sent', async () => {
|
|
117
|
+
mockResolveSession.mockResolvedValue({ session: stubSession(), toolContextFactory: stubFactory() });
|
|
118
|
+
const mockRunMessage = vi.fn().mockImplementation(function* () {
|
|
119
|
+
yield { type: SSEEventType.Init, session_id: 'sess-1', timestamp: '' };
|
|
120
|
+
throw new Error('mid-stream failure');
|
|
121
|
+
});
|
|
122
|
+
const app = express();
|
|
123
|
+
app.use(express.json());
|
|
124
|
+
app.use(createChatStreamRouter({
|
|
125
|
+
sessionManager: { runMessage: mockRunMessage, persist: vi.fn().mockResolvedValue(undefined) },
|
|
126
|
+
bundleResolver: {},
|
|
127
|
+
shared: { storeBackend: null, mcpManager: null, logger: {} },
|
|
128
|
+
}));
|
|
129
|
+
const res = await request(app).post('/chat/stream').send({ message: 'hello' });
|
|
130
|
+
expect(res.status).toBe(200);
|
|
131
|
+
expect(res.text).toContain(SSEEventType.Error);
|
|
132
|
+
expect(res.text).toContain('mid-stream failure');
|
|
133
|
+
});
|
|
134
|
+
});
|
|
135
|
+
//# sourceMappingURL=chat-stream-new.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"chat-stream-new.test.js","sourceRoot":"","sources":["../../../src/routes/chat-stream-new.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;GAKG;AAEH,OAAO,EAAC,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAC,MAAM,QAAQ,CAAC;AAC5D,OAAO,OAAO,MAAM,SAAS,CAAC;AAC9B,OAAO,OAAO,MAAM,WAAW,CAAC;AAChC,OAAO,EAAC,sBAAsB,EAAC,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAC,YAAY,EAAC,MAAM,gCAAgC,CAAC;AAC5D,OAAO,EAAC,YAAY,EAAC,MAAM,aAAa,CAAC;AAGzC,8EAA8E;AAC9E,QAAQ;AACR,8EAA8E;AAE9E,MAAM,kBAAkB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AACnC,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,cAAc,EAAE,CAAC,GAAG,IAAe,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,IAAI,CAAC;CACpE,CAAC,CAAC,CAAC;AAEJ,SAAS,WAAW,CAAC,EAAE,GAAG,QAAQ;IAChC,OAAO;QACL,EAAE;QACF,KAAK,EAAE,YAAY;QACnB,YAAY,EAAE,eAAe;KAC9B,CAAC;AACJ,CAAC;AAED,SAAS,WAAW;IAClB,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC;AACjB,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,UAAU,CAAC,MAAkB;IAC3C,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,SAAS,SAAS,CAAC,YAAqC,EAAE;IACxD,MAAM,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAC/B,MAAM,cAAc,GAAG,EAAC,UAAU,EAAE,cAAc,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,EAAE,GAAG,SAAS,EAAC,CAAC;IAEjH,MAAM,GAAG,GAAG,OAAO,EAAE,CAAC;IACtB,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IACxB,GAAG,CAAC,GAAG,CAAC,sBAAsB,CAAC;QAC7B,cAAc,EAAE,cAAuB;QACvC,cAAc,EAAE,EAAE;QAClB,MAAM,EAAE,EAAC,YAAY,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,EAAE,EAAW,EAAC;KACpE,CAAC,CAAC,CAAC;IACJ,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;IAEtB,OAAO,EAAC,GAAG,EAAE,cAAc,EAAC,CAAC;AAC/B,CAAC;AAED,8EAA8E;AAC9E,QAAQ;AACR,8EAA8E;AAE9E,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC7C,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,OAAO,GAAG,WAAW,EAAE,CAAC;QAC9B,kBAAkB,CAAC,iBAAiB,CAAC,EAAC,OAAO,EAAE,WAAW,EAAE,EAAE,kBAAkB,EAAE,OAAO,EAAC,CAAC,CAAC;QAE5F,MAAM,EAAC,GAAG,EAAE,cAAc,EAAC,GAAG,SAAS,EAAE,CAAC;QAC1C,cAAc,CAAC,eAAe,CAC5B,UAAU,CAAC;YACT,EAAC,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,SAAS,EAAE,sBAAsB,EAAa;YAC9F,EAAC,IAAI,EAAE,YAAY,CAAC,SAAS,EAAE,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,sBAAsB,EAAa;YAChG,EAAC,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,SAAS,EAAE,sBAAsB,EAAa;SACzE,CAAC,CACH,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;aAC3B,IAAI,CAAC,cAAc,CAAC;aACpB,IAAI,CAAC,EAAC,OAAO,EAAE,OAAO,EAAC,CAAC,CAAC;QAE5B,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QACnE,MAAM,KAAK,GAAG,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;QAC/C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,OAAO,GAAG,WAAW,EAAE,CAAC;QAC9B,kBAAkB,CAAC,iBAAiB,CAAC,EAAC,OAAO,EAAE,WAAW,EAAE,EAAE,kBAAkB,EAAE,OAAO,EAAC,CAAC,CAAC;QAE5F,MAAM,EAAC,GAAG,EAAE,cAAc,EAAC,GAAG,SAAS,EAAE,CAAC;QAC1C,cAAc,CAAC,eAAe,CAAC,UAAU,CAAC;YACxC,EAAC,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,SAAS,EAAE,sBAAsB,EAAa;SACzE,CAAC,CAAC,CAAC;QAEJ,MAAM,OAAO,CAAC,GAAG,CAAC;aACf,IAAI,CAAC,cAAc,CAAC;aACpB,IAAI,CAAC,EAAC,OAAO,EAAE,OAAO,EAAC,CAAC,CAAC;QAE5B,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CACzC,QAAQ,EACR,OAAO,EACP,MAAM,CAAC,gBAAgB,CAAC,EAAC,gBAAgB,EAAE,OAAO,EAAC,CAAC,CACrD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,OAAO,GAAG,WAAW,EAAE,CAAC;QAC9B,kBAAkB,CAAC,iBAAiB,CAAC,EAAC,OAAO,EAAE,WAAW,EAAE,EAAE,kBAAkB,EAAE,OAAO,EAAC,CAAC,CAAC;QAE5F,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,MAAM,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,UAAU,CAAC;YACxD,EAAC,IAAI,EAAE,YAAY,CAAC,aAAa,EAAE,SAAS,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAa;YACpH,EAAC,IAAI,EAAE,YAAY,CAAC,cAAc,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,EAAE,EAAa;YAClG,EAAC,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,EAAa;SACrD,CAAC,CAAC,CAAC;QAEJ,MAAM,GAAG,GAAG,OAAO,EAAE,CAAC;QACtB,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QACxB,GAAG,CAAC,GAAG,CAAC,sBAAsB,CAAC;YAC7B,cAAc,EAAE,EAAC,UAAU,EAAE,cAAc,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,EAAU;YACpG,cAAc,EAAE,EAAE;YAClB,MAAM,EAAE,EAAC,YAAY,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,EAAE,EAAW,EAAC;YACnE,iBAAiB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAC,UAAU,EAAC,CAAC;SACxC,CAAC,CAAC,CAAC;QAEJ,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,EAAC,OAAO,EAAE,MAAM,EAAC,CAAC,CAAC;QAEhE,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CACrC,MAAM,CAAC,gBAAgB,CAAC;YACtB,KAAK,EAAE,mBAAmB;YAC1B,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;gBAC/B,UAAU,EAAE,CAAC,EAAC,SAAS,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAC,CAAC;aACzE,CAAC;SACH,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,kBAAkB,CAAC,iBAAiB,CAAC,EAAC,OAAO,EAAE,WAAW,EAAE,EAAE,kBAAkB,EAAE,WAAW,EAAE,EAAC,CAAC,CAAC;QAElG,MAAM,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,QAAQ,CAAC;YACzD,MAAM,EAAC,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,EAAC,CAAC;YACrE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,MAAM,GAAG,GAAG,OAAO,EAAE,CAAC;QACtB,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QACxB,GAAG,CAAC,GAAG,CAAC,sBAAsB,CAAC;YAC7B,cAAc,EAAE,EAAC,UAAU,EAAE,cAAc,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,EAAU;YACpG,cAAc,EAAE,EAAE;YAClB,MAAM,EAAE,EAAC,YAAY,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,EAAE,EAAW,EAAC;SACpE,CAAC,CAAC,CAAC;QAEJ,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,EAAC,OAAO,EAAE,OAAO,EAAC,CAAC,CAAC;QAE7E,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;QAC/C,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1,14 +1,24 @@
|
|
|
1
1
|
/**
|
|
2
2
|
* @license
|
|
3
|
-
* Copyright
|
|
3
|
+
* Copyright 2026 Amodal Labs, Inc.
|
|
4
4
|
* SPDX-License-Identifier: MIT
|
|
5
5
|
*/
|
|
6
|
+
/**
|
|
7
|
+
* Streaming chat route.
|
|
8
|
+
*
|
|
9
|
+
* Accepts POST to /chat and /chat/stream, resolves a session via the
|
|
10
|
+
* standalone session manager, runs the message through the agent loop,
|
|
11
|
+
* and streams SSE events to the client.
|
|
12
|
+
*/
|
|
6
13
|
import { Router } from 'express';
|
|
7
14
|
import type { AuthContext } from '../middleware/auth.js';
|
|
8
|
-
import type {
|
|
9
|
-
import {
|
|
15
|
+
import type { StandaloneSessionManager } from '../session/manager.js';
|
|
16
|
+
import type { StreamHooks } from '../session/stream-hooks.js';
|
|
17
|
+
import type { BundleResolver, SharedResources } from './session-resolver.js';
|
|
10
18
|
export interface ChatStreamRouterOptions {
|
|
11
|
-
sessionManager:
|
|
19
|
+
sessionManager: StandaloneSessionManager;
|
|
20
|
+
bundleResolver: BundleResolver;
|
|
21
|
+
shared: SharedResources;
|
|
12
22
|
/** Factory that builds per-request stream hooks from the auth context */
|
|
13
23
|
createStreamHooks?: (auth?: AuthContext) => StreamHooks;
|
|
14
24
|
}
|