@posthog/ai 5.0.1 → 5.2.0

package/lib/gemini/index.cjs

@@ -0,0 +1,364 @@
+'use strict';
+
+Object.defineProperty(exports, '__esModule', { value: true });
+
+var genai = require('@google/genai');
+var uuid = require('uuid');
+var buffer = require('buffer');
+
+const STRING_FORMAT = 'utf8';
+const getModelParams = params => {
+  if (!params) {
+    return {};
+  }
+  const modelParams = {};
+  const paramKeys = ['temperature', 'max_tokens', 'max_completion_tokens', 'top_p', 'frequency_penalty', 'presence_penalty', 'n', 'stop', 'stream', 'streaming'];
+  for (const key of paramKeys) {
+    if (key in params && params[key] !== undefined) {
+      modelParams[key] = params[key];
+    }
+  }
+  return modelParams;
+};
+const withPrivacyMode = (client, privacyMode, input) => {
+  return client.privacy_mode || privacyMode ? null : input;
+};
+function sanitizeValues(obj) {
+  if (obj === undefined || obj === null) {
+    return obj;
+  }
+  const jsonSafe = JSON.parse(JSON.stringify(obj));
+  if (typeof jsonSafe === 'string') {
+    return buffer.Buffer.from(jsonSafe, STRING_FORMAT).toString(STRING_FORMAT);
+  } else if (Array.isArray(jsonSafe)) {
+    return jsonSafe.map(sanitizeValues);
+  } else if (jsonSafe && typeof jsonSafe === 'object') {
+    return Object.fromEntries(Object.entries(jsonSafe).map(([k, v]) => [k, sanitizeValues(v)]));
+  }
+  return jsonSafe;
+}
+const sendEventToPosthog = async ({
+  client,
+  distinctId,
+  traceId,
+  model,
+  provider,
+  input,
+  output,
+  latency,
+  baseURL,
+  params,
+  httpStatus = 200,
+  usage = {},
+  isError = false,
+  error,
+  tools,
+  captureImmediate = false
+}) => {
+  if (!client.capture) {
+    return Promise.resolve();
+  }
+  // sanitize input and output for UTF-8 validity
+  const safeInput = sanitizeValues(input);
+  const safeOutput = sanitizeValues(output);
+  const safeError = sanitizeValues(error);
+  let errorData = {};
+  if (isError) {
+    errorData = {
+      $ai_is_error: true,
+      $ai_error: safeError
+    };
+  }
+  let costOverrideData = {};
+  if (params.posthogCostOverride) {
+    const inputCostUSD = (params.posthogCostOverride.inputCost ?? 0) * (usage.inputTokens ?? 0);
+    const outputCostUSD = (params.posthogCostOverride.outputCost ?? 0) * (usage.outputTokens ?? 0);
+    costOverrideData = {
+      $ai_input_cost_usd: inputCostUSD,
+      $ai_output_cost_usd: outputCostUSD,
+      $ai_total_cost_usd: inputCostUSD + outputCostUSD
+    };
+  }
+  const additionalTokenValues = {
+    ...(usage.reasoningTokens ? {
+      $ai_reasoning_tokens: usage.reasoningTokens
+    } : {}),
+    ...(usage.cacheReadInputTokens ? {
+      $ai_cache_read_input_tokens: usage.cacheReadInputTokens
+    } : {}),
+    ...(usage.cacheCreationInputTokens ? {
+      $ai_cache_creation_input_tokens: usage.cacheCreationInputTokens
+    } : {})
+  };
+  const properties = {
+    $ai_provider: params.posthogProviderOverride ?? provider,
+    $ai_model: params.posthogModelOverride ?? model,
+    $ai_model_parameters: getModelParams(params),
+    $ai_input: withPrivacyMode(client, params.posthogPrivacyMode ?? false, safeInput),
+    $ai_output_choices: withPrivacyMode(client, params.posthogPrivacyMode ?? false, safeOutput),
+    $ai_http_status: httpStatus,
+    $ai_input_tokens: usage.inputTokens ?? 0,
+    $ai_output_tokens: usage.outputTokens ?? 0,
+    ...additionalTokenValues,
+    $ai_latency: latency,
+    $ai_trace_id: traceId,
+    $ai_base_url: baseURL,
+    ...params.posthogProperties,
+    ...(distinctId ? {} : {
+      $process_person_profile: false
+    }),
+    ...(tools ? {
+      $ai_tools: tools
+    } : {}),
+    ...errorData,
+    ...costOverrideData
+  };
+  const event = {
+    distinctId: distinctId ?? traceId,
+    event: '$ai_generation',
+    properties,
+    groups: params.posthogGroups
+  };
+  if (captureImmediate) {
+    // await capture promise to send single event in serverless environments
+    await client.captureImmediate(event);
+  } else {
+    client.capture(event);
+  }
+};
+
+// Types from @google/genai
+
+class PostHogGoogleGenAI {
+  constructor(config) {
+    const {
+      posthog,
+      ...geminiConfig
+    } = config;
+    this.phClient = posthog;
+    this.client = new genai.GoogleGenAI(geminiConfig);
+    this.models = new WrappedModels(this.client, this.phClient);
+  }
+}
+class WrappedModels {
+  constructor(client, phClient) {
+    this.client = client;
+    this.phClient = phClient;
+  }
+  async generateContent(params) {
+    const {
+      posthogDistinctId,
+      posthogTraceId,
+      posthogProperties,
+      posthogGroups,
+      posthogCaptureImmediate,
+      ...geminiParams
+    } = params;
+    const traceId = posthogTraceId ?? uuid.v4();
+    const startTime = Date.now();
+    try {
+      const response = await this.client.models.generateContent(geminiParams);
+      const latency = (Date.now() - startTime) / 1000;
+      await sendEventToPosthog({
+        client: this.phClient,
+        distinctId: posthogDistinctId,
+        traceId,
+        model: geminiParams.model,
+        provider: 'gemini',
+        input: this.formatInput(geminiParams.contents),
+        output: this.formatOutput(response),
+        latency,
+        baseURL: 'https://generativelanguage.googleapis.com',
+        params: params,
+        httpStatus: 200,
+        usage: {
+          inputTokens: response.usageMetadata?.promptTokenCount ?? 0,
+          outputTokens: response.usageMetadata?.candidatesTokenCount ?? 0
+        },
+        captureImmediate: posthogCaptureImmediate
+      });
+      return response;
+    } catch (error) {
+      const latency = (Date.now() - startTime) / 1000;
+      await sendEventToPosthog({
+        client: this.phClient,
+        distinctId: posthogDistinctId,
+        traceId,
+        model: geminiParams.model,
+        provider: 'gemini',
+        input: this.formatInput(geminiParams.contents),
+        output: [],
+        latency,
+        baseURL: 'https://generativelanguage.googleapis.com',
+        params: params,
+        httpStatus: error?.status ?? 500,
+        usage: {
+          inputTokens: 0,
+          outputTokens: 0
+        },
+        isError: true,
+        error: JSON.stringify(error),
+        captureImmediate: posthogCaptureImmediate
+      });
+      throw error;
+    }
+  }
+  async *generateContentStream(params) {
+    const {
+      posthogDistinctId,
+      posthogTraceId,
+      posthogProperties,
+      posthogGroups,
+      posthogCaptureImmediate,
+      ...geminiParams
+    } = params;
+    const traceId = posthogTraceId ?? uuid.v4();
+    const startTime = Date.now();
+    let accumulatedContent = '';
+    let usage = {
+      inputTokens: 0,
+      outputTokens: 0
+    };
+    try {
+      const stream = await this.client.models.generateContentStream(geminiParams);
+      for await (const chunk of stream) {
+        if (chunk.text) {
+          accumulatedContent += chunk.text;
+        }
+        if (chunk.usageMetadata) {
+          usage = {
+            inputTokens: chunk.usageMetadata.promptTokenCount ?? 0,
+            outputTokens: chunk.usageMetadata.candidatesTokenCount ?? 0
+          };
+        }
+        yield chunk;
+      }
+      const latency = (Date.now() - startTime) / 1000;
+      await sendEventToPosthog({
+        client: this.phClient,
+        distinctId: posthogDistinctId,
+        traceId,
+        model: geminiParams.model,
+        provider: 'gemini',
+        input: this.formatInput(geminiParams.contents),
+        output: [{
+          content: accumulatedContent,
+          role: 'assistant'
+        }],
+        latency,
+        baseURL: 'https://generativelanguage.googleapis.com',
+        params: params,
+        httpStatus: 200,
+        usage,
+        captureImmediate: posthogCaptureImmediate
+      });
+    } catch (error) {
+      const latency = (Date.now() - startTime) / 1000;
+      await sendEventToPosthog({
+        client: this.phClient,
+        distinctId: posthogDistinctId,
+        traceId,
+        model: geminiParams.model,
+        provider: 'gemini',
+        input: this.formatInput(geminiParams.contents),
+        output: [],
+        latency,
+        baseURL: 'https://generativelanguage.googleapis.com',
+        params: params,
+        httpStatus: error?.status ?? 500,
+        usage: {
+          inputTokens: 0,
+          outputTokens: 0
+        },
+        isError: true,
+        error: JSON.stringify(error),
+        captureImmediate: posthogCaptureImmediate
+      });
+      throw error;
+    }
+  }
+  formatInput(contents) {
+    if (typeof contents === 'string') {
+      return [{
+        role: 'user',
+        content: contents
+      }];
+    }
+    if (Array.isArray(contents)) {
+      return contents.map(item => {
+        if (typeof item === 'string') {
+          return {
+            role: 'user',
+            content: item
+          };
+        }
+        if (item && typeof item === 'object') {
+          if (item.text) {
+            return {
+              role: item.role || 'user',
+              content: item.text
+            };
+          }
+          if (item.content) {
+            return {
+              role: item.role || 'user',
+              content: item.content
+            };
+          }
+        }
+        return {
+          role: 'user',
+          content: String(item)
+        };
+      });
+    }
+    if (contents && typeof contents === 'object') {
+      if (contents.text) {
+        return [{
+          role: 'user',
+          content: contents.text
+        }];
+      }
+      if (contents.content) {
+        return [{
+          role: 'user',
+          content: contents.content
+        }];
+      }
+    }
+    return [{
+      role: 'user',
+      content: String(contents)
+    }];
+  }
+  formatOutput(response) {
+    if (response.text) {
+      return [{
+        role: 'assistant',
+        content: response.text
+      }];
+    }
+    if (response.candidates && Array.isArray(response.candidates)) {
+      return response.candidates.map(candidate => {
+        if (candidate.content && candidate.content.parts) {
+          const text = candidate.content.parts.filter(part => part.text).map(part => part.text).join('');
+          return {
+            role: 'assistant',
+            content: text
+          };
+        }
+        return {
+          role: 'assistant',
+          content: String(candidate)
+        };
+      });
+    }
+    return [];
+  }
+}
+
+exports.Gemini = PostHogGoogleGenAI;
+exports.PostHogGoogleGenAI = PostHogGoogleGenAI;
+exports.WrappedModels = WrappedModels;
+exports["default"] = PostHogGoogleGenAI;
+//# sourceMappingURL=index.cjs.map

