@amodalai/runtime 0.1.0

package/dist/src/routes/ask-user-response.js

@@ -0,0 +1,61 @@
+/**
+ * @license
+ * Copyright 2025 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import { Router } from 'express';
+import { z } from 'zod';
+const AskUserResponseSchema = z.object({
+    ask_id: z.string().min(1),
+    answers: z.record(z.string()),
+});
+/**
+ * POST /chat/sessions/:session_id/ask-user-response
+ *
+ * Accepts user answers to an ask_user prompt. Resolves the pending
+ * deferred promise in the session so the stream can continue.
+ */
+export function createAskUserResponseRouter(options) {
+    const router = Router();
+    router.post('/chat/sessions/:session_id/ask-user-response', (req, res, next) => {
+        try {
+            const sessionId = req.params['session_id'];
+            if (!sessionId) {
+                res.status(400).json({
+                    error: { code: 'MISSING_SESSION_ID', message: 'session_id is required' },
+                });
+                return;
+            }
+            const parsed = AskUserResponseSchema.safeParse(req.body);
+            if (!parsed.success) {
+                res.status(400).json({
+                    error: {
+                        code: 'INVALID_BODY',
+                        message: parsed.error.errors.map((e) => e.message).join(', '),
+                    },
+                });
+                return;
+            }
+            const session = options.sessionManager.get(sessionId);
+            if (!session) {
+                res.status(404).json({
+                    error: { code: 'SESSION_NOT_FOUND', message: `Session '${sessionId}' not found` },
+                });
+                return;
+            }
+            const resolved = options.sessionManager.resolveAskUser(session, parsed.data.ask_id, parsed.data.answers);
+            if (!resolved) {
+                res.status(404).json({
+                    error: { code: 'ASK_NOT_FOUND', message: `No pending ask_user with id '${parsed.data.ask_id}'` },
+                });
+                return;
+            }
+            res.json({ ok: true });
+        }
+        catch (err) {
+            next(err);
+        }
+    });
+    return router;
+}
+//# sourceMappingURL=ask-user-response.js.map

package/dist/src/routes/ask-user-response.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ask-user-response.js","sourceRoot":"","sources":["../../../src/routes/ask-user-response.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,MAAM,qBAAqB,GAAG,CAAC,CAAC,MAAM,CAAC;IACrC,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IACzB,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CAC9B,CAAC,CAAC;AAQH;;;;;GAKG;AACH,MAAM,UAAU,2BAA2B,CACzC,OAAqC;IAErC,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC;IAExB,MAAM,CAAC,IAAI,CACT,8CAA8C,EAC9C,CAAC,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QACjB,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;YAC3C,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;oBACnB,KAAK,EAAE,EAAE,IAAI,EAAE,oBAAoB,EAAE,OAAO,EAAE,wBAAwB,EAAE;iBACzE,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAED,MAAM,MAAM,GAAG,qBAAqB,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACzD,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACpB,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;oBACnB,KAAK,EAAE;wBACL,IAAI,EAAE,cAAc;wBACpB,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;qBAC9D;iBACF,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAED,MAAM,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACtD,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;oBACnB,KAAK,EAAE,EAAE,IAAI,EAAE,mBAAmB,EAAE,OAAO,EAAE,YAAY,SAAS,aAAa,EAAE;iBAClF,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAED,MAAM,QAAQ,GAAG,OAAO,CAAC,cAAc,CAAC,cAAc,CACpD,OAAO,EACP,MAAM,CAAC,IAAI,CAAC,MAAM,EAClB,MAAM,CAAC,IAAI,CAAC,OAAO,CACpB,CAAC;YAEF,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;oBACnB,KAAK,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,gCAAgC,MAAM,CAAC,IAAI,CAAC,MAAM,GAAG,EAAE;iBACjG,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAED,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;QACzB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,GAAG,CAAC,CAAC;QACZ,CAAC;IACH,CAAC,CACF,CAAC;IAEF,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/src/routes/ask-user-response.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+export {};

package/dist/src/routes/ask-user-response.test.js

