@kognitivedev/vercel-ai-provider 0.2.21 → 0.2.26

package/dist/index.js

@@ -1,4 +1,11 @@
 "use strict";
+var __asyncValues = (this && this.__asyncValues) || function (o) {
+    if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
+    var m = o[Symbol.asyncIterator], i;
+    return m ? m.call(o) : (o = typeof __values === "function" ? __values(o) : o[Symbol.iterator](), i = {}, verb("next"), verb("throw"), verb("return"), i[Symbol.asyncIterator] = function () { return this; }, i);
+    function verb(n) { i[n] = o[n] && function (v) { return new Promise(function (resolve, reject) { v = o[n](v), settle(resolve, reject, v.done, v.value); }); }; }
+    function settle(resolve, reject, d, v) { Promise.resolve(v).then(function(v) { resolve({ value: v, done: d }); }, reject); }
+};
 var __rest = (this && this.__rest) || function (s, e) {
     var t = {};
     for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
@@ -12,9 +19,13 @@ var __rest = (this && this.__rest) || function (s, e) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.renderTemplate = void 0;
+exports.toMessageEventStream = toMessageEventStream;
+exports.toGeneratedMessageEventStream = toGeneratedMessageEventStream;
 exports.createCognitiveLayer = createCognitiveLayer;
 const ai_1 = require("ai");
-const crypto_1 = require("crypto");
+const memory_1 = require("@kognitivedev/memory");
+const tools_1 = require("@kognitivedev/tools");
+const shared_1 = require("@kognitivedev/shared");
 const prompthub_1 = require("@kognitivedev/prompthub");
 var template_1 = require("./template");
 Object.defineProperty(exports, "renderTemplate", { enumerable: true, get: function () { return template_1.renderTemplate; } });
@@ -72,30 +83,6 @@ function createLogger(logLevel) {
         },
     };
 }
-function getContentText(content) {
-    if (typeof content === "string") {
-        if (content.includes("data:image/") && content.includes("base64,"))
-            return "[Image]";
-        return content;
-    }
-    if (!Array.isArray(content))
-        return "";
-    return content.map((part) => {
-        if (!part || typeof part !== "object")
-            return "";
-        if (typeof part.text === "string")
-            return part.text;
-        if (part.type === "tool-call" && typeof part.toolName === "string")
-            return `Called ${part.toolName}`;
-        if (part.type === "tool-result")
-            return "Received tool result";
-        if (part.type === "image" || part.type === "image_url")
-            return "[Image]";
-        if (part.type === "file")
-            return "[File]";
-        return "";
-    }).filter(Boolean).join(" ");
-}
 /**
  * Unwraps V2/V3 ToolResultOutput discriminated union to a displayable value.
  * Stream ToolResult uses plain `result` (passthrough), while prompt ToolResultPart
@@ -123,14 +110,6 @@ function extractOutputValue(raw) {
             return raw;
     }
 }
-function buildTracePreviews(messages) {
-    const request = [...messages].reverse().find((message) => (message === null || message === void 0 ? void 0 : message.role) === "user");
-    const response = [...messages].reverse().find((message) => (message === null || message === void 0 ? void 0 : message.role) === "assistant");
-    return {
-        requestPreview: request ? getContentText(request.content).slice(0, 220) : "No request captured",
-        responsePreview: response ? getContentText(response.content).slice(0, 240) : "No response captured",
-    };
-}
 function buildTraceSpansFromMessages(messages) {
     var _a, _b;
     const resultMap = new Map();
@@ -191,6 +170,143 @@ const MEMORY_TAG_REGEX = /<MemoryContext>/i;
 const SESSION_KEY = Symbol.for("cl:session");
 // Session key → prompt metadata (populated by cl.streamText/cl.generateText, read by middleware)
 const sessionPromptMetadata = new Map();
+function toAISDKTopicMemoryTool(userId, baseUrl, apiKey) {
+    const tool = (0, tools_1.createTopicMemoryTool)({ apiKey, baseUrl });
+    return {
+        description: tool.description,
+        inputSchema: tool.inputSchema,
+        execute: async (input) => {
+            const result = await tool.execute(input, {
+                abortSignal: new AbortController().signal,
+                resourceId: { userId },
+                metadata: undefined,
+                emit: () => { },
+            });
+            return tool.toModelOutput ? tool.toModelOutput(result) : result;
+        },
+    };
+}
+function withAutoInjectedTools(params, session, baseUrl, apiKey, options) {
+    var _a;
+    if (!(options === null || options === void 0 ? void 0 : options.autoTopicMemoryTool) || !isValidId(session === null || session === void 0 ? void 0 : session.userId)) {
+        return params;
+    }
+    const existingTools = params.tools;
+    if (existingTools && (typeof existingTools !== "object" || Array.isArray(existingTools))) {
+        return params;
+    }
+    const toolSet = Object.assign({}, ((_a = existingTools) !== null && _a !== void 0 ? _a : {}));
+    if (tools_1.TOPIC_MEMORY_TOOL_ID in toolSet) {
+        return params;
+    }
+    toolSet[tools_1.TOPIC_MEMORY_TOOL_ID] = toAISDKTopicMemoryTool(session.userId, baseUrl, apiKey);
+    return Object.assign(Object.assign({}, params), { tools: toolSet });
+}
+function toMessageEventStream(result) {
+    return new ReadableStream({
+        async start(controller) {
+            var _a, e_1, _b, _c;
+            try {
+                try {
+                    for (var _d = true, _e = __asyncValues(result.fullStream), _f; _f = await _e.next(), _a = _f.done, !_a; _d = true) {
+                        _c = _f.value;
+                        _d = false;
+                        const chunk = _c;
+                        if (chunk.type === "text-delta" && chunk.delta) {
+                            controller.enqueue({
+                                event: "messages",
+                                data: { token: chunk.delta },
+                            });
+                            continue;
+                        }
+                        if (chunk.type === "tool-call") {
+                            controller.enqueue({
+                                event: "messages",
+                                data: {
+                                    type: "tool-call",
+                                    id: chunk.toolCallId,
+                                    name: chunk.toolName,
+                                    input: chunk.input,
+                                },
+                            });
+                            continue;
+                        }
+                        if (chunk.type === "tool-result") {
+                            controller.enqueue({
+                                event: "messages",
+                                data: {
+                                    type: "tool-result",
+                                    id: chunk.toolCallId,
+                                    name: chunk.toolName,
+                                    result: chunk.result,
+                                },
+                            });
+                            continue;
+                        }
+                        if (chunk.type === "error") {
+                            controller.enqueue({
+                                event: "debug",
+                                data: {
+                                    type: "error",
+                                    message: chunk.error instanceof Error ? chunk.error.message : String(chunk.error),
+                                },
+                            });
+                        }
+                    }
+                }
+                catch (e_1_1) { e_1 = { error: e_1_1 }; }
+                finally {
+                    try {
+                        if (!_d && !_a && (_b = _e.return)) await _b.call(_e);
+                    }
+                    finally { if (e_1) throw e_1.error; }
+                }
+            }
+            finally {
+                controller.close();
+            }
+        },
+    });
+}
+function toGeneratedMessageEventStream(result) {
+    return new ReadableStream({
+        start(controller) {
+            var _a, _b, _c, _d, _e, _f;
+            const steps = ((_a = result.steps) === null || _a === void 0 ? void 0 : _a.length)
+                ? result.steps
+                : [{
+                        toolCalls: result.toolCalls,
+                        toolResults: result.toolResults,
+                    }];
+            for (const step of steps) {
+                for (const toolCall of (_b = step.toolCalls) !== null && _b !== void 0 ? _b : []) {
+                    controller.enqueue({
+                        event: "messages",
+                        data: {
+                            type: "tool-call",
+                            id: toolCall.toolCallId,
+                            name: (_c = toolCall.toolName) !== null && _c !== void 0 ? _c : "tool",
+                            input: toolCall.input,
+                        },
+                    });
+                }
+                for (const toolResult of (_d = step.toolResults) !== null && _d !== void 0 ? _d : []) {
+                    controller.enqueue({
+                        event: "messages",
+                        data: Object.assign({ type: "tool-result", id: toolResult.toolCallId, name: (_e = toolResult.toolName) !== null && _e !== void 0 ? _e : "tool", result: (_f = toolResult.output) !== null && _f !== void 0 ? _f : toolResult.result }, (toolResult.isError ? { isError: true } : {})),
+                    });
+                }
+            }
+            if (result.text) {
+                controller.enqueue({
+                    event: "messages",
+                    data: { token: result.text },
+                });
+            }
+            controller.close();
+        },
+    });
+}
 /**
  * Check if any system message already contains a <MemoryContext> block.
  */
