@corgelabs/corge 0.0.1 → 1.0.1

package/dist/chat/service.js

@@ -0,0 +1,279 @@
+import { stepCountIs, streamText, tool as aiTool } from 'ai';
+import { registerAttachmentTypes } from '../attachments';
+import { CORE_CONFIG } from '../config';
+import { Core } from '../index';
+import { ChatContextEngine } from './context';
+import { ReasoningExpertInputDto, ReasoningExpertOutputDto, webSearchInput } from './schema';
+import { ChatStreamEngine } from './stream';
+import { buildDynamicContextBlock, extractLatestUserPrompt, FORBIDDEN_TOOL_SCHEMA_KEYS, isSafePublicUrl, sanitizeToolInputSchema } from './utils';
+const toolRegistry = new Map();
+const agentRegistry = new Map();
+export class ChatService {
+    #core;
+    #contextEngine;
+    #streamEngine;
+    constructor(coreClient) {
+        this.#core = coreClient;
+        this.#contextEngine = new ChatContextEngine(coreClient);
+        this.#streamEngine = new ChatStreamEngine(coreClient, agentRegistry);
+    }
+    #registerWorkflowTool = (name, spec) => {
+        toolRegistry.set(name, {
+            ...spec,
+            mcp: spec.mcp === true,
+            tool: spec.tool !== false,
+            input: sanitizeToolInputSchema(spec.input)
+        });
+    };
+    #clearWorkflowTools = () => {
+        toolRegistry.clear();
+    };
+    #getMcpTools = () => new Map([...toolRegistry.entries()].filter(([, spec]) => spec.mcp === true));
+    #registerAgents = (defs, reset = false) => {
+        if (reset) {
+            agentRegistry.clear();
+            registerAttachmentTypes(undefined, true);
+        }
+        for (const [id, cfg] of Object.entries(defs)) {
+            agentRegistry.set(id, { ...cfg, name: cfg.name ?? id });
+            if (cfg.attachmentTypes)
+                registerAttachmentTypes(cfg.attachmentTypes);
+        }
+    };
+    #getRegisteredAgents = () => Object.fromEntries(agentRegistry);
+    #getAgentRegistry = () => agentRegistry;
+    getContext = (prompt, catalog, options) => this.#contextEngine.getContext(prompt, catalog, options);
+    retrieveContext = (opts) => this.#contextEngine.retrieveContext(opts);
+    messages = (chatId, opts) => this.#streamEngine.messages(chatId, opts);
+    cancel = (chatId) => this.#streamEngine.cancel(chatId);
+    #webSearchTool = (abortSignal) => aiTool({
+        description: 'Search the web (DuckDuckGo) or open a URL and read it as markdown. Use for current information.',
+        inputSchema: webSearchInput,
+        execute: async ({ query, url }) => {
+            const useUrl = isSafePublicUrl(url);
+            const target = useUrl ? url : `https://html.duckduckgo.com/html/?q=${encodeURIComponent(query ?? url ?? '')}`;
+            const maxChars = useUrl ? 6000 : 4000;
+            try {
+                const timeoutSignal = AbortSignal.timeout(20_000);
+                const signal = abortSignal ? AbortSignal.any([abortSignal, timeoutSignal]) : timeoutSignal;
+                const jinaKey = this.#core.env.JINA_API_KEY;
+                const res = await fetch(`https://r.jina.ai/${target}`, {
+                    headers: {
+                        Accept: 'text/plain, text/markdown;q=0.9, */*;q=0.8',
+                        ...(jinaKey ? { Authorization: `Bearer ${jinaKey}` } : {})
+                    },
+                    signal
+                });
+                if (!res.ok)
+                    return { text: `Web ${useUrl ? 'page' : 'search'} failed (HTTP ${res.status}).` };
+                const reader = res.body?.getReader();
+                if (!reader)
+                    return { text: '' };
+                const decoder = new TextDecoder();
+                let text = '';
+                while (text.length < maxChars + 256) {
+                    const { done, value } = await reader.read();
+                    if (done)
+                        break;
+                    text += decoder.decode(value, { stream: true });
+                }
+                await reader.cancel().catch(() => { });
+                return { text: text.slice(0, maxChars).replace(/\s+\S*$/, '') };
+            }
+            catch {
+                return { text: 'Web search failed.' };
+            }
+        }
+    });
+    #createReasoningExpertTool = (ctx) => aiTool({
+        description: 'Delegate complex architectural, legal, financial, or multi-constraint reasoning problems to a frontier reasoning model. Use only when standard analysis is insufficient.',
+        inputSchema: ReasoningExpertInputDto,
+        execute: async ({ problemStatement, constraintsAndContext }) => {
+            const result = await ctx.core.ai.generate(`Problem:\n${problemStatement}\n\nConstraints & Context:\n${constraintsAndContext}`, {
+                schema: ReasoningExpertOutputDto,
+                model: CORE_CONFIG.models.vertexExpert ?? 'claude-sonnet-4-5',
+                signal: ctx.abortSignal
+            });
+            return result.output;
+        }
+    });
+    #buildAgentToolsForTurn = (toolKeys = [], ctx) => {
+        const tools = {};
+        const { core: turnCore, abortSignal } = ctx;
+        const coreState = {
+            env: turnCore.env,
+            userId: turnCore.userId,
+            role: turnCore.role,
+            step: turnCore.step
+        };
+        for (const name of toolKeys) {
+            if (name === 'web_search') {
+                tools[name] = this.#webSearchTool(abortSignal);
+            }
+            else if (name === 'consult_reasoning_expert') {
+                tools[name] = this.#createReasoningExpertTool(ctx);
+            }
+            else {
+                const wf = toolRegistry.get(name);
+                if (wf && wf.tool !== false) {
+                    tools[name] = aiTool({
+                        description: wf.description,
+                        inputSchema: sanitizeToolInputSchema(wf.input),
+                        execute: (rawInput) => this.#core.internals.context.run(coreState, async () => {
+                            const safeInput = rawInput && typeof rawInput === 'object'
+                                ? Object.fromEntries(Object.entries(rawInput).filter(([k]) => !FORBIDDEN_TOOL_SCHEMA_KEYS.has(k)))
+                                : rawInput;
+                            return wf.execute(safeInput, turnCore.userId);
+                        })
+                    });
+                }
+            }
+        }
+        return tools;
+    };
+    #resolveStaticSystemPrompt = async (agent) => {
+        const raw = typeof agent.systemPrompt === 'function'
+            ? await agent.systemPrompt({
+                today: 'see dynamic context',
+                userId: 'authenticated-user',
+                env: this.#core.env
+            })
+            : agent.systemPrompt;
+        return raw
+            .replace(/\s*Today is \{\{today\}\}\.?/gi, '')
+            .replace(/\s*Current userId is \{\{userId\}\}\.?/gi, '')
+            .replaceAll('{{today}}', 'see dynamic context')
+            .replaceAll('{{userId}}', 'authenticated-user')
+            .trim();
+    };
+    #resolveStepActiveTools = (agent, allToolNames, stepNumber) => {
+        if (allToolNames.length === 0)
+            return [];
+        const presets = agent.toolPresets;
+        if (!presets)
+            return allToolNames;
+        if (stepNumber === 0 && presets.discovery?.length && presets.workflow?.length) {
+            return [...new Set([...presets.discovery, ...presets.workflow])].filter((k) => allToolNames.includes(k));
+        }
+        if (stepNumber > 0 && presets.workflow?.length) {
+            return presets.workflow.filter((k) => allToolNames.includes(k));
+        }
+        return allToolNames;
+    };
+    stream = async (agent, messages, userId = this.#core.userId, opts) => {
+        await this.#core.internals.payments.enforceAiQuota(userId);
+        const [, staticCoreAgentInstructions] = await Promise.all([
+            this.#core.internals.resilience.acquireLimit('ai'),
+            this.#resolveStaticSystemPrompt(agent)
+        ]);
+        const role = opts?.core?.role ?? this.#core.internals.auth.resolveUserRole(userId, this.#core.role);
+        const verifiedCore = opts?.core ?? new Core({ env: this.#core.env, userId, role });
+        const tools = this.#buildAgentToolsForTurn(agent.tools ?? [], {
+            core: verifiedCore,
+            chatId: opts?.chatId,
+            abortSignal: opts?.abortSignal
+        });
+        const allToolNames = Object.keys(tools);
+        const dynamicBlock = buildDynamicContextBlock({
+            currentTimeIso: opts?.dynamicContext?.currentTimeIso ?? new Date().toISOString(),
+            recalledMemories: opts?.dynamicContext?.recalledMemories,
+            infobaseContextEntries: opts?.dynamicContext?.infobaseContextEntries,
+            stagedAttachmentSummaries: opts?.dynamicContext?.stagedAttachmentSummaries,
+            userContext: opts?.dynamicContext?.userContext
+        });
+        const systemInstructions = [
+            {
+                role: 'system',
+                content: staticCoreAgentInstructions,
+                providerOptions: {
+                    anthropic: { cacheControl: { type: 'ephemeral' } },
+                    vertex: { cacheControl: { type: 'ephemeral' } }
+                }
+            },
+            {
+                role: 'system',
+                content: dynamicBlock
+            }
+        ];
+        const messagesForLlm = messages.filter((m) => m.role !== 'system');
+        const selectedModel = opts?.modelOverride || agent.model || CORE_CONFIG.models.vertex;
+        return streamText({
+            model: this.#core.ai.languageModel(selectedModel),
+            system: systemInstructions,
+            messages: messagesForLlm,
+            tools,
+            ...(allToolNames.length > 0
+                ? {
+                    activeTools: allToolNames,
+                    prepareStep: ({ stepNumber, messages: stepMessages }) => ({
+                        activeTools: this.#resolveStepActiveTools(agent, allToolNames, stepNumber),
+                        ...(stepNumber > 2 && stepMessages.length > CORE_CONFIG.limits.chatMessagesLimit
+                            ? { messages: stepMessages.slice(-CORE_CONFIG.limits.chatMessagesLimit) }
+                            : {})
+                    })
+                }
+                : {}),
+            stopWhen: stepCountIs(agent.maxSteps ?? CORE_CONFIG.models.defaultAgentMaxSteps),
+            ...(opts?.abortSignal ? { abortSignal: opts.abortSignal } : {}),
+            onFinish: async (event) => {
+                const usageObj = event.totalUsage ?? event.usage;
+                await this.#core.internals.payments.trackAiUsage({
+                    userId,
+                    model: selectedModel,
+                    inputTokens: usageObj?.inputTokens ?? 0,
+                    outputTokens: usageObj?.outputTokens ?? 0,
+                    ...(typeof usageObj?.totalTokens === 'number' ? { totalTokens: usageObj.totalTokens } : {}),
+                    source: 'chat'
+                });
+                await opts?.onFinish?.(event);
+            }
+        });
+    };
+    ask = async (agentOrId, prompt, options) => {
+        const agent = typeof agentOrId === 'string' ? agentRegistry.get(agentOrId) : agentOrId;
+        if (!agent)
+            return this.#core.errors.httpError(404, `Unknown agent: ${String(agentOrId)}`, 'NOT_FOUND');
+        const userId = options?.userId ?? this.#core.userId ?? CORE_CONFIG.auth.adminUserId;
+        const rawMessages = typeof prompt === 'string' ? [{ role: 'user', content: [{ type: 'text', text: prompt }] }] : [...prompt];
+        const latestPrompt = typeof prompt === 'string' ? prompt : extractLatestUserPrompt(rawMessages);
+        const hybrid = options?.includeContext !== false
+            ? await this.retrieveContext({
+                userId,
+                prompt: latestPrompt,
+                useContext: agent.useContext !== false,
+                agent,
+                abortSignal: options?.abortSignal
+            })
+            : { recalledMemories: [], infobaseContextEntries: [], userContext: undefined };
+        const turn = await this.stream(agent, rawMessages.slice(-15), userId, {
+            abortSignal: options?.abortSignal,
+            dynamicContext: {
+                recalledMemories: hybrid.recalledMemories,
+                infobaseContextEntries: hybrid.infobaseContextEntries,
+                userContext: hybrid.userContext
+            }
+        });
+        const [text, res] = await Promise.all([turn.text, turn.response]);
+        return {
+            text,
+            messages: [...rawMessages, ...res.messages]
+        };
+    };
+    get internals() {
+        return {
+            registerWorkflowTool: this.#registerWorkflowTool,
+            clearWorkflowTools: this.#clearWorkflowTools,
+            getMcpTools: this.#getMcpTools,
+            registerAgents: this.#registerAgents,
+            getRegisteredAgents: this.#getRegisteredAgents,
+            getAgentRegistry: this.#getAgentRegistry,
+            registerContextCatalog: this.#contextEngine.registerContextCatalog,
+            enforceAgentAccess: this.#streamEngine.enforceAgentAccess,
+            stageAndScanAttachments: this.#streamEngine.stageAndScanAttachments,
+            startChatStreamTurn: this.#streamEngine.startChatStreamTurn,
+            saveChat: this.#streamEngine.saveChat,
+            streamChatSse: this.#streamEngine.streamChatSse
+        };
+    }
+}
+//# sourceMappingURL=service.js.map

