@corgelabs/corge 0.0.1 → 1.0.0

package/dist/chat/stream.js

@@ -0,0 +1,339 @@
+import { CORE_CONFIG } from '../config';
+import { entryId } from '../db';
+import { Core } from '../index';
+import { streamChatBody } from './schema';
+import { buildCompletedTurnWriteBatch, buildUIMessageResponse, deriveTitle, extractMessageText, extractToolInvocations, extractTurnAttachmentParts, formatAttachmentLine, mapFirestoreMessageDoc, PUBLIC_MESSAGE_FIELDS } from './utils';
+const activeGenerations = new Map();
+const genKey = (isDev, chatId) => `${isDev ? 'dev' : 'prod'}:${chatId}`;
+export class ChatStreamEngine {
+    #core;
+    #agentRegistry;
+    constructor(coreClient, agentRegistry) {
+        this.#core = coreClient;
+        this.#agentRegistry = agentRegistry;
+    }
+    #assertChatAccess = async (chatId) => {
+        const caller = this.#core.caller;
+        if (caller.role !== 'admin' && caller.userId !== CORE_CONFIG.auth.adminUserId) {
+            const chatDoc = await this.#core.db.read('chat', chatId, false, {
+                userId: caller.userId,
+                userRole: caller.role
+            });
+            if (!chatDoc)
+                this.#core.errors.httpError(404, `Chat not found: ${chatId}`, 'NOT_FOUND');
+        }
+    };
+    enforceAgentAccess = async (c, agent, scopeKey) => {
+        if (agent.auth) {
+            await this.#core.internals.context.authenticateRequest(c, agent.auth);
+        }
+        if (agent.rateLimit !== undefined) {
+            this.#core.internals.context.applyRateLimit(c, agent.rateLimit, `${scopeKey}:${agent.name ?? 'agent'}`);
+        }
+    };
+    stageAndScanAttachments = async (userId, parts, types) => {
+        if (parts.length === 0)
+            return { staged: [], attachments: [] };
+        if (parts.length > CORE_CONFIG.limits.maxChatAttachments) {
+            this.#core.errors.httpError(400, `Too many attachments (max ${CORE_CONFIG.limits.maxChatAttachments})`, 'TOO_MANY_ATTACHMENTS');
+        }
+        const staged = await Promise.all(parts.map(async (part) => {
+            if (part.ref) {
+                if (!this.#core.files.canAccess(part.ref, userId)) {
+                    this.#core.errors.httpError(403, 'Forbidden', 'FORBIDDEN');
+                }
+                return { ref: part.ref, filename: part.filename };
+            }
+            if ((part.dataUrl?.length ?? 0) > 28 * 1024 * 1024) {
+                this.#core.errors.httpError(413, 'Attachment exceeds 20MiB limit', 'PAYLOAD_TOO_LARGE');
+            }
+            const match = /^data:([^;,]+);base64,(.+)$/.exec(part.dataUrl ?? '');
+            if (!match?.[1] || !match[2]) {
+                return this.#core.errors.httpError(400, 'Invalid base64 data URL for attachment', 'INVALID_FILE');
+            }
+            const contentType = match[1];
+            const bytes = new Uint8Array(Buffer.from(match[2], 'base64'));
+            const { ref } = await this.#core.files.upload({ type: contentType, name: part.filename, bytes }, userId);
+            return { ref, filename: part.filename };
+        }));
+        const { attachments } = await this.#core.attachments.process({ refs: staged.map((s) => s.ref) }, { types });
+        return { staged, attachments };
+    };
+    saveChat = async ({ userId = this.#core.userId, input }) => {
+        const targetId = input.id ?? input.chatId;
+        const existing = targetId ? await this.#core.db.read('chat', targetId, true, { bypassRbac: true }) : undefined;
+        if (existing && userId !== CORE_CONFIG.auth.adminUserId && existing.userId !== userId) {
+            return this.#core.errors.httpError(403, 'Forbidden', 'FORBIDDEN');
+        }
+        const defaultAgentId = this.#agentRegistry.keys().next().value ?? 'chat';
+        if (existing && input.agentId && input.agentId !== existing.agentId) {
+            return this.#core.errors.httpError(400, 'Cannot switch agentId on an existing chat', 'AGENT_IMMUTABLE');
+        }
+        const title = input.title ?? existing?.title ?? 'New Chat';
+        const agentId = existing ? existing.agentId : (input.agentId ?? defaultAgentId);
+        if (!this.#agentRegistry.has(agentId)) {
+            return this.#core.errors.httpError(404, `Unknown agent: ${agentId}`, 'NOT_FOUND');
+        }
+        const model = input.model ?? existing?.model ?? CORE_CONFIG.models.vertex;
+        if (existing && targetId) {
+            const patch = { title, model, ...(input.summary !== undefined ? { summary: input.summary } : {}), deleted: false };
+            await this.#core.db.update('chat', targetId, patch, { bypassRbac: true });
+            return { ...existing, ...patch, updatedAt: new Date() };
+        }
+        return this.#core.db.create('chat', {
+            ...(targetId ? { id: targetId } : {}),
+            userId,
+            agentId,
+            title,
+            model,
+            messageCount: 0,
+            summary: input.summary ?? null,
+            activeGenerationId: null
+        }, { bypassRbac: true });
+    };
+    messages = async (chatId, opts) => {
+        await this.#assertChatAccess(chatId);
+        const limit = Math.min(Math.max(opts?.limit ?? CORE_CONFIG.limits.chatMessagesLimit, 1), CORE_CONFIG.limits.defaultListLimit);
+        const db = this.#core.db.firestore();
+        const messagesCol = db.collection('chat').doc(chatId).collection('messages');
+        await this.#core.internals.resilience.acquireLimit('firestore');
+        let query = messagesCol.select(...PUBLIC_MESSAGE_FIELDS).orderBy('createdAt', 'desc').limit(limit);
+        if (opts?.before) {
+            const beforeSnap = await messagesCol.doc(opts.before).get();
+            if (beforeSnap.exists)
+                query = query.startAfter(beforeSnap);
+        }
+        const snap = await query.get();
+        return snap.docs.map((d) => mapFirestoreMessageDoc(d.id, d.data() ?? {})).reverse();
+    };
+    cancel = async (chatId) => {
+        await this.#assertChatAccess(chatId);
+        const key = genKey(Boolean(this.#core.env.isDev), chatId);
+        const localGen = activeGenerations.get(key);
+        await this.#core.db.update('chat', chatId, { activeGenerationId: null }, { bypassRbac: true }).catch(() => { });
+        if (localGen) {
+            localGen.controller.abort(new Error('Cancelled by user'));
+            if (localGen.completionPromise) {
+                await Promise.race([localGen.completionPromise, new Promise((r) => setTimeout(r, 2_500))]);
+            }
+        }
+        return { chatId, cancelled: true };
+    };
+    #resolveStreamTurnTarget = async (c, userId, body) => {
+        const paramValue = c.req.param('chatId');
+        const paramIsAgent = Boolean(paramValue && this.#agentRegistry.has(paramValue) && !body.agentId && !body.chatId && !body.id);
+        const defaultAgentId = this.#agentRegistry.keys().next().value ?? 'chat';
+        const db = this.#core.db.firestore();
+        const candidateChatId = !paramIsAgent ? (paramValue ?? body.chatId ?? body.id) : (body.chatId ?? body.id);
+        const chatId = candidateChatId && entryId.safeParse(candidateChatId).success
+            ? candidateChatId
+            : db.collection('chat').doc().id;
+        const existing = await this.#core.db.read('chat', chatId, true, { bypassRbac: true });
+        if (existing && userId !== CORE_CONFIG.auth.adminUserId && existing.userId !== userId) {
+            this.#core.errors.httpError(403, 'Forbidden', 'FORBIDDEN');
+        }
+        const requestedAgentId = body.agentId ?? (paramIsAgent ? paramValue : undefined);
+        if (existing && requestedAgentId && requestedAgentId !== existing.agentId) {
+            this.#core.errors.httpError(400, 'Cannot switch agentId on an existing chat', 'AGENT_IMMUTABLE');
+        }
+        const resolvedAgentId = existing ? existing.agentId : (requestedAgentId ?? defaultAgentId);
+        const agent = this.#agentRegistry.get(resolvedAgentId);
+        if (!agent)
+            return this.#core.errors.httpError(404, `Unknown agent: ${resolvedAgentId}`, 'NOT_FOUND');
+        await this.enforceAgentAccess(c, agent, 'chat.stream');
+        const attachmentParts = extractTurnAttachmentParts(body.message);
+        const rawUserText = extractMessageText(body.message);
+        if (!rawUserText && attachmentParts.length === 0) {
+            this.#core.errors.httpError(400, 'message must contain non-empty text or attachments', 'VALIDATION_ERROR');
+        }
+        const key = genKey(Boolean(this.#core.env.isDev), chatId);
+        const activeTimeoutMs = CORE_CONFIG.cloudRun.http.timeoutSeconds * 1000;
+        const hasFreshRemoteLock = Boolean(existing?.activeGenerationId) &&
+            Date.now() - (existing?.updatedAt instanceof Date ? existing.updatedAt.getTime() : 0) < activeTimeoutMs;
+        if (activeGenerations.has(key) || hasFreshRemoteLock) {
+            this.#core.errors.httpError(409, 'A generation is already in progress for this chat', 'GENERATION_IN_PROGRESS');
+        }
+        return { chatId, existing, resolvedAgentId, agent, attachmentParts, rawUserText, key };
+    };
+    #persistTurnCompletion = async (opts) => {
+        let assistantText = '';
+        let totalTokens;
+        const toolInvocations = [];
+        try {
+            try {
+                const [text, response, usage] = await Promise.all([opts.turn.text, opts.turn.response, opts.turn.usage]);
+                assistantText = text || opts.getAccumulatedText();
+                totalTokens = usage?.totalTokens ?? ((usage?.inputTokens ?? 0) + (usage?.outputTokens ?? 0) || undefined);
+                toolInvocations.push(...extractToolInvocations(response.messages));
+            }
+            catch {
+                assistantText = opts.getAccumulatedText();
+            }
+            const finalAssistantText = assistantText.trim();
+            const [userVector, assistantVector] = await Promise.all([
+                opts.userVecPromise,
+                this.#core.ai.embed(finalAssistantText).catch(() => null)
+            ]);
+            const batch = buildCompletedTurnWriteBatch(this.#core.db.firestore(), {
+                ...opts,
+                finalAssistantText,
+                toolInvocations,
+                ...(totalTokens !== undefined ? { totalTokens } : {}),
+                cancelled: opts.signal.aborted,
+                userVector,
+                assistantVector
+            });
+            await this.#core.internals.resilience.acquireLimit('firestore');
+            await batch.commit().catch((err) => console.error('[chat] failed to commit turn batch:', err));
+        }
+        finally {
+            try {
+                opts.unsubCancelWatch?.();
+            }
+            catch { }
+            activeGenerations.delete(opts.key);
+        }
+    };
+    #upsertChatTurnLock = async (existing, chatDocRef, chatId, userId, resolvedAgentId, title, selectedModel, generationId, generationController) => {
+        if (existing) {
+            await this.#core.db.update('chat', chatId, { title, agentId: resolvedAgentId, model: selectedModel, activeGenerationId: generationId, deleted: false }, { bypassRbac: true });
+        }
+        else {
+            await this.#core.db.create('chat', {
+                id: chatId,
+                userId,
+                agentId: resolvedAgentId,
+                title,
+                model: selectedModel,
+                messageCount: 0,
+                summary: null,
+                activeGenerationId: generationId
+            }, { bypassRbac: true });
+        }
+        return chatDocRef.onSnapshot((snap) => {
+            if (snap.data()?.activeGenerationId !== generationId && !generationController.signal.aborted) {
+                generationController.abort(new Error('Generation cancelled via Firestore state'));
+            }
+        }, () => { });
+    };
+    startChatStreamTurn = async (opts) => {
+        const { userId, role, agent, attachmentParts, rawUserText, signal, chatId, messageCount, selectedModel, priorMessages } = opts;
+        const [{ staged, attachments }, hybrid] = await Promise.all([
+            this.stageAndScanAttachments(userId, attachmentParts, agent.attachmentTypes ?? {}),
+            this.#core.chat.retrieveContext({
+                userId,
+                ...(chatId ? { chatId } : {}),
+                ...(messageCount !== undefined ? { messageCount } : {}),
+                prompt: rawUserText,
+                useContext: agent.useContext !== false,
+                agent,
+                role,
+                abortSignal: signal
+            })
+        ]);
+        const stagedAttachmentSummaries = attachments.map((item, idx) => formatAttachmentLine(staged[idx]?.filename ?? 'file', item).trim());
+        const userCreatedAtMs = Date.now();
+        const messagesForTurn = priorMessages
+            ? [
+                ...priorMessages,
+                ...(stagedAttachmentSummaries.length > 0
+                    ? [{ role: 'user', content: [{ type: 'text', text: stagedAttachmentSummaries.join('\n\n') }] }]
+                    : [])
+            ].slice(-CORE_CONFIG.limits.chatMessagesLimit)
+            : [
+                ...hybrid.slidingWindowMessages,
+                {
+                    role: 'user',
+                    content: [{ type: 'text', text: [rawUserText, ...stagedAttachmentSummaries].filter(Boolean).join('\n\n') || 'Attached files.' }]
+                }
+            ];
+        const turn = await this.#core.chat.stream(agent, messagesForTurn, userId, {
+            ...(selectedModel ? { modelOverride: selectedModel } : {}),
+            ...(chatId ? { chatId } : {}),
+            abortSignal: signal,
+            core: new Core({ env: this.#core.env, userId, role }),
+            dynamicContext: {
+                currentTimeIso: new Date(userCreatedAtMs).toISOString(),
+                recalledMemories: hybrid.recalledMemories,
+                infobaseContextEntries: hybrid.infobaseContextEntries,
+                stagedAttachmentSummaries,
+                userContext: hybrid.userContext
+            }
+        });
+        return { turn, stagedRefs: staged.map((s) => s.ref), stagedAttachmentSummaries, userCreatedAtMs };
+    };
+    streamChatSse = async ({ c, userId = this.#core.userId, input }) => {
+        const parsedBody = streamChatBody.safeParse(input ?? (await c.req.json().catch(() => ({}))));
+        if (!parsedBody.success) {
+            return this.#core.errors.httpError(400, parsedBody.error.issues.map((i) => `${i.path.join('.') || 'input'}: ${i.message}`).join('; '), 'VALIDATION_ERROR');
+        }
+        const body = parsedBody.data;
+        const { chatId, existing, resolvedAgentId, agent, attachmentParts, rawUserText, key } = await this.#resolveStreamTurnTarget(c, userId, body);
+        const generationId = `gen_${crypto.randomUUID().replace(/-/g, '').slice(0, 20)}`;
+        const generationController = new AbortController();
+        let accumulatedText = '';
+        const activeEntry = {
+            generationId,
+            controller: generationController,
+            partialText: () => accumulatedText
+        };
+        activeGenerations.set(key, activeEntry);
+        const title = body.title ?? existing?.title ?? deriveTitle(body.message);
+        const selectedModel = body.model ?? existing?.model ?? agent.model ?? CORE_CONFIG.models.vertex;
+        const chatDocRef = this.#core.db.firestore().collection('chat').doc(chatId);
+        let unsubCancelWatch;
+        try {
+            unsubCancelWatch = await this.#upsertChatTurnLock(existing, chatDocRef, chatId, userId, resolvedAgentId, title, selectedModel, generationId, generationController);
+            const role = this.#core.internals.auth.resolveUserRole(userId, c.get('userRole'));
+            const { turn, stagedRefs, stagedAttachmentSummaries, userCreatedAtMs } = await this.startChatStreamTurn({
+                userId,
+                role,
+                chatId,
+                messageCount: existing?.messageCount ?? 0,
+                agent,
+                selectedModel,
+                attachmentParts,
+                rawUserText,
+                signal: generationController.signal
+            });
+            const finalUserText = rawUserText || stagedAttachmentSummaries.join('\n');
+            const msgCol = chatDocRef.collection('messages');
+            const userMsgId = body.message.id && entryId.safeParse(body.message.id).success ? body.message.id : msgCol.doc().id;
+            const userPartsForStorage = body.message.parts?.length ? body.message.parts : [{ type: 'text', text: rawUserText }];
+            const completionPromise = this.#persistTurnCompletion({
+                turn,
+                getAccumulatedText: () => accumulatedText,
+                signal: generationController.signal,
+                userVecPromise: this.#core.ai.embed(finalUserText).catch(() => null),
+                chatDocRef,
+                chatId,
+                userId,
+                resolvedAgentId,
+                title,
+                selectedModel,
+                userMsgId,
+                assistantMsgId: msgCol.doc().id,
+                userCreatedAtMs,
+                finalUserText,
+                userPartsForStorage,
+                stagedRefs,
+                unsubCancelWatch,
+                key
+            });
+            activeEntry.completionPromise = completionPromise;
+            return buildUIMessageResponse(turn, chatId, generationId, completionPromise, (d) => {
+                accumulatedText += d;
+            });
+        }
+        catch (err) {
+            try {
+                unsubCancelWatch?.();
+            }
+            catch { }
+            activeGenerations.delete(key);
+            await this.#core.db.update('chat', chatId, { activeGenerationId: null }, { bypassRbac: true }).catch(() => { });
+            throw err;
+        }
+    };
+}
+//# sourceMappingURL=stream.js.map

