@kognitivedev/vercel-ai-provider 0.1.6 → 0.1.7

package/README.md

@@ -55,7 +55,7 @@ const { text } = await generateText({
 | `apiKey` | `string` | - | - | API key for authentication (if required) |
 | `processDelayMs` | `number` | - | `500` | Delay before triggering memory processing (set to 0 to disable) |
 | `logLevel` | `LogLevel` | - | `'info'` | Controls console logging verbosity |
-| `providerFactory` | `(baseURL: string) => (modelId: string) => LanguageModel` | - | - | Factory for creating a provider that routes through a gateway URL |
+| `providerFactory` | `(baseURL: string, apiKey: string) => (modelId: string) => LanguageModel` | - | - | Factory for creating a provider that routes through a gateway URL |
 
 ### `LogLevel`
 
@@ -317,8 +317,8 @@ const cl = createCognitiveLayer({
     clConfig: {
         appId: "my-app",
         baseUrl: "https://api.kognitive.dev",
-        providerFactory: (baseURL) => {
-            const gatewayProvider = createOpenAI({ baseURL });
+        providerFactory: (baseURL, apiKey) => {
+            const gatewayProvider = createOpenAI({ baseURL, apiKey });
             return (modelId) => gatewayProvider(modelId);
         }
     }

package/dist/index.d.ts

@@ -27,7 +27,7 @@ export interface CognitiveLayerConfig {
     /**
      * Factory for creating a provider that routes through a gateway URL.
      */
-    providerFactory?: (baseURL: string) => (modelId: string) => LanguageModel;
+    providerFactory?: (baseURL: string, apiKey: string) => (modelId: string) => LanguageModel;
 }
 export type CLModelWrapper = (modelId: string, settings?: {
     userId?: string;

package/dist/index.js

@@ -13,6 +13,12 @@ var __rest = (this && this.__rest) || function (s, e) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.createCognitiveLayer = createCognitiveLayer;
 const ai_1 = require("ai");
+function isValidId(value) {
+    if (value == null || typeof value !== "string")
+        return false;
+    const trimmed = value.trim();
+    return trimmed !== "" && trimmed !== "null" && trimmed !== "undefined";
+}
 const LOG_LEVEL_PRIORITY = {
     none: 0,
     error: 1,
@@ -166,7 +172,7 @@ function createCognitiveLayer(config) {
     const buildMiddleware = (userId, projectId, sessionId, modelId) => ({
         specificationVersion: 'v3',
         async transformParams({ params }) {
-            if (!userId)
+            if (!isValidId(userId))
                 return params;
             const incomingMessages = Array.isArray(params.prompt)
                 ? params.prompt
@@ -254,7 +260,7 @@ ${userContextBlock || "None"}
                 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));
                 throw err;
             }
-            if (userId && sessionId) {
+            if (isValidId(userId) && isValidId(sessionId)) {
                 const sessionKey = `${userId}:${projectId}:${sessionId}`;
                 const promptMeta = sessionPromptMetadata.get(sessionKey);
                 const messagesInput = params.messages || params.prompt || [];
@@ -288,7 +294,7 @@ ${userContextBlock || "None"}
                 logger.error("doStream failed", err);
                 throw err;
             }
-            if (userId && sessionId) {
+            if (isValidId(userId) && isValidId(sessionId)) {
                 const sessionKey = `${userId}:${projectId}:${sessionId}`;
                 const promptMeta = sessionPromptMetadata.get(sessionKey);
                 const messagesInput = params.messages || params.prompt || [];
@@ -337,7 +343,7 @@ ${userContextBlock || "None"}
         try {
             const gatewayURL = `${baseUrl}/api/cognitive/gateway/${gatewaySlug}`;
             const modelId = originalModel.modelId || 'default';
-            const rawModel = clConfig.providerFactory(gatewayURL)(modelId);
+            const rawModel = clConfig.providerFactory(gatewayURL, clConfig.apiKey)(modelId);
             const session = originalModel[SESSION_KEY];
             if (!session)
                 return rawModel;
@@ -361,7 +367,7 @@ ${userContextBlock || "None"}
         const userId = settings === null || settings === void 0 ? void 0 : settings.userId;
         const projectId = (settings === null || settings === void 0 ? void 0 : settings.projectId) || clConfig.projectId || "default";
         const sessionId = settings === null || settings === void 0 ? void 0 : settings.sessionId;
-        const sessionMissing = !!userId && !sessionId;
+        const sessionMissing = isValidId(userId) && !isValidId(sessionId);
         if (sessionMissing) {
             logger.warn("sessionId is required to log and process memories; skipping logging until provided.");
         }
@@ -370,7 +376,7 @@ ${userContextBlock || "None"}
             middleware: buildMiddleware(userId, projectId, sessionId, modelId),
         });
         // Track session settings on the model for use in cl.streamText/cl.generateText
-        if (userId && sessionId) {
+        if (isValidId(userId) && isValidId(sessionId)) {
             wrappedModel[SESSION_KEY] = { userId, projectId, sessionId };
         }
         return wrappedModel;

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@kognitivedev/vercel-ai-provider",
-    "version": "0.1.6",
+    "version": "0.1.7",
     "main": "dist/index.js",
     "types": "dist/index.d.ts",
     "publishConfig": {

package/src/index.ts

@@ -15,6 +15,12 @@ import {
  */
 export type LogLevel = 'none' | 'error' | 'warn' | 'info' | 'debug';
 
+function isValidId(value: string | undefined | null): value is string {
+    if (value == null || typeof value !== "string") return false;
+    const trimmed = value.trim();
+    return trimmed !== "" && trimmed !== "null" && trimmed !== "undefined";
+}
+
 const LOG_LEVEL_PRIORITY: Record<LogLevel, number> = {
     none: 0,
     error: 1,
@@ -72,7 +78,7 @@ export interface CognitiveLayerConfig {
     /**
      * Factory for creating a provider that routes through a gateway URL.
      */
-    providerFactory?: (baseURL: string) => (modelId: string) => LanguageModel;
+    providerFactory?: (baseURL: string, apiKey: string) => (modelId: string) => LanguageModel;
 }
 
 export type CLModelWrapper = (
@@ -275,7 +281,7 @@ export function createCognitiveLayer(config: {
     const buildMiddleware = (userId: string | undefined, projectId: string, sessionId: string | undefined, modelId: string) => ({
         specificationVersion: 'v3' as const,
         async transformParams({ params }: { params: any }) {
-            if (!userId) return params;
+            if (!isValidId(userId)) return params;
 
             const incomingMessages = Array.isArray((params as any).prompt)
                 ? (params as any).prompt
@@ -373,7 +379,7 @@ ${userContextBlock || "None"}
                 throw err;
             }
 
-            if (userId && sessionId) {
+            if (isValidId(userId) && isValidId(sessionId)) {
                 const sessionKey = `${userId}:${projectId}:${sessionId}`;
                 const promptMeta = sessionPromptMetadata.get(sessionKey);
 
@@ -416,7 +422,7 @@ ${userContextBlock || "None"}
                 throw err;
             }
 
-            if (userId && sessionId) {
+            if (isValidId(userId) && isValidId(sessionId)) {
                 const sessionKey = `${userId}:${projectId}:${sessionId}`;
                 const promptMeta = sessionPromptMetadata.get(sessionKey);
 
@@ -479,7 +485,7 @@ ${userContextBlock || "None"}
         try {
             const gatewayURL = `${baseUrl}/api/cognitive/gateway/${gatewaySlug}`;
             const modelId = (originalModel as any).modelId || 'default';
-            const rawModel = clConfig.providerFactory(gatewayURL)(modelId);
+            const rawModel = clConfig.providerFactory(gatewayURL, clConfig.apiKey)(modelId);
 
             const session = (originalModel as any)[SESSION_KEY];
             if (!session) return rawModel as LanguageModel;
@@ -510,7 +516,7 @@ ${userContextBlock || "None"}
         const userId = settings?.userId;
         const projectId = settings?.projectId || clConfig.projectId || "default";
         const sessionId = settings?.sessionId;
-        const sessionMissing = !!userId && !sessionId;
+        const sessionMissing = isValidId(userId) && !isValidId(sessionId);
 
         if (sessionMissing) {
             logger.warn("sessionId is required to log and process memories; skipping logging until provided.");
@@ -522,7 +528,7 @@ ${userContextBlock || "None"}
         }) as LanguageModel;
 
         // Track session settings on the model for use in cl.streamText/cl.generateText
-        if (userId && sessionId) {
+        if (isValidId(userId) && isValidId(sessionId)) {
             (wrappedModel as any)[SESSION_KEY] = { userId, projectId, sessionId };
         }