@kognitivedev/vercel-ai-provider 0.2.1 → 0.2.3

package/CHANGELOG.md

@@ -0,0 +1,11 @@
+# @kognitivedev/vercel-ai-provider
+
+## 0.2.3
+
+### Patch Changes
+
+- fix: resolve workspace dependencies for external consumers
+
+- Updated dependencies []:
+  - @kognitivedev/prompthub@0.1.1
+  - @kognitivedev/shared@0.2.3

package/dist/__tests__/wrap-stream-logging.test.js

@@ -116,6 +116,68 @@ const test_1 = require("ai/test");
         (0, vitest_1.expect)(toolSpan.inputPreview).toContain("London");
         (0, vitest_1.expect)(toolSpan.outputPreview).toContain("15");
     });
+    (0, vitest_1.it)("should include tool definitions in logged conversation when tools are present", async () => {
+        const mockModel = new test_1.MockLanguageModelV3({
+            doStream: async () => ({
+                stream: (0, test_1.convertArrayToReadableStream)([
+                    { type: "text-start", id: "t1" },
+                    { type: "text-delta", id: "t1", delta: "Sure" },
+                    { type: "text-end", id: "t1" },
+                    {
+                        type: "finish",
+                        finishReason: {
+                            unified: "stop",
+                            raw: undefined,
+                        },
+                        usage: {
+                            inputTokens: { total: 10, noCache: undefined, cacheRead: undefined, cacheWrite: undefined },
+                            outputTokens: { total: 5, text: undefined, reasoning: undefined },
+                        },
+                    },
+                ]),
+            }),
+        });
+        const mockProvider = () => mockModel;
+        const cl = (0, index_1.createCognitiveLayer)({
+            provider: mockProvider,
+            clConfig: {
+                apiKey: "test-api-key",
+                appId: "test-app",
+                projectId: "test-project",
+                processDelayMs: 0,
+                logLevel: "none",
+            },
+        });
+        const model = cl("mock-model", {
+            userId: "user-1",
+            projectId: "project-1",
+            sessionId: "session-1",
+        });
+        const result = (0, ai_1.streamText)({
+            model,
+            messages: [{ role: "user", content: "What's the weather?" }],
+            tools: {
+                get_weather: {
+                    description: "Get the current weather for a location",
+                    parameters: {
+                        type: "object",
+                        properties: {
+                            city: { type: "string", description: "City name" },
+                        },
+                        required: ["city"],
+                    },
+                },
+            },
+        });
+        await result.text;
+        await new Promise((r) => setTimeout(r, 100));
+        const logCall = fetchCalls.find((c) => c.url.includes("/api/cognitive/log"));
+        (0, vitest_1.expect)(logCall).toBeDefined();
+        (0, vitest_1.expect)(logCall.body.tools).toBeDefined();
+        (0, vitest_1.expect)(logCall.body.tools).toEqual(vitest_1.expect.arrayContaining([
+            vitest_1.expect.objectContaining({ name: "get_weather" }),
+        ]));
+    });
     (0, vitest_1.it)("should include assistant message in logged conversation after streaming", async () => {
         const mockModel = new test_1.MockLanguageModelV3({
             doStream: async () => ({

package/dist/index.d.ts

@@ -64,6 +64,12 @@ export interface LogConversationPayload {
     startedAt?: string;
     endedAt?: string;
     durationMs?: number;
+    tools?: Array<{
+        name: string;
+        description?: string;
+        parameters?: Record<string, unknown>;
+    }>;
+    agentRunId?: string;
     metadata?: Record<string, unknown>;
     spans?: Array<{
         spanKey: string;

package/dist/index.js

@@ -15,6 +15,7 @@ exports.renderTemplate = void 0;
 exports.createCognitiveLayer = createCognitiveLayer;
 const ai_1 = require("ai");
 const crypto_1 = require("crypto");
+const prompthub_1 = require("@kognitivedev/prompthub");
 var template_1 = require("./template");
 Object.defineProperty(exports, "renderTemplate", { enumerable: true, get: function () { return template_1.renderTemplate; } });
 const template_2 = require("./template");
@@ -71,7 +72,6 @@ function createLogger(logLevel) {
         },
     };
 }
-const PROMPT_CACHE_TTL_MS = 60000; // 1 minute
 function getContentText(content) {
     if (typeof content === "string")
         return content;
@@ -158,6 +158,21 @@ function buildTraceSpansFromMessages(messages) {
     }
     return spans;
 }
+function extractToolDefinitions(params) {
+    const tools = params.tools;
+    if (!Array.isArray(tools) || tools.length === 0)
+        return undefined;
+    const defs = [];
+    for (const tool of tools) {
+        if (tool.type === 'function') {
+            defs.push(Object.assign(Object.assign({ name: tool.name }, (tool.description && { description: tool.description })), (tool.inputSchema && { parameters: tool.inputSchema })));
+        }
+        else if (tool.type === 'provider' || tool.type === 'provider-defined') {
+            defs.push(Object.assign({ name: tool.name }, (tool.args && { parameters: tool.args })));
+        }
+    }
+    return defs.length > 0 ? defs : undefined;
+}
 // Session-scoped snapshot cache: sessionKey → formatted memory block
 const sessionSnapshots = new Map();
 // Regex to detect if memory has already been injected
@@ -194,48 +209,20 @@ function createCognitiveLayer(config) {
         "Content-Type": "application/json",
         "Authorization": `Bearer ${clConfig.apiKey}`,
     };
-    // Prompt cache: slug → CachedPrompt
-    const promptCache = new Map();
+    const promptClient = (0, prompthub_1.createPromptHubClient)({
+        baseUrl,
+        apiKey: clConfig.apiKey,
+        logger,
+    });
     const resolvePrompt = async (slug, userId) => {
         var _a;
-        const cacheKey = userId ? `${slug}:${userId}` : slug;
-        const cached = promptCache.get(cacheKey);
-        if (cached && Date.now() - cached.fetchedAt < PROMPT_CACHE_TTL_MS) {
-            logger.debug("Using cached prompt", { slug, version: cached.version });
-            return cached;
-        }
-        const url = new URL(`${baseUrl}/api/cognitive/prompt`);
-        url.searchParams.set("slug", slug);
-        if (userId)
-            url.searchParams.set("userId", userId);
         logger.debug("Resolving prompt from backend", {
             slug,
             userId,
-            url: url.toString(),
             baseUrl,
             apiKeyHint: maskSecret(clConfig.apiKey),
         });
-        const res = await fetch(url.toString(), {
-            headers: { "Authorization": `Bearer ${clConfig.apiKey}` },
-        });
-        logger.debug("Prompt resolve response received", {
-            slug,
-            userId,
-            status: res.status,
-            ok: res.ok,
-            contentType: res.headers.get("content-type"),
-        });
-        if (!res.ok) {
-            const body = await res.text();
-            logger.debug("Prompt resolve response body preview", {
-                slug,
-                userId,
-                status: res.status,
-                bodyPreview: previewText(body),
-            });
-            throw new Error(`Failed to resolve prompt "${slug}": ${res.status} ${body}`);
-        }
-        const data = await res.json();
+        const data = await promptClient.resolvePrompt({ slug, userId });
         const entry = {
             promptId: data.promptId,
             slug: data.slug,
@@ -244,7 +231,6 @@ function createCognitiveLayer(config) {
             fetchedAt: Date.now(),
             gatewaySlug: data.gatewaySlug,
         };
-        promptCache.set(cacheKey, entry);
         logger.debug("Prompt resolved payload", {
             slug,
             resolvedSlug: entry.slug,
@@ -404,7 +390,7 @@ ${userContextBlock || "None"}
             return Object.assign(Object.assign({}, nextParams), { prompt: messagesWithMemory });
         },
         async wrapGenerate({ doGenerate, params }) {
-            var _a;
+            var _a, _b, _c;
             const startedAt = new Date();
             let result;
             try {
@@ -419,6 +405,7 @@ ${userContextBlock || "None"}
                 const endedAt = new Date();
                 const sessionKey = `${userId}:${projectId}:${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 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 : [];
@@ -450,13 +437,14 @@ ${userContextBlock || "None"}
                 const finalMessages = [...messagesInput, ...assistantMessage];
                 const { requestPreview, responsePreview } = buildTracePreviews(finalMessages);
                 const spans = buildTraceSpansFromMessages(finalMessages);
-                logConversation(Object.assign(Object.assign({ userId,
+                const toolDefs = extractToolDefinitions(params);
+                logConversation(Object.assign(Object.assign(Object.assign(Object.assign({ userId,
                     projectId,
                     sessionId, messages: finalMessages, modelId, usage: result.usage }, (promptMeta && {
                     promptSlug: promptMeta.promptSlug,
                     promptVersion: promptMeta.promptVersion,
                     promptId: promptMeta.promptId,
-                })), { traceId: (0, crypto_1.randomUUID)(), requestPreview,
+                })), (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: {
                         appId: clConfig.appId,
                     }, spans })).then(() => triggerProcessing(userId, projectId, sessionId));
@@ -464,7 +452,7 @@ ${userContextBlock || "None"}
             return result;
         },
         async wrapStream({ doStream, params }) {
-            var _a;
+            var _a, _b, _c;
             const startedAt = new Date();
             const traceId = (0, crypto_1.randomUUID)();
             let result;
@@ -481,8 +469,9 @@ ${userContextBlock || "None"}
             if (isValidId(userId) && isValidId(sessionId)) {
                 const sessionKey = `${userId}:${projectId}:${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 messagesInput = params.prompt || params.messages || [];
-                const resultMessages = (_a = result === null || result === void 0 ? void 0 : result.response) === null || _a === void 0 ? void 0 : _a.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
                     ? resultMessages
                     : messagesInput;
@@ -554,18 +543,21 @@ ${userContextBlock || "None"}
                         }
                         const { requestPreview, responsePreview } = buildTracePreviews(allMessages);
                         const spans = buildTraceSpansFromMessages(allMessages);
-                        await logConversation(Object.assign(Object.assign({ userId,
+                        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,
                             projectId,
                             sessionId, messages: allMessages, modelId, usage: streamUsage }, (promptMeta && {
                             promptSlug: promptMeta.promptSlug,
                             promptVersion: promptMeta.promptVersion,
                             promptId: promptMeta.promptId,
-                        })), { traceId,
+                        })), (toolDefs && { tools: toolDefs })), (agentRunId && { agentRunId })), { traceId,
                             requestPreview,
                             responsePreview, state: "completed", startedAt: startedAt.toISOString(), endedAt: endedAt.toISOString(), durationMs: endedAt.getTime() - startedAt.getTime(), metadata: {
                                 appId: clConfig.appId,
-                            }, spans }));
-                        triggerProcessing(userId, projectId, sessionId);
+                            }, spans })).then(() => triggerProcessing(userId, projectId, sessionId))
+                            .catch((e) => logger.error("Stream log failed", e));
                     }
                 });
                 result.stream = originalStream.pipeThrough(transformStream);