package/dist/chat/stream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream.js","sourceRoot":"","sources":["../../src/chat/stream.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,WAAW,EAA4B,MAAM,WAAW,CAAA;AACjE,OAAO,EAAE,OAAO,EAAkC,MAAM,OAAO,CAAA;AAE/D,OAAO,EAAE,IAAI,EAAE,MAAM,UAAU,CAAA;AAC/B,OAAO,EAEL,cAAc,EAMf,MAAM,UAAU,CAAA;AACjB,OAAO,EACL,4BAA4B,EAC5B,sBAAsB,EACtB,WAAW,EACX,kBAAkB,EAClB,sBAAsB,EACtB,0BAA0B,EAC1B,oBAAoB,EACpB,sBAAsB,EACtB,qBAAqB,EACtB,MAAM,SAAS,CAAA;AAShB,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAiC,CAAA;AAClE,MAAM,MAAM,GAAG,CAAC,KAAc,EAAE,MAAc,EAAU,EAAE,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,IAAI,MAAM,EAAE,CAAA;AAEhG,MAAM,OAAO,gBAAgB;IAClB,KAAK,CAAM;IACX,cAAc,CAA0B;IAEjD,YAAY,UAAgB,EAAE,aAAuC;QACnE,IAAI,CAAC,KAAK,GAAG,UAAU,CAAA;QACvB,IAAI,CAAC,cAAc,GAAG,aAAa,CAAA;IACrC,CAAC;IAED,iBAAiB,GAAG,KAAK,EAAE,MAAe,EAAiB,EAAE;QAC3D,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAA;QAChC,IAAI,MAAM,CAAC,IAAI,KAAK,OAAO,IAAI,MAAM,CAAC,MAAM,KAAK,WAAW,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YAC9E,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE;gBAC9D,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,QAAQ,EAAE,MAAM,CAAC,IAAI;aACtB,CAAC,CAAA;YACF,IAAI,CAAC,OAAO;gBAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,mBAAmB,MAAM,EAAE,EAAE,WAAW,CAAC,CAAA;QAC1F,CAAC;IACH,CAAC,CAAA;IAED,kBAAkB,GAAG,KAAK,EAAE,CAAkB,EAAE,KAAkB,EAAE,QAAgB,EAAiB,EAAE;QACrG,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;YACf,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC,EAAE,KAAK,CAAC,IAAI,CAAC,CAAA;QACvE,CAAC;QACD,IAAI,KAAK,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YAClC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,EAAE,KAAK,CAAC,SAAS,EAAE,GAAG,QAAQ,IAAI,KAAK,CAAC,IAAI,IAAI,OAAO,EAAE,CAAC,CAAA;QACzG,CAAC;IACH,CAAC,CAAA;IAED,uBAAuB,GAAG,KAAK,EAC7B,MAAe,EACf,KAA0B,EAC1B,KAA0B,EACqE,EAAE;QACjG,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,CAAA;QAC9D,IAAI,KAAK,CAAC,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;YACzD,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CACzB,GAAG,EACH,6BAA6B,WAAW,CAAC,MAAM,CAAC,kBAAkB,GAAG,EACrE,sBAAsB,CACvB,CAAA;QACH,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,GAAG,CAC9B,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;YACvB,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC,EAAE,CAAC;oBAClD,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,WAAW,EAAE,WAAW,CAAC,CAAA;gBAC5D,CAAC;gBACD,OAAO,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAA;YACnD,CAAC;YACD,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI,EAAE,CAAC;gBACnD,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,gCAAgC,EAAE,mBAAmB,CAAC,CAAA;YACzF,CAAC;YACD,MAAM,KAAK,GAAG,6BAA6B,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAA;YACpE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC7B,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,wCAAwC,EAAE,cAAc,CAAC,CAAA;YACnG,CAAC;YACD,MAAM,WAAW,GAAG,KAAK,CAAC,CAAC,CAAa,CAAA;YACxC,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,CAAA;YAC7D,MAAM,EAAE,GAAG,EAAE,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,CAAC,QAAQ,EAAE,KAAK,EAAE,EAAE,MAAM,CAAC,CAAA;YACxG,OAAO,EAAE,GAAG,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAA;QACzC,CAAC,CAAC,CACH,CAAA;QACD,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,CAAA;QAC3G,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,CAAA;IAChC,CAAC,CAAA;IAED,QAAQ,GAAG,KAAK,EAAE,EAChB,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAC1B,KAAK,EAIN,EAAgC,EAAE;QACjC,MAAM,QAAQ,GAAG,KAAK,CAAC,EAAE,IAAI,KAAK,CAAC,MAAM,CAAA;QACzC,MAAM,QAAQ,GAAG,QAAQ,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAA;QAC9G,IAAI,QAAQ,IAAI,MAAM,KAAK,WAAW,CAAC,IAAI,CAAC,WAAW,IAAI,QAAQ,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;YACtF,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,WAAW,EAAE,WAAW,CAAC,CAAA;QACnE,CAAC;QACD,MAAM,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,IAAI,MAAM,CAAA;QACxE,IAAI,QAAQ,IAAI,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,KAAK,QAAQ,CAAC,OAAO,EAAE,CAAC;YACpE,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,2CAA2C,EAAE,iBAAiB,CAAC,CAAA;QACzG,CAAC;QACD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,IAAI,QAAQ,EAAE,KAAK,IAAI,UAAU,CAAA;QAC1D,MAAM,OAAO,GAAG,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,IAAI,cAAc,CAAC,CAAA;QAC/E,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;YACtC,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,kBAAkB,OAAO,EAAE,EAAE,WAAW,CAAC,CAAA;QACnF,CAAC;QACD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,IAAI,QAAQ,EAAE,KAAK,IAAI,WAAW,CAAC,MAAM,CAAC,MAAM,CAAA;QAEzE,IAAI,QAAQ,IAAI,QAAQ,EAAE,CAAC;YACzB,MAAM,KAAK,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAA;YAClH,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAA;YACzE,OAAO,EAAE,GAAG,QAAQ,EAAE,GAAG,KAAK,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,EAAE,CAAA;QACzD,CAAC;QAED,OAAO,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,CACzB,MAAM,EACN;YACE,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACrC,MAAM;YACN,OAAO;YACP,KAAK;YACL,KAAK;YACL,YAAY,EAAE,CAAC;YACf,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,IAAI;YAC9B,kBAAkB,EAAE,IAAI;SACzB,EACD,EAAE,UAAU,EAAE,IAAI,EAAE,CACrB,CAAA;IACH,CAAC,CAAA;IAED,QAAQ,GAAG,KAAK,EAAE,MAAe,EAAE,IAA8B,EAAgC,EAAE;QACjG,MAAM,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAA;QACpC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CACpB,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,IAAI,WAAW,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC,CAAC,EAChE,WAAW,CAAC,MAAM,CAAC,gBAAgB,CACpC,CAAA;QACD,MAAM,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,SAAS,EAAE,CAAA;QACpC,MAAM,WAAW,GAAG,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,UAAU,CAAC,UAAU,CAAC,CAAA;QAE5E,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,UAAU,CAAC,YAAY,CAAC,WAAW,CAAC,CAAA;QAC/D,IAAI,KAAK,GAAG,WAAW,CAAC,MAAM,CAAC,GAAG,qBAAqB,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;QAClG,IAAI,IAAI,EAAE,MAAM,EAAE,CAAC;YACjB,MAAM,UAAU,GAAG,MAAM,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAA;YAC3D,IAAI,UAAU,CAAC,MAAM;gBAAE,KAAK,GAAG,KAAK,CAAC,UAAU,CAAC,UAAU,CAAC,CAAA;QAC7D,CAAC;QACD,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,CAAA;QAC9B,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,sBAAsB,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,CAAA;IACrF,CAAC,CAAA;IAED,MAAM,GAAG,KAAK,EAAE,MAAe,EAAgD,EAAE;QAC/E,MAAM,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAA;QACpC,MAAM,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC,CAAA;QACzD,MAAM,QAAQ,GAAG,iBAAiB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;QAE3C,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE,kBAAkB,EAAE,IAAI,EAAE,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA;QAE9G,IAAI,QAAQ,EAAE,CAAC;YACb,QAAQ,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC,CAAA;YACzD,IAAI,QAAQ,CAAC,iBAAiB,EAAE,CAAC;gBAC/B,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,iBAAiB,EAAE,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAA;YAC5F,CAAC;QACH,CAAC;QAED,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,CAAA;IACpC,CAAC,CAAA;IAED,wBAAwB,GAAG,KAAK,EAAE,CAAkB,EAAE,MAAe,EAAE,IAAoC,EAAE,EAAE;QAC7G,MAAM,UAAU,GAAG,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAA;QACxC,MAAM,YAAY,GAAG,OAAO,CAC1B,UAAU,IAAI,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,CAC/F,CAAA;QACD,MAAM,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,IAAI,MAAM,CAAA;QACxE,MAAM,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,SAAS,EAAE,CAAA;QACpC,MAAM,eAAe,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,EAAE,CAAC,CAAA;QACzG,MAAM,MAAM,GACV,eAAe,IAAI,OAAO,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC,OAAO;YAC3D,CAAC,CAAC,eAAe;YACjB,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,CAAA;QAEpC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAA;QACrF,IAAI,QAAQ,IAAI,MAAM,KAAK,WAAW,CAAC,IAAI,CAAC,WAAW,IAAI,QAAQ,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;YACtF,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,WAAW,EAAE,WAAW,CAAC,CAAA;QAC5D,CAAC;QAED,MAAM,gBAAgB,GAAG,IAAI,CAAC,OAAO,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC,CAAA;QAChF,IAAI,QAAQ,IAAI,gBAAgB,IAAI,gBAAgB,KAAK,QAAQ,CAAC,OAAO,EAAE,CAAC;YAC1E,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,2CAA2C,EAAE,iBAAiB,CAAC,CAAA;QAClG,CAAC;QACD,MAAM,eAAe,GAAG,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,gBAAgB,IAAI,cAAc,CAAC,CAAA;QAC1F,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,eAAe,CAAC,CAAA;QACtD,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,kBAAkB,eAAe,EAAE,EAAE,WAAW,CAAC,CAAA;QACrG,MAAM,IAAI,CAAC,kBAAkB,CAAC,CAAC,EAAE,KAAK,EAAE,aAAa,CAAC,CAAA;QAEtD,MAAM,eAAe,GAAG,0BAA0B,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAChE,MAAM,WAAW,GAAG,kBAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QACpD,IAAI,CAAC,WAAW,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjD,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,oDAAoD,EAAE,kBAAkB,CAAC,CAAA;QAC5G,CAAC;QAED,MAAM,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC,CAAA;QACzD,MAAM,eAAe,GAAG,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAA;QACvE,MAAM,kBAAkB,GACtB,OAAO,CAAC,QAAQ,EAAE,kBAAkB,CAAC;YACrC,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,QAAQ,EAAE,SAAS,YAAY,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,eAAe,CAAA;QAEzG,IAAI,iBAAiB,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,kBAAkB,EAAE,CAAC;YACrD,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,mDAAmD,EAAE,wBAAwB,CAAC,CAAA;QACjH,CAAC;QAED,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,KAAK,EAAE,eAAe,EAAE,WAAW,EAAE,GAAG,EAAE,CAAA;IACxF,CAAC,CAAA;IAED,sBAAsB,GAAG,KAAK,EAAE,IAA+B,EAAiB,EAAE;QAChF,IAAI,aAAa,GAAG,EAAE,CAAA;QACtB,IAAI,WAA+B,CAAA;QACnC,MAAM,eAAe,GAAc,EAAE,CAAA;QACrC,IAAI,CAAC;YACH,IAAI,CAAC;gBACH,MAAM,CAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAA;gBACxG,aAAa,GAAG,IAAI,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAA;gBACjD,WAAW,GAAG,KAAK,EAAE,WAAW,IAAI,CAAC,CAAC,KAAK,EAAE,WAAW,IAAI,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC,CAAC,IAAI,SAAS,CAAC,CAAA;gBACzG,eAAe,CAAC,IAAI,CAAC,GAAG,sBAAsB,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAA;YACpE,CAAC;YAAC,MAAM,CAAC;gBACP,aAAa,GAAG,IAAI,CAAC,kBAAkB,EAAE,CAAA;YAC3C,CAAC;YAED,MAAM,kBAAkB,GAAG,aAAa,CAAC,IAAI,EAAE,CAAA;YAC/C,MAAM,CAAC,UAAU,EAAE,eAAe,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBACtD,IAAI,CAAC,cAAc;gBACnB,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;aAC1D,CAAC,CAAA;YAEF,MAAM,KAAK,GAAG,4BAA4B,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,SAAS,EAAE,EAAE;gBACpE,GAAG,IAAI;gBACP,kBAAkB;gBAClB,eAAe;gBACf,GAAG,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACrD,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;gBAC9B,UAAU;gBACV,eAAe;aAChB,CAAC,CAAA;YAEF,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,UAAU,CAAC,YAAY,CAAC,WAAW,CAAC,CAAA;YAC/D,MAAM,KAAK,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,qCAAqC,EAAE,GAAG,CAAC,CAAC,CAAA;QAChG,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC;gBACH,IAAI,CAAC,gBAAgB,EAAE,EAAE,CAAA;YAC3B,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;YACV,iBAAiB,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;QACpC,CAAC;IACH,CAAC,CAAA;IAED,mBAAmB,GAAG,KAAK,EACzB,QAAyC,EACzC,UAA6B,EAC7B,MAAc,EACd,MAAe,EACf,eAAuB,EACvB,KAAa,EACb,aAAqB,EACrB,YAAoB,EACpB,oBAAqC,EAChB,EAAE;QACvB,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,CACxB,MAAM,EACN,MAAM,EACN,EAAE,KAAK,EAAE,OAAO,EAAE,eAAe,EAAE,KAAK,EAAE,aAAa,EAAE,kBAAkB,EAAE,YAAY,EAAE,OAAO,EAAE,KAAK,EAAE,EAC3G,EAAE,UAAU,EAAE,IAAI,EAAE,CACrB,CAAA;QACH,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,CACxB,MAAM,EACN;gBACE,EAAE,EAAE,MAAM;gBACV,MAAM;gBACN,OAAO,EAAE,eAAe;gBACxB,KAAK;gBACL,KAAK,EAAE,aAAa;gBACpB,YAAY,EAAE,CAAC;gBACf,OAAO,EAAE,IAAI;gBACb,kBAAkB,EAAE,YAAY;aACjC,EACD,EAAE,UAAU,EAAE,IAAI,EAAE,CACrB,CAAA;QACH,CAAC;QAED,OAAO,UAAU,CAAC,UAAU,CAC1B,CAAC,IAAI,EAAE,EAAE;YACP,IAAI,IAAI,CAAC,IAAI,EAAE,EAAE,kBAAkB,KAAK,YAAY,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBAC7F,oBAAoB,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC,CAAA;YACnF,CAAC;QACH,CAAC,EACD,GAAG,EAAE,GAAE,CAAC,CACT,CAAA;IACH,CAAC,CAAA;IAED,mBAAmB,GAAG,KAAK,EAAE,IAW5B,EAAE,EAAE;QACH,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,eAAe,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE,aAAa,EAAE,aAAa,EAAE,GACrH,IAAI,CAAA;QACN,MAAM,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,MAAM,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YAC1D,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,eAAe,EAAE,KAAK,CAAC,eAAe,IAAI,EAAE,CAAC;YAClF,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC;gBAC9B,MAAM;gBACN,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC7B,GAAG,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACvD,MAAM,EAAE,WAAW;gBACnB,UAAU,EAAE,KAAK,CAAC,UAAU,KAAK,KAAK;gBACtC,KAAK;gBACL,IAAI;gBACJ,WAAW,EAAE,MAAM;aACpB,CAAC;SACH,CAAC,CAAA;QAEF,MAAM,yBAAyB,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE,CAC9D,oBAAoB,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,QAAQ,IAAI,MAAM,EAAE,IAAI,CAAC,CAAC,IAAI,EAAE,CACnE,CAAA;QACD,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;QAClC,MAAM,eAAe,GAAG,aAAa;YACnC,CAAC,CAAC;gBACE,GAAG,aAAa;gBAChB,GAAG,CAAC,yBAAyB,CAAC,MAAM,GAAG,CAAC;oBACtC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,yBAAyB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC;oBACjH,CAAC,CAAC,EAAE,CAAC;aACR,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC,iBAAiB,CAAC;YAChD,CAAC,CAAC;gBACE,GAAG,MAAM,CAAC,qBAAqB;gBAC/B;oBACE,IAAI,EAAE,MAAe;oBACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,CAAC,WAAW,EAAE,GAAG,yBAAyB,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,iBAAiB,EAAE,CAAC;iBAC1I;aACF,CAAA;QAEL,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,eAAe,EAAE,MAAM,EAAE;YACxE,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC1D,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7B,WAAW,EAAE,MAAM;YACnB,IAAI,EAAE,IAAI,IAAI,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;YACrD,cAAc,EAAE;gBACd,cAAc,EAAE,IAAI,IAAI,CAAC,eAAe,CAAC,CAAC,WAAW,EAAE;gBACvD,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;gBACzC,sBAAsB,EAAE,MAAM,CAAC,sBAAsB;gBACrD,yBAAyB;gBACzB,WAAW,EAAE,MAAM,CAAC,WAAW;aAChC;SACF,CAAC,CAAA;QACF,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,yBAAyB,EAAE,eAAe,EAAE,CAAA;IACnG,CAAC,CAAA;IAED,aAAa,GAAG,KAAK,EAAE,EACrB,CAAC,EACD,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAC1B,KAAK,EAKN,EAAqB,EAAE;QACtB,MAAM,UAAU,GAAG,cAAc,CAAC,SAAS,CAAC,KAAK,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAA;QAC5F,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAChC,GAAG,EACH,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,OAAO,KAAK,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAC7F,kBAAkB,CACnB,CAAA;QACH,CAAC;QACD,MAAM,IAAI,GAAG,UAAU,CAAC,IAAI,CAAA;QAC5B,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,KAAK,EAAE,eAAe,EAAE,WAAW,EAAE,GAAG,EAAE,GACnF,MAAM,IAAI,CAAC,wBAAwB,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,CAAA;QAEtD,MAAM,YAAY,GAAG,OAAO,MAAM,CAAC,UAAU,EAAE,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAA;QAChF,MAAM,oBAAoB,GAAG,IAAI,eAAe,EAAE,CAAA;QAClD,IAAI,eAAe,GAAG,EAAE,CAAA;QACxB,MAAM,WAAW,GAA0B;YACzC,YAAY;YACZ,UAAU,EAAE,oBAAoB;YAChC,WAAW,EAAE,GAAG,EAAE,CAAC,eAAe;SACnC,CAAA;QACD,iBAAiB,CAAC,GAAG,CAAC,GAAG,EAAE,WAAW,CAAC,CAAA;QAEvC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,IAAI,QAAQ,EAAE,KAAK,IAAI,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QACxE,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,IAAI,QAAQ,EAAE,KAAK,IAAI,KAAK,CAAC,KAAK,IAAI,WAAW,CAAC,MAAM,CAAC,MAAM,CAAA;QAC/F,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;QAC3E,IAAI,gBAA0C,CAAA;QAE9C,IAAI,CAAC;YACH,gBAAgB,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAC/C,QAAQ,EACR,UAAU,EACV,MAAM,EACN,MAAM,EACN,eAAe,EACf,KAAK,EACL,aAAa,EACb,YAAY,EACZ,oBAAoB,CACrB,CAAA;YACD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAA;YACjF,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,yBAAyB,EAAE,eAAe,EAAE,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC;gBACtG,MAAM;gBACN,IAAI;gBACJ,MAAM;gBACN,YAAY,EAAE,QAAQ,EAAE,YAAY,IAAI,CAAC;gBACzC,KAAK;gBACL,aAAa;gBACb,eAAe;gBACf,WAAW;gBACX,MAAM,EAAE,oBAAoB,CAAC,MAAM;aACpC,CAAC,CAAA;YAEF,MAAM,aAAa,GAAG,WAAW,IAAI,yBAAyB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YACzE,MAAM,MAAM,GAAG,UAAU,CAAC,UAAU,CAAC,UAAU,CAAC,CAAA;YAChD,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,OAAO,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,EAAE,CAAA;YACnH,MAAM,mBAAmB,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAA;YACnH,MAAM,iBAAiB,GAAG,IAAI,CAAC,sBAAsB,CAAC;gBACpD,IAAI;gBACJ,kBAAkB,EAAE,GAAG,EAAE,CAAC,eAAe;gBACzC,MAAM,EAAE,oBAAoB,CAAC,MAAM;gBACnC,cAAc,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;gBACpE,UAAU;gBACV,MAAM;gBACN,MAAM;gBACN,eAAe;gBACf,KAAK;gBACL,aAAa;gBACb,SAAS;gBACT,cAAc,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC,EAAE;gBAC/B,eAAe;gBACf,aAAa;gBACb,mBAAmB;gBACnB,UAAU;gBACV,gBAAgB;gBAChB,GAAG;aACJ,CAAC,CAAA;YACF,WAAW,CAAC,iBAAiB,GAAG,iBAAiB,CAAA;YACjD,OAAO,sBAAsB,CAAC,IAAI,EAAE,MAAM,EAAE,YAAY,EAAE,iBAAiB,EAAE,CAAC,CAAC,EAAE,EAAE;gBACjF,eAAe,IAAI,CAAC,CAAA;YACtB,CAAC,CAAC,CAAA;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC;gBACH,gBAAgB,EAAE,EAAE,CAAA;YACtB,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;YACV,iBAAiB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAA;YAC7B,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE,kBAAkB,EAAE,IAAI,EAAE,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA;YAC9G,MAAM,GAAG,CAAA;QACX,CAAC;IACH,CAAC,CAAA;CACF"}

