@posthog/ai 6.4.1 → 6.4.3

package/dist/index.d.ts

@@ -188,6 +188,7 @@ declare class WrappedModels {
     generateContent(params: GenerateContentParameters & MonitoringParams): Promise<GenerateContentResponse>;
     generateContentStream(params: GenerateContentParameters & MonitoringParams): AsyncGenerator<GenerateContentResponse, void, unknown>;
     private formatInput;
+    private extractSystemInstruction;
     private formatInputForPostHog;
 }
 

package/dist/index.mjs

@@ -6,7 +6,15 @@ import { wrapLanguageModel } from 'ai';
 import AnthropicOriginal from '@anthropic-ai/sdk';
 import { GoogleGenAI } from '@google/genai';
 
-var version = "6.4.1";
+var version = "6.4.3";
+
+// Type guards for safer type checking
+const isString = value => {
+  return typeof value === 'string';
+};
+const isObject = value => {
+  return value !== null && typeof value === 'object' && !Array.isArray(value);
+};
 
 // limit large outputs by truncating to 200kb (approx 200k bytes)
 const MAX_OUTPUT_SIZE = 200000;
@@ -420,14 +428,49 @@ const sendEventToPosthog = async ({
     client.capture(event);
   }
 };
-
-// Type guards for safer type checking
-const isString = value => {
-  return typeof value === 'string';
-};
-const isObject = value => {
-  return value !== null && typeof value === 'object' && !Array.isArray(value);
-};
+function formatOpenAIResponsesInput(input, instructions) {
+  const messages = [];
+  if (instructions) {
+    messages.push({
+      role: 'system',
+      content: instructions
+    });
+  }
+  if (Array.isArray(input)) {
+    for (const item of input) {
+      if (typeof item === 'string') {
+        messages.push({
+          role: 'user',
+          content: item
+        });
+      } else if (item && typeof item === 'object') {
+        const obj = item;
+        const role = isString(obj.role) ? obj.role : 'user';
+        const content = obj.content || obj.text || String(item);
+        messages.push({
+          role,
+          content: String(content)
+        });
+      } else {
+        messages.push({
+          role: 'user',
+          content: String(item)
+        });
+      }
+    }
+  } else if (typeof input === 'string') {
+    messages.push({
+      role: 'user',
+      content: input
+    });
+  } else if (input) {
+    messages.push({
+      role: 'user',
+      content: String(input)
+    });
+  }
+  return messages;
+}
 
 const REDACTED_IMAGE_PLACEHOLDER = '[base64 image redacted]';
 // ============================================
@@ -507,17 +550,6 @@ const sanitizeOpenAIImage = item => {
   }
   return item;
 };