@@ -708,7 +700,7 @@ ${userContextBlock || "None"}
         resolvePrompt,
         logConversation,
         triggerProcessing,
-        clearPromptCache: () => promptCache.clear(),
+        clearPromptCache: () => promptClient.clearPromptCache(),
         clearSessionCache: (sessionKey) => {
             if (sessionKey) {
                 sessionSnapshots.delete(sessionKey);

package/dist/template.d.ts

@@ -1,2 +1 @@
-export type TemplateVariables = Record<string, string | boolean>;
-export declare function renderTemplate(template: string, variables: TemplateVariables): string;
+export { renderTemplate, type TemplateVariables } from "@kognitivedev/shared";

package/dist/template.js

@@ -1,13 +1,5 @@
 "use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.renderTemplate = renderTemplate;
-// Use the pre-built dist to avoid `require.extensions` warning in webpack/Next.js
-// eslint-disable-next-line @typescript-eslint/no-require-imports
-const handlebars_1 = __importDefault(require("handlebars/dist/cjs/handlebars"));
-function renderTemplate(template, variables) {
-    const compiled = handlebars_1.default.compile(template, { noEscape: true });
-    return compiled(variables);
-}
+exports.renderTemplate = void 0;
+var shared_1 = require("@kognitivedev/shared");
+Object.defineProperty(exports, "renderTemplate", { enumerable: true, get: function () { return shared_1.renderTemplate; } });

package/package.json

@@ -1,27 +1,42 @@
 {
-    "name": "@kognitivedev/vercel-ai-provider",
-    "version": "0.2.1",
-    "main": "dist/index.js",
-    "types": "dist/index.d.ts",
-    "publishConfig": {
-        "access": "public"
-    },
-    "scripts": {
-        "build": "tsc",
-        "dev": "tsc -w",
-        "test": "vitest run",
-        "prepublishOnly": "npm run build"
-    },
-    "dependencies": {
-        "handlebars": "^4.7.8"
-    },
-    "peerDependencies": {
-        "ai": "^5.0.0 || ^6.0.0"
-    },
-    "devDependencies": {
-        "typescript": "^5.0.0",
-        "ai": "^6.0.0",
-        "@types/node": "^20.0.0",
-        "vitest": "^3.0.0"
-    }
-}
+  "name": "@kognitivedev/vercel-ai-provider",
+  "version": "0.2.3",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsc -w",
+    "test": "vitest run",
+    "prepublishOnly": "npm run build"
+  },
+  "dependencies": {
+    "@kognitivedev/prompthub": "workspace:*",
+    "@kognitivedev/shared": "workspace:*"
+  },
+  "peerDependencies": {
+    "ai": "^5.0.0 || ^6.0.0"
+  },
+  "devDependencies": {
+    "typescript": "^5.0.0",
+    "ai": "^6.0.0",
+    "@types/node": "^20.0.0",
+    "vitest": "^3.0.0"
+  },
+  "description": "Vercel AI SDK middleware for memory injection and prompt management",
+  "keywords": [
+    "kognitive",
+    "vercel-ai-sdk",
+    "memory",
+    "middleware"
+  ],
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/kognitivedev/kognitive",
+    "directory": "packages/provider"
+  },
+  "homepage": "https://kognitive.dev"
+}