package/dist/chat/utils.d.ts

@@ -0,0 +1,61 @@
+import { type DocumentReference, type Firestore } from 'firebase-admin/firestore';
+import { type LanguageModelUsage, type ModelMessage, type streamText } from 'ai';
+import { z } from 'zod';
+import type { ProcessedAttachment } from '../attachments';
+import type { EntryId } from '../db';
+import type { FileRef } from '../files';
+import type { ChatMessagePublic, ContextEntry, ContextItemSpec, DynamicContextBlockInput, RawAttachmentPart, UserContext, streamChatBody } from './schema';
+export declare const FORBIDDEN_TOOL_SCHEMA_KEYS: Set<string>;
+export declare const sanitizeToolInputSchema: (schema: z.ZodType<any>) => z.ZodType<any>;
+export declare const isUserContextShape: (value: unknown) => value is UserContext;
+export declare const formatAttachmentLine: (filename: string, item: ProcessedAttachment) => string;
+export declare const isSafePublicUrl: (raw?: string) => boolean;
+export declare const extractPart: (p: Record<string, unknown>) => RawAttachmentPart | undefined;
+export declare const extractTurnAttachmentParts: (msg: z.infer<typeof streamChatBody>['message']) => RawAttachmentPart[];
+export declare const extractMessageText: (msg: {
+    content?: unknown;
+    parts?: readonly unknown[];
+}) => string;
+export declare const deriveTitle: (msgOrList: {
+    content?: unknown;
+    parts?: readonly unknown[];
+    role?: string;
+} | readonly Record<string, unknown>[]) => string;
+export declare const extractLatestUserPrompt: (messages: readonly (Record<string, unknown> | ModelMessage)[]) => string;
+export declare const formatUsage: (usage: LanguageModelUsage | null) => {
+    prompt_tokens: number;
+    completion_tokens: number;
+    total_tokens: number;
+};
+export declare const isContextItemVisible: (item: ContextItemSpec, audiences?: readonly string[]) => boolean;
+export declare const formatContextBlock: (entries: readonly ContextEntry[]) => string;
+export declare const buildDynamicContextBlock: (input: DynamicContextBlockInput) => string;
+export declare const PUBLIC_MESSAGE_FIELDS: readonly ['id', 'chatId', 'userId', 'role', 'content', 'parts', 'toolInvocations', 'attachments', 'tokenCount', 'cancelled', 'createdAt'];
+export declare const RECALL_MESSAGE_FIELDS: readonly ['id', 'chatId', 'role', 'content'];
+export declare const mapFirestoreMessageDoc: (id: string, raw: Record<string, unknown>) => ChatMessagePublic;
+export declare const toModelMessageFromDoc: (doc: ChatMessagePublic) => ModelMessage | undefined;
+export declare const extractToolInvocations: (messages: readonly {
+    content: unknown;
+}[]) => unknown[];
+export declare const buildUIMessageResponse: (turn: ReturnType<typeof streamText>, chatId: string, generationId: string, completionPromise: Promise<void>, onDelta: (delta: string) => void) => Response;
+export declare const buildCompletedTurnWriteBatch: (db: Firestore, opts: {
+    chatDocRef: DocumentReference;
+    chatId: string;
+    userId: EntryId;
+    resolvedAgentId: string;
+    title: string;
+    selectedModel: string;
+    userMsgId: string;
+    assistantMsgId: string;
+    userCreatedAtMs: number;
+    finalUserText: string;
+    userPartsForStorage: unknown[];
+    stagedRefs: FileRef[];
+    finalAssistantText: string;
+    toolInvocations: unknown[];
+    totalTokens?: number;
+    cancelled: boolean;
+    userVector: number[] | null;
+    assistantVector: number[] | null;
+}) => FirebaseFirestore.WriteBatch;
+//# sourceMappingURL=utils.d.ts.map

