@posthog/ai 6.0.1 → 6.1.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/anthropic/index.cjs +61 -4
- package/dist/anthropic/index.cjs.map +1 -1
- package/dist/anthropic/index.mjs +61 -4
- package/dist/anthropic/index.mjs.map +1 -1
- package/dist/gemini/index.cjs +65 -6
- package/dist/gemini/index.cjs.map +1 -1
- package/dist/gemini/index.d.ts +3 -0
- package/dist/gemini/index.mjs +65 -6
- package/dist/gemini/index.mjs.map +1 -1
- package/dist/index.cjs +268 -52
- package/dist/index.cjs.map +1 -1
- package/dist/index.d.ts +3 -0
- package/dist/index.mjs +268 -52
- package/dist/index.mjs.map +1 -1
- package/dist/langchain/index.cjs +132 -2
- package/dist/langchain/index.cjs.map +1 -1
- package/dist/langchain/index.mjs +132 -2
- package/dist/langchain/index.mjs.map +1 -1
- package/dist/openai/index.cjs +133 -10
- package/dist/openai/index.cjs.map +1 -1
- package/dist/openai/index.mjs +133 -10
- package/dist/openai/index.mjs.map +1 -1
- package/dist/vercel/index.cjs +83 -30
- package/dist/vercel/index.cjs.map +1 -1
- package/dist/vercel/index.mjs +83 -30
- package/dist/vercel/index.mjs.map +1 -1
- package/package.json +1 -1
package/dist/gemini/index.cjs
CHANGED
|
@@ -189,6 +189,51 @@ const sendEventToPosthog = async ({
|
|
|
189
189
|
}
|
|
190
190
|
};
|
|
191
191
|
|
|
192
|
+
// Type guards for safer type checking
|
|
193
|
+
|
|
194
|
+
const isObject = value => {
|
|
195
|
+
return value !== null && typeof value === 'object' && !Array.isArray(value);
|
|
196
|
+
};
|
|
197
|
+
|
|
198
|
+
const REDACTED_IMAGE_PLACEHOLDER = '[base64 image redacted]';
|
|
199
|
+
const sanitizeGeminiPart = part => {
|
|
200
|
+
if (!isObject(part)) return part;
|
|
201
|
+
|
|
202
|
+
// Handle Gemini's inline data format
|
|
203
|
+
if ('inlineData' in part && isObject(part.inlineData) && 'data' in part.inlineData) {
|
|
204
|
+
return {
|
|
205
|
+
...part,
|
|
206
|
+
inlineData: {
|
|
207
|
+
...part.inlineData,
|
|
208
|
+
data: REDACTED_IMAGE_PLACEHOLDER
|
|
209
|
+
}
|
|
210
|
+
};
|
|
211
|
+
}
|
|
212
|
+
return part;
|
|
213
|
+
};
|
|
214
|
+
const processGeminiItem = item => {
|
|
215
|
+
if (!isObject(item)) return item;
|
|
216
|
+
|
|
217
|
+
// If it has parts, process them
|
|
218
|
+
if ('parts' in item && item.parts) {
|
|
219
|
+
const parts = Array.isArray(item.parts) ? item.parts.map(sanitizeGeminiPart) : sanitizeGeminiPart(item.parts);
|
|
220
|
+
return {
|
|
221
|
+
...item,
|
|
222
|
+
parts
|
|
223
|
+
};
|
|
224
|
+
}
|
|
225
|
+
return item;
|
|
226
|
+
};
|
|
227
|
+
const sanitizeGemini = data => {
|
|
228
|
+
// Gemini has a different structure with 'parts' directly on items instead of 'content'
|
|
229
|
+
// So we need custom processing instead of using processMessages
|
|
230
|
+
if (!data) return data;
|
|
231
|
+
if (Array.isArray(data)) {
|
|
232
|
+
return data.map(processGeminiItem);
|
|
233
|
+
}
|
|
234
|
+
return processGeminiItem(data);
|
|
235
|
+
};
|
|
236
|
+
|
|
192
237
|
// Types from @google/genai
|
|
193
238
|
|
|
194
239
|
class PostHogGoogleGenAI {
|
|
@@ -228,7 +273,7 @@ class WrappedModels {
|
|
|
228
273
|
traceId,
|
|
229
274
|
model: geminiParams.model,
|
|
230
275
|
provider: 'gemini',
|
|
231
|
-
input: this.
|
|
276
|
+
input: this.formatInputForPostHog(geminiParams.contents),
|
|
232
277
|
output: formatResponseGemini(response),
|
|
233
278
|
latency,
|
|
234
279
|
baseURL: 'https://generativelanguage.googleapis.com',
|
|
@@ -236,7 +281,9 @@ class WrappedModels {
|
|
|
236
281
|
httpStatus: 200,
|
|
237
282
|
usage: {
|
|
238
283
|
inputTokens: response.usageMetadata?.promptTokenCount ?? 0,
|
|
239
|
-
outputTokens: response.usageMetadata?.candidatesTokenCount ?? 0
|
|
284
|
+
outputTokens: response.usageMetadata?.candidatesTokenCount ?? 0,
|
|
285
|
+
reasoningTokens: response.usageMetadata?.thoughtsTokenCount ?? 0,
|
|
286
|
+
cacheReadInputTokens: response.usageMetadata?.cachedContentTokenCount ?? 0
|
|
240
287
|
},
|
|
241
288
|
tools: availableTools,
|
|
242
289
|
captureImmediate: posthogCaptureImmediate
|
|
@@ -250,7 +297,7 @@ class WrappedModels {
|
|
|
250
297
|
traceId,
|
|
251
298
|
model: geminiParams.model,
|
|
252
299
|
provider: 'gemini',
|
|
253
|
-
input: this.
|
|
300
|
+
input: this.formatInputForPostHog(geminiParams.contents),
|
|
254
301
|
output: [],
|
|
255
302
|
latency,
|
|
256
303
|
baseURL: 'https://generativelanguage.googleapis.com',
|
|
@@ -292,7 +339,9 @@ class WrappedModels {
|
|
|
292
339
|
if (chunk.usageMetadata) {
|
|
293
340
|
usage = {
|
|
294
341
|
inputTokens: chunk.usageMetadata.promptTokenCount ?? 0,
|
|
295
|
-
outputTokens: chunk.usageMetadata.candidatesTokenCount ?? 0
|
|
342
|
+
outputTokens: chunk.usageMetadata.candidatesTokenCount ?? 0,
|
|
343
|
+
reasoningTokens: chunk.usageMetadata.thoughtsTokenCount ?? 0,
|
|
344
|
+
cacheReadInputTokens: chunk.usageMetadata.cachedContentTokenCount ?? 0
|
|
296
345
|
};
|
|
297
346
|
}
|
|
298
347
|
yield chunk;
|
|
@@ -305,7 +354,7 @@ class WrappedModels {
|
|
|
305
354
|
traceId,
|
|
306
355
|
model: geminiParams.model,
|
|
307
356
|
provider: 'gemini',
|
|
308
|
-
input: this.
|
|
357
|
+
input: this.formatInputForPostHog(geminiParams.contents),
|
|
309
358
|
output: [{
|
|
310
359
|
content: accumulatedContent,
|
|
311
360
|
role: 'assistant'
|
|
@@ -326,7 +375,7 @@ class WrappedModels {
|
|
|
326
375
|
traceId,
|
|
327
376
|
model: geminiParams.model,
|
|
328
377
|
provider: 'gemini',
|
|
329
|
-
input: this.
|
|
378
|
+
input: this.formatInputForPostHog(geminiParams.contents),
|
|
330
379
|
output: [],
|
|
331
380
|
latency,
|
|
332
381
|
baseURL: 'https://generativelanguage.googleapis.com',
|
|
@@ -371,6 +420,12 @@ class WrappedModels {
|
|
|
371
420
|
content: item.content
|
|
372
421
|
};
|
|
373
422
|
}
|
|
423
|
+
if (item.parts) {
|
|
424
|
+
return {
|
|
425
|
+
role: item.role || 'user',
|
|
426
|
+
content: item.parts.map(part => part.text ? part.text : part)
|
|
427
|
+
};
|
|
428
|
+
}
|
|
374
429
|
}
|
|
375
430
|
return {
|
|
376
431
|
role: 'user',
|
|
@@ -397,6 +452,10 @@ class WrappedModels {
|
|
|
397
452
|
content: String(contents)
|
|
398
453
|
}];
|
|
399
454
|
}
|
|
455
|
+
formatInputForPostHog(contents) {
|
|
456
|
+
const sanitized = sanitizeGemini(contents);
|
|
457
|
+
return this.formatInput(sanitized);
|
|
458
|
+
}
|
|
400
459
|
}
|
|
401
460
|
|
|
402
461
|
exports.Gemini = PostHogGoogleGenAI;
|
|
@@ -1 +1 @@
|
|
|
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'\nimport type { ChatCompletionTool } from 'openai/resources/chat/completions'\nimport type { Tool as GeminiTool } from '@google/genai'\nimport type { FormattedMessage, FormattedContent, TokenUsage } from './types'\n\ntype ChatCompletionCreateParamsBase = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParams\ntype MessageCreateParams = AnthropicOriginal.Messages.MessageCreateParams\ntype ResponseCreateParams = OpenAIOrignal.Responses.ResponseCreateParams\ntype AnthropicTool = AnthropicOriginal.Tool\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): FormattedMessage[] => {\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): FormattedMessage[] => {\n const output: FormattedMessage[] = []\n const content: FormattedContent = []\n\n for (const choice of response.content ?? []) {\n if (choice?.type === 'text' && choice?.text) {\n content.push({ type: 'text', text: choice.text })\n } else if (choice?.type === 'tool_use' && choice?.name && choice?.id) {\n content.push({\n type: 'function',\n id: choice.id,\n function: {\n name: choice.name,\n arguments: choice.input || {},\n },\n })\n }\n }\n\n if (content.length > 0) {\n output.push({\n role: 'assistant',\n content,\n })\n }\n\n return output\n}\n\nexport const formatResponseOpenAI = (response: any): FormattedMessage[] => {\n const output: FormattedMessage[] = []\n\n if (response.choices) {\n for (const choice of response.choices) {\n const content: FormattedContent = []\n let role = 'assistant'\n\n if (choice.message) {\n if (choice.message.role) {\n role = choice.message.role\n }\n\n if (choice.message.content) {\n content.push({ type: 'text', text: choice.message.content })\n }\n\n if (choice.message.tool_calls) {\n for (const toolCall of choice.message.tool_calls) {\n content.push({\n type: 'function',\n id: toolCall.id,\n function: {\n name: toolCall.function.name,\n arguments: toolCall.function.arguments,\n },\n })\n }\n }\n }\n\n if (content.length > 0) {\n output.push({\n role,\n content,\n })\n }\n }\n }\n\n // Handle Responses API format\n if (response.output) {\n const content: FormattedContent = []\n let role = 'assistant'\n\n for (const item of response.output) {\n if (item.type === 'message') {\n role = item.role\n\n if (item.content && Array.isArray(item.content)) {\n for (const contentItem of item.content) {\n if (contentItem.type === 'output_text' && contentItem.text) {\n content.push({ type: 'text', text: contentItem.text })\n } else if (contentItem.text) {\n content.push({ type: 'text', text: contentItem.text })\n } else if (contentItem.type === 'input_image' && contentItem.image_url) {\n content.push({\n type: 'image',\n image: contentItem.image_url,\n })\n }\n }\n } else if (item.content) {\n content.push({ type: 'text', text: String(item.content) })\n }\n } else if (item.type === 'function_call') {\n content.push({\n type: 'function',\n id: item.call_id || item.id || '',\n function: {\n name: item.name,\n arguments: item.arguments || {},\n },\n })\n }\n }\n\n if (content.length > 0) {\n output.push({\n role,\n content,\n })\n }\n }\n\n return output\n}\n\nexport const formatResponseGemini = (response: any): FormattedMessage[] => {\n const output: FormattedMessage[] = []\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 content: FormattedContent = []\n\n for (const part of candidate.content.parts) {\n if (part.text) {\n content.push({ type: 'text', text: part.text })\n } else if (part.functionCall) {\n content.push({\n type: 'function',\n function: {\n name: part.functionCall.name,\n arguments: part.functionCall.args,\n },\n })\n }\n }\n\n if (content.length > 0) {\n output.push({\n role: 'assistant',\n content,\n })\n }\n } else if (candidate.text) {\n output.push({\n role: 'assistant',\n content: [{ type: 'text', text: candidate.text }],\n })\n }\n }\n } else if (response.text) {\n output.push({\n role: 'assistant',\n content: [{ type: 'text', text: response.text }],\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\n/**\n * Extract available tool calls from the request parameters.\n * These are the tools provided to the LLM, not the tool calls in the response.\n */\nexport const extractAvailableToolCalls = (\n provider: string,\n params: any\n): ChatCompletionTool[] | AnthropicTool[] | GeminiTool[] | null => {\n if (provider === 'anthropic') {\n if (params.tools) {\n return params.tools\n }\n\n return null\n } else if (provider === 'gemini') {\n if (params.config && params.config.tools) {\n return params.config.tools\n }\n\n return null\n } else if (provider === 'openai') {\n if (params.tools) {\n return params.tools\n }\n\n return null\n } else if (provider === 'vercel') {\n // Vercel AI SDK stores tools in params.mode.tools when mode type is 'regular'\n if (params.mode?.type === 'regular' && params.mode.tools) {\n return params.mode.tools\n }\n\n return null\n }\n\n return null\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?: TokenUsage\n params: (ChatCompletionCreateParamsBase | MessageCreateParams | ResponseCreateParams) & MonitoringParams\n isError?: boolean\n error?: string\n tools?: ChatCompletionTool[] | AnthropicTool[] | GeminiTool[] | null\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, extractAvailableToolCalls, formatResponseGemini } 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 const availableTools = extractAvailableToolCalls('gemini', geminiParams)\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: formatResponseGemini(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 tools: availableTools,\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\n const availableTools = extractAvailableToolCalls('gemini', geminiParams)\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: [{ content: accumulatedContent, role: 'assistant' }],\n latency,\n baseURL: 'https://generativelanguage.googleapis.com',\n params: params as any,\n httpStatus: 200,\n usage,\n tools: availableTools,\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\nexport default PostHogGoogleGenAI\nexport { PostHogGoogleGenAI as Gemini }\n"],"names":["STRING_FORMAT","getModelParams","params","modelParams","paramKeys","key","undefined","formatResponseGemini","response","output","candidates","Array","isArray","candidate","content","parts","part","text","push","type","functionCall","function","name","arguments","args","length","role","withPrivacyMode","client","privacyMode","input","privacy_mode","extractAvailableToolCalls","provider","config","tools","sanitizeValues","obj","jsonSafe","JSON","parse","stringify","Buffer","from","toString","map","Object","fromEntries","entries","k","v","sendEventToPosthog","distinctId","traceId","model","latency","baseURL","httpStatus","usage","isError","error","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","posthog","geminiConfig","phClient","GoogleGenAI","models","WrappedModels","generateContent","posthogDistinctId","posthogTraceId","posthogCaptureImmediate","geminiParams","uuidv4","startTime","Date","now","availableTools","formatInput","contents","usageMetadata","promptTokenCount","candidatesTokenCount","status","generateContentStream","accumulatedContent","stream","chunk","item","String"],"mappings":";;;;;;;;AAeA,MAAMA,aAAa,GAAG,MAAM;AAmBrB,MAAMC,cAAc,GACzBC,MAAiH,IACzF;EACxB,IAAI,CAACA,MAAM,EAAE;AACX,IAAA,OAAO,EAAE;AACX,EAAA;EACA,MAAMC,WAAgC,GAAG,EAAE;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;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;AACzC,IAAA;AACF,EAAA;AACA,EAAA,OAAOF,WAAW;AACpB,CAAC;AAwIM,MAAMI,oBAAoB,GAAIC,QAAa,IAAyB;EACzE,MAAMC,MAA0B,GAAG,EAAE;AAErC,EAAA,IAAID,QAAQ,CAACE,UAAU,IAAIC,KAAK,CAACC,OAAO,CAACJ,QAAQ,CAACE,UAAU,CAAC,EAAE;AAC7D,IAAA,KAAK,MAAMG,SAAS,IAAIL,QAAQ,CAACE,UAAU,EAAE;MAC3C,IAAIG,SAAS,CAACC,OAAO,IAAID,SAAS,CAACC,OAAO,CAACC,KAAK,EAAE;QAChD,MAAMD,OAAyB,GAAG,EAAE;QAEpC,KAAK,MAAME,IAAI,IAAIH,SAAS,CAACC,OAAO,CAACC,KAAK,EAAE;UAC1C,IAAIC,IAAI,CAACC,IAAI,EAAE;YACbH,OAAO,CAACI,IAAI,CAAC;AAAEC,cAAAA,IAAI,EAAE,MAAM;cAAEF,IAAI,EAAED,IAAI,CAACC;AAAK,aAAC,CAAC;AACjD,UAAA,CAAC,MAAM,IAAID,IAAI,CAACI,YAAY,EAAE;YAC5BN,OAAO,CAACI,IAAI,CAAC;AACXC,cAAAA,IAAI,EAAE,UAAU;AAChBE,cAAAA,QAAQ,EAAE;AACRC,gBAAAA,IAAI,EAAEN,IAAI,CAACI,YAAY,CAACE,IAAI;AAC5BC,gBAAAA,SAAS,EAAEP,IAAI,CAACI,YAAY,CAACI;AAC/B;AACF,aAAC,CAAC;AACJ,UAAA;AACF,QAAA;AAEA,QAAA,IAAIV,OAAO,CAACW,MAAM,GAAG,CAAC,EAAE;UACtBhB,MAAM,CAACS,IAAI,CAAC;AACVQ,YAAAA,IAAI,EAAE,WAAW;AACjBZ,YAAAA;AACF,WAAC,CAAC;AACJ,QAAA;AACF,MAAA,CAAC,MAAM,IAAID,SAAS,CAACI,IAAI,EAAE;QACzBR,MAAM,CAACS,IAAI,CAAC;AACVQ,UAAAA,IAAI,EAAE,WAAW;AACjBZ,UAAAA,OAAO,EAAE,CAAC;AAAEK,YAAAA,IAAI,EAAE,MAAM;YAAEF,IAAI,EAAEJ,SAAS,CAACI;WAAM;AAClD,SAAC,CAAC;AACJ,MAAA;AACF,IAAA;AACF,EAAA,CAAC,MAAM,IAAIT,QAAQ,CAACS,IAAI,EAAE;IACxBR,MAAM,CAACS,IAAI,CAAC;AACVQ,MAAAA,IAAI,EAAE,WAAW;AACjBZ,MAAAA,OAAO,EAAE,CAAC;AAAEK,QAAAA,IAAI,EAAE,MAAM;QAAEF,IAAI,EAAET,QAAQ,CAACS;OAAM;AACjD,KAAC,CAAC;AACJ,EAAA;AAEA,EAAA,OAAOR,MAAM;AACf,CAAC;AAcM,MAAMkB,eAAe,GAAGA,CAACC,MAAe,EAAEC,WAAoB,EAAEC,KAAU,KAAU;EACzF,OAAQF,MAAM,CAASG,YAAY,IAAIF,WAAW,GAAG,IAAI,GAAGC,KAAK;AACnE,CAAC;;AAgBD;AACA;AACA;AACA;AACO,MAAME,yBAAyB,GAAGA,CACvCC,QAAgB,EAChB/B,MAAW,KACsD;EAO/B;IAChC,IAAIA,MAAM,CAACgC,MAAM,IAAIhC,MAAM,CAACgC,MAAM,CAACC,KAAK,EAAE;AACxC,MAAA,OAAOjC,MAAM,CAACgC,MAAM,CAACC,KAAK;AAC5B,IAAA;AAEA,IAAA,OAAO,IAAI;AACb,EAAA;AAgBF,CAAC;AAqBD,SAASC,cAAcA,CAACC,GAAQ,EAAO;AACrC,EAAA,IAAIA,GAAG,KAAK/B,SAAS,IAAI+B,GAAG,KAAK,IAAI,EAAE;AACrC,IAAA,OAAOA,GAAG;AACZ,EAAA;AACA,EAAA,MAAMC,QAAQ,GAAGC,IAAI,CAACC,KAAK,CAACD,IAAI,CAACE,SAAS,CAACJ,GAAG,CAAC,CAAC;AAChD,EAAA,IAAI,OAAOC,QAAQ,KAAK,QAAQ,EAAE;AAChC,IAAA,OAAOI,aAAM,CAACC,IAAI,CAACL,QAAQ,EAAEtC,aAAa,CAAC,CAAC4C,QAAQ,CAAC5C,aAAa,CAAC;EACrE,CAAC,MAAM,IAAIW,KAAK,CAACC,OAAO,CAAC0B,QAAQ,CAAC,EAAE;AAClC,IAAA,OAAOA,QAAQ,CAACO,GAAG,CAACT,cAAc,CAAC;EACrC,CAAC,MAAM,IAAIE,QAAQ,IAAI,OAAOA,QAAQ,KAAK,QAAQ,EAAE;AACnD,IAAA,OAAOQ,MAAM,CAACC,WAAW,CAACD,MAAM,CAACE,OAAO,CAACV,QAAQ,CAAC,CAACO,GAAG,CAAC,CAAC,CAACI,CAAC,EAAEC,CAAC,CAAC,KAAK,CAACD,CAAC,EAAEb,cAAc,CAACc,CAAC,CAAC,CAAC,CAAC,CAAC;AAC7F,EAAA;AACA,EAAA,OAAOZ,QAAQ;AACjB;AAEO,MAAMa,kBAAkB,GAAG,OAAO;EACvCvB,MAAM;EACNwB,UAAU;EACVC,OAAO;EACPC,KAAK;EACLrB,QAAQ;EACRH,KAAK;EACLrB,MAAM;EACN8C,OAAO;EACPC,OAAO;EACPtD,MAAM;AACNuD,EAAAA,UAAU,GAAG,GAAG;EAChBC,KAAK,GAAG,EAAE;AACVC,EAAAA,OAAO,GAAG,KAAK;EACfC,KAAK;EACLzB,KAAK;AACL0B,EAAAA,gBAAgB,GAAG;AACK,CAAC,KAAoB;AAC7C,EAAA,IAAI,CAACjC,MAAM,CAACkC,OAAO,EAAE;AACnB,IAAA,OAAOC,OAAO,CAACC,OAAO,EAAE;AAC1B,EAAA;AACA;AACA,EAAA,MAAMC,SAAS,GAAG7B,cAAc,CAACN,KAAK,CAAC;AACvC,EAAA,MAAMoC,UAAU,GAAG9B,cAAc,CAAC3B,MAAM,CAAC;AACzC,EAAA,MAAM0D,SAAS,GAAG/B,cAAc,CAACwB,KAAK,CAAC;EAEvC,IAAIQ,SAAS,GAAG,EAAE;AAClB,EAAA,IAAIT,OAAO,EAAE;AACXS,IAAAA,SAAS,GAAG;AACVC,MAAAA,YAAY,EAAE,IAAI;AAClBC,MAAAA,SAAS,EAAEH;KACZ;AACH,EAAA;EACA,IAAII,gBAAgB,GAAG,EAAE;EACzB,IAAIrE,MAAM,CAACsE,mBAAmB,EAAE;AAC9B,IAAA,MAAMC,YAAY,GAAG,CAACvE,MAAM,CAACsE,mBAAmB,CAACE,SAAS,IAAI,CAAC,KAAKhB,KAAK,CAACiB,WAAW,IAAI,CAAC,CAAC;AAC3F,IAAA,MAAMC,aAAa,GAAG,CAAC1E,MAAM,CAACsE,mBAAmB,CAACK,UAAU,IAAI,CAAC,KAAKnB,KAAK,CAACoB,YAAY,IAAI,CAAC,CAAC;AAC9FP,IAAAA,gBAAgB,GAAG;AACjBQ,MAAAA,kBAAkB,EAAEN,YAAY;AAChCO,MAAAA,mBAAmB,EAAEJ,aAAa;MAClCK,kBAAkB,EAAER,YAAY,GAAGG;KACpC;AACH,EAAA;AAEA,EAAA,MAAMM,qBAAqB,GAAG;IAC5B,IAAIxB,KAAK,CAACyB,eAAe,GAAG;MAAEC,oBAAoB,EAAE1B,KAAK,CAACyB;KAAiB,GAAG,EAAE,CAAC;IACjF,IAAIzB,KAAK,CAAC2B,oBAAoB,GAAG;MAAEC,2BAA2B,EAAE5B,KAAK,CAAC2B;KAAsB,GAAG,EAAE,CAAC;IAClG,IAAI3B,KAAK,CAAC6B,wBAAwB,GAAG;MAAEC,+BAA+B,EAAE9B,KAAK,CAAC6B;KAA0B,GAAG,EAAE;GAC9G;AAED,EAAA,MAAME,UAAU,GAAG;AACjBC,IAAAA,YAAY,EAAExF,MAAM,CAACyF,uBAAuB,IAAI1D,QAAQ;AACxD2D,IAAAA,SAAS,EAAE1F,MAAM,CAAC2F,oBAAoB,IAAIvC,KAAK;AAC/CwC,IAAAA,oBAAoB,EAAE7F,cAAc,CAACC,MAAM,CAAC;AAC5C6F,IAAAA,SAAS,EAAEpE,eAAe,CAACC,MAAM,EAAE1B,MAAM,CAAC8F,kBAAkB,IAAI,KAAK,EAAE/B,SAAS,CAAC;AACjFgC,IAAAA,kBAAkB,EAAEtE,eAAe,CAACC,MAAM,EAAE1B,MAAM,CAAC8F,kBAAkB,IAAI,KAAK,EAAE9B,UAAU,CAAC;AAC3FgC,IAAAA,eAAe,EAAEzC,UAAU;AAC3B0C,IAAAA,gBAAgB,EAAEzC,KAAK,CAACiB,WAAW,IAAI,CAAC;AACxCyB,IAAAA,iBAAiB,EAAE1C,KAAK,CAACoB,YAAY,IAAI,CAAC;AAC1C,IAAA,GAAGI,qBAAqB;AACxBmB,IAAAA,WAAW,EAAE9C,OAAO;AACpB+C,IAAAA,YAAY,EAAEjD,OAAO;AACrBkD,IAAAA,YAAY,EAAE/C,OAAO;IACrB,GAAGtD,MAAM,CAACsG,iBAAiB;AAC3B,IAAA,IAAIpD,UAAU,GAAG,EAAE,GAAG;AAAEqD,MAAAA,uBAAuB,EAAE;AAAM,KAAC,CAAC;AACzD,IAAA,IAAItE,KAAK,GAAG;AAAEuE,MAAAA,SAAS,EAAEvE;KAAO,GAAG,EAAE,CAAC;AACtC,IAAA,GAAGiC,SAAS;IACZ,GAAGG;GACJ;AAED,EAAA,MAAMoC,KAAK,GAAG;IACZvD,UAAU,EAAEA,UAAU,IAAIC,OAAO;AACjCsD,IAAAA,KAAK,EAAE,gBAAgB;IACvBlB,UAAU;IACVmB,MAAM,EAAE1G,MAAM,CAAC2G;GAChB;AAED,EAAA,IAAIhD,gBAAgB,EAAE;AACpB;AACA,IAAA,MAAMjC,MAAM,CAACiC,gBAAgB,CAAC8C,KAAK,CAAC;AACtC,EAAA,CAAC,MAAM;AACL/E,IAAAA,MAAM,CAACkC,OAAO,CAAC6C,KAAK,CAAC;AACvB,EAAA;AACF,CAAC;;ACraD;;AA4BO,MAAMG,kBAAkB,CAAC;EAK9BC,WAAWA,CAAC7E,MAA8B,EAAE;IAC1C,MAAM;MAAE8E,OAAO;MAAE,GAAGC;AAAa,KAAC,GAAG/E,MAAM;IAC3C,IAAI,CAACgF,QAAQ,GAAGF,OAAO;AACvB,IAAA,IAAI,CAACpF,MAAM,GAAG,IAAIuF,iBAAW,CAACF,YAAY,CAAC;AAC3C,IAAA,IAAI,CAACG,MAAM,GAAG,IAAIC,aAAa,CAAC,IAAI,CAACzF,MAAM,EAAE,IAAI,CAACsF,QAAQ,CAAC;AAC7D,EAAA;AACF;AAEO,MAAMG,aAAa,CAAC;AAIzBN,EAAAA,WAAWA,CAACnF,MAAmB,EAAEsF,QAAiB,EAAE;IAClD,IAAI,CAACtF,MAAM,GAAGA,MAAM;IACpB,IAAI,CAACsF,QAAQ,GAAGA,QAAQ;AAC1B,EAAA;EAEA,MAAaI,eAAeA,CAACpH,MAAiD,EAAoC;IAChH,MAAM;MACJqH,iBAAiB;MACjBC,cAAc;MACdhB,iBAAiB;MACjBK,aAAa;MACbY,uBAAuB;MACvB,GAAGC;AACL,KAAC,GAAGxH,MAAM;AAEV,IAAA,MAAMmD,OAAO,GAAGmE,cAAc,IAAIG,OAAM,EAAE;AAC1C,IAAA,MAAMC,SAAS,GAAGC,IAAI,CAACC,GAAG,EAAE;IAE5B,IAAI;AACF,MAAA,MAAMtH,QAAQ,GAAG,MAAM,IAAI,CAACoB,MAAM,CAACwF,MAAM,CAACE,eAAe,CAACI,YAAY,CAAC;MACvE,MAAMnE,OAAO,GAAG,CAACsE,IAAI,CAACC,GAAG,EAAE,GAAGF,SAAS,IAAI,IAAI;AAE/C,MAAA,MAAMG,cAAc,GAAG/F,yBAAyB,CAAC,QAAQ,EAAE0F,YAAY,CAAC;AAExE,MAAA,MAAMvE,kBAAkB,CAAC;QACvBvB,MAAM,EAAE,IAAI,CAACsF,QAAQ;AACrB9D,QAAAA,UAAU,EAAEmE,iBAAiB;QAC7BlE,OAAO;QACPC,KAAK,EAAEoE,YAAY,CAACpE,KAAK;AACzBrB,QAAAA,QAAQ,EAAE,QAAQ;QAClBH,KAAK,EAAE,IAAI,CAACkG,WAAW,CAACN,YAAY,CAACO,QAAQ,CAAC;AAC9CxH,QAAAA,MAAM,EAAEF,oBAAoB,CAACC,QAAQ,CAAC;QACtC+C,OAAO;AACPC,QAAAA,OAAO,EAAE,2CAA2C;AACpDtD,QAAAA,MAAM,EAAEA,MAAa;AACrBuD,QAAAA,UAAU,EAAE,GAAG;AACfC,QAAAA,KAAK,EAAE;AACLiB,UAAAA,WAAW,EAAEnE,QAAQ,CAAC0H,aAAa,EAAEC,gBAAgB,IAAI,CAAC;AAC1DrD,UAAAA,YAAY,EAAEtE,QAAQ,CAAC0H,aAAa,EAAEE,oBAAoB,IAAI;SAC/D;AACDjG,QAAAA,KAAK,EAAE4F,cAAc;AACrBlE,QAAAA,gBAAgB,EAAE4D;AACpB,OAAC,CAAC;AAEF,MAAA,OAAOjH,QAAQ;IACjB,CAAC,CAAC,OAAOoD,KAAU,EAAE;MACnB,MAAML,OAAO,GAAG,CAACsE,IAAI,CAACC,GAAG,EAAE,GAAGF,SAAS,IAAI,IAAI;AAC/C,MAAA,MAAMzE,kBAAkB,CAAC;QACvBvB,MAAM,EAAE,IAAI,CAACsF,QAAQ;AACrB9D,QAAAA,UAAU,EAAEmE,iBAAiB;QAC7BlE,OAAO;QACPC,KAAK,EAAEoE,YAAY,CAACpE,KAAK;AACzBrB,QAAAA,QAAQ,EAAE,QAAQ;QAClBH,KAAK,EAAE,IAAI,CAACkG,WAAW,CAACN,YAAY,CAACO,QAAQ,CAAC;AAC9CxH,QAAAA,MAAM,EAAE,EAAE;QACV8C,OAAO;AACPC,QAAAA,OAAO,EAAE,2CAA2C;AACpDtD,QAAAA,MAAM,EAAEA,MAAa;AACrBuD,QAAAA,UAAU,EAAEG,KAAK,EAAEyE,MAAM,IAAI,GAAG;AAChC3E,QAAAA,KAAK,EAAE;AACLiB,UAAAA,WAAW,EAAE,CAAC;AACdG,UAAAA,YAAY,EAAE;SACf;AACDnB,QAAAA,OAAO,EAAE,IAAI;AACbC,QAAAA,KAAK,EAAErB,IAAI,CAACE,SAAS,CAACmB,KAAK,CAAC;AAC5BC,QAAAA,gBAAgB,EAAE4D;AACpB,OAAC,CAAC;AACF,MAAA,MAAM7D,KAAK;AACb,IAAA;AACF,EAAA;EAEA,OAAc0E,qBAAqBA,CACjCpI,MAAiD,EACb;IACpC,MAAM;MACJqH,iBAAiB;MACjBC,cAAc;MACdhB,iBAAiB;MACjBK,aAAa;MACbY,uBAAuB;MACvB,GAAGC;AACL,KAAC,GAAGxH,MAAM;AAEV,IAAA,MAAMmD,OAAO,GAAGmE,cAAc,IAAIG,OAAM,EAAE;AAC1C,IAAA,MAAMC,SAAS,GAAGC,IAAI,CAACC,GAAG,EAAE;IAC5B,IAAIS,kBAAkB,GAAG,EAAE;AAC3B,IAAA,IAAI7E,KAAK,GAAG;AACViB,MAAAA,WAAW,EAAE,CAAC;AACdG,MAAAA,YAAY,EAAE;KACf;IAED,IAAI;AACF,MAAA,MAAM0D,MAAM,GAAG,MAAM,IAAI,CAAC5G,MAAM,CAACwF,MAAM,CAACkB,qBAAqB,CAACZ,YAAY,CAAC;AAE3E,MAAA,WAAW,MAAMe,KAAK,IAAID,MAAM,EAAE;QAChC,IAAIC,KAAK,CAACxH,IAAI,EAAE;UACdsH,kBAAkB,IAAIE,KAAK,CAACxH,IAAI;AAClC,QAAA;QACA,IAAIwH,KAAK,CAACP,aAAa,EAAE;AACvBxE,UAAAA,KAAK,GAAG;AACNiB,YAAAA,WAAW,EAAE8D,KAAK,CAACP,aAAa,CAACC,gBAAgB,IAAI,CAAC;AACtDrD,YAAAA,YAAY,EAAE2D,KAAK,CAACP,aAAa,CAACE,oBAAoB,IAAI;WAC3D;AACH,QAAA;AACA,QAAA,MAAMK,KAAK;AACb,MAAA;MAEA,MAAMlF,OAAO,GAAG,CAACsE,IAAI,CAACC,GAAG,EAAE,GAAGF,SAAS,IAAI,IAAI;AAE/C,MAAA,MAAMG,cAAc,GAAG/F,yBAAyB,CAAC,QAAQ,EAAE0F,YAAY,CAAC;AAExE,MAAA,MAAMvE,kBAAkB,CAAC;QACvBvB,MAAM,EAAE,IAAI,CAACsF,QAAQ;AACrB9D,QAAAA,UAAU,EAAEmE,iBAAiB;QAC7BlE,OAAO;QACPC,KAAK,EAAEoE,YAAY,CAACpE,KAAK;AACzBrB,QAAAA,QAAQ,EAAE,QAAQ;QAClBH,KAAK,EAAE,IAAI,CAACkG,WAAW,CAACN,YAAY,CAACO,QAAQ,CAAC;AAC9CxH,QAAAA,MAAM,EAAE,CAAC;AAAEK,UAAAA,OAAO,EAAEyH,kBAAkB;AAAE7G,UAAAA,IAAI,EAAE;AAAY,SAAC,CAAC;QAC5D6B,OAAO;AACPC,QAAAA,OAAO,EAAE,2CAA2C;AACpDtD,QAAAA,MAAM,EAAEA,MAAa;AACrBuD,QAAAA,UAAU,EAAE,GAAG;QACfC,KAAK;AACLvB,QAAAA,KAAK,EAAE4F,cAAc;AACrBlE,QAAAA,gBAAgB,EAAE4D;AACpB,OAAC,CAAC;IACJ,CAAC,CAAC,OAAO7D,KAAU,EAAE;MACnB,MAAML,OAAO,GAAG,CAACsE,IAAI,CAACC,GAAG,EAAE,GAAGF,SAAS,IAAI,IAAI;AAC/C,MAAA,MAAMzE,kBAAkB,CAAC;QACvBvB,MAAM,EAAE,IAAI,CAACsF,QAAQ;AACrB9D,QAAAA,UAAU,EAAEmE,iBAAiB;QAC7BlE,OAAO;QACPC,KAAK,EAAEoE,YAAY,CAACpE,KAAK;AACzBrB,QAAAA,QAAQ,EAAE,QAAQ;QAClBH,KAAK,EAAE,IAAI,CAACkG,WAAW,CAACN,YAAY,CAACO,QAAQ,CAAC;AAC9CxH,QAAAA,MAAM,EAAE,EAAE;QACV8C,OAAO;AACPC,QAAAA,OAAO,EAAE,2CAA2C;AACpDtD,QAAAA,MAAM,EAAEA,MAAa;AACrBuD,QAAAA,UAAU,EAAEG,KAAK,EAAEyE,MAAM,IAAI,GAAG;AAChC3E,QAAAA,KAAK,EAAE;AACLiB,UAAAA,WAAW,EAAE,CAAC;AACdG,UAAAA,YAAY,EAAE;SACf;AACDnB,QAAAA,OAAO,EAAE,IAAI;AACbC,QAAAA,KAAK,EAAErB,IAAI,CAACE,SAAS,CAACmB,KAAK,CAAC;AAC5BC,QAAAA,gBAAgB,EAAE4D;AACpB,OAAC,CAAC;AACF,MAAA,MAAM7D,KAAK;AACb,IAAA;AACF,EAAA;EAEQoE,WAAWA,CAACC,QAAa,EAA4C;AAC3E,IAAA,IAAI,OAAOA,QAAQ,KAAK,QAAQ,EAAE;AAChC,MAAA,OAAO,CAAC;AAAEvG,QAAAA,IAAI,EAAE,MAAM;AAAEZ,QAAAA,OAAO,EAAEmH;AAAS,OAAC,CAAC;AAC9C,IAAA;AAEA,IAAA,IAAItH,KAAK,CAACC,OAAO,CAACqH,QAAQ,CAAC,EAAE;AAC3B,MAAA,OAAOA,QAAQ,CAACpF,GAAG,CAAE6F,IAAI,IAAK;AAC5B,QAAA,IAAI,OAAOA,IAAI,KAAK,QAAQ,EAAE;UAC5B,OAAO;AAAEhH,YAAAA,IAAI,EAAE,MAAM;AAAEZ,YAAAA,OAAO,EAAE4H;WAAM;AACxC,QAAA;AACA,QAAA,IAAIA,IAAI,IAAI,OAAOA,IAAI,KAAK,QAAQ,EAAE;UACpC,IAAIA,IAAI,CAACzH,IAAI,EAAE;YACb,OAAO;AAAES,cAAAA,IAAI,EAAEgH,IAAI,CAAChH,IAAI,IAAI,MAAM;cAAEZ,OAAO,EAAE4H,IAAI,CAACzH;aAAM;AAC1D,UAAA;UACA,IAAIyH,IAAI,CAAC5H,OAAO,EAAE;YAChB,OAAO;AAAEY,cAAAA,IAAI,EAAEgH,IAAI,CAAChH,IAAI,IAAI,MAAM;cAAEZ,OAAO,EAAE4H,IAAI,CAAC5H;aAAS;AAC7D,UAAA;AACF,QAAA;QACA,OAAO;AAAEY,UAAAA,IAAI,EAAE,MAAM;UAAEZ,OAAO,EAAE6H,MAAM,CAACD,IAAI;SAAG;AAChD,MAAA,CAAC,CAAC;AACJ,IAAA;AAEA,IAAA,IAAIT,QAAQ,IAAI,OAAOA,QAAQ,KAAK,QAAQ,EAAE;MAC5C,IAAIA,QAAQ,CAAChH,IAAI,EAAE;AACjB,QAAA,OAAO,CAAC;AAAES,UAAAA,IAAI,EAAE,MAAM;UAAEZ,OAAO,EAAEmH,QAAQ,CAAChH;AAAK,SAAC,CAAC;AACnD,MAAA;MACA,IAAIgH,QAAQ,CAACnH,OAAO,EAAE;AACpB,QAAA,OAAO,CAAC;AAAEY,UAAAA,IAAI,EAAE,MAAM;UAAEZ,OAAO,EAAEmH,QAAQ,CAACnH;AAAQ,SAAC,CAAC;AACtD,MAAA;AACF,IAAA;AAEA,IAAA,OAAO,CAAC;AAAEY,MAAAA,IAAI,EAAE,MAAM;MAAEZ,OAAO,EAAE6H,MAAM,CAACV,QAAQ;AAAE,KAAC,CAAC;AACtD,EAAA;AACF;;;;;;;"}
|
|
1
|
+
{"version":3,"file":"index.cjs","sources":["../../src/utils.ts","../../src/typeGuards.ts","../../src/sanitization.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'\nimport type { ChatCompletionTool } from 'openai/resources/chat/completions'\nimport type { Tool as GeminiTool } from '@google/genai'\nimport type { FormattedMessage, FormattedContent, TokenUsage } from './types'\n\ntype ChatCompletionCreateParamsBase = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParams\ntype MessageCreateParams = AnthropicOriginal.Messages.MessageCreateParams\ntype ResponseCreateParams = OpenAIOrignal.Responses.ResponseCreateParams\ntype AnthropicTool = AnthropicOriginal.Tool\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): FormattedMessage[] => {\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): FormattedMessage[] => {\n const output: FormattedMessage[] = []\n const content: FormattedContent = []\n\n for (const choice of response.content ?? []) {\n if (choice?.type === 'text' && choice?.text) {\n content.push({ type: 'text', text: choice.text })\n } else if (choice?.type === 'tool_use' && choice?.name && choice?.id) {\n content.push({\n type: 'function',\n id: choice.id,\n function: {\n name: choice.name,\n arguments: choice.input || {},\n },\n })\n }\n }\n\n if (content.length > 0) {\n output.push({\n role: 'assistant',\n content,\n })\n }\n\n return output\n}\n\nexport const formatResponseOpenAI = (response: any): FormattedMessage[] => {\n const output: FormattedMessage[] = []\n\n if (response.choices) {\n for (const choice of response.choices) {\n const content: FormattedContent = []\n let role = 'assistant'\n\n if (choice.message) {\n if (choice.message.role) {\n role = choice.message.role\n }\n\n if (choice.message.content) {\n content.push({ type: 'text', text: choice.message.content })\n }\n\n if (choice.message.tool_calls) {\n for (const toolCall of choice.message.tool_calls) {\n content.push({\n type: 'function',\n id: toolCall.id,\n function: {\n name: toolCall.function.name,\n arguments: toolCall.function.arguments,\n },\n })\n }\n }\n }\n\n if (content.length > 0) {\n output.push({\n role,\n content,\n })\n }\n }\n }\n\n // Handle Responses API format\n if (response.output) {\n const content: FormattedContent = []\n let role = 'assistant'\n\n for (const item of response.output) {\n if (item.type === 'message') {\n role = item.role\n\n if (item.content && Array.isArray(item.content)) {\n for (const contentItem of item.content) {\n if (contentItem.type === 'output_text' && contentItem.text) {\n content.push({ type: 'text', text: contentItem.text })\n } else if (contentItem.text) {\n content.push({ type: 'text', text: contentItem.text })\n } else if (contentItem.type === 'input_image' && contentItem.image_url) {\n content.push({\n type: 'image',\n image: contentItem.image_url,\n })\n }\n }\n } else if (item.content) {\n content.push({ type: 'text', text: String(item.content) })\n }\n } else if (item.type === 'function_call') {\n content.push({\n type: 'function',\n id: item.call_id || item.id || '',\n function: {\n name: item.name,\n arguments: item.arguments || {},\n },\n })\n }\n }\n\n if (content.length > 0) {\n output.push({\n role,\n content,\n })\n }\n }\n\n return output\n}\n\nexport const formatResponseGemini = (response: any): FormattedMessage[] => {\n const output: FormattedMessage[] = []\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 content: FormattedContent = []\n\n for (const part of candidate.content.parts) {\n if (part.text) {\n content.push({ type: 'text', text: part.text })\n } else if (part.functionCall) {\n content.push({\n type: 'function',\n function: {\n name: part.functionCall.name,\n arguments: part.functionCall.args,\n },\n })\n }\n }\n\n if (content.length > 0) {\n output.push({\n role: 'assistant',\n content,\n })\n }\n } else if (candidate.text) {\n output.push({\n role: 'assistant',\n content: [{ type: 'text', text: candidate.text }],\n })\n }\n }\n } else if (response.text) {\n output.push({\n role: 'assistant',\n content: [{ type: 'text', text: response.text }],\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\n/**\n * Extract available tool calls from the request parameters.\n * These are the tools provided to the LLM, not the tool calls in the response.\n */\nexport const extractAvailableToolCalls = (\n provider: string,\n params: any\n): ChatCompletionTool[] | AnthropicTool[] | GeminiTool[] | null => {\n if (provider === 'anthropic') {\n if (params.tools) {\n return params.tools\n }\n\n return null\n } else if (provider === 'gemini') {\n if (params.config && params.config.tools) {\n return params.config.tools\n }\n\n return null\n } else if (provider === 'openai') {\n if (params.tools) {\n return params.tools\n }\n\n return null\n } else if (provider === 'vercel') {\n // Vercel AI SDK stores tools in params.mode.tools when mode type is 'regular'\n if (params.mode?.type === 'regular' && params.mode.tools) {\n return params.mode.tools\n }\n\n return null\n }\n\n return null\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?: TokenUsage\n params: (ChatCompletionCreateParamsBase | MessageCreateParams | ResponseCreateParams) & MonitoringParams\n isError?: boolean\n error?: string\n tools?: ChatCompletionTool[] | AnthropicTool[] | GeminiTool[] | null\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","// Type guards for safer type checking\n\nexport const isString = (value: unknown): value is string => {\n return typeof value === 'string'\n}\n\nexport const isObject = (value: unknown): value is Record<string, unknown> => {\n return value !== null && typeof value === 'object' && !Array.isArray(value)\n}\n","import { isString, isObject } from './typeGuards'\n\nconst REDACTED_IMAGE_PLACEHOLDER = '[base64 image redacted]'\n\n// ============================================\n// Base64 Detection Helpers\n// ============================================\n\nconst isBase64DataUrl = (str: string): boolean => {\n return /^data:([^;]+);base64,/.test(str)\n}\n\nconst isValidUrl = (str: string): boolean => {\n try {\n new URL(str)\n return true\n } catch {\n // Not an absolute URL, check if it's a relative URL or path\n return str.startsWith('/') || str.startsWith('./') || str.startsWith('../')\n }\n}\n\nconst isRawBase64 = (str: string): boolean => {\n // Skip if it's a valid URL or path\n if (isValidUrl(str)) {\n return false\n }\n\n // Check if it's a valid base64 string\n // Base64 images are typically at least a few hundred chars, but we'll be conservative\n return str.length > 20 && /^[A-Za-z0-9+/]+=*$/.test(str)\n}\n\nexport function redactBase64DataUrl(str: string): string\nexport function redactBase64DataUrl(str: unknown): unknown\nexport function redactBase64DataUrl(str: unknown): unknown {\n if (!isString(str)) return str\n\n // Check for data URL format\n if (isBase64DataUrl(str)) {\n return REDACTED_IMAGE_PLACEHOLDER\n }\n\n // Check for raw base64 (Vercel sends raw base64 for inline images)\n if (isRawBase64(str)) {\n return REDACTED_IMAGE_PLACEHOLDER\n }\n\n return str\n}\n\n// ============================================\n// Common Message Processing\n// ============================================\n\ntype ContentTransformer = (item: unknown) => unknown\n\nconst processMessages = (messages: unknown, transformContent: ContentTransformer): unknown => {\n if (!messages) return messages\n\n const processContent = (content: unknown): unknown => {\n if (typeof content === 'string') return content\n\n if (!content) return content\n\n if (Array.isArray(content)) {\n return content.map(transformContent)\n }\n\n // Handle single object content\n return transformContent(content)\n }\n\n const processMessage = (msg: unknown): unknown => {\n if (!isObject(msg) || !('content' in msg)) return msg\n return { ...msg, content: processContent(msg.content) }\n }\n\n // Handle both arrays and single messages\n if (Array.isArray(messages)) {\n return messages.map(processMessage)\n }\n\n return processMessage(messages)\n}\n\n// ============================================\n// Provider-Specific Image Sanitizers\n// ============================================\n\nconst sanitizeOpenAIImage = (item: unknown): unknown => {\n if (!isObject(item)) return item\n\n // Handle image_url format\n if (item.type === 'image_url' && 'image_url' in item && isObject(item.image_url) && 'url' in item.image_url) {\n return {\n ...item,\n image_url: {\n ...item.image_url,\n url: redactBase64DataUrl(item.image_url.url),\n },\n }\n }\n\n return item\n}\n\nconst sanitizeOpenAIResponseImage = (item: unknown): unknown => {\n if (!isObject(item)) return item\n\n // Handle input_image format\n if (item.type === 'input_image' && 'image_url' in item) {\n return {\n ...item,\n image_url: redactBase64DataUrl(item.image_url),\n }\n }\n\n return item\n}\n\nconst sanitizeAnthropicImage = (item: unknown): unknown => {\n if (!isObject(item)) return item\n\n // Handle Anthropic's image format\n if (\n item.type === 'image' &&\n 'source' in item &&\n isObject(item.source) &&\n item.source.type === 'base64' &&\n 'data' in item.source\n ) {\n return {\n ...item,\n source: {\n ...item.source,\n data: REDACTED_IMAGE_PLACEHOLDER,\n },\n }\n }\n\n return item\n}\n\nconst sanitizeGeminiPart = (part: unknown): unknown => {\n if (!isObject(part)) return part\n\n // Handle Gemini's inline data format\n if ('inlineData' in part && isObject(part.inlineData) && 'data' in part.inlineData) {\n return {\n ...part,\n inlineData: {\n ...part.inlineData,\n data: REDACTED_IMAGE_PLACEHOLDER,\n },\n }\n }\n\n return part\n}\n\nconst processGeminiItem = (item: unknown): unknown => {\n if (!isObject(item)) return item\n\n // If it has parts, process them\n if ('parts' in item && item.parts) {\n const parts = Array.isArray(item.parts) ? item.parts.map(sanitizeGeminiPart) : sanitizeGeminiPart(item.parts)\n\n return { ...item, parts }\n }\n\n return item\n}\n\nconst sanitizeLangChainImage = (item: unknown): unknown => {\n if (!isObject(item)) return item\n\n // OpenAI style\n if (item.type === 'image_url' && 'image_url' in item && isObject(item.image_url) && 'url' in item.image_url) {\n return {\n ...item,\n image_url: {\n ...item.image_url,\n url: redactBase64DataUrl(item.image_url.url),\n },\n }\n }\n\n // Direct image with data field\n if (item.type === 'image' && 'data' in item) {\n return { ...item, data: redactBase64DataUrl(item.data) }\n }\n\n // Anthropic style\n if (item.type === 'image' && 'source' in item && isObject(item.source) && 'data' in item.source) {\n return {\n ...item,\n source: {\n ...item.source,\n data: redactBase64DataUrl(item.source.data),\n },\n }\n }\n\n // Google style\n if (item.type === 'media' && 'data' in item) {\n return { ...item, data: redactBase64DataUrl(item.data) }\n }\n\n return item\n}\n\n// Export individual sanitizers for tree-shaking\nexport const sanitizeOpenAI = (data: unknown): unknown => {\n return processMessages(data, sanitizeOpenAIImage)\n}\n\nexport const sanitizeOpenAIResponse = (data: unknown): unknown => {\n return processMessages(data, sanitizeOpenAIResponseImage)\n}\n\nexport const sanitizeAnthropic = (data: unknown): unknown => {\n return processMessages(data, sanitizeAnthropicImage)\n}\n\nexport const sanitizeGemini = (data: unknown): unknown => {\n // Gemini has a different structure with 'parts' directly on items instead of 'content'\n // So we need custom processing instead of using processMessages\n if (!data) return data\n\n if (Array.isArray(data)) {\n return data.map(processGeminiItem)\n }\n\n return processGeminiItem(data)\n}\n\nexport const sanitizeLangChain = (data: unknown): unknown => {\n return processMessages(data, sanitizeLangChainImage)\n}\n","import { GoogleGenAI } from '@google/genai'\nimport { PostHog } from 'posthog-node'\nimport { v4 as uuidv4 } from 'uuid'\nimport { MonitoringParams, sendEventToPosthog, extractAvailableToolCalls, formatResponseGemini } from '../utils'\nimport { sanitizeGemini } from '../sanitization'\nimport type { TokenUsage } from '../types'\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 thoughtsTokenCount?: number\n cachedContentTokenCount?: 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 const availableTools = extractAvailableToolCalls('gemini', geminiParams)\n\n await sendEventToPosthog({\n client: this.phClient,\n distinctId: posthogDistinctId,\n traceId,\n model: geminiParams.model,\n provider: 'gemini',\n input: this.formatInputForPostHog(geminiParams.contents),\n output: formatResponseGemini(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 reasoningTokens: response.usageMetadata?.thoughtsTokenCount ?? 0,\n cacheReadInputTokens: response.usageMetadata?.cachedContentTokenCount ?? 0,\n },\n tools: availableTools,\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.formatInputForPostHog(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: TokenUsage = {\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 reasoningTokens: chunk.usageMetadata.thoughtsTokenCount ?? 0,\n cacheReadInputTokens: chunk.usageMetadata.cachedContentTokenCount ?? 0,\n }\n }\n yield chunk\n }\n\n const latency = (Date.now() - startTime) / 1000\n\n const availableTools = extractAvailableToolCalls('gemini', geminiParams)\n\n await sendEventToPosthog({\n client: this.phClient,\n distinctId: posthogDistinctId,\n traceId,\n model: geminiParams.model,\n provider: 'gemini',\n input: this.formatInputForPostHog(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 tools: availableTools,\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.formatInputForPostHog(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\n if (item && typeof item === 'object') {\n if (item.text) {\n return { role: item.role || 'user', content: item.text }\n }\n\n if (item.content) {\n return { role: item.role || 'user', content: item.content }\n }\n\n if (item.parts) {\n return {\n role: item.role || 'user',\n content: item.parts.map((part: any) => (part.text ? part.text : part)),\n }\n }\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\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 formatInputForPostHog(contents: any): any {\n const sanitized = sanitizeGemini(contents)\n return this.formatInput(sanitized)\n }\n}\n\nexport default PostHogGoogleGenAI\nexport { PostHogGoogleGenAI as Gemini }\n"],"names":["STRING_FORMAT","getModelParams","params","modelParams","paramKeys","key","undefined","formatResponseGemini","response","output","candidates","Array","isArray","candidate","content","parts","part","text","push","type","functionCall","function","name","arguments","args","length","role","withPrivacyMode","client","privacyMode","input","privacy_mode","extractAvailableToolCalls","provider","config","tools","sanitizeValues","obj","jsonSafe","JSON","parse","stringify","Buffer","from","toString","map","Object","fromEntries","entries","k","v","sendEventToPosthog","distinctId","traceId","model","latency","baseURL","httpStatus","usage","isError","error","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","isObject","value","REDACTED_IMAGE_PLACEHOLDER","sanitizeGeminiPart","inlineData","data","processGeminiItem","item","sanitizeGemini","PostHogGoogleGenAI","constructor","posthog","geminiConfig","phClient","GoogleGenAI","models","WrappedModels","generateContent","posthogDistinctId","posthogTraceId","posthogCaptureImmediate","geminiParams","uuidv4","startTime","Date","now","availableTools","formatInputForPostHog","contents","usageMetadata","promptTokenCount","candidatesTokenCount","thoughtsTokenCount","cachedContentTokenCount","status","generateContentStream","accumulatedContent","stream","chunk","formatInput","String","sanitized"],"mappings":";;;;;;;;AAeA,MAAMA,aAAa,GAAG,MAAM;AAmBrB,MAAMC,cAAc,GACzBC,MAAiH,IACzF;EACxB,IAAI,CAACA,MAAM,EAAE;AACX,IAAA,OAAO,EAAE;AACX,EAAA;EACA,MAAMC,WAAgC,GAAG,EAAE;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;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;AACzC,IAAA;AACF,EAAA;AACA,EAAA,OAAOF,WAAW;AACpB,CAAC;AAwIM,MAAMI,oBAAoB,GAAIC,QAAa,IAAyB;EACzE,MAAMC,MAA0B,GAAG,EAAE;AAErC,EAAA,IAAID,QAAQ,CAACE,UAAU,IAAIC,KAAK,CAACC,OAAO,CAACJ,QAAQ,CAACE,UAAU,CAAC,EAAE;AAC7D,IAAA,KAAK,MAAMG,SAAS,IAAIL,QAAQ,CAACE,UAAU,EAAE;MAC3C,IAAIG,SAAS,CAACC,OAAO,IAAID,SAAS,CAACC,OAAO,CAACC,KAAK,EAAE;QAChD,MAAMD,OAAyB,GAAG,EAAE;QAEpC,KAAK,MAAME,IAAI,IAAIH,SAAS,CAACC,OAAO,CAACC,KAAK,EAAE;UAC1C,IAAIC,IAAI,CAACC,IAAI,EAAE;YACbH,OAAO,CAACI,IAAI,CAAC;AAAEC,cAAAA,IAAI,EAAE,MAAM;cAAEF,IAAI,EAAED,IAAI,CAACC;AAAK,aAAC,CAAC;AACjD,UAAA,CAAC,MAAM,IAAID,IAAI,CAACI,YAAY,EAAE;YAC5BN,OAAO,CAACI,IAAI,CAAC;AACXC,cAAAA,IAAI,EAAE,UAAU;AAChBE,cAAAA,QAAQ,EAAE;AACRC,gBAAAA,IAAI,EAAEN,IAAI,CAACI,YAAY,CAACE,IAAI;AAC5BC,gBAAAA,SAAS,EAAEP,IAAI,CAACI,YAAY,CAACI;AAC/B;AACF,aAAC,CAAC;AACJ,UAAA;AACF,QAAA;AAEA,QAAA,IAAIV,OAAO,CAACW,MAAM,GAAG,CAAC,EAAE;UACtBhB,MAAM,CAACS,IAAI,CAAC;AACVQ,YAAAA,IAAI,EAAE,WAAW;AACjBZ,YAAAA;AACF,WAAC,CAAC;AACJ,QAAA;AACF,MAAA,CAAC,MAAM,IAAID,SAAS,CAACI,IAAI,EAAE;QACzBR,MAAM,CAACS,IAAI,CAAC;AACVQ,UAAAA,IAAI,EAAE,WAAW;AACjBZ,UAAAA,OAAO,EAAE,CAAC;AAAEK,YAAAA,IAAI,EAAE,MAAM;YAAEF,IAAI,EAAEJ,SAAS,CAACI;WAAM;AAClD,SAAC,CAAC;AACJ,MAAA;AACF,IAAA;AACF,EAAA,CAAC,MAAM,IAAIT,QAAQ,CAACS,IAAI,EAAE;IACxBR,MAAM,CAACS,IAAI,CAAC;AACVQ,MAAAA,IAAI,EAAE,WAAW;AACjBZ,MAAAA,OAAO,EAAE,CAAC;AAAEK,QAAAA,IAAI,EAAE,MAAM;QAAEF,IAAI,EAAET,QAAQ,CAACS;OAAM;AACjD,KAAC,CAAC;AACJ,EAAA;AAEA,EAAA,OAAOR,MAAM;AACf,CAAC;AAcM,MAAMkB,eAAe,GAAGA,CAACC,MAAe,EAAEC,WAAoB,EAAEC,KAAU,KAAU;EACzF,OAAQF,MAAM,CAASG,YAAY,IAAIF,WAAW,GAAG,IAAI,GAAGC,KAAK;AACnE,CAAC;;AAgBD;AACA;AACA;AACA;AACO,MAAME,yBAAyB,GAAGA,CACvCC,QAAgB,EAChB/B,MAAW,KACsD;EAO/B;IAChC,IAAIA,MAAM,CAACgC,MAAM,IAAIhC,MAAM,CAACgC,MAAM,CAACC,KAAK,EAAE;AACxC,MAAA,OAAOjC,MAAM,CAACgC,MAAM,CAACC,KAAK;AAC5B,IAAA;AAEA,IAAA,OAAO,IAAI;AACb,EAAA;AAgBF,CAAC;AAqBD,SAASC,cAAcA,CAACC,GAAQ,EAAO;AACrC,EAAA,IAAIA,GAAG,KAAK/B,SAAS,IAAI+B,GAAG,KAAK,IAAI,EAAE;AACrC,IAAA,OAAOA,GAAG;AACZ,EAAA;AACA,EAAA,MAAMC,QAAQ,GAAGC,IAAI,CAACC,KAAK,CAACD,IAAI,CAACE,SAAS,CAACJ,GAAG,CAAC,CAAC;AAChD,EAAA,IAAI,OAAOC,QAAQ,KAAK,QAAQ,EAAE;AAChC,IAAA,OAAOI,aAAM,CAACC,IAAI,CAACL,QAAQ,EAAEtC,aAAa,CAAC,CAAC4C,QAAQ,CAAC5C,aAAa,CAAC;EACrE,CAAC,MAAM,IAAIW,KAAK,CAACC,OAAO,CAAC0B,QAAQ,CAAC,EAAE;AAClC,IAAA,OAAOA,QAAQ,CAACO,GAAG,CAACT,cAAc,CAAC;EACrC,CAAC,MAAM,IAAIE,QAAQ,IAAI,OAAOA,QAAQ,KAAK,QAAQ,EAAE;AACnD,IAAA,OAAOQ,MAAM,CAACC,WAAW,CAACD,MAAM,CAACE,OAAO,CAACV,QAAQ,CAAC,CAACO,GAAG,CAAC,CAAC,CAACI,CAAC,EAAEC,CAAC,CAAC,KAAK,CAACD,CAAC,EAAEb,cAAc,CAACc,CAAC,CAAC,CAAC,CAAC,CAAC;AAC7F,EAAA;AACA,EAAA,OAAOZ,QAAQ;AACjB;AAEO,MAAMa,kBAAkB,GAAG,OAAO;EACvCvB,MAAM;EACNwB,UAAU;EACVC,OAAO;EACPC,KAAK;EACLrB,QAAQ;EACRH,KAAK;EACLrB,MAAM;EACN8C,OAAO;EACPC,OAAO;EACPtD,MAAM;AACNuD,EAAAA,UAAU,GAAG,GAAG;EAChBC,KAAK,GAAG,EAAE;AACVC,EAAAA,OAAO,GAAG,KAAK;EACfC,KAAK;EACLzB,KAAK;AACL0B,EAAAA,gBAAgB,GAAG;AACK,CAAC,KAAoB;AAC7C,EAAA,IAAI,CAACjC,MAAM,CAACkC,OAAO,EAAE;AACnB,IAAA,OAAOC,OAAO,CAACC,OAAO,EAAE;AAC1B,EAAA;AACA;AACA,EAAA,MAAMC,SAAS,GAAG7B,cAAc,CAACN,KAAK,CAAC;AACvC,EAAA,MAAMoC,UAAU,GAAG9B,cAAc,CAAC3B,MAAM,CAAC;AACzC,EAAA,MAAM0D,SAAS,GAAG/B,cAAc,CAACwB,KAAK,CAAC;EAEvC,IAAIQ,SAAS,GAAG,EAAE;AAClB,EAAA,IAAIT,OAAO,EAAE;AACXS,IAAAA,SAAS,GAAG;AACVC,MAAAA,YAAY,EAAE,IAAI;AAClBC,MAAAA,SAAS,EAAEH;KACZ;AACH,EAAA;EACA,IAAII,gBAAgB,GAAG,EAAE;EACzB,IAAIrE,MAAM,CAACsE,mBAAmB,EAAE;AAC9B,IAAA,MAAMC,YAAY,GAAG,CAACvE,MAAM,CAACsE,mBAAmB,CAACE,SAAS,IAAI,CAAC,KAAKhB,KAAK,CAACiB,WAAW,IAAI,CAAC,CAAC;AAC3F,IAAA,MAAMC,aAAa,GAAG,CAAC1E,MAAM,CAACsE,mBAAmB,CAACK,UAAU,IAAI,CAAC,KAAKnB,KAAK,CAACoB,YAAY,IAAI,CAAC,CAAC;AAC9FP,IAAAA,gBAAgB,GAAG;AACjBQ,MAAAA,kBAAkB,EAAEN,YAAY;AAChCO,MAAAA,mBAAmB,EAAEJ,aAAa;MAClCK,kBAAkB,EAAER,YAAY,GAAGG;KACpC;AACH,EAAA;AAEA,EAAA,MAAMM,qBAAqB,GAAG;IAC5B,IAAIxB,KAAK,CAACyB,eAAe,GAAG;MAAEC,oBAAoB,EAAE1B,KAAK,CAACyB;KAAiB,GAAG,EAAE,CAAC;IACjF,IAAIzB,KAAK,CAAC2B,oBAAoB,GAAG;MAAEC,2BAA2B,EAAE5B,KAAK,CAAC2B;KAAsB,GAAG,EAAE,CAAC;IAClG,IAAI3B,KAAK,CAAC6B,wBAAwB,GAAG;MAAEC,+BAA+B,EAAE9B,KAAK,CAAC6B;KAA0B,GAAG,EAAE;GAC9G;AAED,EAAA,MAAME,UAAU,GAAG;AACjBC,IAAAA,YAAY,EAAExF,MAAM,CAACyF,uBAAuB,IAAI1D,QAAQ;AACxD2D,IAAAA,SAAS,EAAE1F,MAAM,CAAC2F,oBAAoB,IAAIvC,KAAK;AAC/CwC,IAAAA,oBAAoB,EAAE7F,cAAc,CAACC,MAAM,CAAC;AAC5C6F,IAAAA,SAAS,EAAEpE,eAAe,CAACC,MAAM,EAAE1B,MAAM,CAAC8F,kBAAkB,IAAI,KAAK,EAAE/B,SAAS,CAAC;AACjFgC,IAAAA,kBAAkB,EAAEtE,eAAe,CAACC,MAAM,EAAE1B,MAAM,CAAC8F,kBAAkB,IAAI,KAAK,EAAE9B,UAAU,CAAC;AAC3FgC,IAAAA,eAAe,EAAEzC,UAAU;AAC3B0C,IAAAA,gBAAgB,EAAEzC,KAAK,CAACiB,WAAW,IAAI,CAAC;AACxCyB,IAAAA,iBAAiB,EAAE1C,KAAK,CAACoB,YAAY,IAAI,CAAC;AAC1C,IAAA,GAAGI,qBAAqB;AACxBmB,IAAAA,WAAW,EAAE9C,OAAO;AACpB+C,IAAAA,YAAY,EAAEjD,OAAO;AACrBkD,IAAAA,YAAY,EAAE/C,OAAO;IACrB,GAAGtD,MAAM,CAACsG,iBAAiB;AAC3B,IAAA,IAAIpD,UAAU,GAAG,EAAE,GAAG;AAAEqD,MAAAA,uBAAuB,EAAE;AAAM,KAAC,CAAC;AACzD,IAAA,IAAItE,KAAK,GAAG;AAAEuE,MAAAA,SAAS,EAAEvE;KAAO,GAAG,EAAE,CAAC;AACtC,IAAA,GAAGiC,SAAS;IACZ,GAAGG;GACJ;AAED,EAAA,MAAMoC,KAAK,GAAG;IACZvD,UAAU,EAAEA,UAAU,IAAIC,OAAO;AACjCsD,IAAAA,KAAK,EAAE,gBAAgB;IACvBlB,UAAU;IACVmB,MAAM,EAAE1G,MAAM,CAAC2G;GAChB;AAED,EAAA,IAAIhD,gBAAgB,EAAE;AACpB;AACA,IAAA,MAAMjC,MAAM,CAACiC,gBAAgB,CAAC8C,KAAK,CAAC;AACtC,EAAA,CAAC,MAAM;AACL/E,IAAAA,MAAM,CAACkC,OAAO,CAAC6C,KAAK,CAAC;AACvB,EAAA;AACF,CAAC;;AC1aD;;AAMO,MAAMG,QAAQ,GAAIC,KAAc,IAAuC;AAC5E,EAAA,OAAOA,KAAK,KAAK,IAAI,IAAI,OAAOA,KAAK,KAAK,QAAQ,IAAI,CAACpG,KAAK,CAACC,OAAO,CAACmG,KAAK,CAAC;AAC7E,CAAC;;ACND,MAAMC,0BAA0B,GAAG,yBAAyB;AA8I5D,MAAMC,kBAAkB,GAAIjG,IAAa,IAAc;AACrD,EAAA,IAAI,CAAC8F,QAAQ,CAAC9F,IAAI,CAAC,EAAE,OAAOA,IAAI;;AAEhC;AACA,EAAA,IAAI,YAAY,IAAIA,IAAI,IAAI8F,QAAQ,CAAC9F,IAAI,CAACkG,UAAU,CAAC,IAAI,MAAM,IAAIlG,IAAI,CAACkG,UAAU,EAAE;IAClF,OAAO;AACL,MAAA,GAAGlG,IAAI;AACPkG,MAAAA,UAAU,EAAE;QACV,GAAGlG,IAAI,CAACkG,UAAU;AAClBC,QAAAA,IAAI,EAAEH;AACR;KACD;AACH,EAAA;AAEA,EAAA,OAAOhG,IAAI;AACb,CAAC;AAED,MAAMoG,iBAAiB,GAAIC,IAAa,IAAc;AACpD,EAAA,IAAI,CAACP,QAAQ,CAACO,IAAI,CAAC,EAAE,OAAOA,IAAI;;AAEhC;AACA,EAAA,IAAI,OAAO,IAAIA,IAAI,IAAIA,IAAI,CAACtG,KAAK,EAAE;IACjC,MAAMA,KAAK,GAAGJ,KAAK,CAACC,OAAO,CAACyG,IAAI,CAACtG,KAAK,CAAC,GAAGsG,IAAI,CAACtG,KAAK,CAAC8B,GAAG,CAACoE,kBAAkB,CAAC,GAAGA,kBAAkB,CAACI,IAAI,CAACtG,KAAK,CAAC;IAE7G,OAAO;AAAE,MAAA,GAAGsG,IAAI;AAAEtG,MAAAA;KAAO;AAC3B,EAAA;AAEA,EAAA,OAAOsG,IAAI;AACb,CAAC;AAqDM,MAAMC,cAAc,GAAIH,IAAa,IAAc;AACxD;AACA;AACA,EAAA,IAAI,CAACA,IAAI,EAAE,OAAOA,IAAI;AAEtB,EAAA,IAAIxG,KAAK,CAACC,OAAO,CAACuG,IAAI,CAAC,EAAE;AACvB,IAAA,OAAOA,IAAI,CAACtE,GAAG,CAACuE,iBAAiB,CAAC;AACpC,EAAA;EAEA,OAAOA,iBAAiB,CAACD,IAAI,CAAC;AAChC,CAAC;;ACpOD;;AA8BO,MAAMI,kBAAkB,CAAC;EAK9BC,WAAWA,CAACtF,MAA8B,EAAE;IAC1C,MAAM;MAAEuF,OAAO;MAAE,GAAGC;AAAa,KAAC,GAAGxF,MAAM;IAC3C,IAAI,CAACyF,QAAQ,GAAGF,OAAO;AACvB,IAAA,IAAI,CAAC7F,MAAM,GAAG,IAAIgG,iBAAW,CAACF,YAAY,CAAC;AAC3C,IAAA,IAAI,CAACG,MAAM,GAAG,IAAIC,aAAa,CAAC,IAAI,CAAClG,MAAM,EAAE,IAAI,CAAC+F,QAAQ,CAAC;AAC7D,EAAA;AACF;AAEO,MAAMG,aAAa,CAAC;AAIzBN,EAAAA,WAAWA,CAAC5F,MAAmB,EAAE+F,QAAiB,EAAE;IAClD,IAAI,CAAC/F,MAAM,GAAGA,MAAM;IACpB,IAAI,CAAC+F,QAAQ,GAAGA,QAAQ;AAC1B,EAAA;EAEA,MAAaI,eAAeA,CAAC7H,MAAiD,EAAoC;IAChH,MAAM;MACJ8H,iBAAiB;MACjBC,cAAc;MACdzB,iBAAiB;MACjBK,aAAa;MACbqB,uBAAuB;MACvB,GAAGC;AACL,KAAC,GAAGjI,MAAM;AAEV,IAAA,MAAMmD,OAAO,GAAG4E,cAAc,IAAIG,OAAM,EAAE;AAC1C,IAAA,MAAMC,SAAS,GAAGC,IAAI,CAACC,GAAG,EAAE;IAE5B,IAAI;AACF,MAAA,MAAM/H,QAAQ,GAAG,MAAM,IAAI,CAACoB,MAAM,CAACiG,MAAM,CAACE,eAAe,CAACI,YAAY,CAAC;MACvE,MAAM5E,OAAO,GAAG,CAAC+E,IAAI,CAACC,GAAG,EAAE,GAAGF,SAAS,IAAI,IAAI;AAE/C,MAAA,MAAMG,cAAc,GAAGxG,yBAAyB,CAAC,QAAQ,EAAEmG,YAAY,CAAC;AAExE,MAAA,MAAMhF,kBAAkB,CAAC;QACvBvB,MAAM,EAAE,IAAI,CAAC+F,QAAQ;AACrBvE,QAAAA,UAAU,EAAE4E,iBAAiB;QAC7B3E,OAAO;QACPC,KAAK,EAAE6E,YAAY,CAAC7E,KAAK;AACzBrB,QAAAA,QAAQ,EAAE,QAAQ;QAClBH,KAAK,EAAE,IAAI,CAAC2G,qBAAqB,CAACN,YAAY,CAACO,QAAQ,CAAC;AACxDjI,QAAAA,MAAM,EAAEF,oBAAoB,CAACC,QAAQ,CAAC;QACtC+C,OAAO;AACPC,QAAAA,OAAO,EAAE,2CAA2C;AACpDtD,QAAAA,MAAM,EAAEA,MAAa;AACrBuD,QAAAA,UAAU,EAAE,GAAG;AACfC,QAAAA,KAAK,EAAE;AACLiB,UAAAA,WAAW,EAAEnE,QAAQ,CAACmI,aAAa,EAAEC,gBAAgB,IAAI,CAAC;AAC1D9D,UAAAA,YAAY,EAAEtE,QAAQ,CAACmI,aAAa,EAAEE,oBAAoB,IAAI,CAAC;AAC/D1D,UAAAA,eAAe,EAAE3E,QAAQ,CAACmI,aAAa,EAAEG,kBAAkB,IAAI,CAAC;AAChEzD,UAAAA,oBAAoB,EAAE7E,QAAQ,CAACmI,aAAa,EAAEI,uBAAuB,IAAI;SAC1E;AACD5G,QAAAA,KAAK,EAAEqG,cAAc;AACrB3E,QAAAA,gBAAgB,EAAEqE;AACpB,OAAC,CAAC;AAEF,MAAA,OAAO1H,QAAQ;IACjB,CAAC,CAAC,OAAOoD,KAAU,EAAE;MACnB,MAAML,OAAO,GAAG,CAAC+E,IAAI,CAACC,GAAG,EAAE,GAAGF,SAAS,IAAI,IAAI;AAC/C,MAAA,MAAMlF,kBAAkB,CAAC;QACvBvB,MAAM,EAAE,IAAI,CAAC+F,QAAQ;AACrBvE,QAAAA,UAAU,EAAE4E,iBAAiB;QAC7B3E,OAAO;QACPC,KAAK,EAAE6E,YAAY,CAAC7E,KAAK;AACzBrB,QAAAA,QAAQ,EAAE,QAAQ;QAClBH,KAAK,EAAE,IAAI,CAAC2G,qBAAqB,CAACN,YAAY,CAACO,QAAQ,CAAC;AACxDjI,QAAAA,MAAM,EAAE,EAAE;QACV8C,OAAO;AACPC,QAAAA,OAAO,EAAE,2CAA2C;AACpDtD,QAAAA,MAAM,EAAEA,MAAa;AACrBuD,QAAAA,UAAU,EAAEG,KAAK,EAAEoF,MAAM,IAAI,GAAG;AAChCtF,QAAAA,KAAK,EAAE;AACLiB,UAAAA,WAAW,EAAE,CAAC;AACdG,UAAAA,YAAY,EAAE;SACf;AACDnB,QAAAA,OAAO,EAAE,IAAI;AACbC,QAAAA,KAAK,EAAErB,IAAI,CAACE,SAAS,CAACmB,KAAK,CAAC;AAC5BC,QAAAA,gBAAgB,EAAEqE;AACpB,OAAC,CAAC;AACF,MAAA,MAAMtE,KAAK;AACb,IAAA;AACF,EAAA;EAEA,OAAcqF,qBAAqBA,CACjC/I,MAAiD,EACb;IACpC,MAAM;MACJ8H,iBAAiB;MACjBC,cAAc;MACdzB,iBAAiB;MACjBK,aAAa;MACbqB,uBAAuB;MACvB,GAAGC;AACL,KAAC,GAAGjI,MAAM;AAEV,IAAA,MAAMmD,OAAO,GAAG4E,cAAc,IAAIG,OAAM,EAAE;AAC1C,IAAA,MAAMC,SAAS,GAAGC,IAAI,CAACC,GAAG,EAAE;IAC5B,IAAIW,kBAAkB,GAAG,EAAE;AAC3B,IAAA,IAAIxF,KAAiB,GAAG;AACtBiB,MAAAA,WAAW,EAAE,CAAC;AACdG,MAAAA,YAAY,EAAE;KACf;IAED,IAAI;AACF,MAAA,MAAMqE,MAAM,GAAG,MAAM,IAAI,CAACvH,MAAM,CAACiG,MAAM,CAACoB,qBAAqB,CAACd,YAAY,CAAC;AAE3E,MAAA,WAAW,MAAMiB,KAAK,IAAID,MAAM,EAAE;QAChC,IAAIC,KAAK,CAACnI,IAAI,EAAE;UACdiI,kBAAkB,IAAIE,KAAK,CAACnI,IAAI;AAClC,QAAA;QACA,IAAImI,KAAK,CAACT,aAAa,EAAE;AACvBjF,UAAAA,KAAK,GAAG;AACNiB,YAAAA,WAAW,EAAEyE,KAAK,CAACT,aAAa,CAACC,gBAAgB,IAAI,CAAC;AACtD9D,YAAAA,YAAY,EAAEsE,KAAK,CAACT,aAAa,CAACE,oBAAoB,IAAI,CAAC;AAC3D1D,YAAAA,eAAe,EAAEiE,KAAK,CAACT,aAAa,CAACG,kBAAkB,IAAI,CAAC;AAC5DzD,YAAAA,oBAAoB,EAAE+D,KAAK,CAACT,aAAa,CAACI,uBAAuB,IAAI;WACtE;AACH,QAAA;AACA,QAAA,MAAMK,KAAK;AACb,MAAA;MAEA,MAAM7F,OAAO,GAAG,CAAC+E,IAAI,CAACC,GAAG,EAAE,GAAGF,SAAS,IAAI,IAAI;AAE/C,MAAA,MAAMG,cAAc,GAAGxG,yBAAyB,CAAC,QAAQ,EAAEmG,YAAY,CAAC;AAExE,MAAA,MAAMhF,kBAAkB,CAAC;QACvBvB,MAAM,EAAE,IAAI,CAAC+F,QAAQ;AACrBvE,QAAAA,UAAU,EAAE4E,iBAAiB;QAC7B3E,OAAO;QACPC,KAAK,EAAE6E,YAAY,CAAC7E,KAAK;AACzBrB,QAAAA,QAAQ,EAAE,QAAQ;QAClBH,KAAK,EAAE,IAAI,CAAC2G,qBAAqB,CAACN,YAAY,CAACO,QAAQ,CAAC;AACxDjI,QAAAA,MAAM,EAAE,CAAC;AAAEK,UAAAA,OAAO,EAAEoI,kBAAkB;AAAExH,UAAAA,IAAI,EAAE;AAAY,SAAC,CAAC;QAC5D6B,OAAO;AACPC,QAAAA,OAAO,EAAE,2CAA2C;AACpDtD,QAAAA,MAAM,EAAEA,MAAa;AACrBuD,QAAAA,UAAU,EAAE,GAAG;QACfC,KAAK;AACLvB,QAAAA,KAAK,EAAEqG,cAAc;AACrB3E,QAAAA,gBAAgB,EAAEqE;AACpB,OAAC,CAAC;IACJ,CAAC,CAAC,OAAOtE,KAAU,EAAE;MACnB,MAAML,OAAO,GAAG,CAAC+E,IAAI,CAACC,GAAG,EAAE,GAAGF,SAAS,IAAI,IAAI;AAC/C,MAAA,MAAMlF,kBAAkB,CAAC;QACvBvB,MAAM,EAAE,IAAI,CAAC+F,QAAQ;AACrBvE,QAAAA,UAAU,EAAE4E,iBAAiB;QAC7B3E,OAAO;QACPC,KAAK,EAAE6E,YAAY,CAAC7E,KAAK;AACzBrB,QAAAA,QAAQ,EAAE,QAAQ;QAClBH,KAAK,EAAE,IAAI,CAAC2G,qBAAqB,CAACN,YAAY,CAACO,QAAQ,CAAC;AACxDjI,QAAAA,MAAM,EAAE,EAAE;QACV8C,OAAO;AACPC,QAAAA,OAAO,EAAE,2CAA2C;AACpDtD,QAAAA,MAAM,EAAEA,MAAa;AACrBuD,QAAAA,UAAU,EAAEG,KAAK,EAAEoF,MAAM,IAAI,GAAG;AAChCtF,QAAAA,KAAK,EAAE;AACLiB,UAAAA,WAAW,EAAE,CAAC;AACdG,UAAAA,YAAY,EAAE;SACf;AACDnB,QAAAA,OAAO,EAAE,IAAI;AACbC,QAAAA,KAAK,EAAErB,IAAI,CAACE,SAAS,CAACmB,KAAK,CAAC;AAC5BC,QAAAA,gBAAgB,EAAEqE;AACpB,OAAC,CAAC;AACF,MAAA,MAAMtE,KAAK;AACb,IAAA;AACF,EAAA;EAEQyF,WAAWA,CAACX,QAAa,EAA4C;AAC3E,IAAA,IAAI,OAAOA,QAAQ,KAAK,QAAQ,EAAE;AAChC,MAAA,OAAO,CAAC;AAAEhH,QAAAA,IAAI,EAAE,MAAM;AAAEZ,QAAAA,OAAO,EAAE4H;AAAS,OAAC,CAAC;AAC9C,IAAA;AAEA,IAAA,IAAI/H,KAAK,CAACC,OAAO,CAAC8H,QAAQ,CAAC,EAAE;AAC3B,MAAA,OAAOA,QAAQ,CAAC7F,GAAG,CAAEwE,IAAI,IAAK;AAC5B,QAAA,IAAI,OAAOA,IAAI,KAAK,QAAQ,EAAE;UAC5B,OAAO;AAAE3F,YAAAA,IAAI,EAAE,MAAM;AAAEZ,YAAAA,OAAO,EAAEuG;WAAM;AACxC,QAAA;AAEA,QAAA,IAAIA,IAAI,IAAI,OAAOA,IAAI,KAAK,QAAQ,EAAE;UACpC,IAAIA,IAAI,CAACpG,IAAI,EAAE;YACb,OAAO;AAAES,cAAAA,IAAI,EAAE2F,IAAI,CAAC3F,IAAI,IAAI,MAAM;cAAEZ,OAAO,EAAEuG,IAAI,CAACpG;aAAM;AAC1D,UAAA;UAEA,IAAIoG,IAAI,CAACvG,OAAO,EAAE;YAChB,OAAO;AAAEY,cAAAA,IAAI,EAAE2F,IAAI,CAAC3F,IAAI,IAAI,MAAM;cAAEZ,OAAO,EAAEuG,IAAI,CAACvG;aAAS;AAC7D,UAAA;UAEA,IAAIuG,IAAI,CAACtG,KAAK,EAAE;YACd,OAAO;AACLW,cAAAA,IAAI,EAAE2F,IAAI,CAAC3F,IAAI,IAAI,MAAM;AACzBZ,cAAAA,OAAO,EAAEuG,IAAI,CAACtG,KAAK,CAAC8B,GAAG,CAAE7B,IAAS,IAAMA,IAAI,CAACC,IAAI,GAAGD,IAAI,CAACC,IAAI,GAAGD,IAAK;aACtE;AACH,UAAA;AACF,QAAA;QAEA,OAAO;AAAEU,UAAAA,IAAI,EAAE,MAAM;UAAEZ,OAAO,EAAEwI,MAAM,CAACjC,IAAI;SAAG;AAChD,MAAA,CAAC,CAAC;AACJ,IAAA;AAEA,IAAA,IAAIqB,QAAQ,IAAI,OAAOA,QAAQ,KAAK,QAAQ,EAAE;MAC5C,IAAIA,QAAQ,CAACzH,IAAI,EAAE;AACjB,QAAA,OAAO,CAAC;AAAES,UAAAA,IAAI,EAAE,MAAM;UAAEZ,OAAO,EAAE4H,QAAQ,CAACzH;AAAK,SAAC,CAAC;AACnD,MAAA;MAEA,IAAIyH,QAAQ,CAAC5H,OAAO,EAAE;AACpB,QAAA,OAAO,CAAC;AAAEY,UAAAA,IAAI,EAAE,MAAM;UAAEZ,OAAO,EAAE4H,QAAQ,CAAC5H;AAAQ,SAAC,CAAC;AACtD,MAAA;AACF,IAAA;AAEA,IAAA,OAAO,CAAC;AAAEY,MAAAA,IAAI,EAAE,MAAM;MAAEZ,OAAO,EAAEwI,MAAM,CAACZ,QAAQ;AAAE,KAAC,CAAC;AACtD,EAAA;EAEQD,qBAAqBA,CAACC,QAAa,EAAO;AAChD,IAAA,MAAMa,SAAS,GAAGjC,cAAc,CAACoB,QAAQ,CAAC;AAC1C,IAAA,OAAO,IAAI,CAACW,WAAW,CAACE,SAAS,CAAC;AACpC,EAAA;AACF;;;;;;;"}
|
package/dist/gemini/index.d.ts
CHANGED
|
@@ -30,6 +30,8 @@ type GenerateContentResponse = {
|
|
|
30
30
|
promptTokenCount?: number;
|
|
31
31
|
candidatesTokenCount?: number;
|
|
32
32
|
totalTokenCount?: number;
|
|
33
|
+
thoughtsTokenCount?: number;
|
|
34
|
+
cachedContentTokenCount?: number;
|
|
33
35
|
};
|
|
34
36
|
[key: string]: any;
|
|
35
37
|
};
|
|
@@ -54,6 +56,7 @@ declare class WrappedModels {
|
|
|
54
56
|
generateContent(params: GenerateContentRequest & MonitoringParams): Promise<GenerateContentResponse>;
|
|
55
57
|
generateContentStream(params: GenerateContentRequest & MonitoringParams): AsyncGenerator<any, void, unknown>;
|
|
56
58
|
private formatInput;
|
|
59
|
+
private formatInputForPostHog;
|
|
57
60
|
}
|
|
58
61
|
|
|
59
62
|
export { PostHogGoogleGenAI as Gemini, PostHogGoogleGenAI, WrappedModels, PostHogGoogleGenAI as default };
|
package/dist/gemini/index.mjs
CHANGED
|
@@ -185,6 +185,51 @@ const sendEventToPosthog = async ({
|
|
|
185
185
|
}
|
|
186
186
|
};
|
|
187
187
|
|
|
188
|
+
// Type guards for safer type checking
|
|
189
|
+
|
|
190
|
+
const isObject = value => {
|
|
191
|
+
return value !== null && typeof value === 'object' && !Array.isArray(value);
|
|
192
|
+
};
|
|
193
|
+
|
|
194
|
+
const REDACTED_IMAGE_PLACEHOLDER = '[base64 image redacted]';
|
|
195
|
+
const sanitizeGeminiPart = part => {
|
|
196
|
+
if (!isObject(part)) return part;
|
|
197
|
+
|
|
198
|
+
// Handle Gemini's inline data format
|
|
199
|
+
if ('inlineData' in part && isObject(part.inlineData) && 'data' in part.inlineData) {
|
|
200
|
+
return {
|
|
201
|
+
...part,
|
|
202
|
+
inlineData: {
|
|
203
|
+
...part.inlineData,
|
|
204
|
+
data: REDACTED_IMAGE_PLACEHOLDER
|
|
205
|
+
}
|
|
206
|
+
};
|
|
207
|
+
}
|
|
208
|
+
return part;
|
|
209
|
+
};
|
|
210
|
+
const processGeminiItem = item => {
|
|
211
|
+
if (!isObject(item)) return item;
|
|
212
|
+
|
|
213
|
+
// If it has parts, process them
|
|
214
|
+
if ('parts' in item && item.parts) {
|
|
215
|
+
const parts = Array.isArray(item.parts) ? item.parts.map(sanitizeGeminiPart) : sanitizeGeminiPart(item.parts);
|
|
216
|
+
return {
|
|
217
|
+
...item,
|
|
218
|
+
parts
|
|
219
|
+
};
|
|
220
|
+
}
|
|
221
|
+
return item;
|
|
222
|
+
};
|
|
223
|
+
const sanitizeGemini = data => {
|
|
224
|
+
// Gemini has a different structure with 'parts' directly on items instead of 'content'
|
|
225
|
+
// So we need custom processing instead of using processMessages
|
|
226
|
+
if (!data) return data;
|
|
227
|
+
if (Array.isArray(data)) {
|
|
228
|
+
return data.map(processGeminiItem);
|
|
229
|
+
}
|
|
230
|
+
return processGeminiItem(data);
|
|
231
|
+
};
|
|
232
|
+
|
|
188
233
|
// Types from @google/genai
|
|
189
234
|
|
|
190
235
|
class PostHogGoogleGenAI {
|
|
@@ -224,7 +269,7 @@ class WrappedModels {
|
|
|
224
269
|
traceId,
|
|
225
270
|
model: geminiParams.model,
|
|
226
271
|
provider: 'gemini',
|
|
227
|
-
input: this.
|
|
272
|
+
input: this.formatInputForPostHog(geminiParams.contents),
|
|
228
273
|
output: formatResponseGemini(response),
|
|
229
274
|
latency,
|
|
230
275
|
baseURL: 'https://generativelanguage.googleapis.com',
|
|
@@ -232,7 +277,9 @@ class WrappedModels {
|
|
|
232
277
|
httpStatus: 200,
|
|
233
278
|
usage: {
|
|
234
279
|
inputTokens: response.usageMetadata?.promptTokenCount ?? 0,
|
|
235
|
-
outputTokens: response.usageMetadata?.candidatesTokenCount ?? 0
|
|
280
|
+
outputTokens: response.usageMetadata?.candidatesTokenCount ?? 0,
|
|
281
|
+
reasoningTokens: response.usageMetadata?.thoughtsTokenCount ?? 0,
|
|
282
|
+
cacheReadInputTokens: response.usageMetadata?.cachedContentTokenCount ?? 0
|
|
236
283
|
},
|
|
237
284
|
tools: availableTools,
|
|
238
285
|
captureImmediate: posthogCaptureImmediate
|
|
@@ -246,7 +293,7 @@ class WrappedModels {
|
|
|
246
293
|
traceId,
|
|
247
294
|
model: geminiParams.model,
|
|
248
295
|
provider: 'gemini',
|
|
249
|
-
input: this.
|
|
296
|
+
input: this.formatInputForPostHog(geminiParams.contents),
|
|
250
297
|
output: [],
|
|
251
298
|
latency,
|
|
252
299
|
baseURL: 'https://generativelanguage.googleapis.com',
|
|
@@ -288,7 +335,9 @@ class WrappedModels {
|
|
|
288
335
|
if (chunk.usageMetadata) {
|
|
289
336
|
usage = {
|
|
290
337
|
inputTokens: chunk.usageMetadata.promptTokenCount ?? 0,
|
|
291
|
-
outputTokens: chunk.usageMetadata.candidatesTokenCount ?? 0
|
|
338
|
+
outputTokens: chunk.usageMetadata.candidatesTokenCount ?? 0,
|
|
339
|
+
reasoningTokens: chunk.usageMetadata.thoughtsTokenCount ?? 0,
|
|
340
|
+
cacheReadInputTokens: chunk.usageMetadata.cachedContentTokenCount ?? 0
|
|
292
341
|
};
|
|
293
342
|
}
|
|
294
343
|
yield chunk;
|
|
@@ -301,7 +350,7 @@ class WrappedModels {
|
|
|
301
350
|
traceId,
|
|
302
351
|
model: geminiParams.model,
|
|
303
352
|
provider: 'gemini',
|
|
304
|
-
input: this.
|
|
353
|
+
input: this.formatInputForPostHog(geminiParams.contents),
|
|
305
354
|
output: [{
|
|
306
355
|
content: accumulatedContent,
|
|
307
356
|
role: 'assistant'
|
|
@@ -322,7 +371,7 @@ class WrappedModels {
|
|
|
322
371
|
traceId,
|
|
323
372
|
model: geminiParams.model,
|
|
324
373
|
provider: 'gemini',
|
|
325
|
-
input: this.
|
|
374
|
+
input: this.formatInputForPostHog(geminiParams.contents),
|
|
326
375
|
output: [],
|
|
327
376
|
latency,
|
|
328
377
|
baseURL: 'https://generativelanguage.googleapis.com',
|
|
@@ -367,6 +416,12 @@ class WrappedModels {
|
|
|
367
416
|
content: item.content
|
|
368
417
|
};
|
|
369
418
|
}
|
|
419
|
+
if (item.parts) {
|
|
420
|
+
return {
|
|
421
|
+
role: item.role || 'user',
|
|
422
|
+
content: item.parts.map(part => part.text ? part.text : part)
|
|
423
|
+
};
|
|
424
|
+
}
|
|
370
425
|
}
|
|
371
426
|
return {
|
|
372
427
|
role: 'user',
|
|
@@ -393,6 +448,10 @@ class WrappedModels {
|
|
|
393
448
|
content: String(contents)
|
|
394
449
|
}];
|
|
395
450
|
}
|
|
451
|
+
formatInputForPostHog(contents) {
|
|
452
|
+
const sanitized = sanitizeGemini(contents);
|
|
453
|
+
return this.formatInput(sanitized);
|
|
454
|
+
}
|
|
396
455
|
}
|
|
397
456
|
|
|
398
457
|
export { PostHogGoogleGenAI as Gemini, PostHogGoogleGenAI, WrappedModels, PostHogGoogleGenAI as default };
|