package/lib/gemini/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.cjs","sources":["../../src/utils.ts","../../src/gemini/index.ts"],"sourcesContent":["import { PostHog } from 'posthog-node'\nimport { Buffer } from 'buffer'\nimport OpenAIOrignal from 'openai'\nimport AnthropicOriginal from '@anthropic-ai/sdk'\n\ntype ChatCompletionCreateParamsBase = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParams\ntype MessageCreateParams = AnthropicOriginal.Messages.MessageCreateParams\ntype ResponseCreateParams = OpenAIOrignal.Responses.ResponseCreateParams\n\n// limit large outputs by truncating to 200kb (approx 200k bytes)\nexport const MAX_OUTPUT_SIZE = 200000\nconst STRING_FORMAT = 'utf8'\n\nexport interface MonitoringParams {\n  posthogDistinctId?: string\n  posthogTraceId?: string\n  posthogProperties?: Record<string, any>\n  posthogPrivacyMode?: boolean\n  posthogGroups?: Record<string, any>\n  posthogModelOverride?: string\n  posthogProviderOverride?: string\n  posthogCostOverride?: CostOverride\n  posthogCaptureImmediate?: boolean\n}\n\nexport interface CostOverride {\n  inputCost: number\n  outputCost: number\n}\n\nexport const getModelParams = (\n  params: ((ChatCompletionCreateParamsBase | MessageCreateParams | ResponseCreateParams) & MonitoringParams) | null\n): Record<string, any> => {\n  if (!params) {\n    return {}\n  }\n  const modelParams: Record<string, any> = {}\n  const paramKeys = [\n    'temperature',\n    'max_tokens',\n    'max_completion_tokens',\n    'top_p',\n    'frequency_penalty',\n    'presence_penalty',\n    'n',\n    'stop',\n    'stream',\n    'streaming',\n  ] as const\n\n  for (const key of paramKeys) {\n    if (key in params && (params as any)[key] !== undefined) {\n      modelParams[key] = (params as any)[key]\n    }\n  }\n  return modelParams\n}\n\n/**\n * Helper to format responses (non-streaming) for consumption, mirroring Python's openai vs. anthropic approach.\n */\nexport const formatResponse = (response: any, provider: string): Array<{ role: string; content: string }> => {\n  if (!response) {\n    return []\n  }\n  if (provider === 'anthropic') {\n    return formatResponseAnthropic(response)\n  } else if (provider === 'openai') {\n    return formatResponseOpenAI(response)\n  } else if (provider === 'gemini') {\n    return formatResponseGemini(response)\n  }\n  return []\n}\n\nexport const formatResponseAnthropic = (response: any): Array<{ role: string; content: string }> => {\n  // Example approach if \"response.content\" holds array of text segments, etc.\n  const output: Array<{ role: string; content: string }> = []\n  for (const choice of response.content ?? []) {\n    if (choice?.text) {\n      output.push({\n        role: 'assistant',\n        content: choice.text,\n      })\n    }\n  }\n  return output\n}\n\nexport const formatResponseOpenAI = (response: any): Array<{ role: string; content: string }> => {\n  const output: Array<{ role: string; content: string }> = []\n  for (const choice of response.choices ?? []) {\n    if (choice.message?.content) {\n      output.push({\n        role: choice.message.role,\n        content: choice.message.content,\n      })\n    }\n  }\n  return output\n}\n\nexport const formatResponseGemini = (response: any): Array<{ role: string; content: string }> => {\n  const output: Array<{ role: string; content: string }> = []\n\n  if (response.text) {\n    output.push({\n      role: 'assistant',\n      content: response.text,\n    })\n    return output\n  }\n\n  if (response.candidates && Array.isArray(response.candidates)) {\n    for (const candidate of response.candidates) {\n      if (candidate.content && candidate.content.parts) {\n        const text = candidate.content.parts\n          .filter((part: any) => part.text)\n          .map((part: any) => part.text)\n          .join('')\n        if (text) {\n          output.push({\n            role: 'assistant',\n            content: text,\n          })\n        }\n      }\n    }\n  }\n\n  return output\n}\n\nexport const mergeSystemPrompt = (params: MessageCreateParams & MonitoringParams, provider: string): any => {\n  if (provider == 'anthropic') {\n    const messages = params.messages || []\n    if (!(params as any).system) {\n      return messages\n    }\n    const systemMessage = (params as any).system\n    return [{ role: 'system', content: systemMessage }, ...messages]\n  }\n  return params.messages\n}\n\nexport const withPrivacyMode = (client: PostHog, privacyMode: boolean, input: any): any => {\n  return (client as any).privacy_mode || privacyMode ? null : input\n}\n\nexport const truncate = (str: string): string => {\n  try {\n    const buffer = Buffer.from(str, STRING_FORMAT)\n    if (buffer.length <= MAX_OUTPUT_SIZE) {\n      return str\n    }\n    const truncatedBuffer = buffer.slice(0, MAX_OUTPUT_SIZE)\n    return `${truncatedBuffer.toString(STRING_FORMAT)}... [truncated]`\n  } catch (error) {\n    console.error('Error truncating, likely not a string')\n    return str\n  }\n}\n\nexport type SendEventToPosthogParams = {\n  client: PostHog\n  distinctId?: string\n  traceId: string\n  model: string\n  provider: string\n  input: any\n  output: any\n  latency: number\n  baseURL: string\n  httpStatus: number\n  usage?: {\n    inputTokens?: number\n    outputTokens?: number\n    reasoningTokens?: any\n    cacheReadInputTokens?: any\n    cacheCreationInputTokens?: any\n  }\n  params: (ChatCompletionCreateParamsBase | MessageCreateParams | ResponseCreateParams) & MonitoringParams\n  isError?: boolean\n  error?: string\n  tools?: any\n  captureImmediate?: boolean\n}\n\nfunction sanitizeValues(obj: any): any {\n  if (obj === undefined || obj === null) {\n    return obj\n  }\n  const jsonSafe = JSON.parse(JSON.stringify(obj))\n  if (typeof jsonSafe === 'string') {\n    return Buffer.from(jsonSafe, STRING_FORMAT).toString(STRING_FORMAT)\n  } else if (Array.isArray(jsonSafe)) {\n    return jsonSafe.map(sanitizeValues)\n  } else if (jsonSafe && typeof jsonSafe === 'object') {\n    return Object.fromEntries(Object.entries(jsonSafe).map(([k, v]) => [k, sanitizeValues(v)]))\n  }\n  return jsonSafe\n}\n\nexport const sendEventToPosthog = async ({\n  client,\n  distinctId,\n  traceId,\n  model,\n  provider,\n  input,\n  output,\n  latency,\n  baseURL,\n  params,\n  httpStatus = 200,\n  usage = {},\n  isError = false,\n  error,\n  tools,\n  captureImmediate = false,\n}: SendEventToPosthogParams): Promise<void> => {\n  if (!client.capture) {\n    return Promise.resolve()\n  }\n  // sanitize input and output for UTF-8 validity\n  const safeInput = sanitizeValues(input)\n  const safeOutput = sanitizeValues(output)\n  const safeError = sanitizeValues(error)\n\n  let errorData = {}\n  if (isError) {\n    errorData = {\n      $ai_is_error: true,\n      $ai_error: safeError,\n    }\n  }\n  let costOverrideData = {}\n  if (params.posthogCostOverride) {\n    const inputCostUSD = (params.posthogCostOverride.inputCost ?? 0) * (usage.inputTokens ?? 0)\n    const outputCostUSD = (params.posthogCostOverride.outputCost ?? 0) * (usage.outputTokens ?? 0)\n    costOverrideData = {\n      $ai_input_cost_usd: inputCostUSD,\n      $ai_output_cost_usd: outputCostUSD,\n      $ai_total_cost_usd: inputCostUSD + outputCostUSD,\n    }\n  }\n\n  const additionalTokenValues = {\n    ...(usage.reasoningTokens ? { $ai_reasoning_tokens: usage.reasoningTokens } : {}),\n    ...(usage.cacheReadInputTokens ? { $ai_cache_read_input_tokens: usage.cacheReadInputTokens } : {}),\n    ...(usage.cacheCreationInputTokens ? { $ai_cache_creation_input_tokens: usage.cacheCreationInputTokens } : {}),\n  }\n\n  const properties = {\n    $ai_provider: params.posthogProviderOverride ?? provider,\n    $ai_model: params.posthogModelOverride ?? model,\n    $ai_model_parameters: getModelParams(params),\n    $ai_input: withPrivacyMode(client, params.posthogPrivacyMode ?? false, safeInput),\n    $ai_output_choices: withPrivacyMode(client, params.posthogPrivacyMode ?? false, safeOutput),\n    $ai_http_status: httpStatus,\n    $ai_input_tokens: usage.inputTokens ?? 0,\n    $ai_output_tokens: usage.outputTokens ?? 0,\n    ...additionalTokenValues,\n    $ai_latency: latency,\n    $ai_trace_id: traceId,\n    $ai_base_url: baseURL,\n    ...params.posthogProperties,\n    ...(distinctId ? {} : { $process_person_profile: false }),\n    ...(tools ? { $ai_tools: tools } : {}),\n    ...errorData,\n    ...costOverrideData,\n  }\n\n  const event = {\n    distinctId: distinctId ?? traceId,\n    event: '$ai_generation',\n    properties,\n    groups: params.posthogGroups,\n  }\n\n  if (captureImmediate) {\n    // await capture promise to send single event in serverless environments\n    await client.captureImmediate(event)\n  } else {\n    client.capture(event)\n  }\n}\n","import { GoogleGenAI } from '@google/genai'\nimport { PostHog } from 'posthog-node'\nimport { v4 as uuidv4 } from 'uuid'\nimport { MonitoringParams, sendEventToPosthog } from '../utils'\n\n// Types from @google/genai\ntype GenerateContentRequest = {\n  model: string\n  contents: any\n  config?: any\n  [key: string]: any\n}\n\ntype GenerateContentResponse = {\n  text?: string\n  candidates?: any[]\n  usageMetadata?: {\n    promptTokenCount?: number\n    candidatesTokenCount?: number\n    totalTokenCount?: number\n  }\n  [key: string]: any\n}\n\ninterface MonitoringGeminiConfig {\n  apiKey?: string\n  vertexai?: boolean\n  project?: string\n  location?: string\n  apiVersion?: string\n  posthog: PostHog\n}\n\nexport class PostHogGoogleGenAI {\n  private readonly phClient: PostHog\n  private readonly client: GoogleGenAI\n  public models: WrappedModels\n\n  constructor(config: MonitoringGeminiConfig) {\n    const { posthog, ...geminiConfig } = config\n    this.phClient = posthog\n    this.client = new GoogleGenAI(geminiConfig)\n    this.models = new WrappedModels(this.client, this.phClient)\n  }\n}\n\nexport class WrappedModels {\n  private readonly phClient: PostHog\n  private readonly client: GoogleGenAI\n\n  constructor(client: GoogleGenAI, phClient: PostHog) {\n    this.client = client\n    this.phClient = phClient\n  }\n\n  public async generateContent(params: GenerateContentRequest & MonitoringParams): Promise<GenerateContentResponse> {\n    const {\n      posthogDistinctId,\n      posthogTraceId,\n      posthogProperties,\n      posthogGroups,\n      posthogCaptureImmediate,\n      ...geminiParams\n    } = params\n\n    const traceId = posthogTraceId ?? uuidv4()\n    const startTime = Date.now()\n\n    try {\n      const response = await this.client.models.generateContent(geminiParams)\n      const latency = (Date.now() - startTime) / 1000\n\n      await sendEventToPosthog({\n        client: this.phClient,\n        distinctId: posthogDistinctId,\n        traceId,\n        model: geminiParams.model,\n        provider: 'gemini',\n        input: this.formatInput(geminiParams.contents),\n        output: this.formatOutput(response),\n        latency,\n        baseURL: 'https://generativelanguage.googleapis.com',\n        params: params as any,\n        httpStatus: 200,\n        usage: {\n          inputTokens: response.usageMetadata?.promptTokenCount ?? 0,\n          outputTokens: response.usageMetadata?.candidatesTokenCount ?? 0,\n        },\n        captureImmediate: posthogCaptureImmediate,\n      })\n\n      return response\n    } catch (error: any) {\n      const latency = (Date.now() - startTime) / 1000\n      await sendEventToPosthog({\n        client: this.phClient,\n        distinctId: posthogDistinctId,\n        traceId,\n        model: geminiParams.model,\n        provider: 'gemini',\n        input: this.formatInput(geminiParams.contents),\n        output: [],\n        latency,\n        baseURL: 'https://generativelanguage.googleapis.com',\n        params: params as any,\n        httpStatus: error?.status ?? 500,\n        usage: {\n          inputTokens: 0,\n          outputTokens: 0,\n        },\n        isError: true,\n        error: JSON.stringify(error),\n        captureImmediate: posthogCaptureImmediate,\n      })\n      throw error\n    }\n  }\n\n  public async *generateContentStream(\n    params: GenerateContentRequest & MonitoringParams\n  ): AsyncGenerator<any, void, unknown> {\n    const {\n      posthogDistinctId,\n      posthogTraceId,\n      posthogProperties,\n      posthogGroups,\n      posthogCaptureImmediate,\n      ...geminiParams\n    } = params\n\n    const traceId = posthogTraceId ?? uuidv4()\n    const startTime = Date.now()\n    let accumulatedContent = ''\n    let usage = {\n      inputTokens: 0,\n      outputTokens: 0,\n    }\n\n    try {\n      const stream = await this.client.models.generateContentStream(geminiParams)\n\n      for await (const chunk of stream) {\n        if (chunk.text) {\n          accumulatedContent += chunk.text\n        }\n        if (chunk.usageMetadata) {\n          usage = {\n            inputTokens: chunk.usageMetadata.promptTokenCount ?? 0,\n            outputTokens: chunk.usageMetadata.candidatesTokenCount ?? 0,\n          }\n        }\n        yield chunk\n      }\n\n      const latency = (Date.now() - startTime) / 1000\n      await sendEventToPosthog({\n        client: this.phClient,\n        distinctId: posthogDistinctId,\n        traceId,\n        model: geminiParams.model,\n        provider: 'gemini',\n        input: this.formatInput(geminiParams.contents),\n        output: [{ content: accumulatedContent, role: 'assistant' }],\n        latency,\n        baseURL: 'https://generativelanguage.googleapis.com',\n        params: params as any,\n        httpStatus: 200,\n        usage,\n        captureImmediate: posthogCaptureImmediate,\n      })\n    } catch (error: any) {\n      const latency = (Date.now() - startTime) / 1000\n      await sendEventToPosthog({\n        client: this.phClient,\n        distinctId: posthogDistinctId,\n        traceId,\n        model: geminiParams.model,\n        provider: 'gemini',\n        input: this.formatInput(geminiParams.contents),\n        output: [],\n        latency,\n        baseURL: 'https://generativelanguage.googleapis.com',\n        params: params as any,\n        httpStatus: error?.status ?? 500,\n        usage: {\n          inputTokens: 0,\n          outputTokens: 0,\n        },\n        isError: true,\n        error: JSON.stringify(error),\n        captureImmediate: posthogCaptureImmediate,\n      })\n      throw error\n    }\n  }\n\n  private formatInput(contents: any): Array<{ role: string; content: string }> {\n    if (typeof contents === 'string') {\n      return [{ role: 'user', content: contents }]\n    }\n\n    if (Array.isArray(contents)) {\n      return contents.map((item) => {\n        if (typeof item === 'string') {\n          return { role: 'user', content: item }\n        }\n        if (item && typeof item === 'object') {\n          if (item.text) {\n            return { role: item.role || 'user', content: item.text }\n          }\n          if (item.content) {\n            return { role: item.role || 'user', content: item.content }\n          }\n        }\n        return { role: 'user', content: String(item) }\n      })\n    }\n\n    if (contents && typeof contents === 'object') {\n      if (contents.text) {\n        return [{ role: 'user', content: contents.text }]\n      }\n      if (contents.content) {\n        return [{ role: 'user', content: contents.content }]\n      }\n    }\n\n    return [{ role: 'user', content: String(contents) }]\n  }\n\n  private formatOutput(response: GenerateContentResponse): Array<{ role: string; content: string }> {\n    if (response.text) {\n      return [{ role: 'assistant', content: response.text }]\n    }\n\n    if (response.candidates && Array.isArray(response.candidates)) {\n      return response.candidates.map((candidate) => {\n        if (candidate.content && candidate.content.parts) {\n          const text = candidate.content.parts\n            .filter((part: any) => part.text)\n            .map((part: any) => part.text)\n            .join('')\n          return { role: 'assistant', content: text }\n        }\n        return { role: 'assistant', content: String(candidate) }\n      })\n    }\n\n    return []\n  }\n}\n\nexport default PostHogGoogleGenAI\nexport { PostHogGoogleGenAI as Gemini }\n"],"names":["STRING_FORMAT","getModelParams","params","modelParams","paramKeys","key","undefined","withPrivacyMode","client","privacyMode","input","privacy_mode","sanitizeValues","obj","jsonSafe","JSON","parse","stringify","Buffer","from","toString","Array","isArray","map","Object","fromEntries","entries","k","v","sendEventToPosthog","distinctId","traceId","model","provider","output","latency","baseURL","httpStatus","usage","isError","error","tools","captureImmediate","capture","Promise","resolve","safeInput","safeOutput","safeError","errorData","$ai_is_error","$ai_error","costOverrideData","posthogCostOverride","inputCostUSD","inputCost","inputTokens","outputCostUSD","outputCost","outputTokens","$ai_input_cost_usd","$ai_output_cost_usd","$ai_total_cost_usd","additionalTokenValues","reasoningTokens","$ai_reasoning_tokens","cacheReadInputTokens","$ai_cache_read_input_tokens","cacheCreationInputTokens","$ai_cache_creation_input_tokens","properties","$ai_provider","posthogProviderOverride","$ai_model","posthogModelOverride","$ai_model_parameters","$ai_input","posthogPrivacyMode","$ai_output_choices","$ai_http_status","$ai_input_tokens","$ai_output_tokens","$ai_latency","$ai_trace_id","$ai_base_url","posthogProperties","$process_person_profile","$ai_tools","event","groups","posthogGroups","PostHogGoogleGenAI","constructor","config","posthog","geminiConfig","phClient","GoogleGenAI","models","WrappedModels","generateContent","posthogDistinctId","posthogTraceId","posthogCaptureImmediate","geminiParams","uuidv4","startTime","Date","now","response","formatInput","contents","formatOutput","usageMetadata","promptTokenCount","candidatesTokenCount","status","generateContentStream","accumulatedContent","stream","chunk","text","content","role","item","String","candidates","candidate","parts","filter","part","join"],"mappings":";;;;;;;;AAWA,MAAMA,aAAa,GAAG,MAAM,CAAA;AAmBrB,MAAMC,cAAc,GACzBC,MAAiH,IACzF;EACxB,IAAI,CAACA,MAAM,EAAE;AACX,IAAA,OAAO,EAAE,CAAA;AACX,GAAA;EACA,MAAMC,WAAgC,GAAG,EAAE,CAAA;EAC3C,MAAMC,SAAS,GAAG,CAChB,aAAa,EACb,YAAY,EACZ,uBAAuB,EACvB,OAAO,EACP,mBAAmB,EACnB,kBAAkB,EAClB,GAAG,EACH,MAAM,EACN,QAAQ,EACR,WAAW,CACH,CAAA;AAEV,EAAA,KAAK,MAAMC,GAAG,IAAID,SAAS,EAAE;IAC3B,IAAIC,GAAG,IAAIH,MAAM,IAAKA,MAAM,CAASG,GAAG,CAAC,KAAKC,SAAS,EAAE;AACvDH,MAAAA,WAAW,CAACE,GAAG,CAAC,GAAIH,MAAM,CAASG,GAAG,CAAC,CAAA;AACzC,KAAA;AACF,GAAA;AACA,EAAA,OAAOF,WAAW,CAAA;AACpB,CAAC,CAAA;AAyFM,MAAMI,eAAe,GAAGA,CAACC,MAAe,EAAEC,WAAoB,EAAEC,KAAU,KAAU;EACzF,OAAQF,MAAM,CAASG,YAAY,IAAIF,WAAW,GAAG,IAAI,GAAGC,KAAK,CAAA;AACnE,CAAC,CAAA;AAyCD,SAASE,cAAcA,CAACC,GAAQ,EAAO;AACrC,EAAA,IAAIA,GAAG,KAAKP,SAAS,IAAIO,GAAG,KAAK,IAAI,EAAE;AACrC,IAAA,OAAOA,GAAG,CAAA;AACZ,GAAA;AACA,EAAA,MAAMC,QAAQ,GAAGC,IAAI,CAACC,KAAK,CAACD,IAAI,CAACE,SAAS,CAACJ,GAAG,CAAC,CAAC,CAAA;AAChD,EAAA,IAAI,OAAOC,QAAQ,KAAK,QAAQ,EAAE;AAChC,IAAA,OAAOI,aAAM,CAACC,IAAI,CAACL,QAAQ,EAAEd,aAAa,CAAC,CAACoB,QAAQ,CAACpB,aAAa,CAAC,CAAA;GACpE,MAAM,IAAIqB,KAAK,CAACC,OAAO,CAACR,QAAQ,CAAC,EAAE;AAClC,IAAA,OAAOA,QAAQ,CAACS,GAAG,CAACX,cAAc,CAAC,CAAA;GACpC,MAAM,IAAIE,QAAQ,IAAI,OAAOA,QAAQ,KAAK,QAAQ,EAAE;AACnD,IAAA,OAAOU,MAAM,CAACC,WAAW,CAACD,MAAM,CAACE,OAAO,CAACZ,QAAQ,CAAC,CAACS,GAAG,CAAC,CAAC,CAACI,CAAC,EAAEC,CAAC,CAAC,KAAK,CAACD,CAAC,EAAEf,cAAc,CAACgB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;AAC7F,GAAA;AACA,EAAA,OAAOd,QAAQ,CAAA;AACjB,CAAA;AAEO,MAAMe,kBAAkB,GAAG,OAAO;EACvCrB,MAAM;EACNsB,UAAU;EACVC,OAAO;EACPC,KAAK;EACLC,QAAQ;EACRvB,KAAK;EACLwB,MAAM;EACNC,OAAO;EACPC,OAAO;EACPlC,MAAM;AACNmC,EAAAA,UAAU,GAAG,GAAG;EAChBC,KAAK,GAAG,EAAE;AACVC,EAAAA,OAAO,GAAG,KAAK;EACfC,KAAK;EACLC,KAAK;AACLC,EAAAA,gBAAgB,GAAG,KAAA;AACK,CAAC,KAAoB;AAC7C,EAAA,IAAI,CAAClC,MAAM,CAACmC,OAAO,EAAE;AACnB,IAAA,OAAOC,OAAO,CAACC,OAAO,EAAE,CAAA;AAC1B,GAAA;AACA;AACA,EAAA,MAAMC,SAAS,GAAGlC,cAAc,CAACF,KAAK,CAAC,CAAA;AACvC,EAAA,MAAMqC,UAAU,GAAGnC,cAAc,CAACsB,MAAM,CAAC,CAAA;AACzC,EAAA,MAAMc,SAAS,GAAGpC,cAAc,CAAC4B,KAAK,CAAC,CAAA;EAEvC,IAAIS,SAAS,GAAG,EAAE,CAAA;AAClB,EAAA,IAAIV,OAAO,EAAE;AACXU,IAAAA,SAAS,GAAG;AACVC,MAAAA,YAAY,EAAE,IAAI;AAClBC,MAAAA,SAAS,EAAEH,SAAAA;KACZ,CAAA;AACH,GAAA;EACA,IAAII,gBAAgB,GAAG,EAAE,CAAA;EACzB,IAAIlD,MAAM,CAACmD,mBAAmB,EAAE;AAC9B,IAAA,MAAMC,YAAY,GAAG,CAACpD,MAAM,CAACmD,mBAAmB,CAACE,SAAS,IAAI,CAAC,KAAKjB,KAAK,CAACkB,WAAW,IAAI,CAAC,CAAC,CAAA;AAC3F,IAAA,MAAMC,aAAa,GAAG,CAACvD,MAAM,CAACmD,mBAAmB,CAACK,UAAU,IAAI,CAAC,KAAKpB,KAAK,CAACqB,YAAY,IAAI,CAAC,CAAC,CAAA;AAC9FP,IAAAA,gBAAgB,GAAG;AACjBQ,MAAAA,kBAAkB,EAAEN,YAAY;AAChCO,MAAAA,mBAAmB,EAAEJ,aAAa;MAClCK,kBAAkB,EAAER,YAAY,GAAGG,aAAAA;KACpC,CAAA;AACH,GAAA;AAEA,EAAA,MAAMM,qBAAqB,GAAG;IAC5B,IAAIzB,KAAK,CAAC0B,eAAe,GAAG;MAAEC,oBAAoB,EAAE3B,KAAK,CAAC0B,eAAAA;KAAiB,GAAG,EAAE,CAAC;IACjF,IAAI1B,KAAK,CAAC4B,oBAAoB,GAAG;MAAEC,2BAA2B,EAAE7B,KAAK,CAAC4B,oBAAAA;KAAsB,GAAG,EAAE,CAAC;IAClG,IAAI5B,KAAK,CAAC8B,wBAAwB,GAAG;MAAEC,+BAA+B,EAAE/B,KAAK,CAAC8B,wBAAAA;KAA0B,GAAG,EAAE,CAAA;GAC9G,CAAA;AAED,EAAA,MAAME,UAAU,GAAG;AACjBC,IAAAA,YAAY,EAAErE,MAAM,CAACsE,uBAAuB,IAAIvC,QAAQ;AACxDwC,IAAAA,SAAS,EAAEvE,MAAM,CAACwE,oBAAoB,IAAI1C,KAAK;AAC/C2C,IAAAA,oBAAoB,EAAE1E,cAAc,CAACC,MAAM,CAAC;AAC5C0E,IAAAA,SAAS,EAAErE,eAAe,CAACC,MAAM,EAAEN,MAAM,CAAC2E,kBAAkB,IAAI,KAAK,EAAE/B,SAAS,CAAC;AACjFgC,IAAAA,kBAAkB,EAAEvE,eAAe,CAACC,MAAM,EAAEN,MAAM,CAAC2E,kBAAkB,IAAI,KAAK,EAAE9B,UAAU,CAAC;AAC3FgC,IAAAA,eAAe,EAAE1C,UAAU;AAC3B2C,IAAAA,gBAAgB,EAAE1C,KAAK,CAACkB,WAAW,IAAI,CAAC;AACxCyB,IAAAA,iBAAiB,EAAE3C,KAAK,CAACqB,YAAY,IAAI,CAAC;AAC1C,IAAA,GAAGI,qBAAqB;AACxBmB,IAAAA,WAAW,EAAE/C,OAAO;AACpBgD,IAAAA,YAAY,EAAEpD,OAAO;AACrBqD,IAAAA,YAAY,EAAEhD,OAAO;IACrB,GAAGlC,MAAM,CAACmF,iBAAiB;AAC3B,IAAA,IAAIvD,UAAU,GAAG,EAAE,GAAG;AAAEwD,MAAAA,uBAAuB,EAAE,KAAA;AAAM,KAAC,CAAC;AACzD,IAAA,IAAI7C,KAAK,GAAG;AAAE8C,MAAAA,SAAS,EAAE9C,KAAAA;KAAO,GAAG,EAAE,CAAC;AACtC,IAAA,GAAGQ,SAAS;IACZ,GAAGG,gBAAAA;GACJ,CAAA;AAED,EAAA,MAAMoC,KAAK,GAAG;IACZ1D,UAAU,EAAEA,UAAU,IAAIC,OAAO;AACjCyD,IAAAA,KAAK,EAAE,gBAAgB;IACvBlB,UAAU;IACVmB,MAAM,EAAEvF,MAAM,CAACwF,aAAAA;GAChB,CAAA;AAED,EAAA,IAAIhD,gBAAgB,EAAE;AACpB;AACA,IAAA,MAAMlC,MAAM,CAACkC,gBAAgB,CAAC8C,KAAK,CAAC,CAAA;AACtC,GAAC,MAAM;AACLhF,IAAAA,MAAM,CAACmC,OAAO,CAAC6C,KAAK,CAAC,CAAA;AACvB,GAAA;AACF,CAAC;;ACzRD;;AA4BO,MAAMG,kBAAkB,CAAC;EAK9BC,WAAWA,CAACC,MAA8B,EAAE;IAC1C,MAAM;MAAEC,OAAO;MAAE,GAAGC,YAAAA;AAAa,KAAC,GAAGF,MAAM,CAAA;IAC3C,IAAI,CAACG,QAAQ,GAAGF,OAAO,CAAA;AACvB,IAAA,IAAI,CAACtF,MAAM,GAAG,IAAIyF,iBAAW,CAACF,YAAY,CAAC,CAAA;AAC3C,IAAA,IAAI,CAACG,MAAM,GAAG,IAAIC,aAAa,CAAC,IAAI,CAAC3F,MAAM,EAAE,IAAI,CAACwF,QAAQ,CAAC,CAAA;AAC7D,GAAA;AACF,CAAA;AAEO,MAAMG,aAAa,CAAC;AAIzBP,EAAAA,WAAWA,CAACpF,MAAmB,EAAEwF,QAAiB,EAAE;IAClD,IAAI,CAACxF,MAAM,GAAGA,MAAM,CAAA;IACpB,IAAI,CAACwF,QAAQ,GAAGA,QAAQ,CAAA;AAC1B,GAAA;EAEA,MAAaI,eAAeA,CAAClG,MAAiD,EAAoC;IAChH,MAAM;MACJmG,iBAAiB;MACjBC,cAAc;MACdjB,iBAAiB;MACjBK,aAAa;MACba,uBAAuB;MACvB,GAAGC,YAAAA;AACL,KAAC,GAAGtG,MAAM,CAAA;AAEV,IAAA,MAAM6B,OAAO,GAAGuE,cAAc,IAAIG,OAAM,EAAE,CAAA;AAC1C,IAAA,MAAMC,SAAS,GAAGC,IAAI,CAACC,GAAG,EAAE,CAAA;IAE5B,IAAI;AACF,MAAA,MAAMC,QAAQ,GAAG,MAAM,IAAI,CAACrG,MAAM,CAAC0F,MAAM,CAACE,eAAe,CAACI,YAAY,CAAC,CAAA;MACvE,MAAMrE,OAAO,GAAG,CAACwE,IAAI,CAACC,GAAG,EAAE,GAAGF,SAAS,IAAI,IAAI,CAAA;AAE/C,MAAA,MAAM7E,kBAAkB,CAAC;QACvBrB,MAAM,EAAE,IAAI,CAACwF,QAAQ;AACrBlE,QAAAA,UAAU,EAAEuE,iBAAiB;QAC7BtE,OAAO;QACPC,KAAK,EAAEwE,YAAY,CAACxE,KAAK;AACzBC,QAAAA,QAAQ,EAAE,QAAQ;QAClBvB,KAAK,EAAE,IAAI,CAACoG,WAAW,CAACN,YAAY,CAACO,QAAQ,CAAC;AAC9C7E,QAAAA,MAAM,EAAE,IAAI,CAAC8E,YAAY,CAACH,QAAQ,CAAC;QACnC1E,OAAO;AACPC,QAAAA,OAAO,EAAE,2CAA2C;AACpDlC,QAAAA,MAAM,EAAEA,MAAa;AACrBmC,QAAAA,UAAU,EAAE,GAAG;AACfC,QAAAA,KAAK,EAAE;AACLkB,UAAAA,WAAW,EAAEqD,QAAQ,CAACI,aAAa,EAAEC,gBAAgB,IAAI,CAAC;AAC1DvD,UAAAA,YAAY,EAAEkD,QAAQ,CAACI,aAAa,EAAEE,oBAAoB,IAAI,CAAA;SAC/D;AACDzE,QAAAA,gBAAgB,EAAE6D,uBAAAA;AACpB,OAAC,CAAC,CAAA;AAEF,MAAA,OAAOM,QAAQ,CAAA;KAChB,CAAC,OAAOrE,KAAU,EAAE;MACnB,MAAML,OAAO,GAAG,CAACwE,IAAI,CAACC,GAAG,EAAE,GAAGF,SAAS,IAAI,IAAI,CAAA;AAC/C,MAAA,MAAM7E,kBAAkB,CAAC;QACvBrB,MAAM,EAAE,IAAI,CAACwF,QAAQ;AACrBlE,QAAAA,UAAU,EAAEuE,iBAAiB;QAC7BtE,OAAO;QACPC,KAAK,EAAEwE,YAAY,CAACxE,KAAK;AACzBC,QAAAA,QAAQ,EAAE,QAAQ;QAClBvB,KAAK,EAAE,IAAI,CAACoG,WAAW,CAACN,YAAY,CAACO,QAAQ,CAAC;AAC9C7E,QAAAA,MAAM,EAAE,EAAE;QACVC,OAAO;AACPC,QAAAA,OAAO,EAAE,2CAA2C;AACpDlC,QAAAA,MAAM,EAAEA,MAAa;AACrBmC,QAAAA,UAAU,EAAEG,KAAK,EAAE4E,MAAM,IAAI,GAAG;AAChC9E,QAAAA,KAAK,EAAE;AACLkB,UAAAA,WAAW,EAAE,CAAC;AACdG,UAAAA,YAAY,EAAE,CAAA;SACf;AACDpB,QAAAA,OAAO,EAAE,IAAI;AACbC,QAAAA,KAAK,EAAEzB,IAAI,CAACE,SAAS,CAACuB,KAAK,CAAC;AAC5BE,QAAAA,gBAAgB,EAAE6D,uBAAAA;AACpB,OAAC,CAAC,CAAA;AACF,MAAA,MAAM/D,KAAK,CAAA;AACb,KAAA;AACF,GAAA;EAEA,OAAc6E,qBAAqBA,CACjCnH,MAAiD,EACb;IACpC,MAAM;MACJmG,iBAAiB;MACjBC,cAAc;MACdjB,iBAAiB;MACjBK,aAAa;MACba,uBAAuB;MACvB,GAAGC,YAAAA;AACL,KAAC,GAAGtG,MAAM,CAAA;AAEV,IAAA,MAAM6B,OAAO,GAAGuE,cAAc,IAAIG,OAAM,EAAE,CAAA;AAC1C,IAAA,MAAMC,SAAS,GAAGC,IAAI,CAACC,GAAG,EAAE,CAAA;IAC5B,IAAIU,kBAAkB,GAAG,EAAE,CAAA;AAC3B,IAAA,IAAIhF,KAAK,GAAG;AACVkB,MAAAA,WAAW,EAAE,CAAC;AACdG,MAAAA,YAAY,EAAE,CAAA;KACf,CAAA;IAED,IAAI;AACF,MAAA,MAAM4D,MAAM,GAAG,MAAM,IAAI,CAAC/G,MAAM,CAAC0F,MAAM,CAACmB,qBAAqB,CAACb,YAAY,CAAC,CAAA;AAE3E,MAAA,WAAW,MAAMgB,KAAK,IAAID,MAAM,EAAE;QAChC,IAAIC,KAAK,CAACC,IAAI,EAAE;UACdH,kBAAkB,IAAIE,KAAK,CAACC,IAAI,CAAA;AAClC,SAAA;QACA,IAAID,KAAK,CAACP,aAAa,EAAE;AACvB3E,UAAAA,KAAK,GAAG;AACNkB,YAAAA,WAAW,EAAEgE,KAAK,CAACP,aAAa,CAACC,gBAAgB,IAAI,CAAC;AACtDvD,YAAAA,YAAY,EAAE6D,KAAK,CAACP,aAAa,CAACE,oBAAoB,IAAI,CAAA;WAC3D,CAAA;AACH,SAAA;AACA,QAAA,MAAMK,KAAK,CAAA;AACb,OAAA;MAEA,MAAMrF,OAAO,GAAG,CAACwE,IAAI,CAACC,GAAG,EAAE,GAAGF,SAAS,IAAI,IAAI,CAAA;AAC/C,MAAA,MAAM7E,kBAAkB,CAAC;QACvBrB,MAAM,EAAE,IAAI,CAACwF,QAAQ;AACrBlE,QAAAA,UAAU,EAAEuE,iBAAiB;QAC7BtE,OAAO;QACPC,KAAK,EAAEwE,YAAY,CAACxE,KAAK;AACzBC,QAAAA,QAAQ,EAAE,QAAQ;QAClBvB,KAAK,EAAE,IAAI,CAACoG,WAAW,CAACN,YAAY,CAACO,QAAQ,CAAC;AAC9C7E,QAAAA,MAAM,EAAE,CAAC;AAAEwF,UAAAA,OAAO,EAAEJ,kBAAkB;AAAEK,UAAAA,IAAI,EAAE,WAAA;AAAY,SAAC,CAAC;QAC5DxF,OAAO;AACPC,QAAAA,OAAO,EAAE,2CAA2C;AACpDlC,QAAAA,MAAM,EAAEA,MAAa;AACrBmC,QAAAA,UAAU,EAAE,GAAG;QACfC,KAAK;AACLI,QAAAA,gBAAgB,EAAE6D,uBAAAA;AACpB,OAAC,CAAC,CAAA;KACH,CAAC,OAAO/D,KAAU,EAAE;MACnB,MAAML,OAAO,GAAG,CAACwE,IAAI,CAACC,GAAG,EAAE,GAAGF,SAAS,IAAI,IAAI,CAAA;AAC/C,MAAA,MAAM7E,kBAAkB,CAAC;QACvBrB,MAAM,EAAE,IAAI,CAACwF,QAAQ;AACrBlE,QAAAA,UAAU,EAAEuE,iBAAiB;QAC7BtE,OAAO;QACPC,KAAK,EAAEwE,YAAY,CAACxE,KAAK;AACzBC,QAAAA,QAAQ,EAAE,QAAQ;QAClBvB,KAAK,EAAE,IAAI,CAACoG,WAAW,CAACN,YAAY,CAACO,QAAQ,CAAC;AAC9C7E,QAAAA,MAAM,EAAE,EAAE;QACVC,OAAO;AACPC,QAAAA,OAAO,EAAE,2CAA2C;AACpDlC,QAAAA,MAAM,EAAEA,MAAa;AACrBmC,QAAAA,UAAU,EAAEG,KAAK,EAAE4E,MAAM,IAAI,GAAG;AAChC9E,QAAAA,KAAK,EAAE;AACLkB,UAAAA,WAAW,EAAE,CAAC;AACdG,UAAAA,YAAY,EAAE,CAAA;SACf;AACDpB,QAAAA,OAAO,EAAE,IAAI;AACbC,QAAAA,KAAK,EAAEzB,IAAI,CAACE,SAAS,CAACuB,KAAK,CAAC;AAC5BE,QAAAA,gBAAgB,EAAE6D,uBAAAA;AACpB,OAAC,CAAC,CAAA;AACF,MAAA,MAAM/D,KAAK,CAAA;AACb,KAAA;AACF,GAAA;EAEQsE,WAAWA,CAACC,QAAa,EAA4C;AAC3E,IAAA,IAAI,OAAOA,QAAQ,KAAK,QAAQ,EAAE;AAChC,MAAA,OAAO,CAAC;AAAEY,QAAAA,IAAI,EAAE,MAAM;AAAED,QAAAA,OAAO,EAAEX,QAAAA;AAAS,OAAC,CAAC,CAAA;AAC9C,KAAA;AAEA,IAAA,IAAI1F,KAAK,CAACC,OAAO,CAACyF,QAAQ,CAAC,EAAE;AAC3B,MAAA,OAAOA,QAAQ,CAACxF,GAAG,CAAEqG,IAAI,IAAK;AAC5B,QAAA,IAAI,OAAOA,IAAI,KAAK,QAAQ,EAAE;UAC5B,OAAO;AAAED,YAAAA,IAAI,EAAE,MAAM;AAAED,YAAAA,OAAO,EAAEE,IAAAA;WAAM,CAAA;AACxC,SAAA;AACA,QAAA,IAAIA,IAAI,IAAI,OAAOA,IAAI,KAAK,QAAQ,EAAE;UACpC,IAAIA,IAAI,CAACH,IAAI,EAAE;YACb,OAAO;AAAEE,cAAAA,IAAI,EAAEC,IAAI,CAACD,IAAI,IAAI,MAAM;cAAED,OAAO,EAAEE,IAAI,CAACH,IAAAA;aAAM,CAAA;AAC1D,WAAA;UACA,IAAIG,IAAI,CAACF,OAAO,EAAE;YAChB,OAAO;AAAEC,cAAAA,IAAI,EAAEC,IAAI,CAACD,IAAI,IAAI,MAAM;cAAED,OAAO,EAAEE,IAAI,CAACF,OAAAA;aAAS,CAAA;AAC7D,WAAA;AACF,SAAA;QACA,OAAO;AAAEC,UAAAA,IAAI,EAAE,MAAM;UAAED,OAAO,EAAEG,MAAM,CAACD,IAAI,CAAA;SAAG,CAAA;AAChD,OAAC,CAAC,CAAA;AACJ,KAAA;AAEA,IAAA,IAAIb,QAAQ,IAAI,OAAOA,QAAQ,KAAK,QAAQ,EAAE;MAC5C,IAAIA,QAAQ,CAACU,IAAI,EAAE;AACjB,QAAA,OAAO,CAAC;AAAEE,UAAAA,IAAI,EAAE,MAAM;UAAED,OAAO,EAAEX,QAAQ,CAACU,IAAAA;AAAK,SAAC,CAAC,CAAA;AACnD,OAAA;MACA,IAAIV,QAAQ,CAACW,OAAO,EAAE;AACpB,QAAA,OAAO,CAAC;AAAEC,UAAAA,IAAI,EAAE,MAAM;UAAED,OAAO,EAAEX,QAAQ,CAACW,OAAAA;AAAQ,SAAC,CAAC,CAAA;AACtD,OAAA;AACF,KAAA;AAEA,IAAA,OAAO,CAAC;AAAEC,MAAAA,IAAI,EAAE,MAAM;MAAED,OAAO,EAAEG,MAAM,CAACd,QAAQ,CAAA;AAAE,KAAC,CAAC,CAAA;AACtD,GAAA;EAEQC,YAAYA,CAACH,QAAiC,EAA4C;IAChG,IAAIA,QAAQ,CAACY,IAAI,EAAE;AACjB,MAAA,OAAO,CAAC;AAAEE,QAAAA,IAAI,EAAE,WAAW;QAAED,OAAO,EAAEb,QAAQ,CAACY,IAAAA;AAAK,OAAC,CAAC,CAAA;AACxD,KAAA;AAEA,IAAA,IAAIZ,QAAQ,CAACiB,UAAU,IAAIzG,KAAK,CAACC,OAAO,CAACuF,QAAQ,CAACiB,UAAU,CAAC,EAAE;AAC7D,MAAA,OAAOjB,QAAQ,CAACiB,UAAU,CAACvG,GAAG,CAAEwG,SAAS,IAAK;QAC5C,IAAIA,SAAS,CAACL,OAAO,IAAIK,SAAS,CAACL,OAAO,CAACM,KAAK,EAAE;AAChD,UAAA,MAAMP,IAAI,GAAGM,SAAS,CAACL,OAAO,CAACM,KAAK,CACjCC,MAAM,CAAEC,IAAS,IAAKA,IAAI,CAACT,IAAI,CAAC,CAChClG,GAAG,CAAE2G,IAAS,IAAKA,IAAI,CAACT,IAAI,CAAC,CAC7BU,IAAI,CAAC,EAAE,CAAC,CAAA;UACX,OAAO;AAAER,YAAAA,IAAI,EAAE,WAAW;AAAED,YAAAA,OAAO,EAAED,IAAAA;WAAM,CAAA;AAC7C,SAAA;QACA,OAAO;AAAEE,UAAAA,IAAI,EAAE,WAAW;UAAED,OAAO,EAAEG,MAAM,CAACE,SAAS,CAAA;SAAG,CAAA;AAC1D,OAAC,CAAC,CAAA;AACJ,KAAA;AAEA,IAAA,OAAO,EAAE,CAAA;AACX,GAAA;AACF;;;;;;;"}