package/dist/chat/service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"service.js","sourceRoot":"","sources":["../../src/chat/service.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,WAAW,EACX,UAAU,EACV,IAAI,IAAI,MAAM,EAIf,MAAM,IAAI,CAAA;AACX,OAAO,EAAE,uBAAuB,EAAE,MAAM,gBAAgB,CAAA;AACxD,OAAO,EAAE,WAAW,EAAE,MAAM,WAAW,CAAA;AAEvC,OAAO,EAAE,IAAI,EAAE,MAAM,UAAU,CAAA;AAC/B,OAAO,EAAE,iBAAiB,EAAE,MAAM,WAAW,CAAA;AAC7C,OAAO,EACL,uBAAuB,EACvB,wBAAwB,EACxB,cAAc,EAaf,MAAM,UAAU,CAAA;AACjB,OAAO,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAA;AAC3C,OAAO,EACL,wBAAwB,EACxB,uBAAuB,EACvB,0BAA0B,EAC1B,eAAe,EACf,uBAAuB,EACxB,MAAM,SAAS,CAAA;AAEhB,MAAM,YAAY,GAAG,IAAI,GAAG,EAA4B,CAAA;AACxD,MAAM,aAAa,GAAG,IAAI,GAAG,EAAuB,CAAA;AAEpD,MAAM,OAAO,WAAW;IACb,KAAK,CAAM;IACX,cAAc,CAAmB;IACjC,aAAa,CAAkB;IAExC,YAAY,UAAgB;QAC1B,IAAI,CAAC,KAAK,GAAG,UAAU,CAAA;QACvB,IAAI,CAAC,cAAc,GAAG,IAAI,iBAAiB,CAAC,UAAU,CAAC,CAAA;QACvD,IAAI,CAAC,aAAa,GAAG,IAAI,gBAAgB,CAAC,UAAU,EAAE,aAAa,CAAC,CAAA;IACtE,CAAC;IAED,qBAAqB,GAAG,CAAC,IAAY,EAAE,IAAsB,EAAQ,EAAE;QACrE,YAAY,CAAC,GAAG,CAAC,IAAI,EAAE;YACrB,GAAG,IAAI;YACP,GAAG,EAAE,IAAI,CAAC,GAAG,KAAK,IAAI;YACtB,IAAI,EAAE,IAAI,CAAC,IAAI,KAAK,KAAK;YACzB,KAAK,EAAE,uBAAuB,CAAC,IAAI,CAAC,KAAK,CAAC;SAC3C,CAAC,CAAA;IACJ,CAAC,CAAA;IAED,mBAAmB,GAAG,GAAS,EAAE;QAC/B,YAAY,CAAC,KAAK,EAAE,CAAA;IACtB,CAAC,CAAA;IAED,YAAY,GAAG,GAAkC,EAAE,CACjD,IAAI,GAAG,CAAC,CAAC,GAAG,YAAY,CAAC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,KAAK,IAAI,CAAC,CAAC,CAAA;IAE9E,eAAe,GAAG,CAAC,IAAiC,EAAE,KAAK,GAAG,KAAK,EAAQ,EAAE;QAC3E,IAAI,KAAK,EAAE,CAAC;YACV,aAAa,CAAC,KAAK,EAAE,CAAA;YACrB,uBAAuB,CAAC,SAAS,EAAE,IAAI,CAAC,CAAA;QAC1C,CAAC;QACD,KAAK,MAAM,CAAC,EAAE,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;YAC7C,aAAa,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,GAAG,GAAG,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,IAAI,EAAE,EAAE,CAAC,CAAA;YACvD,IAAI,GAAG,CAAC,eAAe;gBAAE,uBAAuB,CAAC,GAAG,CAAC,eAAe,CAAC,CAAA;QACvE,CAAC;IACH,CAAC,CAAA;IAED,oBAAoB,GAAG,GAAgC,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,CAAA;IAE3F,iBAAiB,GAAG,GAA6B,EAAE,CAAC,aAAa,CAAA;IAEjE,UAAU,GAAG,CACX,MAAc,EACd,OAAoC,EACpC,OAA2B,EACF,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,CAAA;IAEtF,eAAe,GAAG,CAAC,IAA4B,EAAkC,EAAE,CACjF,IAAI,CAAC,cAAc,CAAC,eAAe,CAAC,IAAI,CAAC,CAAA;IAE3C,QAAQ,GAAG,CAAC,MAAe,EAAE,IAA8B,EAAgC,EAAE,CAC3F,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAC,CAAA;IAE3C,MAAM,GAAG,CAAC,MAAe,EAAgD,EAAE,CACzE,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAA;IAEnC,cAAc,GAAG,CAAC,WAAyB,EAAE,EAAE,CAC7C,MAAM,CAAC;QACL,WAAW,EAAE,iGAAiG;QAC9G,WAAW,EAAE,cAAc;QAC3B,OAAO,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,EAAE;YAChC,MAAM,MAAM,GAAG,eAAe,CAAC,GAAG,CAAC,CAAA;YACnC,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,GAAI,CAAC,CAAC,CAAC,uCAAuC,kBAAkB,CAAC,KAAK,IAAI,GAAG,IAAI,EAAE,CAAC,EAAE,CAAA;YAC9G,MAAM,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAA;YACrC,IAAI,CAAC;gBACH,MAAM,aAAa,GAAG,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;gBACjD,MAAM,MAAM,GAAG,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAA;gBAC1F,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,YAAY,CAAA;gBAC3C,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,qBAAqB,MAAM,EAAE,EAAE;oBACrD,OAAO,EAAE;wBACP,MAAM,EAAE,4CAA4C;wBACpD,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,UAAU,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;qBAC3D;oBACD,MAAM;iBACP,CAAC,CAAA;gBACF,IAAI,CAAC,GAAG,CAAC,EAAE;oBAAE,OAAO,EAAE,IAAI,EAAE,OAAO,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,iBAAiB,GAAG,CAAC,MAAM,IAAI,EAAE,CAAA;gBAC9F,MAAM,MAAM,GAAG,GAAG,CAAC,IAAI,EAAE,SAAS,EAAE,CAAA;gBACpC,IAAI,CAAC,MAAM;oBAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,CAAA;gBAChC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAA;gBACjC,IAAI,IAAI,GAAG,EAAE,CAAA;gBACb,OAAO,IAAI,CAAC,MAAM,GAAG,QAAQ,GAAG,GAAG,EAAE,CAAC;oBACpC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAA;oBAC3C,IAAI,IAAI;wBAAE,MAAK;oBACf,IAAI,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAA;gBACjD,CAAC;gBACD,MAAM,MAAM,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA;gBACrC,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,EAAE,CAAA;YACjE,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAA;YACvC,CAAC;QACH,CAAC;KACF,CAAC,CAAA;IAEJ,0BAA0B,GAAG,CAAC,GAA6B,EAAE,EAAE,CAC7D,MAAM,CAAC;QACL,WAAW,EACT,0KAA0K;QAC5K,WAAW,EAAE,uBAAuB;QACpC,OAAO,EAAE,KAAK,EAAE,EAAE,gBAAgB,EAAE,qBAAqB,EAAE,EAAE,EAAE;YAC7D,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,QAAQ,CACvC,aAAa,gBAAgB,+BAA+B,qBAAqB,EAAE,EACnF;gBACE,MAAM,EAAE,wBAAwB;gBAChC,KAAK,EAAE,WAAW,CAAC,MAAM,CAAC,YAAY,IAAI,mBAAmB;gBAC7D,MAAM,EAAE,GAAG,CAAC,WAAW;aACxB,CACF,CAAA;YACD,OAAO,MAAM,CAAC,MAAM,CAAA;QACtB,CAAC;KACF,CAAC,CAAA;IAEJ,uBAAuB,GAAG,CAAC,QAAQ,GAAa,EAAE,EAAE,GAA6B,EAAW,EAAE;QAC5F,MAAM,KAAK,GAAY,EAAE,CAAA;QACzB,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,GAAG,CAAA;QAC3C,MAAM,SAAS,GAAG;YAChB,GAAG,EAAE,QAAQ,CAAC,GAAG;YACjB,MAAM,EAAE,QAAQ,CAAC,MAAM;YACvB,IAAI,EAAE,QAAQ,CAAC,IAAI;YACnB,IAAI,EAAE,QAAQ,CAAC,IAAI;SACpB,CAAA;QAED,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;YAC5B,IAAI,IAAI,KAAK,YAAY,EAAE,CAAC;gBAC1B,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,CAAA;YAChD,CAAC;iBAAM,IAAI,IAAI,KAAK,0BAA0B,EAAE,CAAC;gBAC/C,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,CAAA;YACpD,CAAC;iBAAM,CAAC;gBACN,MAAM,EAAE,GAAG,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;gBACjC,IAAI,EAAE,IAAI,EAAE,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;oBAC5B,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC;wBACnB,WAAW,EAAE,EAAE,CAAC,WAAW;wBAC3B,WAAW,EAAE,uBAAuB,CAAC,EAAE,CAAC,KAAK,CAAC;wBAC9C,OAAO,EAAE,CAAC,QAAQ,EAAE,EAAE,CACpB,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,IAAI,EAAE;4BACrD,MAAM,SAAS,GACb,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ;gCACtC,CAAC,CAAC,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,OAAO,CAAC,QAAmC,CAAC,CAAC,MAAM,CACxD,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,0BAA0B,CAAC,GAAG,CAAC,CAAC,CAAC,CAC5C,CACF;gCACH,CAAC,CAAC,QAAQ,CAAA;4BACd,OAAO,EAAE,CAAC,OAAO,CAAC,SAAS,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAA;wBAC/C,CAAC,CAAC;qBACL,CAAC,CAAA;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAA;IACd,CAAC,CAAA;IAED,0BAA0B,GAAG,KAAK,EAAE,KAAkB,EAAmB,EAAE;QACzE,MAAM,GAAG,GACP,OAAO,KAAK,CAAC,YAAY,KAAK,UAAU;YACtC,CAAC,CAAC,MAAM,KAAK,CAAC,YAAY,CAAC;gBACvB,KAAK,EAAE,qBAAqB;gBAC5B,MAAM,EAAE,oBAAoB;gBAC5B,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG;aACpB,CAAC;YACJ,CAAC,CAAC,KAAK,CAAC,YAAY,CAAA;QACxB,OAAO,GAAG;aACP,OAAO,CAAC,gCAAgC,EAAE,EAAE,CAAC;aAC7C,OAAO,CAAC,0CAA0C,EAAE,EAAE,CAAC;aACvD,UAAU,CAAC,WAAW,EAAE,qBAAqB,CAAC;aAC9C,UAAU,CAAC,YAAY,EAAE,oBAAoB,CAAC;aAC9C,IAAI,EAAE,CAAA;IACX,CAAC,CAAA;IAED,uBAAuB,GAAG,CAAC,KAAkB,EAAE,YAAsB,EAAE,UAAkB,EAAY,EAAE;QACrG,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAA;QACxC,MAAM,OAAO,GAAG,KAAK,CAAC,WAAW,CAAA;QACjC,IAAI,CAAC,OAAO;YAAE,OAAO,YAAY,CAAA;QACjC,IAAI,UAAU,KAAK,CAAC,IAAI,OAAO,CAAC,SAAS,EAAE,MAAM,IAAI,OAAO,CAAC,QAAQ,EAAE,MAAM,EAAE,CAAC;YAC9E,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,SAAS,EAAE,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAA;QAC1G,CAAC;QACD,IAAI,UAAU,GAAG,CAAC,IAAI,OAAO,CAAC,QAAQ,EAAE,MAAM,EAAE,CAAC;YAC/C,OAAO,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAA;QACjE,CAAC;QACD,OAAO,YAAY,CAAA;IACrB,CAAC,CAAA;IAED,MAAM,GAAG,KAAK,EACZ,KAAkB,EAClB,QAAwB,EACxB,MAAM,GAAY,IAAI,CAAC,KAAK,CAAC,MAAM,EACnC,IAAsB,EACkB,EAAE;QAC1C,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,QAAQ,CAAC,cAAc,CAAC,MAAM,CAAC,CAAA;QAC1D,MAAM,CAAC,EAAE,2BAA2B,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACxD,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,UAAU,CAAC,YAAY,CAAC,IAAI,CAAC;YAClD,IAAI,CAAC,0BAA0B,CAAC,KAAK,CAAC;SACvC,CAAC,CAAA;QAEF,MAAM,IAAI,GAAG,IAAI,EAAE,IAAI,EAAE,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QACnG,MAAM,YAAY,GAAG,IAAI,EAAE,IAAI,IAAI,IAAI,IAAI,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAA;QAClF,MAAM,KAAK,GAAG,IAAI,CAAC,uBAAuB,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE,EAAE;YAC5D,IAAI,EAAE,YAAY;YAClB,MAAM,EAAE,IAAI,EAAE,MAAM;YACpB,WAAW,EAAE,IAAI,EAAE,WAAW;SAC/B,CAAC,CAAA;QACF,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAEvC,MAAM,YAAY,GAAG,wBAAwB,CAAC;YAC5C,cAAc,EAAE,IAAI,EAAE,cAAc,EAAE,cAAc,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YAChF,gBAAgB,EAAE,IAAI,EAAE,cAAc,EAAE,gBAAgB;YACxD,sBAAsB,EAAE,IAAI,EAAE,cAAc,EAAE,sBAAsB;YACpE,yBAAyB,EAAE,IAAI,EAAE,cAAc,EAAE,yBAAyB;YAC1E,WAAW,EAAE,IAAI,EAAE,cAAc,EAAE,WAAW;SAC/C,CAAC,CAAA;QAEF,MAAM,kBAAkB,GAAyB;YAC/C;gBACE,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,2BAA2B;gBACpC,eAAe,EAAE;oBACf,SAAS,EAAE,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE;oBAClD,MAAM,EAAE,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE;iBAChD;aACF;YACD;gBACE,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,YAAY;aACtB;SACF,CAAA;QAED,MAAM,cAAc,GAAmB,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAA;QAClF,MAAM,aAAa,GAAG,IAAI,EAAE,aAAa,IAAI,KAAK,CAAC,KAAK,IAAI,WAAW,CAAC,MAAM,CAAC,MAAM,CAAA;QAErF,OAAO,UAAU,CAAC;YAChB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,aAAa,CAAC,aAAa,CAAC;YACjD,MAAM,EAAE,kBAAkB;YAC1B,QAAQ,EAAE,cAAc;YACxB,KAAK;YACL,GAAG,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC;gBACzB,CAAC,CAAC;oBACE,WAAW,EAAE,YAAY;oBACzB,WAAW,EAAE,CAAC,EAAE,UAAU,EAAE,QAAQ,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC;wBACxD,WAAW,EAAE,IAAI,CAAC,uBAAuB,CAAC,KAAK,EAAE,YAAY,EAAE,UAAU,CAAC;wBAC1E,GAAG,CAAC,UAAU,GAAG,CAAC,IAAI,YAAY,CAAC,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC,iBAAiB;4BAC9E,CAAC,CAAC,EAAE,QAAQ,EAAE,YAAY,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC,iBAAiB,CAAC,EAAE;4BACzE,CAAC,CAAC,EAAE,CAAC;qBACR,CAAC;iBACH;gBACH,CAAC,CAAC,EAAE,CAAC;YACP,QAAQ,EAAE,WAAW,CAAC,KAAK,CAAC,QAAQ,IAAI,WAAW,CAAC,MAAM,CAAC,oBAAoB,CAAC;YAChF,GAAG,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC/D,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;gBACxB,MAAM,QAAQ,GAAI,KAA+K,CAAC,UAAU,IAAI,KAAK,CAAC,KAAK,CAAA;gBAC3N,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,QAAQ,CAAC,YAAY,CAAC;oBAC/C,MAAM;oBACN,KAAK,EAAE,aAAa;oBACpB,WAAW,EAAE,QAAQ,EAAE,WAAW,IAAI,CAAC;oBACvC,YAAY,EAAE,QAAQ,EAAE,YAAY,IAAI,CAAC;oBACzC,GAAG,CAAC,OAAO,QAAQ,EAAE,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC3F,MAAM,EAAE,MAAM;iBACf,CAAC,CAAA;gBACF,MAAM,IAAI,EAAE,QAAQ,EAAE,CAAC,KAAK,CAAC,CAAA;YAC/B,CAAC;SACF,CAAC,CAAA;IACJ,CAAC,CAAA;IAED,GAAG,GAAG,KAAK,EACT,SAA+B,EAC/B,MAA+B,EAC/B,OAAyB,EAC4B,EAAE;QACvD,MAAM,KAAK,GAAG,OAAO,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAA;QACtF,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,kBAAkB,MAAM,CAAC,SAAS,CAAC,EAAE,EAAE,WAAW,CAAC,CAAA;QAEvG,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,WAAW,CAAC,IAAI,CAAC,WAAW,CAAA;QACnF,MAAM,WAAW,GACf,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAA;QAC1G,MAAM,YAAY,GAAG,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,uBAAuB,CAAC,WAAW,CAAC,CAAA;QAE/F,MAAM,MAAM,GACV,OAAO,EAAE,cAAc,KAAK,KAAK;YAC/B,CAAC,CAAC,MAAM,IAAI,CAAC,eAAe,CAAC;gBACzB,MAAM;gBACN,MAAM,EAAE,YAAY;gBACpB,UAAU,EAAE,KAAK,CAAC,UAAU,KAAK,KAAK;gBACtC,KAAK;gBACL,WAAW,EAAE,OAAO,EAAE,WAAW;aAClC,CAAC;YACJ,CAAC,CAAC,EAAE,gBAAgB,EAAE,EAAE,EAAE,sBAAsB,EAAE,EAAE,EAAE,WAAW,EAAE,SAAS,EAAE,CAAA;QAElF,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE;YACpE,WAAW,EAAE,OAAO,EAAE,WAAW;YACjC,cAAc,EAAE;gBACd,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;gBACzC,sBAAsB,EAAE,MAAM,CAAC,sBAAsB;gBACrD,WAAW,EAAE,MAAM,CAAC,WAAW;aAChC;SACF,CAAC,CAAA;QACF,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAA;QACjE,OAAO;YACL,IAAI;YACJ,QAAQ,EAAE,CAAC,GAAG,WAAW,EAAE,GAAI,GAAG,CAAC,QAAsC,CAAC;SAC3E,CAAA;IACH,CAAC,CAAA;IAED,IAAI,SAAS;QACX,OAAO;YACL,oBAAoB,EAAE,IAAI,CAAC,qBAAqB;YAChD,kBAAkB,EAAE,IAAI,CAAC,mBAAmB;YAC5C,WAAW,EAAE,IAAI,CAAC,YAAY;YAC9B,cAAc,EAAE,IAAI,CAAC,eAAe;YACpC,mBAAmB,EAAE,IAAI,CAAC,oBAAoB;YAC9C,gBAAgB,EAAE,IAAI,CAAC,iBAAiB;YACxC,sBAAsB,EAAE,IAAI,CAAC,cAAc,CAAC,sBAAsB;YAClE,kBAAkB,EAAE,IAAI,CAAC,aAAa,CAAC,kBAAkB;YACzD,uBAAuB,EAAE,IAAI,CAAC,aAAa,CAAC,uBAAuB;YACnE,mBAAmB,EAAE,IAAI,CAAC,aAAa,CAAC,mBAAmB;YAC3D,QAAQ,EAAE,IAAI,CAAC,aAAa,CAAC,QAAQ;YACrC,aAAa,EAAE,IAAI,CAAC,aAAa,CAAC,aAAa;SAChD,CAAA;IACH,CAAC;CACF"}