package/src/__tests__/wrap-stream-logging.test.ts

@@ -140,6 +140,77 @@ describe("wrapStream logging", () => {
         expect(toolSpan.outputPreview).toContain("15");
     });
 
+    it("should include tool definitions in logged conversation when tools are present", async () => {
+        const mockModel = new MockLanguageModelV3({
+            doStream: async () => ({
+                stream: convertArrayToReadableStream([
+                    { type: "text-start" as const, id: "t1" },
+                    { type: "text-delta" as const, id: "t1", delta: "Sure" },
+                    { type: "text-end" as const, id: "t1" },
+                    {
+                        type: "finish" as const,
+                        finishReason: {
+                            unified: "stop" as const,
+                            raw: undefined,
+                        },
+                        usage: {
+                            inputTokens: { total: 10, noCache: undefined, cacheRead: undefined, cacheWrite: undefined },
+                            outputTokens: { total: 5, text: undefined, reasoning: undefined },
+                        },
+                    },
+                ] satisfies import("@ai-sdk/provider").LanguageModelV3StreamPart[]),
+            }),
+        });
+
+        const mockProvider = () => mockModel;
+
+        const cl = createCognitiveLayer({
+            provider: mockProvider,
+            clConfig: {
+                apiKey: "test-api-key",
+                appId: "test-app",
+                projectId: "test-project",
+                processDelayMs: 0,
+                logLevel: "none",
+            },
+        });
+
+        const model = cl("mock-model", {
+            userId: "user-1",
+            projectId: "project-1",
+            sessionId: "session-1",
+        });
+
+        const result = streamText({
+            model,
+            messages: [{ role: "user", content: "What's the weather?" }],
+            tools: {
+                get_weather: {
+                    description: "Get the current weather for a location",
+                    parameters: {
+                        type: "object",
+                        properties: {
+                            city: { type: "string", description: "City name" },
+                        },
+                        required: ["city"],
+                    },
+                },
+            } as any,
+        });
+
+        await result.text;
+        await new Promise((r) => setTimeout(r, 100));
+
+        const logCall = fetchCalls.find((c) => c.url.includes("/api/cognitive/log"));
+        expect(logCall).toBeDefined();
+        expect(logCall!.body.tools).toBeDefined();
+        expect(logCall!.body.tools).toEqual(
+            expect.arrayContaining([
+                expect.objectContaining({ name: "get_weather" }),
+            ])
+        );
+    });
+
     it("should include assistant message in logged conversation after streaming", async () => {
         const mockModel = new MockLanguageModelV3({
             doStream: async () => ({

package/src/index.ts

@@ -5,6 +5,7 @@ import {
     type LanguageModel,
 } from "ai";
 import { randomUUID } from "crypto";
+import { createPromptHubClient } from "@kognitivedev/prompthub";
 export { renderTemplate, type TemplateVariables } from "./template";
 import { renderTemplate } from "./template";
 
@@ -133,6 +134,8 @@ export interface LogConversationPayload {
     startedAt?: string;
     endedAt?: string;
     durationMs?: number;
+    tools?: Array<{ name: string; description?: string; parameters?: Record<string, unknown> }>;
+    agentRunId?: string;
     metadata?: Record<string, unknown>;
     spans?: Array<{
         spanKey: string;
@@ -169,8 +172,6 @@ export interface CachedPrompt {
     gatewaySlug?: string;
 }
 
-const PROMPT_CACHE_TTL_MS = 60_000; // 1 minute
-
 function getContentText(content: any): string {
     if (typeof content === "string") return content;
     if (!Array.isArray(content)) return "";
@@ -276,6 +277,28 @@ function buildTraceSpansFromMessages(messages: any[]): Array<{
     return spans;
 }
 
+function extractToolDefinitions(params: any): Array<{ name: string; description?: string; parameters?: Record<string, unknown> }> | undefined {
+    const tools = (params as any).tools;
+    if (!Array.isArray(tools) || tools.length === 0) return undefined;
+
+    const defs: Array<{ name: string; description?: string; parameters?: Record<string, unknown> }> = [];
+    for (const tool of tools) {
+        if (tool.type === 'function') {
+            defs.push({
+                name: tool.name,
+                ...(tool.description && { description: tool.description }),
+                ...(tool.inputSchema && { parameters: tool.inputSchema as Record<string, unknown> }),
+            });
+        } else if (tool.type === 'provider' || tool.type === 'provider-defined') {
+            defs.push({
+                name: tool.name,
+                ...(tool.args && { parameters: tool.args as Record<string, unknown> }),
+            });
+        }
+    }
+    return defs.length > 0 ? defs : undefined;
+}
+
 // Session-scoped snapshot cache: sessionKey → formatted memory block
 const sessionSnapshots = new Map<string, string>();
 
@@ -323,51 +346,21 @@ export function createCognitiveLayer(config: {
         "Authorization": `Bearer ${clConfig.apiKey}`,
     };
 
-    // Prompt cache: slug → CachedPrompt
-    const promptCache = new Map<string, CachedPrompt>();
+    const promptClient = createPromptHubClient({
+        baseUrl,
+        apiKey: clConfig.apiKey,
+        logger,
+    });
 
     const resolvePrompt = async (slug: string, userId?: string): Promise<CachedPrompt> => {
-        const cacheKey = userId ? `${slug}:${userId}` : slug;
-        const cached = promptCache.get(cacheKey);
-        if (cached && Date.now() - cached.fetchedAt < PROMPT_CACHE_TTL_MS) {
-            logger.debug("Using cached prompt", { slug, version: cached.version });
-            return cached;
-        }
-
-        const url = new URL(`${baseUrl}/api/cognitive/prompt`);
-        url.searchParams.set("slug", slug);
-        if (userId) url.searchParams.set("userId", userId);
-
         logger.debug("Resolving prompt from backend", {
             slug,
             userId,
-            url: url.toString(),
             baseUrl,
             apiKeyHint: maskSecret(clConfig.apiKey),
         });
 
-        const res = await fetch(url.toString(), {
-            headers: { "Authorization": `Bearer ${clConfig.apiKey}` },
-        });
-        logger.debug("Prompt resolve response received", {
-            slug,
-            userId,
-            status: res.status,
-            ok: res.ok,
-            contentType: res.headers.get("content-type"),
-        });
-        if (!res.ok) {
-            const body = await res.text();
-            logger.debug("Prompt resolve response body preview", {
-                slug,
-                userId,
-                status: res.status,
-                bodyPreview: previewText(body),
-            });
-            throw new Error(`Failed to resolve prompt "${slug}": ${res.status} ${body}`);
-        }
-
-        const data = await res.json();
+        const data = await promptClient.resolvePrompt({ slug, userId });
         const entry: CachedPrompt = {
             promptId: data.promptId,
             slug: data.slug,
@@ -376,7 +369,6 @@ export function createCognitiveLayer(config: {
             fetchedAt: Date.now(),
             gatewaySlug: data.gatewaySlug,
         };
-        promptCache.set(cacheKey, entry);
         logger.debug("Prompt resolved payload", {
             slug,
             resolvedSlug: entry.slug,
@@ -574,6 +566,7 @@ ${userContextBlock || "None"}
                 const endedAt = new Date();
                 const sessionKey = `${userId}:${projectId}:${sessionId}`;
                 const promptMeta = sessionPromptMetadata.get(sessionKey);
+                const agentRunId = (params as any)?.providerMetadata?.kognitive?.agentRunId as string | undefined;
 
                 const messagesInput = (params as any).prompt || (params as any).messages || [];
 
@@ -605,6 +598,7 @@ ${userContextBlock || "None"}
                 const finalMessages = [...messagesInput, ...assistantMessage];
                 const { requestPreview, responsePreview } = buildTracePreviews(finalMessages);
                 const spans = buildTraceSpansFromMessages(finalMessages);
+                const toolDefs = extractToolDefinitions(params);
 
                 logConversation({
                     userId,
@@ -618,6 +612,8 @@ ${userContextBlock || "None"}
                         promptVersion: promptMeta.promptVersion,
                         promptId: promptMeta.promptId,
                     }),
+                    ...(toolDefs && { tools: toolDefs }),
+                    ...(agentRunId && { agentRunId }),
                     traceId: randomUUID(),
                     requestPreview,
                     responsePreview,
@@ -652,6 +648,7 @@ ${userContextBlock || "None"}
             if (isValidId(userId) && isValidId(sessionId)) {
                 const sessionKey = `${userId}:${projectId}:${sessionId}`;
                 const promptMeta = sessionPromptMetadata.get(sessionKey);
+                const agentRunId = (params as any)?.providerMetadata?.kognitive?.agentRunId as string | undefined;
 
                 const messagesInput = (params as any).prompt || (params as any).messages || [];
                 const resultMessages = (result as any)?.response?.messages;
@@ -730,8 +727,11 @@ ${userContextBlock || "None"}
 
                         const { requestPreview, responsePreview } = buildTracePreviews(allMessages);
                         const spans = buildTraceSpansFromMessages(allMessages);
+                        const toolDefs = extractToolDefinitions(params);
 
-                        await logConversation({
+                        // Fire-and-forget: do not await so the stream closes immediately,
+                        // allowing the AI SDK's multi-step continuation logic to proceed.
+                        logConversation({
                             userId,
                             projectId,
                             sessionId,
@@ -743,6 +743,8 @@ ${userContextBlock || "None"}
                                 promptVersion: promptMeta.promptVersion,
                                 promptId: promptMeta.promptId,
                             }),
+                            ...(toolDefs && { tools: toolDefs }),
+                            ...(agentRunId && { agentRunId }),
                             traceId,
                             requestPreview,
                             responsePreview,
@@ -754,8 +756,8 @@ ${userContextBlock || "None"}
                                 appId: clConfig.appId,
                             },
                             spans,
-                        });
-                        triggerProcessing(userId, projectId, sessionId);
+                        }).then(() => triggerProcessing(userId, projectId, sessionId))
+                          .catch((e) => logger.error("Stream log failed", e));
                     }
                 });
 
@@ -924,7 +926,7 @@ ${userContextBlock || "None"}
         resolvePrompt,
         logConversation,
         triggerProcessing,
-        clearPromptCache: () => promptCache.clear(),
+        clearPromptCache: () => promptClient.clearPromptCache(),
         clearSessionCache: (sessionKey?: string) => {
             if (sessionKey) {
                 sessionSnapshots.delete(sessionKey);

package/src/template.ts

@@ -1,10 +1 @@
-// Use the pre-built dist to avoid `require.extensions` warning in webpack/Next.js
-// eslint-disable-next-line @typescript-eslint/no-require-imports
-import Handlebars from "handlebars/dist/cjs/handlebars";
-
-export type TemplateVariables = Record<string, string | boolean>;
-
-export function renderTemplate(template: string, variables: TemplateVariables): string {
-    const compiled = Handlebars.compile(template, { noEscape: true });
-    return compiled(variables);
-}
+export { renderTemplate, type TemplateVariables } from "@kognitivedev/shared";

package/tsconfig.json

@@ -2,9 +2,11 @@
     "extends": "../../tsconfig.json",
     "compilerOptions": {
         "module": "commonjs",
+        "rootDir": "src",
         "outDir": "dist",
         "declaration": true,
-        "noEmit": false
+        "noEmit": false,
+        "incremental": false
     },
     "include": [
         "src"