package/lib/gemini/index.d.ts

@@ -0,0 +1,60 @@
+import { GoogleGenAI } from '@google/genai';
+import { PostHog } from 'posthog-node';
+
+interface MonitoringParams {
+    posthogDistinctId?: string;
+    posthogTraceId?: string;
+    posthogProperties?: Record<string, any>;
+    posthogPrivacyMode?: boolean;
+    posthogGroups?: Record<string, any>;
+    posthogModelOverride?: string;
+    posthogProviderOverride?: string;
+    posthogCostOverride?: CostOverride;
+    posthogCaptureImmediate?: boolean;
+}
+interface CostOverride {
+    inputCost: number;
+    outputCost: number;
+}
+
+type GenerateContentRequest = {
+    model: string;
+    contents: any;
+    config?: any;
+    [key: string]: any;
+};
+type GenerateContentResponse = {
+    text?: string;
+    candidates?: any[];
+    usageMetadata?: {
+        promptTokenCount?: number;
+        candidatesTokenCount?: number;
+        totalTokenCount?: number;
+    };
+    [key: string]: any;
+};
+interface MonitoringGeminiConfig {
+    apiKey?: string;
+    vertexai?: boolean;
+    project?: string;
+    location?: string;
+    apiVersion?: string;
+    posthog: PostHog;
+}
+declare class PostHogGoogleGenAI {
+    private readonly phClient;
+    private readonly client;
+    models: WrappedModels;
+    constructor(config: MonitoringGeminiConfig);
+}
+declare class WrappedModels {
+    private readonly phClient;
+    private readonly client;
+    constructor(client: GoogleGenAI, phClient: PostHog);
+    generateContent(params: GenerateContentRequest & MonitoringParams): Promise<GenerateContentResponse>;
+    generateContentStream(params: GenerateContentRequest & MonitoringParams): AsyncGenerator<any, void, unknown>;
+    private formatInput;
+    private formatOutput;
+}
+
+export { PostHogGoogleGenAI as Gemini, PostHogGoogleGenAI, WrappedModels, PostHogGoogleGenAI as default };