package/dist/chat/stream.d.ts

@@ -0,0 +1,53 @@
+import { type ModelMessage } from 'ai';
+import type { Context } from 'hono';
+import type { z } from 'zod';
+import type { AttachmentTypesMap, ProcessedAttachment } from '../attachments';
+import { type AppEnv, type DbRole } from '../config';
+import { type EntryId, type SelectModel } from '../db';
+import type { FileRef } from '../files';
+import { Core } from '../index';
+import { saveChatBody, streamChatBody, type AgentConfig, type ChatMessagePublic, type ListChatMessagesOptions, type RawAttachmentPart } from './schema';
+export declare class ChatStreamEngine {
+    #private;
+    constructor(coreClient: Core, agentRegistry: Map<string, AgentConfig>);
+    enforceAgentAccess: (c: Context<AppEnv>, agent: AgentConfig, scopeKey: string) => Promise<void>;
+    stageAndScanAttachments: (userId: EntryId, parts: RawAttachmentPart[], types?: AttachmentTypesMap) => Promise<{
+        staged: {
+            ref: FileRef;
+            filename: string;
+        }[];
+        attachments: ProcessedAttachment[];
+    }>;
+    saveChat: ({ userId, input }: {
+        userId?: EntryId;
+        input: z.infer<typeof saveChatBody>;
+    }) => Promise<SelectModel<'chat'>>;
+    messages: (chatId: EntryId, opts?: ListChatMessagesOptions) => Promise<ChatMessagePublic[]>;
+    cancel: (chatId: EntryId) => Promise<{
+        chatId: string;
+        cancelled: true;
+    }>;
+    startChatStreamTurn: (opts: {
+        userId: EntryId;
+        role: DbRole;
+        agent: AgentConfig;
+        attachmentParts: RawAttachmentPart[];
+        rawUserText: string;
+        signal: AbortSignal;
+        chatId?: string;
+        messageCount?: number;
+        selectedModel?: string;
+        priorMessages?: ModelMessage[];
+    }) => Promise<{
+        turn: import("ai").StreamTextResult<import("ai").ToolSet, import("@ai-sdk/provider-utils").Context, import("ai").OutputInterface<any, any, any>>;
+        stagedRefs: string[];
+        stagedAttachmentSummaries: string[];
+        userCreatedAtMs: number;
+    }>;
+    streamChatSse: ({ c, userId, input }: {
+        c: Context<AppEnv>;
+        userId?: EntryId;
+        input?: z.infer<typeof streamChatBody>;
+    }) => Promise<Response>;
+}
+//# sourceMappingURL=stream.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"stream.d.ts","sourceRoot":"","sources":["../../src/chat/stream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,YAAY,EAAmB,MAAM,IAAI,CAAA;AAEvD,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,MAAM,CAAA;AACnC,OAAO,KAAK,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAC5B,OAAO,KAAK,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAA;AAC7E,OAAO,EAAe,KAAK,MAAM,EAAE,KAAK,MAAM,EAAE,MAAM,WAAW,CAAA;AACjE,OAAO,EAAW,KAAK,OAAO,EAAE,KAAK,WAAW,EAAE,MAAM,OAAO,CAAA;AAC/D,OAAO,KAAK,EAAE,OAAO,EAAY,MAAM,UAAU,CAAA;AACjD,OAAO,EAAE,IAAI,EAAE,MAAM,UAAU,CAAA;AAC/B,OAAO,EACL,YAAY,EACZ,cAAc,EACd,KAAK,WAAW,EAChB,KAAK,iBAAiB,EACtB,KAAK,uBAAuB,EAE5B,KAAK,iBAAiB,EACvB,MAAM,UAAU,CAAA;AAuBjB,qBAAa,gBAAgB;;IAI3B,YAAY,UAAU,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,CAAC,MAAM,EAAE,WAAW,CAAC,EAGpE;IAaD,kBAAkB,MAAa,OAAO,CAAC,MAAM,CAAC,SAAS,WAAW,YAAY,MAAM,KAAG,OAAO,CAAC,IAAI,CAAC,CAOnG;IAED,uBAAuB,WACb,OAAO,SACR,iBAAiB,EAAE,UAClB,kBAAkB,KACzB,OAAO,CAAC;QAAE,MAAM,EAAE;YAAE,GAAG,EAAE,OAAO,CAAC;YAAC,QAAQ,EAAE,MAAM,CAAA;SAAE,EAAE,CAAC;QAAC,WAAW,EAAE,mBAAmB,EAAE,CAAA;KAAE,CAAC,CAgC/F;IAED,QAAQ,sBAGL;QACD,MAAM,CAAC,EAAE,OAAO,CAAA;QAChB,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,YAAY,CAAC,CAAA;KACpC,KAAG,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAqC/B;IAED,QAAQ,WAAkB,OAAO,SAAS,uBAAuB,KAAG,OAAO,CAAC,iBAAiB,EAAE,CAAC,CAiB/F;IAED,MAAM,WAAkB,OAAO,KAAG,OAAO,CAAC;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,IAAI,CAAA;KAAE,CAAC,CAe7E;IAqID,mBAAmB,SAAgB;QACjC,MAAM,EAAE,OAAO,CAAA;QACf,IAAI,EAAE,MAAM,CAAA;QACZ,KAAK,EAAE,WAAW,CAAA;QAClB,eAAe,EAAE,iBAAiB,EAAE,CAAA;QACpC,WAAW,EAAE,MAAM,CAAA;QACnB,MAAM,EAAE,WAAW,CAAA;QACnB,MAAM,CAAC,EAAE,MAAM,CAAA;QACf,YAAY,CAAC,EAAE,MAAM,CAAA;QACrB,aAAa,CAAC,EAAE,MAAM,CAAA;QACtB,aAAa,CAAC,EAAE,YAAY,EAAE,CAAA;KAC/B;;;;;OAkDA;IAED,aAAa,yBAIV;QACD,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAA;QAClB,MAAM,CAAC,EAAE,OAAO,CAAA;QAChB,KAAK,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,cAAc,CAAC,CAAA;KACvC,KAAG,OAAO,CAAC,QAAQ,CAAC,CAyFpB;CACF"}