package/dist/chat/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/chat/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAyB,KAAK,iBAAiB,EAAE,KAAK,SAAS,EAAE,MAAM,0BAA0B,CAAA;AACxG,OAAO,EAAoD,KAAK,kBAAkB,EAAE,KAAK,YAAY,EAAE,KAAK,UAAU,EAAE,MAAM,IAAI,CAAA;AAClI,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AACvB,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAA;AACzD,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,OAAO,CAAA;AACpC,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,UAAU,CAAA;AACvC,OAAO,KAAK,EACV,iBAAiB,EAEjB,YAAY,EACZ,eAAe,EACf,wBAAwB,EACxB,iBAAiB,EACjB,WAAW,EACX,cAAc,EACf,MAAM,UAAU,CAAA;AAEjB,eAAO,MAAM,0BAA0B,aASrC,CAAA;AAEF,eAAO,MAAM,uBAAuB,WAAY,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,KAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAgB7E,CAAA;AAED,eAAO,MAAM,kBAAkB,UAAW,OAAO,KAAG,KAAK,IAAI,WAItB,CAAA;AAEvC,eAAO,MAAM,oBAAoB,aAAc,MAAM,QAAQ,mBAAmB,KAAG,MAGsF,CAAA;AAEzK,eAAO,MAAM,eAAe,SAAU,MAAM,KAAG,OAgB9C,CAAA;AAED,eAAO,MAAM,WAAW,MAAO,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAG,iBAAiB,GAAG,SAM5E,CAAA;AAED,eAAO,MAAM,0BAA0B,QAAS,CAAC,CAAC,KAAK,CAAC,OAAO,cAAc,CAAC,CAAC,SAAS,CAAC,KAAG,iBAAiB,EAS5G,CAAA;AAED,eAAO,MAAM,kBAAkB,QAAS;IAAE,OAAO,CAAC,EAAE,OAAO,CAAC;IAAC,KAAK,CAAC,EAAE,SAAS,OAAO,EAAE,CAAA;CAAE,KAAG,MAW3F,CAAA;AAED,eAAO,MAAM,WAAW,cACX;IAAE,OAAO,CAAC,EAAE,OAAO,CAAC;IAAC,KAAK,CAAC,EAAE,SAAS,OAAO,EAAE,CAAC;IAAC,IAAI,CAAC,EAAE,MAAM,CAAA;CAAE,GAAG,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,KAC/G,MAQF,CAAA;AAED,eAAO,MAAM,uBAAuB,aAAc,SAAS,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,YAAY,CAAC,EAAE,KAAG,MAQvG,CAAA;AAED,eAAO,MAAM,WAAW,UAAW,kBAAkB,GAAG,IAAI;IAC1D,aAAa;IACb,iBAAiB;IACjB,YAAY;CACZ,CAAA;AAEF,eAAO,MAAM,oBAAoB,SAAU,eAAe,cAAc,SAAS,MAAM,EAAE,KAAG,OACR,CAAA;AAEpF,eAAO,MAAM,kBAAkB,YAAa,SAAS,YAAY,EAAE,KAAG,MASrE,CAAA;AAED,eAAO,MAAM,wBAAwB,UAAW,wBAAwB,KAAG,MA8B1E,CAAA;AAED,eAAO,MAAM,qBAAqB,YAChC,IAAI,EACJ,QAAQ,EACR,QAAQ,EACR,MAAM,EACN,SAAS,EACT,OAAO,EACP,iBAAiB,EACjB,aAAa,EACb,YAAY,EACZ,WAAW,EACX,WAAW,CACH,CAAA;AAEV,eAAO,MAAM,qBAAqB,YAAI,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,CAAU,CAAA;AAEjF,eAAO,MAAM,sBAAsB,OAC7B,MAAM,OACL,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC3B,iBA6BF,CAAA;AAED,eAAO,MAAM,qBAAqB,QAAS,iBAAiB,KAAG,YAAY,GAAG,SAU7E,CAAA;AAED,eAAO,MAAM,sBAAsB,aAAc,SAAS;IAAE,OAAO,EAAE,OAAO,CAAA;CAAE,EAAE,KAAG,OAAO,EAWzF,CAAA;AAED,eAAO,MAAM,sBAAsB,SAC3B,UAAU,CAAC,OAAO,UAAU,CAAC,UAC3B,MAAM,gBACA,MAAM,qBACD,OAAO,CAAC,IAAI,CAAC,WACvB,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,KAC/B,QAsCF,CAAA;AAED,eAAO,MAAM,4BAA4B,OACnC,SAAS,QACP;IACJ,UAAU,EAAE,iBAAiB,CAAA;IAC7B,MAAM,EAAE,MAAM,CAAA;IACd,MAAM,EAAE,OAAO,CAAA;IACf,eAAe,EAAE,MAAM,CAAA;IACvB,KAAK,EAAE,MAAM,CAAA;IACb,aAAa,EAAE,MAAM,CAAA;IACrB,SAAS,EAAE,MAAM,CAAA;IACjB,cAAc,EAAE,MAAM,CAAA;IACtB,eAAe,EAAE,MAAM,CAAA;IACvB,aAAa,EAAE,MAAM,CAAA;IACrB,mBAAmB,EAAE,OAAO,EAAE,CAAA;IAC9B,UAAU,EAAE,OAAO,EAAE,CAAA;IACrB,kBAAkB,EAAE,MAAM,CAAA;IAC1B,eAAe,EAAE,OAAO,EAAE,CAAA;IAC1B,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,SAAS,EAAE,OAAO,CAAA;IAClB,UAAU,EAAE,MAAM,EAAE,GAAG,IAAI,CAAA;IAC3B,eAAe,EAAE,MAAM,EAAE,GAAG,IAAI,CAAA;CACjC,iCA8CF,CAAA"}