@@ -0,0 +1,88 @@
+/**
+ * @license
+ * Copyright 2025 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import express from 'express';
+import request from 'supertest';
+import { createAskUserResponseRouter } from './ask-user-response.js';
+function createMockSessionManager() {
+    const sessions = new Map();
+    const mockGet = vi.fn((id) => sessions.get(id));
+    const mockResolveAskUser = vi.fn().mockReturnValue(true);
+    return {
+        manager: {
+            get: mockGet,
+            resolveAskUser: mockResolveAskUser,
+        },
+        sessions,
+        mockGet,
+        mockResolveAskUser,
+    };
+}
+function createApp(sessionManager) {
+    const app = express();
+    app.use(express.json());
+    app.use(createAskUserResponseRouter({ sessionManager }));
+    return app;
+}
+describe('ask-user-response route', () => {
+    let mock;
+    let app;
+    beforeEach(() => {
+        mock = createMockSessionManager();
+        app = createApp(mock.manager);
+        // Add a session
+        mock.sessions.set('sess-1', {
+            id: 'sess-1',
+            pendingAskUser: new Map(),
+        });
+    });
+    it('returns 400 for missing session_id', async () => {
+        // This won't match the route pattern, so Express returns 404
+        const res = await request(app)
+            .post('/chat/sessions//ask-user-response')
+            .send({ ask_id: 'a1', answers: {} });
+        // Double-slash won't match Express route pattern
+        expect(res.status).toBe(404);
+    });
+    it('returns 400 for invalid body', async () => {
+        const res = await request(app)
+            .post('/chat/sessions/sess-1/ask-user-response')
+            .send({ bad: 'data' });
+        expect(res.status).toBe(400);
+        expect(res.body.error.code).toBe('INVALID_BODY');
+    });
+    it('returns 400 for empty ask_id', async () => {
+        const res = await request(app)
+            .post('/chat/sessions/sess-1/ask-user-response')
+            .send({ ask_id: '', answers: {} });
+        expect(res.status).toBe(400);
+    });
+    it('returns 404 for non-existent session', async () => {
+        const res = await request(app)
+            .post('/chat/sessions/no-such-session/ask-user-response')
+            .send({ ask_id: 'a1', answers: { '0': 'yes' } });
+        expect(res.status).toBe(404);
+        expect(res.body.error.code).toBe('SESSION_NOT_FOUND');
+    });
+    it('returns 404 when ask_id is not pending', async () => {
+        mock.mockResolveAskUser.mockReturnValue(false);
+        const res = await request(app)
+            .post('/chat/sessions/sess-1/ask-user-response')
+            .send({ ask_id: 'not-pending', answers: { '0': 'yes' } });
+        expect(res.status).toBe(404);
+        expect(res.body.error.code).toBe('ASK_NOT_FOUND');
+    });
+    it('resolves ask_user and returns ok', async () => {
+        mock.mockResolveAskUser.mockReturnValue(true);
+        const res = await request(app)
+            .post('/chat/sessions/sess-1/ask-user-response')
+            .send({ ask_id: 'ask-123', answers: { '0': 'yes', '1': 'Option A' } });
+        expect(res.status).toBe(200);
+        expect(res.body).toEqual({ ok: true });
+        expect(mock.mockResolveAskUser).toHaveBeenCalledWith(expect.objectContaining({ id: 'sess-1' }), 'ask-123', { '0': 'yes', '1': 'Option A' });
+    });
+});
+//# sourceMappingURL=ask-user-response.test.js.map

package/dist/src/routes/ask-user-response.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ask-user-response.test.js","sourceRoot":"","sources":["../../../src/routes/ask-user-response.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,OAAO,MAAM,SAAS,CAAC;AAC9B,OAAO,OAAO,MAAM,WAAW,CAAC;AAChC,OAAO,EAAE,2BAA2B,EAAE,MAAM,wBAAwB,CAAC;AAIrE,SAAS,wBAAwB;IAC/B,MAAM,QAAQ,GAAG,IAAI,GAAG,EAA0B,CAAC;IACnD,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,EAAU,EAAE,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;IACxD,MAAM,kBAAkB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;IAEzD,OAAO;QACL,OAAO,EAAE;YACP,GAAG,EAAE,OAAO;YACZ,cAAc,EAAE,kBAAkB;SACN;QAC9B,QAAQ;QACR,OAAO;QACP,kBAAkB;KACnB,CAAC;AACJ,CAAC;AAED,SAAS,SAAS,CAAC,cAA8B;IAC/C,MAAM,GAAG,GAAG,OAAO,EAAE,CAAC;IACtB,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IACxB,GAAG,CAAC,GAAG,CAAC,2BAA2B,CAAC,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC;IACzD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,IAAI,IAAiD,CAAC;IACtD,IAAI,GAAoB,CAAC;IAEzB,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,GAAG,wBAAwB,EAAE,CAAC;QAClC,GAAG,GAAG,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9B,gBAAgB;QAChB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,EAAE;YAC1B,EAAE,EAAE,QAAQ;YACZ,cAAc,EAAE,IAAI,GAAG,EAAE;SACG,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,6DAA6D;QAC7D,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;aAC3B,IAAI,CAAC,mCAAmC,CAAC;aACzC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;QACvC,iDAAiD;QACjD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;aAC3B,IAAI,CAAC,yCAAyC,CAAC;aAC/C,IAAI,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC;QACzB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;aAC3B,IAAI,CAAC,yCAAyC,CAAC;aAC/C,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;QACrC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;aAC3B,IAAI,CAAC,kDAAkD,CAAC;aACxD,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;QACnD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,IAAI,CAAC,kBAAkB,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAC/C,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;aAC3B,IAAI,CAAC,yCAAyC,CAAC;aAC/C,IAAI,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,OAAO,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;QAC5D,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAChD,IAAI,CAAC,kBAAkB,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;aAC3B,IAAI,CAAC,yCAAyC,CAAC;aAC/C,IAAI,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,UAAU,EAAE,EAAE,CAAC,CAAC;QACzE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;QACvC,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,oBAAoB,CAClD,MAAM,CAAC,gBAAgB,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EACzC,SAAS,EACT,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,UAAU,EAAE,CAChC,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/routes/chat-stream.d.ts

@@ -0,0 +1,14 @@
+/**
+ * @license
+ * Copyright 2025 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import { Router } from 'express';
+import type { SessionManager } from '../session/session-manager.js';
+import type { AuditClient } from '../audit/audit-client.js';
+export interface ChatStreamRouterOptions {
+    sessionManager: SessionManager;
+    auditClient?: AuditClient;
+    platformApiUrl?: string;
+}
+export declare function createChatStreamRouter(options: ChatStreamRouterOptions): Router;

package/dist/src/routes/chat-stream.js

@@ -0,0 +1,84 @@
+/**
+ * @license
+ * Copyright 2025 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import { Router } from 'express';
+import { ChatRequestSchema, SSEEventType } from '../types.js';
+import { validate } from '../middleware/request-validation.js';
+import { getAuthContext } from '../middleware/auth.js';
+import { streamMessage } from '../session/session-runner.js';
+export function createChatStreamRouter(options) {
+    const router = Router();
+    router.post('/chat/stream', validate(ChatRequestSchema), async (req, res, next) => {
+        try {
+            const { message, session_id, role, session_type, deploy_id } = req.body;
+            let session;
+            if (session_id) {
+                session = options.sessionManager.get(session_id);
+                if (!session) {
+                    // Try hydrating from stored conversation history
+                    const auth = getAuthContext(res);
+                    session = await options.sessionManager.hydrate(session_id, role, auth, session_type);
+                }
+                if (!session) {
+                    // History not found — create fresh session
+                    const auth = getAuthContext(res);
+                    session = await options.sessionManager.create(role, auth, session_type, undefined, deploy_id);
+                }
+            }
+            else {
+                const auth = getAuthContext(res);
+                session = await options.sessionManager.create(role, auth, session_type, undefined, deploy_id);
+            }
+            // Set up SSE headers (use setHeader to preserve CORS headers from middleware)
+            res.setHeader('Content-Type', 'text/event-stream');
+            res.setHeader('Cache-Control', 'no-cache');
+            res.setHeader('Connection', 'keep-alive');
+            res.flushHeaders();
+            const controller = new AbortController();
+            // Abort on client disconnect
+            res.on('close', () => controller.abort());
+            // Build audit context if audit client is available
+            let audit;
+            if (options.auditClient) {
+                const auth = getAuthContext(res);
+                if (auth?.token && auth.applicationId) {
+                    audit = {
+                        auditClient: options.auditClient,
+                        appId: auth.applicationId,
+                        token: auth.token,
+                        tenantId: auth.tenantId,
+                        orgId: auth.orgId,
+                        actor: auth.actor,
+                        platformApiUrl: options.platformApiUrl,
+                    };
+                }
+            }
+            const stream = streamMessage(session, message, controller.signal, audit, options.sessionManager);
+            for await (const event of stream) {
+                if (controller.signal.aborted)
+                    break;
+                res.write(`data: ${JSON.stringify(event)}\n\n`);
+            }
+            res.end();
+        }
+        catch (err) {
+            // If headers already sent (SSE started), write error as SSE event
+            if (res.headersSent) {
+                const errorEvent = {
+                    type: SSEEventType.Error,
+                    message: err instanceof Error ? err.message : 'Unknown error',
+                    timestamp: new Date().toISOString(),
+                };
+                res.write(`data: ${JSON.stringify(errorEvent)}\n\n`);
+                res.end();
+            }
+            else {
+                next(err);
+            }
+        }
+    });
+    return router;
+}
+//# sourceMappingURL=chat-stream.js.map

package/dist/src/routes/chat-stream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat-stream.js","sourceRoot":"","sources":["../../../src/routes/chat-stream.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,iBAAiB,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC9D,OAAO,EAAE,QAAQ,EAAE,MAAM,qCAAqC,CAAC;AAC/D,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAEvD,OAAO,EAAE,aAAa,EAA2B,MAAM,8BAA8B,CAAC;AAStF,MAAM,UAAU,sBAAsB,CACpC,OAAgC;IAEhC,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC;IAExB,MAAM,CAAC,IAAI,CACT,cAAc,EACd,QAAQ,CAAC,iBAAiB,CAAC,EAC3B,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QACvB,IAAI,CAAC;YACH,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,YAAY,EAAE,SAAS,EAAE,GAAG,GAAG,CAAC,IAAI,CAAC;YAExE,IAAI,OAAO,CAAC;YACZ,IAAI,UAAU,EAAE,CAAC;gBACf,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;gBACjD,IAAI,CAAC,OAAO,EAAE,CAAC;oBACb,iDAAiD;oBACjD,MAAM,IAAI,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;oBACjC,OAAO,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC,OAAO,CAAC,UAAU,EAAE,IAAI,EAAE,IAAI,EAAE,YAAY,CAAC,CAAC;gBACvF,CAAC;gBACD,IAAI,CAAC,OAAO,EAAE,CAAC;oBACb,2CAA2C;oBAC3C,MAAM,IAAI,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;oBACjC,OAAO,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;gBAChG,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;gBACjC,OAAO,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;YAChG,CAAC;YAED,8EAA8E;YAC9E,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,YAAY,EAAE,CAAC;YAEnB,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;YAEzC,6BAA6B;YAC7B,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC,CAAC;YAE1C,mDAAmD;YACnD,IAAI,KAAqC,CAAC;YAC1C,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;gBACxB,MAAM,IAAI,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;gBACjC,IAAI,IAAI,EAAE,KAAK,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;oBACtC,KAAK,GAAG;wBACN,WAAW,EAAE,OAAO,CAAC,WAAW;wBAChC,KAAK,EAAE,IAAI,CAAC,aAAa;wBACzB,KAAK,EAAE,IAAI,CAAC,KAAK;wBACjB,QAAQ,EAAE,IAAI,CAAC,QAAQ;wBACvB,KAAK,EAAE,IAAI,CAAC,KAAK;wBACjB,KAAK,EAAE,IAAI,CAAC,KAAK;wBACjB,cAAc,EAAE,OAAO,CAAC,cAAc;qBACvC,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,EAAE,OAAO,EAAE,UAAU,CAAC,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,cAAc,CAAC,CAAC;YAEjG,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBACjC,IAAI,UAAU,CAAC,MAAM,CAAC,OAAO;oBAAE,MAAM;gBACrC,GAAG,CAAC,KAAK,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAClD,CAAC;YAED,GAAG,CAAC,GAAG,EAAE,CAAC;QACZ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,kEAAkE;YAClE,IAAI,GAAG,CAAC,WAAW,EAAE,CAAC;gBACpB,MAAM,UAAU,GAAG;oBACjB,IAAI,EAAE,YAAY,CAAC,KAAK;oBACxB,OAAO,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;oBAC7D,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;iBACpC,CAAC;gBACF,GAAG,CAAC,KAAK,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;gBACrD,GAAG,CAAC,GAAG,EAAE,CAAC;YACZ,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,GAAG,CAAC,CAAC;YACZ,CAAC;QACH,CAAC;IACH,CAAC,CACF,CAAC;IAEF,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/src/routes/chat-stream.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+export {};

package/dist/src/routes/chat-stream.test.js

@@ -0,0 +1,155 @@
+/**
+ * @license
+ * Copyright 2025 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import express from 'express';
+import request from 'supertest';
+import { createChatStreamRouter } from './chat-stream.js';
+import { errorHandler } from '../middleware/error-handler.js';
+import { SSEEventType } from '../types.js';
+// Mock session runner
+const mockStreamMessage = vi.fn();
+vi.mock('../session/session-runner.js', () => ({
+    streamMessage: (...args) => mockStreamMessage(...args),
+}));
+function createApp(sessionManager) {
+    const app = express();
+    app.use(express.json());
+    app.use(createChatStreamRouter({ sessionManager: sessionManager }));
+    app.use(errorHandler);
+    return app;
+}
+async function* makeEvents(events) {
+    for (const event of events) {
+        yield event;
+    }
+}
+describe('POST /chat/stream', () => {
+    const mockCreate = vi.fn();
+    const mockGet = vi.fn();
+    const mockHydrate = vi.fn();
+    let sessionManager;
+    beforeEach(() => {
+        vi.clearAllMocks();
+        mockCreate.mockResolvedValue({
+            id: 'sess-1',
+            config: {},
+            geminiClient: {},
+            scheduler: {},
+        });
+        mockHydrate.mockResolvedValue(null);
+        sessionManager = {
+            create: mockCreate,
+            get: mockGet,
+            hydrate: mockHydrate,
+        };
+    });
+    it('returns SSE content type', async () => {
+        mockStreamMessage.mockReturnValue(makeEvents([
+            {
+                type: SSEEventType.Init,
+                session_id: 'sess-1',
+                timestamp: new Date().toISOString(),
+            },
+            { type: SSEEventType.Done, timestamp: new Date().toISOString() },
+        ]));
+        const res = await request(createApp(sessionManager))
+            .post('/chat/stream')
+            .send({ message: 'hello' });
+        expect(res.headers['content-type']).toContain('text/event-stream');
+    });
+    it('sends SSE events in the response body', async () => {
+        mockStreamMessage.mockReturnValue(makeEvents([
+            {
+                type: SSEEventType.Init,
+                session_id: 'sess-1',
+                timestamp: '2024-01-01T00:00:00Z',
+            },
+            {
+                type: SSEEventType.TextDelta,
+                content: 'Hello!',
+                timestamp: '2024-01-01T00:00:01Z',
+            },
+            { type: SSEEventType.Done, timestamp: '2024-01-01T00:00:02Z' },
+        ]));
+        const res = await request(createApp(sessionManager))
+            .post('/chat/stream')
+            .send({ message: 'hello' });
+        // Parse SSE events from response text
+        const lines = res.text.split('\n\n').filter(Boolean);
+        expect(lines.length).toBeGreaterThanOrEqual(3);
+        // Each line should start with "data: "
+        for (const line of lines) {
+            expect(line.startsWith('data: ')).toBe(true);
+            const json = JSON.parse(line.replace('data: ', ''));
+            expect(json.type).toBeDefined();
+        }
+    });
+    it('hydrates expired session when session_id not in memory', async () => {
+        mockGet.mockReturnValue(undefined);
+        const hydratedSession = {
+            id: 'old-conv',
+            config: {},
+            geminiClient: {},
+            scheduler: {},
+        };
+        mockHydrate.mockResolvedValue(hydratedSession);
+        mockStreamMessage.mockReturnValue(makeEvents([
+            { type: SSEEventType.Init, session_id: 'old-conv', timestamp: new Date().toISOString() },
+            { type: SSEEventType.Done, timestamp: new Date().toISOString() },
+        ]));
+        const res = await request(createApp(sessionManager))
+            .post('/chat/stream')
+            .send({ message: 'hi', session_id: 'old-conv' });
+        expect(res.status).toBe(200);
+        expect(mockHydrate).toHaveBeenCalledWith('old-conv', undefined, undefined, undefined);
+        expect(mockCreate).not.toHaveBeenCalled();
+    });
+    it('falls back to fresh session when hydration fails', async () => {
+        mockGet.mockReturnValue(undefined);
+        mockHydrate.mockResolvedValue(null);
+        mockStreamMessage.mockReturnValue(makeEvents([
+            { type: SSEEventType.Init, session_id: 'sess-1', timestamp: new Date().toISOString() },
+            { type: SSEEventType.Done, timestamp: new Date().toISOString() },
+        ]));
+        const res = await request(createApp(sessionManager))
+            .post('/chat/stream')
+            .send({ message: 'hi', session_id: 'nonexistent' });
+        expect(res.status).toBe(200);
+        expect(mockHydrate).toHaveBeenCalledOnce();
+        expect(mockCreate).toHaveBeenCalledOnce();
+    });
+    it('streams normally after hydration', async () => {
+        mockGet.mockReturnValue(undefined);
+        const hydratedSession = {
+            id: 'hydrated-conv',
+            config: {},
+            geminiClient: {},
+            scheduler: {},
+        };
+        mockHydrate.mockResolvedValue(hydratedSession);
+        mockStreamMessage.mockReturnValue(makeEvents([
+            { type: SSEEventType.Init, session_id: 'hydrated-conv', timestamp: '2024-01-01T00:00:00Z' },
+            { type: SSEEventType.TextDelta, content: 'Continued response', timestamp: '2024-01-01T00:00:01Z' },
+            { type: SSEEventType.Done, timestamp: '2024-01-01T00:00:02Z' },
+        ]));
+        const res = await request(createApp(sessionManager))
+            .post('/chat/stream')
+            .send({ message: 'continue', session_id: 'hydrated-conv' });
+        expect(res.status).toBe(200);
+        const lines = res.text.split('\n\n').filter(Boolean);
+        expect(lines.length).toBeGreaterThanOrEqual(3);
+        // Verify the stream contains the hydrated session's events
+        const initEvent = JSON.parse(lines[0].replace('data: ', ''));
+        expect(initEvent.session_id).toBe('hydrated-conv');
+    });
+    it('returns 400 for missing message', async () => {
+        const res = await request(createApp(sessionManager))
+            .post('/chat/stream')
+            .send({});
+        expect(res.status).toBe(400);
+    });
+});
+//# sourceMappingURL=chat-stream.test.js.map

package/dist/src/routes/chat-stream.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat-stream.test.js","sourceRoot":"","sources":["../../../src/routes/chat-stream.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,OAAO,MAAM,SAAS,CAAC;AAC9B,OAAO,OAAO,MAAM,WAAW,CAAC;AAChC,OAAO,EAAE,sBAAsB,EAAE,MAAM,kBAAkB,CAAC;AAC1D,OAAO,EAAE,YAAY,EAAE,MAAM,gCAAgC,CAAC;AAC9D,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAE3C,sBAAsB;AACtB,MAAM,iBAAiB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AAClC,EAAE,CAAC,IAAI,CAAC,8BAA8B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC7C,aAAa,EAAE,CAAC,GAAG,IAAe,EAAE,EAAE,CAAC,iBAAiB,CAAC,GAAG,IAAI,CAAC;CAClE,CAAC,CAAC,CAAC;AAEJ,SAAS,SAAS,CAAC,cAAuC;IACxD,MAAM,GAAG,GAAG,OAAO,EAAE,CAAC;IACtB,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IACxB,GAAG,CAAC,GAAG,CAAC,sBAAsB,CAAC,EAAE,cAAc,EAAE,cAAuB,EAAE,CAAC,CAAC,CAAC;IAC7E,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;IACtB,OAAO,GAAG,CAAC;AACb,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,UAAU,CACxB,MAAsC;IAEtC,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAC3B,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IACxB,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAC5B,IAAI,cAAuC,CAAC;IAE5C,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,UAAU,CAAC,iBAAiB,CAAC;YAC3B,EAAE,EAAE,QAAQ;YACZ,MAAM,EAAE,EAAE;YACV,YAAY,EAAE,EAAE;YAChB,SAAS,EAAE,EAAE;SACd,CAAC,CAAC;QACH,WAAW,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QACpC,cAAc,GAAG;YACf,MAAM,EAAE,UAAU;YAClB,GAAG,EAAE,OAAO;YACZ,OAAO,EAAE,WAAW;SACrB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;QACxC,iBAAiB,CAAC,eAAe,CAC/B,UAAU,CAAC;YACT;gBACE,IAAI,EAAE,YAAY,CAAC,IAAI;gBACvB,UAAU,EAAE,QAAQ;gBACpB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACpC;YACD,EAAE,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE;SACjE,CAAC,CACH,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;aACjD,IAAI,CAAC,cAAc,CAAC;aACpB,IAAI,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QAE9B,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,iBAAiB,CAAC,eAAe,CAC/B,UAAU,CAAC;YACT;gBACE,IAAI,EAAE,YAAY,CAAC,IAAI;gBACvB,UAAU,EAAE,QAAQ;gBACpB,SAAS,EAAE,sBAAsB;aAClC;YACD;gBACE,IAAI,EAAE,YAAY,CAAC,SAAS;gBAC5B,OAAO,EAAE,QAAQ;gBACjB,SAAS,EAAE,sBAAsB;aAClC;YACD,EAAE,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,SAAS,EAAE,sBAAsB,EAAE;SAC/D,CAAC,CACH,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;aACjD,IAAI,CAAC,cAAc,CAAC;aACpB,IAAI,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QAE9B,sCAAsC;QACtC,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;QAE/C,uCAAuC;QACvC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC;YACpD,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;QAClC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,OAAO,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;QACnC,MAAM,eAAe,GAAG;YACtB,EAAE,EAAE,UAAU;YACd,MAAM,EAAE,EAAE;YACV,YAAY,EAAE,EAAE;YAChB,SAAS,EAAE,EAAE;SACd,CAAC;QACF,WAAW,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;QAC/C,iBAAiB,CAAC,eAAe,CAC/B,UAAU,CAAC;YACT,EAAE,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE;YACxF,EAAE,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE;SACjE,CAAC,CACH,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;aACjD,IAAI,CAAC,cAAc,CAAC;aACpB,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,CAAC;QAEnD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,UAAU,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;QACtF,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,OAAO,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;QACnC,WAAW,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QACpC,iBAAiB,CAAC,eAAe,CAC/B,UAAU,CAAC;YACT,EAAE,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE;YACtF,EAAE,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE;SACjE,CAAC,CACH,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;aACjD,IAAI,CAAC,cAAc,CAAC;aACpB,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,aAAa,EAAE,CAAC,CAAC;QAEtD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,EAAE,CAAC;QAC3C,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,EAAE,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAChD,OAAO,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;QACnC,MAAM,eAAe,GAAG;YACtB,EAAE,EAAE,eAAe;YACnB,MAAM,EAAE,EAAE;YACV,YAAY,EAAE,EAAE;YAChB,SAAS,EAAE,EAAE;SACd,CAAC;QACF,WAAW,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;QAC/C,iBAAiB,CAAC,eAAe,CAC/B,UAAU,CAAC;YACT,EAAE,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,UAAU,EAAE,eAAe,EAAE,SAAS,EAAE,sBAAsB,EAAE;YAC3F,EAAE,IAAI,EAAE,YAAY,CAAC,SAAS,EAAE,OAAO,EAAE,oBAAoB,EAAE,SAAS,EAAE,sBAAsB,EAAE;YAClG,EAAE,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,SAAS,EAAE,sBAAsB,EAAE;SAC/D,CAAC,CACH,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;aACjD,IAAI,CAAC,cAAc,CAAC;aACpB,IAAI,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC,CAAC;QAE9D,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,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;QAE/C,2DAA2D;QAC3D,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC;QAC7D,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;aACjD,IAAI,CAAC,cAAc,CAAC;aACpB,IAAI,CAAC,EAAE,CAAC,CAAC;QAEZ,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/routes/chat.d.ts

@@ -0,0 +1,13 @@
+/**
+ * @license
+ * Copyright 2025 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import { Router } from 'express';
+import type { SessionManager } from '../session/session-manager.js';
+import type { AuditClient } from '../audit/audit-client.js';
+export interface ChatRouterOptions {
+    sessionManager: SessionManager;
+    auditClient?: AuditClient;
+}
+export declare function createChatRouter(options: ChatRouterOptions): Router;

package/dist/src/routes/chat.js

@@ -0,0 +1,55 @@
+/**
+ * @license
+ * Copyright 2025 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import { Router } from 'express';
+import { ChatRequestSchema } from '../types.js';
+import { validate } from '../middleware/request-validation.js';
+import { AppError } from '../middleware/error-handler.js';
+import { getAuthContext } from '../middleware/auth.js';
+import { runMessage } from '../session/session-runner.js';
+export function createChatRouter(options) {
+    const router = Router();
+    router.post('/chat', validate(ChatRequestSchema), async (req, res, next) => {
+        try {
+            const { message, session_id, role, deploy_id } = req.body;
+            // Get or create session
+            let session;
+            if (session_id) {
+                session = options.sessionManager.get(session_id);
+                if (!session) {
+                    throw new AppError(404, 'SESSION_NOT_FOUND', `Session ${session_id} not found`);
+                }
+            }
+            else {
+                const auth = getAuthContext(res);
+                session = await options.sessionManager.create(role, auth, undefined, undefined, deploy_id);
+            }
+            const controller = new AbortController();
+            // Abort on client disconnect
+            res.on('close', () => controller.abort());
+            // Build audit context if audit client is available
+            let audit;
+            if (options.auditClient) {
+                const auth = getAuthContext(res);
+                if (auth?.token && auth.applicationId) {
+                    audit = {
+                        auditClient: options.auditClient,
+                        appId: auth.applicationId,
+                        token: auth.token,
+                        tenantId: auth.tenantId,
+                        orgId: auth.orgId,
+                    };
+                }
+            }
+            const result = await runMessage(session, message, controller.signal, audit);
+            res.json(result);
+        }
+        catch (err) {
+            next(err);
+        }
+    });
+    return router;
+}
+//# sourceMappingURL=chat.js.map

package/dist/src/routes/chat.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat.js","sourceRoot":"","sources":["../../../src/routes/chat.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAChD,OAAO,EAAE,QAAQ,EAAE,MAAM,qCAAqC,CAAC;AAC/D,OAAO,EAAE,QAAQ,EAAE,MAAM,gCAAgC,CAAC;AAC1D,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAEvD,OAAO,EAAE,UAAU,EAA2B,MAAM,8BAA8B,CAAC;AAQnF,MAAM,UAAU,gBAAgB,CAAC,OAA0B;IACzD,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC;IAExB,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QACzE,IAAI,CAAC;YACH,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE,GAAG,GAAG,CAAC,IAAI,CAAC;YAE1D,wBAAwB;YACxB,IAAI,OAAO,CAAC;YACZ,IAAI,UAAU,EAAE,CAAC;gBACf,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;gBACjD,IAAI,CAAC,OAAO,EAAE,CAAC;oBACb,MAAM,IAAI,QAAQ,CAAC,GAAG,EAAE,mBAAmB,EAAE,WAAW,UAAU,YAAY,CAAC,CAAC;gBAClF,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;gBACjC,OAAO,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;YAC7F,CAAC;YAED,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;YAEzC,6BAA6B;YAC7B,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC,CAAC;YAE1C,mDAAmD;YACnD,IAAI,KAAqC,CAAC;YAC1C,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;gBACxB,MAAM,IAAI,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;gBACjC,IAAI,IAAI,EAAE,KAAK,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;oBACtC,KAAK,GAAG;wBACN,WAAW,EAAE,OAAO,CAAC,WAAW;wBAChC,KAAK,EAAE,IAAI,CAAC,aAAa;wBACzB,KAAK,EAAE,IAAI,CAAC,KAAK;wBACjB,QAAQ,EAAE,IAAI,CAAC,QAAQ;wBACvB,KAAK,EAAE,IAAI,CAAC,KAAK;qBAClB,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,EAAE,OAAO,EAAE,UAAU,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YAC5E,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,GAAG,CAAC,CAAC;QACZ,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/src/routes/chat.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+export {};

package/dist/src/routes/chat.test.js

@@ -0,0 +1,99 @@
+/**
+ * @license
+ * Copyright 2025 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import express from 'express';
+import request from 'supertest';
+import { createChatRouter } from './chat.js';
+import { errorHandler } from '../middleware/error-handler.js';
+// Mock session runner
+const mockRunMessage = vi.fn();
+vi.mock('../session/session-runner.js', () => ({
+    runMessage: (...args) => mockRunMessage(...args),
+}));
+function createApp(sessionManager, auditClient) {
+    const app = express();
+    app.use(express.json());
+    app.use(createChatRouter({ sessionManager: sessionManager, auditClient: auditClient }));
+    app.use(errorHandler);
+    return app;
+}
+describe('POST /chat', () => {
+    const mockCreate = vi.fn();
+    const mockGet = vi.fn();
+    let sessionManager;
+    beforeEach(() => {
+        vi.clearAllMocks();
+        mockRunMessage.mockResolvedValue({
+            session_id: 'sess-1',
+            response: 'Hello!',
+            tool_calls: [],
+        });
+        mockCreate.mockResolvedValue({
+            id: 'sess-1',
+            config: {},
+            geminiClient: {},
+            scheduler: {},
+        });
+        sessionManager = {
+            create: mockCreate,
+            get: mockGet,
+        };
+    });
+    it('creates a new session and returns response', async () => {
+        const app = createApp(sessionManager);
+        const res = await request(app)
+            .post('/chat')
+            .send({ message: 'hello' });
+        expect(res.status).toBe(200);
+        expect(res.body.session_id).toBe('sess-1');
+        expect(res.body.response).toBe('Hello!');
+        expect(mockCreate).toHaveBeenCalledOnce();
+        expect(mockRunMessage).toHaveBeenCalledOnce();
+    });
+    it('reuses existing session when session_id provided', async () => {
+        const existingSession = { id: 'sess-existing' };
+        mockGet.mockReturnValue(existingSession);
+        const app = createApp(sessionManager);
+        const res = await request(app)
+            .post('/chat')
+            .send({ message: 'hi', session_id: 'sess-existing' });
+        expect(res.status).toBe(200);
+        expect(mockGet).toHaveBeenCalledWith('sess-existing');
+        expect(mockCreate).not.toHaveBeenCalled();
+    });
+    it('returns 404 for unknown session_id', async () => {
+        mockGet.mockReturnValue(undefined);
+        const app = createApp(sessionManager);
+        const res = await request(app)
+            .post('/chat')
+            .send({ message: 'hi', session_id: 'nonexistent' });
+        expect(res.status).toBe(404);
+        expect(res.body.error.code).toBe('SESSION_NOT_FOUND');
+    });
+    it('returns 400 for missing message', async () => {
+        const app = createApp(sessionManager);
+        const res = await request(app).post('/chat').send({});
+        expect(res.status).toBe(400);
+        expect(res.body.error.code).toBe('VALIDATION_ERROR');
+    });
+    it('passes role to session create', async () => {
+        const app = createApp(sessionManager);
+        await request(app)
+            .post('/chat')
+            .send({ message: 'hello', role: 'analyst' });
+        expect(mockCreate).toHaveBeenCalledWith('analyst', undefined, undefined, undefined, undefined);
+    });
+    it('returns 500 on internal error', async () => {
+        mockRunMessage.mockRejectedValue(new Error('LLM broke'));
+        const app = createApp(sessionManager);
+        const res = await request(app)
+            .post('/chat')
+            .send({ message: 'hello' });
+        expect(res.status).toBe(500);
+        expect(res.body.error.message).toBe('LLM broke');
+    });
+});
+//# sourceMappingURL=chat.test.js.map

package/dist/src/routes/chat.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat.test.js","sourceRoot":"","sources":["../../../src/routes/chat.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,OAAO,MAAM,SAAS,CAAC;AAC9B,OAAO,OAAO,MAAM,WAAW,CAAC;AAChC,OAAO,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAC7C,OAAO,EAAE,YAAY,EAAE,MAAM,gCAAgC,CAAC;AAE9D,sBAAsB;AACtB,MAAM,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AAC/B,EAAE,CAAC,IAAI,CAAC,8BAA8B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC7C,UAAU,EAAE,CAAC,GAAG,IAAe,EAAE,EAAE,CAAC,cAAc,CAAC,GAAG,IAAI,CAAC;CAC5D,CAAC,CAAC,CAAC;AAEJ,SAAS,SAAS,CAAC,cAAuC,EAAE,WAAqC;IAC/F,MAAM,GAAG,GAAG,OAAO,EAAE,CAAC;IACtB,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IACxB,GAAG,CAAC,GAAG,CAAC,gBAAgB,CAAC,EAAE,cAAc,EAAE,cAAuB,EAAE,WAAW,EAAE,WAAoB,EAAE,CAAC,CAAC,CAAC;IAC1G,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;IACtB,OAAO,GAAG,CAAC;AACb,CAAC;AAED,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAC3B,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IACxB,IAAI,cAAuC,CAAC;IAE5C,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,cAAc,CAAC,iBAAiB,CAAC;YAC/B,UAAU,EAAE,QAAQ;YACpB,QAAQ,EAAE,QAAQ;YAClB,UAAU,EAAE,EAAE;SACf,CAAC,CAAC;QACH,UAAU,CAAC,iBAAiB,CAAC;YAC3B,EAAE,EAAE,QAAQ;YACZ,MAAM,EAAE,EAAE;YACV,YAAY,EAAE,EAAE;YAChB,SAAS,EAAE,EAAE;SACd,CAAC,CAAC;QACH,cAAc,GAAG;YACf,MAAM,EAAE,UAAU;YAClB,GAAG,EAAE,OAAO;SACb,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,GAAG,GAAG,SAAS,CAAC,cAAc,CAAC,CAAC;QACtC,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;aAC3B,IAAI,CAAC,OAAO,CAAC;aACb,IAAI,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QAE9B,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC3C,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,EAAE,CAAC;QAC1C,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,EAAE,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,eAAe,GAAG,EAAE,EAAE,EAAE,eAAe,EAAE,CAAC;QAChD,OAAO,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;QAEzC,MAAM,GAAG,GAAG,SAAS,CAAC,cAAc,CAAC,CAAC;QACtC,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;aAC3B,IAAI,CAAC,OAAO,CAAC;aACb,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC,CAAC;QAExD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,eAAe,CAAC,CAAC;QACtD,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,OAAO,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;QAEnC,MAAM,GAAG,GAAG,SAAS,CAAC,cAAc,CAAC,CAAC;QACtC,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;aAC3B,IAAI,CAAC,OAAO,CAAC;aACb,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,aAAa,EAAE,CAAC,CAAC;QAEtD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,GAAG,GAAG,SAAS,CAAC,cAAc,CAAC,CAAC;QACtC,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEtD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,GAAG,GAAG,SAAS,CAAC,cAAc,CAAC,CAAC;QACtC,MAAM,OAAO,CAAC,GAAG,CAAC;aACf,IAAI,CAAC,OAAO,CAAC;aACb,IAAI,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;QAE/C,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;IACjG,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC7C,cAAc,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;QAEzD,MAAM,GAAG,GAAG,SAAS,CAAC,cAAc,CAAC,CAAC;QACtC,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;aAC3B,IAAI,CAAC,OAAO,CAAC;aACb,IAAI,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QAE9B,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}