-const sanitizeOpenAIResponseImage = item => {
-  if (!isObject(item)) return item;
-  // Handle input_image format
-  if (item.type === 'input_image' && 'image_url' in item) {
-    return {
-      ...item,
-      image_url: redactBase64DataUrl(item.image_url)
-    };
-  }
-  return item;
-};
 const sanitizeAnthropicImage = item => {
   if (!isObject(item)) return item;
   // Handle Anthropic's image format
@@ -600,9 +632,6 @@ const sanitizeLangChainImage = item => {
 const sanitizeOpenAI = data => {
   return processMessages(data, sanitizeOpenAIImage);
 };
-const sanitizeOpenAIResponse = data => {
-  return processMessages(data, sanitizeOpenAIResponseImage);
-};
 const sanitizeAnthropic = data => {
   return processMessages(data, sanitizeAnthropicImage);
 };
@@ -889,7 +918,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
                 //@ts-expect-error
                 model: openAIParams.model,
                 provider: 'openai',
-                input: sanitizeOpenAIResponse(openAIParams.input),
+                input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
                 output: finalContent,
                 latency,
                 baseURL: this.baseURL,
@@ -906,7 +935,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
                 //@ts-expect-error
                 model: openAIParams.model,
                 provider: 'openai',
-                input: sanitizeOpenAIResponse(openAIParams.input),
+                input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
                 output: [],
                 latency: 0,
                 baseURL: this.baseURL,
@@ -936,7 +965,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
             //@ts-expect-error
             model: openAIParams.model,
             provider: 'openai',
-            input: sanitizeOpenAIResponse(openAIParams.input),
+            input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
             output: formatResponseOpenAI({
               output: result.output
             }),
@@ -961,7 +990,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
           ...posthogParams,
           model: String(openAIParams.model ?? ''),
           provider: 'openai',
-          input: sanitizeOpenAIResponse(openAIParams.input),
+          input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
           output: [],
           latency: 0,
           baseURL: this.baseURL,
@@ -998,7 +1027,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
           ...posthogParams,
           model: String(openAIParams.model ?? ''),
           provider: 'openai',
-          input: sanitizeOpenAIResponse(openAIParams.input),
+          input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
           output: result.output,
           latency,
           baseURL: this.baseURL,
@@ -1019,7 +1048,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
           ...posthogParams,
           model: String(openAIParams.model ?? ''),
           provider: 'openai',
-          input: sanitizeOpenAIResponse(openAIParams.input),
+          input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
           output: [],
           latency: 0,
           baseURL: this.baseURL,
@@ -1231,7 +1260,7 @@ class WrappedCompletions extends AzureOpenAI.Chat.Completions {
                 ...posthogParams,
                 model: openAIParams.model,
                 provider: 'azure',
-                input: openAIParams.messages,
+                input: sanitizeOpenAI(openAIParams.messages),
                 output: formattedOutput,
                 latency,
                 baseURL: this.baseURL,
@@ -1246,7 +1275,7 @@ class WrappedCompletions extends AzureOpenAI.Chat.Completions {
                 ...posthogParams,
                 model: openAIParams.model,
                 provider: 'azure',
-                input: openAIParams.messages,
+                input: sanitizeOpenAI(openAIParams.messages),
                 output: [],
                 latency: 0,
                 baseURL: this.baseURL,
@@ -2258,7 +2287,7 @@ class WrappedModels {
         ...posthogParams,
         model: geminiParams.model,
         provider: 'gemini',
-        input: this.formatInputForPostHog(geminiParams.contents),
+        input: this.formatInputForPostHog(geminiParams),
         output: formatResponseGemini(response),
         latency,
         baseURL: 'https://generativelanguage.googleapis.com',
@@ -2280,7 +2309,7 @@ class WrappedModels {
         ...posthogParams,
         model: geminiParams.model,
         provider: 'gemini',
-        input: this.formatInputForPostHog(geminiParams.contents),
+        input: this.formatInputForPostHog(geminiParams),
         output: [],
         latency,
         baseURL: 'https://generativelanguage.googleapis.com',
@@ -2375,7 +2404,7 @@ class WrappedModels {
         ...posthogParams,
         model: geminiParams.model,
         provider: 'gemini',
-        input: this.formatInputForPostHog(geminiParams.contents),
+        input: this.formatInputForPostHog(geminiParams),
         output,
         latency,
         baseURL: 'https://generativelanguage.googleapis.com',
@@ -2391,7 +2420,7 @@ class WrappedModels {
         ...posthogParams,
         model: geminiParams.model,
         provider: 'gemini',
-        input: this.formatInputForPostHog(geminiParams.contents),
+        input: this.formatInputForPostHog(geminiParams),
         output: [],
         latency,
         baseURL: 'https://generativelanguage.googleapis.com',
@@ -2426,19 +2455,19 @@ class WrappedModels {
           const obj = item;
           if ('text' in obj && obj.text) {
             return {
-              role: obj.role || 'user',
+              role: isString(obj.role) ? obj.role : 'user',
               content: obj.text
             };
           }
           if ('content' in obj && obj.content) {
             return {
-              role: obj.role || 'user',
+              role: isString(obj.role) ? obj.role : 'user',
               content: obj.content
             };
           }
           if ('parts' in obj && Array.isArray(obj.parts)) {
             return {
-              role: obj.role || 'user',
+              role: isString(obj.role) ? obj.role : 'user',
               content: obj.parts.map(part => {
                 if (part && typeof part === 'object' && 'text' in part) {
                   return part.text;
@@ -2474,9 +2503,54 @@ class WrappedModels {
       content: String(contents)
     }];
   }
-  formatInputForPostHog(contents) {
-    const sanitized = sanitizeGemini(contents);
-    return this.formatInput(sanitized);
+  extractSystemInstruction(params) {
+    if (!params || typeof params !== 'object' || !params.config) {
+      return null;
+    }
+    const config = params.config;
+    if (!('systemInstruction' in config)) {
+      return null;
+    }
+    const systemInstruction = config.systemInstruction;
+    if (typeof systemInstruction === 'string') {
+      return systemInstruction;
+    }
+    if (systemInstruction && typeof systemInstruction === 'object' && 'text' in systemInstruction) {
+      return systemInstruction.text;
+    }
+    if (systemInstruction && typeof systemInstruction === 'object' && 'parts' in systemInstruction && Array.isArray(systemInstruction.parts)) {
+      for (const part of systemInstruction.parts) {
+        if (part && typeof part === 'object' && 'text' in part && typeof part.text === 'string') {
+          return part.text;
+        }
+      }
+    }
+    if (Array.isArray(systemInstruction)) {
+      for (const part of systemInstruction) {
+        if (typeof part === 'string') {
+          return part;
+        }
+        if (part && typeof part === 'object' && 'text' in part && typeof part.text === 'string') {
+          return part.text;
+        }
+      }
+    }
+    return null;
+  }
+  formatInputForPostHog(params) {
+    const sanitized = sanitizeGemini(params.contents);
+    const messages = this.formatInput(sanitized);
+    const systemInstruction = this.extractSystemInstruction(params);
+    if (systemInstruction) {
+      const hasSystemMessage = messages.some(msg => msg.role === 'system');
+      if (!hasSystemMessage) {
+        return [{
+          role: 'system',
+          content: systemInstruction
+        }, ...messages];
+      }
+    }
+    return messages;
   }
 }