@kognitivedev/vercel-ai-provider 0.2.28 → 0.2.29

package/CHANGELOG.md

@@ -1,5 +1,17 @@
 # @kognitivedev/vercel-ai-provider
 
+## 0.2.29
+
+### Patch Changes
+
+- release
+
+- Updated dependencies []:
+  - @kognitivedev/memory@0.2.29
+  - @kognitivedev/prompthub@0.2.29
+  - @kognitivedev/shared@0.2.29
+  - @kognitivedev/tools@0.2.29
+
 ## 0.2.28
 
 ### Patch Changes

package/dist/index.js

@@ -110,6 +110,18 @@ function extractOutputValue(raw) {
             return raw;
     }
 }
+function resolveRequestedAssistantMessageId(metadata) {
+    if (!metadata || typeof metadata !== "object")
+        return undefined;
+    const value = metadata.kognitiveAssistantMessageId;
+    return typeof value === "string" && value.trim().length > 0 ? value.trim() : undefined;
+}
+function buildAssistantMessageMetadata(metadata) {
+    var _a;
+    return {
+        kognitiveMessageId: (_a = resolveRequestedAssistantMessageId(metadata)) !== null && _a !== void 0 ? _a : crypto.randomUUID(),
+    };
+}
 function buildTraceSpansFromMessages(messages) {
     var _a, _b;
     const resultMap = new Map();
@@ -674,6 +686,7 @@ function createCognitiveLayer(config) {
                 const promptMeta = sessionPromptMetadata.get(sessionKey);
                 const automaticTitle = resolveAutomaticTitleConfig(params);
                 const messagesInput = params.prompt || params.messages || [];
+                const assistantMetadata = buildAssistantMessageMetadata(params.metadata);
                 // Build assistant message from result.content (V2/V3 GenerateResult)
                 const resultContent = Array.isArray(result === null || result === void 0 ? void 0 : result.content) ? result.content : [];
                 const assistantParts = [];
@@ -699,7 +712,7 @@ function createCognitiveLayer(config) {
                     }
                 }
                 const assistantMessage = assistantParts.length > 0
-                    ? [{ role: "assistant", content: assistantParts }]
+                    ? [{ role: "assistant", content: assistantParts, metadata: assistantMetadata }]
                     : [];
                 const finalMessages = [...messagesInput, ...assistantMessage];
                 const { requestPreview, responsePreview } = (0, shared_1.buildRemoteTracePreviews)(finalMessages);
@@ -995,12 +1008,13 @@ function createCognitiveLayer(config) {
                             }
                         }
                         const assistantParts = [];
+                        const assistantMetadata = buildAssistantMessageMetadata(params.metadata);
                         if (accumulatedText)
                             assistantParts.push({ type: "text", text: accumulatedText });
                         for (const tc of completedToolCalls)
                             assistantParts.push(tc);
                         const allMessages = assistantParts.length > 0
-                            ? [...finalMessages, { role: "assistant", content: assistantParts }]
+                            ? [...finalMessages, { role: "assistant", content: assistantParts, metadata: assistantMetadata }]
                             : finalMessages;
                         if (completedToolResults.length > 0) {
                             allMessages.push({ role: "tool", content: completedToolResults });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kognitivedev/vercel-ai-provider",
-  "version": "0.2.28",
+  "version": "0.2.29",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "publishConfig": {
@@ -13,10 +13,10 @@
     "prepublishOnly": "npm run build"
   },
   "dependencies": {
-    "@kognitivedev/memory": "^0.2.28",
-    "@kognitivedev/prompthub": "^0.2.28",
-    "@kognitivedev/shared": "^0.2.28",
-    "@kognitivedev/tools": "^0.2.28"
+    "@kognitivedev/memory": "^0.2.29",
+    "@kognitivedev/prompthub": "^0.2.29",
+    "@kognitivedev/shared": "^0.2.29",
+    "@kognitivedev/tools": "^0.2.29"
   },
   "peerDependencies": {
     "ai": "^5.0.0 || ^6.0.0"
@@ -24,6 +24,7 @@
   "devDependencies": {
     "typescript": "^5.0.0",
     "ai": "^6.0.0",
+    "@ai-sdk/provider": "3.0.8",
     "@types/node": "^20.0.0",
     "vitest": "^3.0.0"
   },

package/src/index.ts

@@ -232,6 +232,18 @@ function extractOutputValue(raw: unknown): unknown {
     }
 }
 
+function resolveRequestedAssistantMessageId(metadata: unknown): string | undefined {
+    if (!metadata || typeof metadata !== "object") return undefined;
+    const value = (metadata as Record<string, unknown>).kognitiveAssistantMessageId;
+    return typeof value === "string" && value.trim().length > 0 ? value.trim() : undefined;
+}
+
+function buildAssistantMessageMetadata(metadata: unknown): Record<string, unknown> {
+    return {
+        kognitiveMessageId: resolveRequestedAssistantMessageId(metadata) ?? crypto.randomUUID(),
+    };
+}
+
 function buildTraceSpansFromMessages(messages: any[]): Array<{
     spanKey: string;
     parentSpanKey?: string;
@@ -910,6 +922,7 @@ export function createCognitiveLayer(config: {
                 const automaticTitle = resolveAutomaticTitleConfig(params);
 
                 const messagesInput = (params as any).prompt || (params as any).messages || [];
+                const assistantMetadata = buildAssistantMessageMetadata((params as { metadata?: unknown }).metadata);
 
                 // Build assistant message from result.content (V2/V3 GenerateResult)
                 const resultContent = Array.isArray(result?.content) ? result.content : [];
@@ -934,7 +947,7 @@ export function createCognitiveLayer(config: {
                     }
                 }
                 const assistantMessage = assistantParts.length > 0
-                    ? [{ role: "assistant", content: assistantParts }]
+                    ? [{ role: "assistant", content: assistantParts, metadata: assistantMetadata }]
                     : [];
                 const finalMessages = [...messagesInput, ...assistantMessage];
                 const { requestPreview, responsePreview } = buildRemoteTracePreviews(finalMessages);
@@ -1263,11 +1276,12 @@ export function createCognitiveLayer(config: {
                         }
 
                         const assistantParts: any[] = [];
+                        const assistantMetadata = buildAssistantMessageMetadata((params as { metadata?: unknown }).metadata);
                         if (accumulatedText) assistantParts.push({ type: "text", text: accumulatedText });
                         for (const tc of completedToolCalls) assistantParts.push(tc);
 
                         const allMessages = assistantParts.length > 0
-                            ? [...finalMessages, { role: "assistant", content: assistantParts }]
+                            ? [...finalMessages, { role: "assistant", content: assistantParts, metadata: assistantMetadata }]
                             : finalMessages;
 
                         if (completedToolResults.length > 0) {