@posthog/ai 6.4.0 → 6.4.2

package/dist/index.d.ts

@@ -2,6 +2,7 @@ import OpenAIOrignal, { OpenAI, APIPromise, ClientOptions as ClientOptions$1, Az
 import { PostHog } from 'posthog-node';
 import { Stream } from 'openai/streaming';
 import { ParsedResponse } from 'openai/resources/responses/responses';
+import { ResponseCreateParamsWithTools, ExtractParsedContentFromParams } from 'openai/lib/ResponsesParser';
 import { LanguageModelV2 } from '@ai-sdk/provider';
 import AnthropicOriginal, { APIPromise as APIPromise$1 } from '@anthropic-ai/sdk';
 import { Stream as Stream$1 } from '@anthropic-ai/sdk/streaming';
@@ -80,7 +81,7 @@ declare class WrappedResponses extends Responses {
     create(body: ResponsesCreateParamsNonStreaming & MonitoringParams, options?: RequestOptions$2): APIPromise<OpenAI.Responses.Response>;
     create(body: ResponsesCreateParamsStreaming & MonitoringParams, options?: RequestOptions$2): APIPromise<Stream<OpenAI.Responses.ResponseStreamEvent>>;
     create(body: ResponsesCreateParamsBase & MonitoringParams, options?: RequestOptions$2): APIPromise<OpenAI.Responses.Response | Stream<OpenAI.Responses.ResponseStreamEvent>>;
-    parse<Params extends ResponsesCreateParamsBase, ParsedT = any>(body: Params & MonitoringParams, options?: RequestOptions$2): APIPromise<ParsedResponse<ParsedT>>;
+    parse<Params extends ResponseCreateParamsWithTools, ParsedT = ExtractParsedContentFromParams<Params>>(body: Params & MonitoringParams, options?: RequestOptions$2): APIPromise<ParsedResponse<ParsedT>>;
 }
 declare class WrappedEmbeddings$1 extends Embeddings {
     private readonly phClient;
@@ -187,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.0";
+var version = "6.4.2";
+
+// 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);
 };
@@ -822,7 +851,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
         await sendEventToPosthog({
           client: this.phClient,
           ...posthogParams,
-          model: openAIParams.model,
+          model: String(openAIParams.model ?? ''),
           provider: 'openai',
           input: sanitizeOpenAI(openAIParams.messages),
           output: [],
@@ -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
             }),
@@ -959,10 +988,9 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
         await sendEventToPosthog({
           client: this.phClient,
           ...posthogParams,
-          //@ts-expect-error
-          model: openAIParams.model,
+          model: String(openAIParams.model ?? ''),
           provider: 'openai',
-          input: sanitizeOpenAIResponse(openAIParams.input),
+          input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
           output: [],
           latency: 0,
           baseURL: this.baseURL,
@@ -986,7 +1014,6 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
       posthogParams
     } = extractPosthogParams(body);
     const startTime = Date.now();
-    // Create a temporary instance that bypasses our wrapped create method
     const originalCreate = super.create.bind(this);
     const originalSelf = this;
     const tempCreate = originalSelf.create;
@@ -998,10 +1025,9 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
         await sendEventToPosthog({
           client: this.phClient,
           ...posthogParams,
-          //@ts-expect-error
-          model: openAIParams.model,
+          model: String(openAIParams.model ?? ''),
           provider: 'openai',
-          input: sanitizeOpenAIResponse(openAIParams.input),
+          input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
           output: result.output,
           latency,
           baseURL: this.baseURL,
@@ -1020,10 +1046,9 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
         await sendEventToPosthog({
           client: this.phClient,
           ...posthogParams,
-          //@ts-expect-error
-          model: openAIParams.model,
+          model: String(openAIParams.model ?? ''),
           provider: 'openai',
-          input: sanitizeOpenAIResponse(openAIParams.input),
+          input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
           output: [],
           latency: 0,
           baseURL: this.baseURL,
@@ -1235,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,
@@ -1250,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,
@@ -1463,8 +1488,7 @@ class WrappedResponses extends AzureOpenAI.Responses {
       await sendEventToPosthog({
         client: this.phClient,
         ...posthogParams,
-        //@ts-expect-error
-        model: openAIParams.model,
+        model: String(openAIParams.model ?? ''),
         provider: 'azure',
         input: openAIParams.input,
         output: result.output,
@@ -1484,8 +1508,7 @@ class WrappedResponses extends AzureOpenAI.Responses {
       await sendEventToPosthog({
         client: this.phClient,
         ...posthogParams,
-        //@ts-expect-error
-        model: openAIParams.model,
+        model: String(openAIParams.model ?? ''),
         provider: 'azure',
         input: openAIParams.input,
         output: [],
@@ -2264,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',
@@ -2286,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',
@@ -2381,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',
@@ -2397,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',
@@ -2432,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;
@@ -2480,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;
   }
 }