@@ -215,6 +331,11 @@ function createCognitiveLayer(config) {
     // Default to 'info' log level
     const logLevel = clConfig.logLevel || 'info';
     const logger = createLogger(logLevel);
+    const memoryClient = new memory_1.MemoryClient({
+        apiKey: clConfig.apiKey,
+        baseUrl,
+        logger: logger,
+    });
     const authHeaders = {
         "Content-Type": "application/json",
         "Authorization": `Bearer ${clConfig.apiKey}`,
@@ -224,6 +345,17 @@ function createCognitiveLayer(config) {
         apiKey: clConfig.apiKey,
         logger,
     });
+    const sessionTurnIndexes = new Map();
+    const reserveTurnIndex = (agentName, userId, sessionId, requestedRunScope) => {
+        var _a;
+        if (!isValidId(userId) || !isValidId(sessionId) || !(0, shared_1.isSessionScopedExecution)(requestedRunScope, sessionId ? "session" : "invocation")) {
+            return undefined;
+        }
+        const key = `${agentName}:${getSessionKey(userId, sessionId)}`;
+        const nextIndex = (_a = sessionTurnIndexes.get(key)) !== null && _a !== void 0 ? _a : 0;
+        sessionTurnIndexes.set(key, nextIndex + 1);
+        return nextIndex;
+    };
     const resolvePrompt = async (slug, userIdOrOptions) => {
         var _a;
         const userId = typeof userIdOrOptions === "string"
@@ -263,23 +395,73 @@ function createCognitiveLayer(config) {
     };
     const logConversation = async (payload) => {
         try {
-            await fetch(`${baseUrl}/api/cognitive/log`, {
+            const response = await fetch(`${baseUrl}/api/cognitive/log`, {
                 method: "POST",
                 headers: authHeaders,
                 body: JSON.stringify(Object.assign(Object.assign({}, payload), { type: "conversation", timestamp: new Date().toISOString() })),
             });
+            if (!response.ok) {
+                logger.warn("Log failed", { status: response.status });
+                return null;
+            }
+            const data = await response.json().catch(() => null);
+            return {
+                generatedTitle: typeof (data === null || data === void 0 ? void 0 : data.generatedTitle) === "string" ? data.generatedTitle : null,
+            };
         }
         catch (e) {
             logger.error("Log failed", e);
+            return null;
         }
     };
-    const triggerProcessing = (userId, sessionId) => {
-        const run = () => {
-            fetch(`${baseUrl}/api/cognitive/process`, {
+    const postAgentRun = async (payload) => {
+        try {
+            const response = await fetch(`${baseUrl}/api/cognitive/agent-run`, {
                 method: "POST",
                 headers: authHeaders,
-                body: JSON.stringify({ userId, sessionId }),
-            }).catch(e => logger.error("Process trigger failed", e));
+                body: JSON.stringify(payload),
+            });
+            if (!response.ok) {
+                logger.warn("Agent run request failed", { status: response.status });
+                return null;
+            }
+            return await response.json().catch(() => null);
+        }
+        catch (e) {
+            logger.error("Agent run request failed", e);
+            return null;
+        }
+    };
+    const postTraceEvents = async (payload) => {
+        try {
+            const response = await fetch(`${baseUrl}/api/cognitive/trace-events`, {
+                method: "POST",
+                headers: authHeaders,
+                body: JSON.stringify(payload),
+            });
+            if (!response.ok) {
+                logger.warn("Trace events request failed", { status: response.status });
+                return null;
+            }
+            return await response.json().catch(() => null);
+        }
+        catch (e) {
+            logger.error("Trace events request failed", e);
+            return null;
+        }
+    };
+    const appendLiveTraceEvents = async (payload) => {
+        await postTraceEvents({
+            traceDbId: payload.traceDbId,
+            traceId: payload.traceId,
+            sessionDbId: payload.sessionDbId,
+            events: payload.events,
+        });
+    };
+    const triggerProcessing = (userId, sessionId) => {
+        const run = () => {
+            memoryClient.triggerProcessing(userId, sessionId)
+                .catch((e) => logger.error("Process trigger failed", e));
         };
         if (processDelay > 0) {
             setTimeout(run, processDelay);
@@ -304,6 +486,30 @@ function createCognitiveLayer(config) {
         const updated = [{ role: "system", content: memoryPrompt }, ...incomingMessages];
         return { nextParams, messages: updated, mode: "prepend-system" };
     };
+    const getKognitiveProviderConfig = (params) => {
+        var _a, _b;
+        const providerOptionsKognitive = (_a = params === null || params === void 0 ? void 0 : params.providerOptions) === null || _a === void 0 ? void 0 : _a.kognitive;
+        if (providerOptionsKognitive && typeof providerOptionsKognitive === "object") {
+            return providerOptionsKognitive;
+        }
+        const providerMetadataKognitive = (_b = params === null || params === void 0 ? void 0 : params.providerMetadata) === null || _b === void 0 ? void 0 : _b.kognitive;
+        if (providerMetadataKognitive && typeof providerMetadataKognitive === "object") {
+            return providerMetadataKognitive;
+        }
+        return undefined;
+    };
+    const resolveRequestedRunScope = (params) => {
+        const direct = typeof (params === null || params === void 0 ? void 0 : params.runScope) === "string" ? params.runScope : undefined;
+        const kognitiveConfig = getKognitiveProviderConfig(params);
+        const providerMetadataScope = typeof (kognitiveConfig === null || kognitiveConfig === void 0 ? void 0 : kognitiveConfig.runScope) === "string"
+            ? kognitiveConfig.runScope
+            : undefined;
+        return direct !== null && direct !== void 0 ? direct : providerMetadataScope;
+    };
+    const resolveAutomaticTitleConfig = (params) => {
+        var _a;
+        return (0, shared_1.normalizeAutomaticThreadTitleConfig)((_a = getKognitiveProviderConfig(params)) === null || _a === void 0 ? void 0 : _a.automaticTitle);
+    };
     const buildMiddleware = (userId, projectId, sessionId, modelId) => ({
         specificationVersion: 'v3',
         async transformParams({ params }) {
@@ -323,67 +529,36 @@ function createCognitiveLayer(config) {
             // 3) Fetch snapshot only if not cached
             if (systemPromptToAdd === undefined) {
                 try {
-                    const url = `${baseUrl}/api/cognitive/snapshot?userId=${userId}`;
                     logger.debug("Fetching snapshot from backend", {
                         userId,
                         sessionId,
-                        url,
+                        url: `${baseUrl}/api/cognitive/snapshot?userId=${userId}`,
                         baseUrl,
                         apiKeyHint: maskSecret(clConfig.apiKey),
                     });
-                    const res = await fetch(url, {
-                        headers: { "Authorization": `Bearer ${clConfig.apiKey}` },
-                    });
-                    logger.debug("Snapshot response received", {
+                    const snapshot = await memoryClient.getSnapshot(userId);
+                    const systemBlock = (snapshot === null || snapshot === void 0 ? void 0 : snapshot.systemBlock) || "";
+                    const userContextBlock = (snapshot === null || snapshot === void 0 ? void 0 : snapshot.userContextBlock) || "";
+                    const topicIndexBlock = (snapshot === null || snapshot === void 0 ? void 0 : snapshot.topicIndexBlock) || "";
+                    const topicContextBlock = (snapshot === null || snapshot === void 0 ? void 0 : snapshot.topicContextBlock) || "";
+                    systemPromptToAdd = snapshot ? memoryClient.buildMemoryBlock(snapshot) : "";
+                    sessionSnapshots.set(sessionKey, systemPromptToAdd);
+                    logger.info("Snapshot fetched and cached", {
                         userId,
                         sessionId,
-                        status: res.status,
-                        ok: res.ok,
-                        contentType: res.headers.get("content-type"),
+                        sessionKey,
+                        systemLen: systemBlock.length,
+                        userLen: userContextBlock.length,
+                        topicIndexLen: topicIndexBlock.length,
+                        topicContextLen: topicContextBlock.length,
+                    });
+                    logger.debug("Full snapshot data", {
+                        systemBlock,
+                        userContextBlock,
+                        topicIndexBlock,
+                        topicContextBlock,
+                        rawData: snapshot,
                     });
-                    if (res.ok) {
-                        const data = await res.json();
-                        const systemBlock = data.systemBlock || "";
-                        const userContextBlock = data.userContextBlock || "";
-                        systemPromptToAdd =
-                            systemBlock !== "" || userContextBlock !== ""
-                                ? `
-<MemoryContext>
-Use the following memory to stay consistent. Prefer UserContext facts for answers; AgentHeuristics guide style, safety, and priorities.
-${systemBlock || "None"}
-${userContextBlock || "None"}
-</MemoryContext>
-                            `.trim()
-                                : "";
-                        // Cache the snapshot for this session
-                        sessionSnapshots.set(sessionKey, systemPromptToAdd);
-                        logger.info("Snapshot fetched and cached", {
-                            userId,
-                            sessionId,
-                            sessionKey,
-                            systemLen: systemBlock.length,
-                            userLen: userContextBlock.length,
-                        });
-                        // At debug level, log the full snapshot data
-                        logger.debug("Full snapshot data", {
-                            systemBlock,
-                            userContextBlock,
-                            rawData: data,
-                        });
-                    }
-                    else {
-                        const body = await res.text();
-                        logger.warn("Snapshot fetch failed", { status: res.status });
-                        logger.debug("Snapshot response body preview", {
-                            userId,
-                            projectId,
-                            sessionId,
-                            status: res.status,
-                            bodyPreview: previewText(body),
-                        });
-                        systemPromptToAdd = "";
-                        sessionSnapshots.set(sessionKey, systemPromptToAdd);
-                    }
                 }
                 catch (e) {
                     logger.warn("Failed to fetch snapshot", e);
@@ -408,20 +583,96 @@ ${userContextBlock || "None"}
         async wrapGenerate({ doGenerate, params }) {
             var _a, _b, _c;
             const startedAt = new Date();
+            const requestedRunScope = resolveRequestedRunScope(params);
+            const remoteAgentName = (_a = clConfig.appId) !== null && _a !== void 0 ? _a : modelId;
+            const turnIndex = reserveTurnIndex(remoteAgentName, userId, sessionId, requestedRunScope);
+            const remoteExecution = (0, shared_1.createRemoteExecutionContext)({
+                agentName: remoteAgentName,
+                sessionId,
+                requestedRunScope,
+                runId: (_b = getKognitiveProviderConfig(params)) === null || _b === void 0 ? void 0 : _b.agentRunId,
+                turnIndex,
+            });
+            const traceId = remoteExecution.traceId;
+            let liveTraceDbId;
+            if (isValidId(userId) && isValidId(sessionId)) {
+                const messagesInput = params.prompt || params.messages || [];
+                const { requestPreview } = (0, shared_1.buildRemoteTracePreviews)(messagesInput);
+                await postAgentRun((0, shared_1.buildRemoteRunPayload)({
+                    execution: remoteExecution,
+                    userId,
+                    sessionId,
+                    modelId,
+                    status: "running",
+                    inputPreview: requestPreview,
+                    startedAt: startedAt.toISOString(),
+                    metadata: {
+                        appId: clConfig.appId,
+                    },
+                    sessionMetadata: {
+                        kind: "chat",
+                        agentName: remoteExecution.agentName,
+                    },
+                }));
+            }
+            if (isValidId(userId) && isValidId(sessionId) && isValidId(projectId)) {
+                const messagesInput = params.prompt || params.messages || [];
+                const { requestPreview } = (0, shared_1.buildRemoteTracePreviews)(messagesInput);
+                const traceResponse = await postTraceEvents((0, shared_1.buildRemoteTraceStartPayload)({
+                    execution: remoteExecution,
+                    userId,
+                    projectId,
+                    sessionId,
+                    requestPreview,
+                    modelId,
+                    metadata: {
+                        appId: clConfig.appId,
+                    },
+                    startedAt: startedAt.toISOString(),
+                }));
+                liveTraceDbId = typeof (traceResponse === null || traceResponse === void 0 ? void 0 : traceResponse.traceDbId) === "string" ? traceResponse.traceDbId : undefined;
+            }
             let result;
             try {
                 result = await doGenerate();
             }
             catch (err) {
+                const errorMessage = err instanceof Error ? err.message : String(err);
+                if (liveTraceDbId) {
+                    await postTraceEvents((0, shared_1.buildRemoteTraceFinishPayload)({
+                        execution: remoteExecution,
+                        traceDbId: liveTraceDbId,
+                        state: "error",
+                        responsePreview: errorMessage.slice(0, 240),
+                        durationMs: Date.now() - startedAt.getTime(),
+                        errorMessage,
+                    }));
+                }
+                if (isValidId(userId) && isValidId(sessionId)) {
+                    await postAgentRun((0, shared_1.buildRemoteRunPayload)({
+                        execution: remoteExecution,
+                        userId,
+                        sessionId,
+                        modelId,
+                        status: "failed",
+                        errorMessage,
+                        durationMs: Date.now() - startedAt.getTime(),
+                        startedAt: startedAt.toISOString(),
+                        completedAt: new Date().toISOString(),
+                        metadata: {
+                            appId: clConfig.appId,
+                        },
+                    }));
+                }
                 logger.error("doGenerate failed", err);
-                logger.error("doGenerate params.prompt", JSON.stringify((_a = params.prompt) === null || _a === void 0 ? void 0 : _a.map((m) => ({ role: m.role, contentType: typeof m.content, contentLength: Array.isArray(m.content) ? m.content.length : undefined })), null, 2));
+                logger.error("doGenerate params.prompt", JSON.stringify((_c = params.prompt) === null || _c === void 0 ? void 0 : _c.map((m) => ({ role: m.role, contentType: typeof m.content, contentLength: Array.isArray(m.content) ? m.content.length : undefined })), null, 2));
                 throw err;
             }
             if (isValidId(userId) && isValidId(sessionId)) {
                 const endedAt = new Date();
                 const sessionKey = getSessionKey(userId, sessionId);
                 const promptMeta = sessionPromptMetadata.get(sessionKey);
-                const agentRunId = (_c = (_b = params === null || params === void 0 ? void 0 : params.providerMetadata) === null || _b === void 0 ? void 0 : _b.kognitive) === null || _c === void 0 ? void 0 : _c.agentRunId;
+                const automaticTitle = resolveAutomaticTitleConfig(params);
                 const messagesInput = params.prompt || params.messages || [];
                 // Build assistant message from result.content (V2/V3 GenerateResult)
                 const resultContent = Array.isArray(result === null || result === void 0 ? void 0 : result.content) ? result.content : [];
@@ -451,10 +702,31 @@ ${userContextBlock || "None"}
                     ? [{ role: "assistant", content: assistantParts }]
                     : [];
                 const finalMessages = [...messagesInput, ...assistantMessage];
-                const { requestPreview, responsePreview } = buildTracePreviews(finalMessages);
+                const { requestPreview, responsePreview } = (0, shared_1.buildRemoteTracePreviews)(finalMessages);
                 const spans = buildTraceSpansFromMessages(finalMessages);
                 const toolDefs = extractToolDefinitions(params);
-                logConversation(Object.assign(Object.assign(Object.assign(Object.assign({ userId,
+                const usage = (0, shared_1.normalizeRemoteUsage)(result.usage);
+                await postAgentRun((0, shared_1.buildRemoteRunPayload)({
+                    execution: remoteExecution,
+                    userId,
+                    sessionId,
+                    modelId,
+                    status: "completed",
+                    inputPreview: requestPreview,
+                    outputPreview: responsePreview,
+                    inputTokens: usage.inputTokens,
+                    outputTokens: usage.outputTokens,
+                    cachedInputTokens: usage.cachedInputTokens,
+                    durationMs: endedAt.getTime() - startedAt.getTime(),
+                    startedAt: startedAt.toISOString(),
+                    completedAt: endedAt.toISOString(),
+                    metadata: Object.assign(Object.assign(Object.assign({ appId: clConfig.appId }, ((promptMeta === null || promptMeta === void 0 ? void 0 : promptMeta.tag) && { promptTag: promptMeta.tag })), ((promptMeta === null || promptMeta === void 0 ? void 0 : promptMeta.abTestId) && { abTestId: promptMeta.abTestId })), ((promptMeta === null || promptMeta === void 0 ? void 0 : promptMeta.variant) && { variant: promptMeta.variant })),
+                    sessionMetadata: {
+                        kind: "chat",
+                        agentName: remoteExecution.agentName,
+                    },
+                }));
+                const logResult = await logConversation((0, shared_1.buildRemoteLogPayload)(Object.assign(Object.assign(Object.assign({ execution: remoteExecution, userId,
                     sessionId, messages: finalMessages, modelId, usage: result.usage }, (promptMeta && {
                     promptSlug: promptMeta.promptSlug,
                     promptVersion: promptMeta.promptVersion,
@@ -462,21 +734,107 @@ ${userContextBlock || "None"}
                     tag: promptMeta.tag,
                     abTestId: promptMeta.abTestId,
                     variant: promptMeta.variant,
-                })), (toolDefs && { tools: toolDefs })), (agentRunId && { agentRunId })), { traceId: (0, crypto_1.randomUUID)(), requestPreview,
-                    responsePreview, state: "completed", startedAt: startedAt.toISOString(), endedAt: endedAt.toISOString(), durationMs: endedAt.getTime() - startedAt.getTime(), metadata: Object.assign(Object.assign(Object.assign({ appId: clConfig.appId }, ((promptMeta === null || promptMeta === void 0 ? void 0 : promptMeta.tag) && { promptTag: promptMeta.tag })), ((promptMeta === null || promptMeta === void 0 ? void 0 : promptMeta.abTestId) && { abTestId: promptMeta.abTestId })), ((promptMeta === null || promptMeta === void 0 ? void 0 : promptMeta.variant) && { variant: promptMeta.variant })), spans })).then(() => triggerProcessing(userId, sessionId));
+                })), (toolDefs && { tools: toolDefs })), { requestPreview,
+                    responsePreview, state: "completed", startedAt: startedAt.toISOString(), endedAt: endedAt.toISOString(), durationMs: endedAt.getTime() - startedAt.getTime(), metadata: Object.assign(Object.assign(Object.assign({ appId: clConfig.appId }, ((promptMeta === null || promptMeta === void 0 ? void 0 : promptMeta.tag) && { promptTag: promptMeta.tag })), ((promptMeta === null || promptMeta === void 0 ? void 0 : promptMeta.abTestId) && { abTestId: promptMeta.abTestId })), ((promptMeta === null || promptMeta === void 0 ? void 0 : promptMeta.variant) && { variant: promptMeta.variant })), spans,
+                    automaticTitle })));
+                if (logResult) {
+                    triggerProcessing(userId, sessionId);
+                }
+                await postTraceEvents((0, shared_1.buildRemoteTraceFinishPayload)({
+                    execution: remoteExecution,
+                    traceDbId: liveTraceDbId,
+                    state: "completed",
+                    responsePreview,
+                    durationMs: endedAt.getTime() - startedAt.getTime(),
+                    usage: result.usage,
+                }));
             }
             return result;
         },
         async wrapStream({ doStream, params }) {
             var _a, _b, _c;
             const startedAt = new Date();
-            const traceId = (0, crypto_1.randomUUID)();
+            const requestedRunScope = resolveRequestedRunScope(params);
+            const remoteAgentName = (_a = clConfig.appId) !== null && _a !== void 0 ? _a : modelId;
+            const turnIndex = reserveTurnIndex(remoteAgentName, userId, sessionId, requestedRunScope);
+            const remoteExecution = (0, shared_1.createRemoteExecutionContext)({
+                agentName: remoteAgentName,
+                sessionId,
+                requestedRunScope,
+                runId: (_b = getKognitiveProviderConfig(params)) === null || _b === void 0 ? void 0 : _b.agentRunId,
+                turnIndex,
+            });
+            const traceId = remoteExecution.traceId;
+            let liveTraceDbId;
+            if (isValidId(userId) && isValidId(sessionId)) {
+                const messagesInput = params.prompt || params.messages || [];
+                const { requestPreview } = (0, shared_1.buildRemoteTracePreviews)(messagesInput);
+                await postAgentRun((0, shared_1.buildRemoteRunPayload)({
+                    execution: remoteExecution,
+                    userId,
+                    sessionId,
+                    modelId,
+                    status: "running",
+                    inputPreview: requestPreview,
+                    startedAt: startedAt.toISOString(),
+                    metadata: {
+                        appId: clConfig.appId,
+                    },
+                    sessionMetadata: {
+                        kind: "chat",
+                        agentName: remoteExecution.agentName,
+                    },
+                }));
+            }
+            if (isValidId(userId) && isValidId(sessionId) && isValidId(projectId)) {
+                const messagesInput = params.prompt || params.messages || [];
+                const { requestPreview } = (0, shared_1.buildRemoteTracePreviews)(messagesInput);
+                const traceResponse = await postTraceEvents((0, shared_1.buildRemoteTraceStartPayload)({
+                    execution: remoteExecution,
+                    userId,
+                    projectId,
+                    sessionId,
+                    requestPreview,
+                    modelId,
+                    metadata: {
+                        appId: clConfig.appId,
+                    },
+                    startedAt: startedAt.toISOString(),
+                }));
+                liveTraceDbId = typeof (traceResponse === null || traceResponse === void 0 ? void 0 : traceResponse.traceDbId) === "string" ? traceResponse.traceDbId : undefined;
+            }
             let result;
             try {
                 logger.debug("Starting doStream with params", JSON.stringify(params, null, 2));
                 result = await doStream();
             }
             catch (err) {
+                if (liveTraceDbId) {
+                    await postTraceEvents((0, shared_1.buildRemoteTraceFinishPayload)({
+                        execution: remoteExecution,
+                        traceDbId: liveTraceDbId,
+                        state: "error",
+                        responsePreview: err instanceof Error ? err.message.slice(0, 240) : "Stream failed",
+                        durationMs: Date.now() - startedAt.getTime(),
+                        errorMessage: err instanceof Error ? err.message : String(err),
+                    }));
+                }
+                if (isValidId(userId) && isValidId(sessionId)) {
+                    await postAgentRun((0, shared_1.buildRemoteRunPayload)({
+                        execution: remoteExecution,
+                        userId,
+                        sessionId,
+                        modelId,
+                        status: "failed",
+                        errorMessage: err instanceof Error ? err.message : String(err),
+                        durationMs: Date.now() - startedAt.getTime(),
+                        startedAt: startedAt.toISOString(),
+                        completedAt: new Date().toISOString(),
+                        metadata: {
+                            appId: clConfig.appId,
+                        },
+                    }));
+                }
                 console.log(err.cause);
                 console.log(err.stack);
                 logger.error("doStream failed", err);
@@ -485,7 +843,7 @@ ${userContextBlock || "None"}
             if (isValidId(userId) && isValidId(sessionId)) {
                 const sessionKey = getSessionKey(userId, sessionId);
                 const promptMeta = sessionPromptMetadata.get(sessionKey);
-                const agentRunId = (_b = (_a = params === null || params === void 0 ? void 0 : params.providerMetadata) === null || _a === void 0 ? void 0 : _a.kognitive) === null || _b === void 0 ? void 0 : _b.agentRunId;
+                const automaticTitle = resolveAutomaticTitleConfig(params);
                 const messagesInput = params.prompt || params.messages || [];
                 const resultMessages = (_c = result === null || result === void 0 ? void 0 : result.response) === null || _c === void 0 ? void 0 : _c.messages;
                 const finalMessages = Array.isArray(resultMessages) && resultMessages.length > 0
@@ -493,7 +851,10 @@ ${userContextBlock || "None"}
                     : messagesInput;
                 let streamUsage;
                 let accumulatedText = '';
+                let pendingProgressDelta = '';
+                let lastProgressAt = Date.now();
                 const toolCallInputs = new Map();
+                const startedToolCalls = new Set();
                 const completedToolCalls = [];
                 const completedToolResults = [];
                 const originalStream = result.stream;
@@ -501,6 +862,23 @@ ${userContextBlock || "None"}
                     transform(chunk, controller) {
                         if (chunk.type === 'text-delta') {
                             accumulatedText += chunk.delta;
+                            pendingProgressDelta += chunk.delta;
+                            if (liveTraceDbId && pendingProgressDelta && (pendingProgressDelta.length >= 80 || Date.now() - lastProgressAt >= 250)) {
+                                void appendLiveTraceEvents({
+                                    traceDbId: liveTraceDbId,
+                                    traceId,
+                                    events: [{
+                                            eventType: "assistant.progress",
+                                            status: "active",
+                                            payload: {
+                                                text: pendingProgressDelta,
+                                                preview: previewText(accumulatedText),
+                                            },
+                                        }],
+                                });
+                                pendingProgressDelta = '';
+                                lastProgressAt = Date.now();
+                            }
                         }
                         if (chunk.type === 'finish' && chunk.usage) {
                             streamUsage = chunk.usage;
@@ -508,6 +886,23 @@ ${userContextBlock || "None"}
                         // Capture tool-call stream chunks (V2/V3 shared types)
                         if (chunk.type === 'tool-input-start') {
                             toolCallInputs.set(chunk.id, { toolName: chunk.toolName, chunks: [] });
+                            if (liveTraceDbId && !startedToolCalls.has(chunk.id)) {
+                                startedToolCalls.add(chunk.id);
+                                void appendLiveTraceEvents({
+                                    traceDbId: liveTraceDbId,
+                                    traceId,
+                                    events: [{
+                                            eventType: "tool.started",
+                                            spanKey: chunk.id,
+                                            status: "active",
+                                            payload: {
+                                                toolCallId: chunk.id,
+                                                toolName: chunk.toolName,
+                                                inputPreview: "",
+                                            },
+                                        }],
+                                });
+                            }
                         }
                         if (chunk.type === 'tool-input-delta') {
                             const entry = toolCallInputs.get(chunk.id);
@@ -521,6 +916,29 @@ ${userContextBlock || "None"}
                                 toolName: chunk.toolName,
                                 input: chunk.input,
                             });
+                            if (!toolCallInputs.has(chunk.toolCallId)) {
+                                toolCallInputs.set(chunk.toolCallId, {
+                                    toolName: chunk.toolName,
+                                    chunks: typeof chunk.input === "string" ? [chunk.input] : [],
+                                });
+                            }
+                            if (liveTraceDbId && !startedToolCalls.has(chunk.toolCallId)) {
+                                startedToolCalls.add(chunk.toolCallId);
+                                void appendLiveTraceEvents({
+                                    traceDbId: liveTraceDbId,
+                                    traceId,
+                                    events: [{
+                                            eventType: "tool.started",
+                                            spanKey: chunk.toolCallId,
+                                            status: "active",
+                                            payload: {
+                                                toolCallId: chunk.toolCallId,
+                                                toolName: chunk.toolName,
+                                                inputPreview: typeof chunk.input === "string" ? previewText(chunk.input) : "",
+                                            },
+                                        }],
+                                });
+                            }
                         }
                         if (chunk.type === 'tool-result') {
                             completedToolResults.push({
@@ -529,11 +947,41 @@ ${userContextBlock || "None"}
                                 toolName: chunk.toolName,
                                 result: chunk.result,
                             });
+                            if (liveTraceDbId) {
+                                void appendLiveTraceEvents({
+                                    traceDbId: liveTraceDbId,
+                                    traceId,
+                                    events: [{
+                                            eventType: "tool.completed",
+                                            spanKey: chunk.toolCallId,
+                                            status: "completed",
+                                            payload: {
+                                                toolCallId: chunk.toolCallId,
+                                                toolName: chunk.toolName,
+                                                outputPreview: previewText(JSON.stringify(chunk.result)),
+                                            },
+                                        }],
+                                });
+                            }
                         }
                         controller.enqueue(chunk);
                     },
                     async flush() {
                         const endedAt = new Date();
+                        if (liveTraceDbId && pendingProgressDelta) {
+                            await appendLiveTraceEvents({
+                                traceDbId: liveTraceDbId,
+                                traceId,
+                                events: [{
+                                        eventType: "assistant.progress",
+                                        status: "active",
+                                        payload: {
+                                            text: pendingProgressDelta,
+                                            preview: previewText(accumulatedText),
+                                        },
+                                    }],
+                            });
+                        }
                         // Finalize any tool calls from incremental input chunks
                         for (const [id, entry] of toolCallInputs) {
                             // Only add if not already captured via a tool-call chunk
@@ -557,12 +1005,31 @@ ${userContextBlock || "None"}
                         if (completedToolResults.length > 0) {
                             allMessages.push({ role: "tool", content: completedToolResults });
                         }
-                        const { requestPreview, responsePreview } = buildTracePreviews(allMessages);
+                        const { requestPreview, responsePreview } = (0, shared_1.buildRemoteTracePreviews)(allMessages);
                         const spans = buildTraceSpansFromMessages(allMessages);
                         const toolDefs = extractToolDefinitions(params);
-                        // Fire-and-forget: do not await so the stream closes immediately,
-                        // allowing the AI SDK's multi-step continuation logic to proceed.
-                        logConversation(Object.assign(Object.assign(Object.assign(Object.assign({ userId,
+                        const usage = (0, shared_1.normalizeRemoteUsage)(streamUsage);
+                        await postAgentRun((0, shared_1.buildRemoteRunPayload)({
+                            execution: remoteExecution,
+                            userId,
+                            sessionId,
+                            modelId,
+                            status: "completed",
+                            inputPreview: requestPreview,
+                            outputPreview: responsePreview,
+                            inputTokens: usage.inputTokens,
+                            outputTokens: usage.outputTokens,
+                            cachedInputTokens: usage.cachedInputTokens,
+                            durationMs: endedAt.getTime() - startedAt.getTime(),
+                            startedAt: startedAt.toISOString(),
+                            completedAt: endedAt.toISOString(),
+                            metadata: Object.assign(Object.assign(Object.assign({ appId: clConfig.appId }, ((promptMeta === null || promptMeta === void 0 ? void 0 : promptMeta.tag) && { promptTag: promptMeta.tag })), ((promptMeta === null || promptMeta === void 0 ? void 0 : promptMeta.abTestId) && { abTestId: promptMeta.abTestId })), ((promptMeta === null || promptMeta === void 0 ? void 0 : promptMeta.variant) && { variant: promptMeta.variant })),
+                            sessionMetadata: {
+                                kind: "chat",
+                                agentName: remoteExecution.agentName,
+                            },
+                        }));
+                        const logResult = await logConversation((0, shared_1.buildRemoteLogPayload)(Object.assign(Object.assign(Object.assign({ execution: remoteExecution, userId,
                             sessionId, messages: allMessages, modelId, usage: streamUsage }, (promptMeta && {
                             promptSlug: promptMeta.promptSlug,
                             promptVersion: promptMeta.promptVersion,
@@ -570,10 +1037,30 @@ ${userContextBlock || "None"}
                             tag: promptMeta.tag,
                             abTestId: promptMeta.abTestId,
                             variant: promptMeta.variant,
-                        })), (toolDefs && { tools: toolDefs })), (agentRunId && { agentRunId })), { traceId,
-                            requestPreview,
-                            responsePreview, state: "completed", startedAt: startedAt.toISOString(), endedAt: endedAt.toISOString(), durationMs: endedAt.getTime() - startedAt.getTime(), metadata: Object.assign(Object.assign(Object.assign({ appId: clConfig.appId }, ((promptMeta === null || promptMeta === void 0 ? void 0 : promptMeta.tag) && { promptTag: promptMeta.tag })), ((promptMeta === null || promptMeta === void 0 ? void 0 : promptMeta.abTestId) && { abTestId: promptMeta.abTestId })), ((promptMeta === null || promptMeta === void 0 ? void 0 : promptMeta.variant) && { variant: promptMeta.variant })), spans })).then(() => triggerProcessing(userId, sessionId))
-                            .catch((e) => logger.error("Stream log failed", e));
+                        })), (toolDefs && { tools: toolDefs })), { requestPreview,
+                            responsePreview, state: "completed", startedAt: startedAt.toISOString(), endedAt: endedAt.toISOString(), durationMs: endedAt.getTime() - startedAt.getTime(), metadata: Object.assign(Object.assign(Object.assign({ appId: clConfig.appId }, ((promptMeta === null || promptMeta === void 0 ? void 0 : promptMeta.tag) && { promptTag: promptMeta.tag })), ((promptMeta === null || promptMeta === void 0 ? void 0 : promptMeta.abTestId) && { abTestId: promptMeta.abTestId })), ((promptMeta === null || promptMeta === void 0 ? void 0 : promptMeta.variant) && { variant: promptMeta.variant })), spans,
+                            automaticTitle })));
+                        if (logResult) {
+                            triggerProcessing(userId, sessionId);
+                        }
+                        if (logResult === null || logResult === void 0 ? void 0 : logResult.generatedTitle) {
+                            const currentProviderMetadata = result.providerMetadata && typeof result.providerMetadata === "object"
+                                ? result.providerMetadata
+                                : {};
+                            const currentKognitiveMetadata = currentProviderMetadata.kognitive
+                                && typeof currentProviderMetadata.kognitive === "object"
+                                ? currentProviderMetadata.kognitive
+                                : {};
+                            result.providerMetadata = Object.assign(Object.assign({}, currentProviderMetadata), { kognitive: Object.assign(Object.assign({}, currentKognitiveMetadata), { generatedTitle: logResult.generatedTitle }) });
+                        }
+                        await postTraceEvents((0, shared_1.buildRemoteTraceFinishPayload)({
+                            execution: remoteExecution,
+                            traceDbId: liveTraceDbId,
+                            state: "completed",
+                            responsePreview,
+                            durationMs: endedAt.getTime() - startedAt.getTime(),
+                            usage: streamUsage,
+                        }));
                     }
                 });
                 result.stream = originalStream.pipeThrough(transformStream);
@@ -632,23 +1119,29 @@ ${userContextBlock || "None"}
         return wrappedModel;
     };
     const clStreamText = async (options) => {
-        const { prompt: promptConfig } = options, rest = __rest(options, ["prompt"]);
+        const { prompt: promptConfig, kognitive } = options, rest = __rest(options, ["prompt", "kognitive"]);
         const session = options.model[SESSION_KEY];
         // Resolve and interpolate prompt (graceful fallback on failure)
         let resolved = null;
-        try {
-            resolved = await resolvePrompt(promptConfig.slug, {
-                userId: session === null || session === void 0 ? void 0 : session.userId,
-                tag: promptConfig.tag,
-            });
-        }
-        catch (err) {
-            logger.warn(`Failed to resolve prompt "${promptConfig.slug}", streaming without system prompt.`, err);
+        if (promptConfig === null || promptConfig === void 0 ? void 0 : promptConfig.slug) {
+            try {
+                resolved = await resolvePrompt(promptConfig.slug, {
+                    userId: session === null || session === void 0 ? void 0 : session.userId,
+                    tag: promptConfig.tag,
+                });
+            }
+            catch (err) {
+                if ((0, shared_1.isModerationError)(err)) {
+                    throw err;
+                }
+                logger.warn(`Failed to resolve prompt "${promptConfig.slug}", streaming without system prompt.`, err);
+            }
         }
-        let system;
+        let system = typeof rest.system === "string" ? rest.system : undefined;
         if (resolved) {
-            system = promptConfig.variables
-                ? (0, template_2.renderTemplate)(resolved.content, promptConfig.variables)
+            const resolvedPromptConfig = promptConfig;
+            system = resolvedPromptConfig.variables
+                ? (0, template_2.renderTemplate)(resolved.content, resolvedPromptConfig.variables)
                 : resolved.content;
             // Store prompt metadata for the session (read by middleware during logging)
             if (session === null || session === void 0 ? void 0 : session.sessionId) {
@@ -663,37 +1156,44 @@ ${userContextBlock || "None"}
                 });
             }
             logger.info("cl.streamText called", {
-                slug: promptConfig.slug,
+                slug: resolvedPromptConfig.slug,
                 version: resolved.version,
                 systemLength: system.length,
             });
         }
-        else {
+        else if (promptConfig === null || promptConfig === void 0 ? void 0 : promptConfig.slug) {
             logger.info("cl.streamText called without resolved prompt", {
                 slug: promptConfig.slug,
             });
         }
         const model = resolveModel(options.model, resolved === null || resolved === void 0 ? void 0 : resolved.gatewaySlug);
-        return (0, ai_1.streamText)(Object.assign(Object.assign(Object.assign({}, rest), { model }), (system && { system })));
+        const nextParams = withAutoInjectedTools(Object.assign(Object.assign(Object.assign({}, rest), { model }), (system && { system })), session, baseUrl, clConfig.apiKey, kognitive);
+        return (0, ai_1.streamText)(nextParams);
     };
     const clGenerateText = async (options) => {
-        const { prompt: promptConfig } = options, rest = __rest(options, ["prompt"]);
+        const { prompt: promptConfig, kognitive } = options, rest = __rest(options, ["prompt", "kognitive"]);
         const session = options.model[SESSION_KEY];
         // Resolve and interpolate prompt (graceful fallback on failure)
         let resolved = null;
-        try {
-            resolved = await resolvePrompt(promptConfig.slug, {
-                userId: session === null || session === void 0 ? void 0 : session.userId,
-                tag: promptConfig.tag,
-            });
-        }
-        catch (err) {
-            logger.warn(`Failed to resolve prompt "${promptConfig.slug}", generating without system prompt.`, err);
+        if (promptConfig === null || promptConfig === void 0 ? void 0 : promptConfig.slug) {
+            try {
+                resolved = await resolvePrompt(promptConfig.slug, {
+                    userId: session === null || session === void 0 ? void 0 : session.userId,
+                    tag: promptConfig.tag,
+                });
+            }
+            catch (err) {
+                if ((0, shared_1.isModerationError)(err)) {
+                    throw err;
+                }
+                logger.warn(`Failed to resolve prompt "${promptConfig.slug}", generating without system prompt.`, err);
+            }
         }
-        let system;
+        let system = typeof rest.system === "string" ? rest.system : undefined;
         if (resolved) {
-            system = promptConfig.variables
-                ? (0, template_2.renderTemplate)(resolved.content, promptConfig.variables)
+            const resolvedPromptConfig = promptConfig;
+            system = resolvedPromptConfig.variables
+                ? (0, template_2.renderTemplate)(resolved.content, resolvedPromptConfig.variables)
                 : resolved.content;
             // Store prompt metadata for the session (read by middleware during logging)
             if (session === null || session === void 0 ? void 0 : session.sessionId) {
@@ -708,18 +1208,19 @@ ${userContextBlock || "None"}
                 });
             }
             logger.info("cl.generateText called", {
-                slug: promptConfig.slug,
+                slug: resolvedPromptConfig.slug,
                 version: resolved.version,
                 systemLength: system.length,
             });
         }
-        else {
+        else if (promptConfig === null || promptConfig === void 0 ? void 0 : promptConfig.slug) {
             logger.info("cl.generateText called without resolved prompt", {
                 slug: promptConfig.slug,
             });
         }
         const model = resolveModel(options.model, resolved === null || resolved === void 0 ? void 0 : resolved.gatewaySlug);
-        return (0, ai_1.generateText)(Object.assign(Object.assign(Object.assign({}, rest), { model }), (system && { system })));
+        const nextParams = withAutoInjectedTools(Object.assign(Object.assign(Object.assign({}, rest), { model }), (system && { system })), session, baseUrl, clConfig.apiKey, kognitive);
+        return (0, ai_1.generateText)(nextParams);
     };
     // Return the model wrapper function with streamText/generateText attached
     return Object.assign(clWrapper, {
@@ -732,10 +1233,15 @@ ${userContextBlock || "None"}
             if (sessionKey) {
                 sessionSnapshots.delete(sessionKey);
                 sessionPromptMetadata.delete(sessionKey);
+                for (const key of sessionTurnIndexes.keys()) {
+                    if (key.endsWith(`:${sessionKey}`))
+                        sessionTurnIndexes.delete(key);
+                }
             }
             else {
                 sessionSnapshots.clear();
                 sessionPromptMetadata.clear();
+                sessionTurnIndexes.clear();
             }
         },
     });