@posthog/ai 1.3.0 → 1.5.0

package/lib/index.cjs.js

@@ -0,0 +1,386 @@
+'use strict';
+
+Object.defineProperty(exports, '__esModule', { value: true });
+
+var OpenAIOrignal = require('openai');
+var uuid = require('uuid');
+var stream = require('stream');
+var ai = require('ai');
+
+function _interopDefaultLegacy (e) { return e && typeof e === 'object' && 'default' in e ? e : { 'default': e }; }
+
+var OpenAIOrignal__default = /*#__PURE__*/_interopDefaultLegacy(OpenAIOrignal);
+
+const getModelParams = params => {
+  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 mergeSystemPrompt = (params, provider) => {
+  if (provider !== 'anthropic') {
+    return params.messages;
+  }
+  const messages = params.messages || [];
+  if (!params.system) {
+    return messages;
+  }
+  const systemMessage = params.system;
+  return [{
+    role: 'system',
+    content: systemMessage
+  }, ...messages];
+};
+const withPrivacyMode = (client, privacyMode, input) => {
+  return client.privacy_mode || privacyMode ? null : input;
+};
+const sendEventToPosthog = ({
+  client,
+  distinctId,
+  traceId,
+  model,
+  provider,
+  input,
+  output,
+  latency,
+  baseURL,
+  params,
+  httpStatus = 200,
+  usage = {}
+}) => {
+  if (client.capture) {
+    client.capture({
+      distinctId: distinctId ?? traceId,
+      event: '$ai_generation',
+      properties: {
+        $ai_provider: provider,
+        $ai_model: model,
+        $ai_model_parameters: getModelParams(params),
+        $ai_input: withPrivacyMode(client, params.posthog_privacy_mode ?? false, input),
+        $ai_output_choices: withPrivacyMode(client, params.posthog_privacy_mode ?? false, output),
+        $ai_http_status: httpStatus,
+        $ai_input_tokens: usage.input_tokens ?? 0,
+        $ai_output_tokens: usage.output_tokens ?? 0,
+        $ai_latency: latency,
+        $ai_trace_id: traceId,
+        $ai_base_url: baseURL,
+        ...params.posthog_properties,
+        ...(distinctId ? {} : {
+          $process_person_profile: false
+        })
+      },
+      groups: params.posthog_groups
+    });
+  }
+};
+
+class PostHogOpenAI extends OpenAIOrignal__default["default"] {
+  constructor(config) {
+    const {
+      posthog,
+      ...openAIConfig
+    } = config;
+    super(openAIConfig);
+    this.phClient = posthog;
+    this.chat = new WrappedChat(this, this.phClient);
+  }
+}
+class WrappedChat extends OpenAIOrignal__default["default"].Chat {
+  constructor(parentClient, phClient) {
+    super(parentClient);
+    this.completions = new WrappedCompletions(parentClient, phClient);
+  }
+}
+class WrappedCompletions extends OpenAIOrignal__default["default"].Chat.Completions {
+  constructor(client, phClient) {
+    super(client);
+    this.phClient = phClient;
+  }
+  // --- Implementation Signature
+  create(body, options) {
+    const {
+      posthog_distinct_id,
+      posthog_trace_id,
+      posthog_properties,
+      posthog_privacy_mode = false,
+      posthog_groups,
+      ...openAIParams
+    } = body;
+    const traceId = posthog_trace_id ?? uuid.v4();
+    const startTime = Date.now();
+    const parentPromise = super.create(openAIParams, options);
+    if (openAIParams.stream) {
+      return parentPromise.then(value => {
+        const passThroughStream = new stream.PassThrough({
+          objectMode: true
+        });
+        let accumulatedContent = '';
+        let usage = {
+          input_tokens: 0,
+          output_tokens: 0
+        };
+        if ('tee' in value) {
+          const openAIStream = value;
+          (async () => {
+            try {
+              for await (const chunk of openAIStream) {
+                const delta = chunk?.choices?.[0]?.delta?.content ?? '';
+                accumulatedContent += delta;
+                if (chunk.usage) {
+                  usage = {
+                    input_tokens: chunk.usage.prompt_tokens ?? 0,
+                    output_tokens: chunk.usage.completion_tokens ?? 0
+                  };
+                }
+                passThroughStream.write(chunk);
+              }
+              const latency = (Date.now() - startTime) / 1000;
+              sendEventToPosthog({
+                client: this.phClient,
+                distinctId: posthog_distinct_id ?? traceId,
+                traceId,
+                model: openAIParams.model,
+                provider: 'openai',
+                input: posthog_privacy_mode ? '' : mergeSystemPrompt(openAIParams, 'openai'),
+                output: [{
+                  content: accumulatedContent,
+                  role: 'assistant'
+                }],
+                latency,
+                baseURL: this.baseURL ?? '',
+                params: body,
+                httpStatus: 200,
+                usage
+              });
+              passThroughStream.end();
+            } catch (error) {
+              // error handling
+              sendEventToPosthog({
+                client: this.phClient,
+                distinctId: posthog_distinct_id ?? traceId,
+                traceId,
+                model: openAIParams.model,
+                provider: 'openai',
+                input: posthog_privacy_mode ? '' : mergeSystemPrompt(openAIParams, 'openai'),
+                output: [],
+                latency: 0,
+                baseURL: this.baseURL ?? '',
+                params: body,
+                httpStatus: 500,
+                usage: {
+                  input_tokens: 0,
+                  output_tokens: 0
+                }
+              });
+              passThroughStream.emit('error', error);
+            }
+          })();
+        }
+        return passThroughStream;
+      });
+    } else {
+      const wrappedPromise = parentPromise.then(result => {
+        if ('choices' in result) {
+          const latency = (Date.now() - startTime) / 1000;
+          sendEventToPosthog({
+            client: this.phClient,
+            distinctId: posthog_distinct_id ?? traceId,
+            traceId,
+            model: openAIParams.model,
+            provider: 'openai',
+            input: posthog_privacy_mode ? '' : mergeSystemPrompt(openAIParams, 'openai'),
+            output: [{
+              content: result.choices[0].message.content,
+              role: 'assistant'
+            }],
+            latency,
+            baseURL: this.baseURL ?? '',
+            params: body,
+            httpStatus: 200,
+            usage: {
+              input_tokens: result.usage?.prompt_tokens ?? 0,
+              output_tokens: result.usage?.completion_tokens ?? 0
+            }
+          });
+        }
+        return result;
+      }, error => {
+        sendEventToPosthog({
+          client: this.phClient,
+          distinctId: posthog_distinct_id ?? traceId,
+          traceId,
+          model: openAIParams.model,
+          provider: 'openai',
+          input: posthog_privacy_mode ? '' : mergeSystemPrompt(openAIParams, 'openai'),
+          output: [],
+          latency: 0,
+          baseURL: this.baseURL ?? '',
+          params: body,
+          httpStatus: 500,
+          usage: {
+            input_tokens: 0,
+            output_tokens: 0
+          }
+        });
+        throw error;
+      });
+      return wrappedPromise;
+    }
+  }
+}
+
+const createInstrumentationMiddleware = (phClient, model, options) => {
+  const middleware = {
+    wrapGenerate: async ({
+      doGenerate,
+      params
+    }) => {
+      const startTime = Date.now();
+      try {
+        const result = await doGenerate();
+        const latency = (Date.now() - startTime) / 1000;
+        sendEventToPosthog({
+          client: phClient,
+          distinctId: options.posthogDistinctId,
+          traceId: options.posthogTraceId,
+          model: model.modelId,
+          provider: 'vercel',
+          input: options.posthogPrivacyMode ? '' : params.prompt,
+          output: [{
+            content: result.text,
+            role: 'assistant'
+          }],
+          latency,
+          baseURL: '',
+          params: {
+            posthog_properties: options
+          },
+          httpStatus: 200,
+          usage: {
+            input_tokens: result.usage.promptTokens,
+            output_tokens: result.usage.completionTokens
+          }
+        });
+        return result;
+      } catch (error) {
+        sendEventToPosthog({
+          client: phClient,
+          distinctId: options.posthogDistinctId,
+          traceId: options.posthogTraceId,
+          model: model.modelId,
+          provider: 'vercel',
+          input: options.posthogPrivacyMode ? '' : params.prompt,
+          output: [],
+          latency: 0,
+          baseURL: '',
+          params: {
+            posthog_properties: options
+          },
+          httpStatus: 500,
+          usage: {
+            input_tokens: 0,
+            output_tokens: 0
+          }
+        });
+        throw error;
+      }
+    },
+    wrapStream: async ({
+      doStream,
+      params
+    }) => {
+      const startTime = Date.now();
+      let generatedText = '';
+      let usage = {};
+      try {
+        const {
+          stream,
+          ...rest
+        } = await doStream();
+        const transformStream = new TransformStream({
+          transform(chunk, controller) {
+            if (chunk.type === 'text-delta') {
+              generatedText += chunk.textDelta;
+            }
+            if (chunk.type === 'finish') {
+              usage = {
+                input_tokens: chunk.usage?.promptTokens,
+                output_tokens: chunk.usage?.completionTokens
+              };
+            }
+            controller.enqueue(chunk);
+          },
+          flush() {
+            const latency = (Date.now() - startTime) / 1000;
+            sendEventToPosthog({
+              client: phClient,
+              distinctId: options.posthogDistinctId,
+              traceId: options.posthogTraceId,
+              model: model.modelId,
+              provider: 'vercel',
+              input: options.posthogPrivacyMode ? '' : params.prompt,
+              output: [{
+                content: generatedText,
+                role: 'assistant'
+              }],
+              latency,
+              baseURL: '',
+              params: {
+                posthog_properties: options
+              },
+              httpStatus: 200,
+              usage
+            });
+          }
+        });
+        return {
+          stream: stream.pipeThrough(transformStream),
+          ...rest
+        };
+      } catch (error) {
+        sendEventToPosthog({
+          client: phClient,
+          distinctId: options.posthogDistinctId,
+          traceId: options.posthogTraceId,
+          model: model.modelId,
+          provider: 'vercel',
+          input: options.posthogPrivacyMode ? '' : params.prompt,
+          output: [],
+          latency: 0,
+          baseURL: '',
+          params: {
+            posthog_properties: options
+          },
+          httpStatus: 500,
+          usage: {
+            input_tokens: 0,
+            output_tokens: 0
+          }
+        });
+        throw error;
+      }
+    }
+  };
+  return middleware;
+};
+const wrapVercelLanguageModel = (model, phClient, options) => {
+  const traceId = options.posthogTraceId ?? uuid.v4();
+  const middleware = createInstrumentationMiddleware(phClient, model, {
+    ...options,
+    posthogTraceId: traceId,
+    posthogDistinctId: options.posthogDistinctId ?? traceId
+  });
+  const wrappedModel = ai.experimental_wrapLanguageModel({
+    model,
+    middleware
+  });
+  return wrappedModel;
+};
+
+exports.OpenAI = PostHogOpenAI;
+exports.withTracing = wrapVercelLanguageModel;
+//# sourceMappingURL=index.cjs.js.map

package/lib/index.cjs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.cjs.js","sources":["../src/utils.ts","../src/openai/index.ts","../src/vercel/middleware.ts"],"sourcesContent":["import { PostHog } from 'posthog-node'\nimport OpenAIOrignal from 'openai'\n\ntype ChatCompletionCreateParamsBase = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParams\n\nexport interface MonitoringParams {\n  posthog_distinct_id?: string\n  posthog_trace_id?: string\n  posthog_properties?: Record<string, any>\n  posthog_privacy_mode?: boolean\n  posthog_groups?: Record<string, any>\n}\n\nexport const getModelParams = (params: ChatCompletionCreateParamsBase & MonitoringParams): Record<string, any> => {\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\nexport const getUsage = (response: any, provider: string): { input_tokens: number; output_tokens: number } => {\n  if (!response?.usage) {\n    return { input_tokens: 0, output_tokens: 0 }\n  }\n\n  if (provider === 'anthropic') {\n    return {\n      input_tokens: response.usage.input_tokens ?? 0,\n      output_tokens: response.usage.output_tokens ?? 0,\n    }\n  } else if (provider === 'openai') {\n    return {\n      input_tokens: response.usage.prompt_tokens ?? 0,\n      output_tokens: response.usage.completion_tokens ?? 0,\n    }\n  }\n\n  return { input_tokens: 0, output_tokens: 0 }\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  }\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 mergeSystemPrompt = (params: ChatCompletionCreateParamsBase & MonitoringParams, provider: string): any => {\n  if (provider !== 'anthropic') {\n    return params.messages\n  }\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\nexport const withPrivacyMode = (client: PostHog, privacyMode: boolean, input: any): any => {\n  return (client as any).privacy_mode || privacyMode ? null : input\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?: { input_tokens?: number; output_tokens?: number }\n  params: ChatCompletionCreateParamsBase & MonitoringParams\n}\n\nexport const sendEventToPosthog = ({\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}: SendEventToPosthogParams): void => {\n  if (client.capture) {\n    client.capture({\n      distinctId: distinctId ?? traceId,\n      event: '$ai_generation',\n      properties: {\n        $ai_provider: provider,\n        $ai_model: model,\n        $ai_model_parameters: getModelParams(params),\n        $ai_input: withPrivacyMode(client, params.posthog_privacy_mode ?? false, input),\n        $ai_output_choices: withPrivacyMode(client, params.posthog_privacy_mode ?? false, output),\n        $ai_http_status: httpStatus,\n        $ai_input_tokens: usage.input_tokens ?? 0,\n        $ai_output_tokens: usage.output_tokens ?? 0,\n        $ai_latency: latency,\n        $ai_trace_id: traceId,\n        $ai_base_url: baseURL,\n        ...params.posthog_properties,\n        ...(distinctId ? {} : { $process_person_profile: false }),\n      },\n      groups: params.posthog_groups,\n    })\n  }\n}\n","import OpenAIOrignal from 'openai'\nimport { PostHog } from 'posthog-node'\nimport { v4 as uuidv4 } from 'uuid'\nimport { PassThrough } from 'stream'\nimport { mergeSystemPrompt, MonitoringParams, sendEventToPosthog } from '../utils'\n\ntype ChatCompletion = OpenAIOrignal.ChatCompletion\ntype ChatCompletionChunk = OpenAIOrignal.ChatCompletionChunk\ntype ChatCompletionCreateParamsBase = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParams\ntype ChatCompletionCreateParamsNonStreaming = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParamsNonStreaming\ntype ChatCompletionCreateParamsStreaming = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParamsStreaming\nimport type { APIPromise, RequestOptions } from 'openai/core'\nimport type { Stream } from 'openai/streaming'\n\ninterface MonitoringOpenAIConfig {\n  apiKey: string\n  posthog: PostHog\n  baseURL?: string\n}\n\nexport class PostHogOpenAI extends OpenAIOrignal {\n  private readonly phClient: PostHog\n\n  constructor(config: MonitoringOpenAIConfig) {\n    const { posthog, ...openAIConfig } = config\n    super(openAIConfig)\n    this.phClient = posthog\n    this.chat = new WrappedChat(this, this.phClient)\n  }\n\n  public chat: WrappedChat\n}\n\nexport class WrappedChat extends OpenAIOrignal.Chat {\n  constructor(parentClient: PostHogOpenAI, phClient: PostHog) {\n    super(parentClient)\n    this.completions = new WrappedCompletions(parentClient, phClient)\n  }\n\n  public completions: WrappedCompletions\n}\n\nexport class WrappedCompletions extends OpenAIOrignal.Chat.Completions {\n  private readonly phClient: PostHog\n\n  constructor(client: OpenAIOrignal, phClient: PostHog) {\n    super(client)\n    this.phClient = phClient\n  }\n\n  // --- Overload #1: Non-streaming\n  public create(\n    body: ChatCompletionCreateParamsNonStreaming & MonitoringParams,\n    options?: RequestOptions\n  ): APIPromise<ChatCompletion>\n\n  // --- Overload #2: Streaming\n  public create(\n    body: ChatCompletionCreateParamsStreaming & MonitoringParams,\n    options?: RequestOptions\n  ): APIPromise<Stream<ChatCompletionChunk>>\n\n  // --- Overload #3: Generic base\n  public create(\n    body: ChatCompletionCreateParamsBase & MonitoringParams,\n    options?: RequestOptions\n  ): APIPromise<ChatCompletion | Stream<ChatCompletionChunk>>\n\n  // --- Implementation Signature\n  public create(\n    body: ChatCompletionCreateParamsBase & MonitoringParams,\n    options?: RequestOptions\n  ): APIPromise<ChatCompletion | Stream<ChatCompletionChunk>> {\n    const {\n      posthog_distinct_id,\n      posthog_trace_id,\n      posthog_properties,\n      posthog_privacy_mode = false,\n      posthog_groups,\n      ...openAIParams\n    } = body\n\n    const traceId = posthog_trace_id ?? uuidv4()\n    const startTime = Date.now()\n\n    const parentPromise = super.create(openAIParams, options)\n\n    if (openAIParams.stream) {\n      return parentPromise.then((value) => {\n        const passThroughStream = new PassThrough({ objectMode: true })\n        let accumulatedContent = ''\n        let usage: { input_tokens: number; output_tokens: number } = {\n          input_tokens: 0,\n          output_tokens: 0,\n        }\n        if ('tee' in value) {\n          const openAIStream = value\n          ;(async () => {\n            try {\n              for await (const chunk of openAIStream) {\n                const delta = chunk?.choices?.[0]?.delta?.content ?? ''\n                accumulatedContent += delta\n                if (chunk.usage) {\n                  usage = {\n                    input_tokens: chunk.usage.prompt_tokens ?? 0,\n                    output_tokens: chunk.usage.completion_tokens ?? 0,\n                  }\n                }\n                passThroughStream.write(chunk)\n              }\n              const latency = (Date.now() - startTime) / 1000\n              sendEventToPosthog({\n                client: this.phClient,\n                distinctId: posthog_distinct_id ?? traceId,\n                traceId,\n                model: openAIParams.model,\n                provider: 'openai',\n                input: posthog_privacy_mode ? '' : mergeSystemPrompt(openAIParams, 'openai'),\n                output: [{ content: accumulatedContent, role: 'assistant' }],\n                latency,\n                baseURL: (this as any).baseURL ?? '',\n                params: body,\n                httpStatus: 200,\n                usage,\n              })\n              passThroughStream.end()\n            } catch (error) {\n              // error handling\n              sendEventToPosthog({\n                client: this.phClient,\n                distinctId: posthog_distinct_id ?? traceId,\n                traceId,\n                model: openAIParams.model,\n                provider: 'openai',\n                input: posthog_privacy_mode ? '' : mergeSystemPrompt(openAIParams, 'openai'),\n                output: [],\n                latency: 0,\n                baseURL: (this as any).baseURL ?? '',\n                params: body,\n                httpStatus: 500,\n                usage: {\n                  input_tokens: 0,\n                  output_tokens: 0,\n                },\n              })\n              passThroughStream.emit('error', error)\n            }\n          })()\n        }\n        return passThroughStream as unknown as Stream<ChatCompletionChunk>\n      }) as APIPromise<Stream<ChatCompletionChunk>>\n    } else {\n      const wrappedPromise = parentPromise.then(\n        (result) => {\n          if ('choices' in result) {\n            const latency = (Date.now() - startTime) / 1000\n            sendEventToPosthog({\n              client: this.phClient,\n              distinctId: posthog_distinct_id ?? traceId,\n              traceId,\n              model: openAIParams.model,\n              provider: 'openai',\n              input: posthog_privacy_mode ? '' : mergeSystemPrompt(openAIParams, 'openai'),\n              output: [{ content: result.choices[0].message.content, role: 'assistant' }],\n              latency,\n              baseURL: (this as any).baseURL ?? '',\n              params: body,\n              httpStatus: 200,\n              usage: {\n                input_tokens: result.usage?.prompt_tokens ?? 0,\n                output_tokens: result.usage?.completion_tokens ?? 0,\n              },\n            })\n          }\n          return result\n        },\n        (error) => {\n          sendEventToPosthog({\n            client: this.phClient,\n            distinctId: posthog_distinct_id ?? traceId,\n            traceId,\n            model: openAIParams.model,\n            provider: 'openai',\n            input: posthog_privacy_mode ? '' : mergeSystemPrompt(openAIParams, 'openai'),\n            output: [],\n            latency: 0,\n            baseURL: (this as any).baseURL ?? '',\n            params: body,\n            httpStatus: 500,\n            usage: {\n              input_tokens: 0,\n              output_tokens: 0,\n            },\n          })\n          throw error\n        }\n      ) as APIPromise<ChatCompletion>\n\n      return wrappedPromise\n    }\n  }\n}\n\nexport default PostHogOpenAI\n","import { experimental_wrapLanguageModel as wrapLanguageModel } from 'ai'\nimport type {\n  LanguageModelV1,\n  Experimental_LanguageModelV1Middleware as LanguageModelV1Middleware,\n  LanguageModelV1StreamPart,\n} from 'ai'\nimport { v4 as uuidv4 } from 'uuid'\nimport type { PostHog } from 'posthog-node'\nimport { sendEventToPosthog } from '../utils'\n\ninterface CreateInstrumentationMiddlewareOptions {\n  posthogDistinctId?: string\n  posthogTraceId: string\n  posthogProperties?: Record<string, any>\n  posthogPrivacyMode?: boolean\n  posthogGroups?: string[]\n}\n\nexport const createInstrumentationMiddleware = (\n  phClient: PostHog,\n  model: LanguageModelV1,\n  options: CreateInstrumentationMiddlewareOptions\n): LanguageModelV1Middleware => {\n  const middleware: LanguageModelV1Middleware = {\n    wrapGenerate: async ({ doGenerate, params }) => {\n      const startTime = Date.now()\n\n      try {\n        const result = await doGenerate()\n        const latency = (Date.now() - startTime) / 1000\n\n        sendEventToPosthog({\n          client: phClient,\n          distinctId: options.posthogDistinctId,\n          traceId: options.posthogTraceId,\n          model: model.modelId,\n          provider: 'vercel',\n          input: options.posthogPrivacyMode ? '' : params.prompt,\n          output: [{ content: result.text, role: 'assistant' }],\n          latency,\n          baseURL: '',\n          params: { posthog_properties: options } as any,\n          httpStatus: 200,\n          usage: {\n            input_tokens: result.usage.promptTokens,\n            output_tokens: result.usage.completionTokens,\n          },\n        })\n\n        return result\n      } catch (error) {\n        sendEventToPosthog({\n          client: phClient,\n          distinctId: options.posthogDistinctId,\n          traceId: options.posthogTraceId,\n          model: model.modelId,\n          provider: 'vercel',\n          input: options.posthogPrivacyMode ? '' : params.prompt,\n          output: [],\n          latency: 0,\n          baseURL: '',\n          params: { posthog_properties: options } as any,\n          httpStatus: 500,\n          usage: {\n            input_tokens: 0,\n            output_tokens: 0,\n          },\n        })\n        throw error\n      }\n    },\n\n    wrapStream: async ({ doStream, params }) => {\n      const startTime = Date.now()\n      let generatedText = ''\n      let usage: { input_tokens?: number; output_tokens?: number } = {}\n\n      try {\n        const { stream, ...rest } = await doStream()\n\n        const transformStream = new TransformStream<LanguageModelV1StreamPart, LanguageModelV1StreamPart>({\n          transform(chunk, controller) {\n            if (chunk.type === 'text-delta') {\n              generatedText += chunk.textDelta\n            }\n            if (chunk.type === 'finish') {\n              usage = {\n                input_tokens: chunk.usage?.promptTokens,\n                output_tokens: chunk.usage?.completionTokens,\n              }\n            }\n            controller.enqueue(chunk)\n          },\n\n          flush() {\n            const latency = (Date.now() - startTime) / 1000\n            sendEventToPosthog({\n              client: phClient,\n              distinctId: options.posthogDistinctId,\n              traceId: options.posthogTraceId,\n              model: model.modelId,\n              provider: 'vercel',\n              input: options.posthogPrivacyMode ? '' : params.prompt,\n              output: [{ content: generatedText, role: 'assistant' }],\n              latency,\n              baseURL: '',\n              params: { posthog_properties: options } as any,\n              httpStatus: 200,\n              usage,\n            })\n          },\n        })\n\n        return {\n          stream: stream.pipeThrough(transformStream),\n          ...rest,\n        }\n      } catch (error) {\n        sendEventToPosthog({\n          client: phClient,\n          distinctId: options.posthogDistinctId,\n          traceId: options.posthogTraceId,\n          model: model.modelId,\n          provider: 'vercel',\n          input: options.posthogPrivacyMode ? '' : params.prompt,\n          output: [],\n          latency: 0,\n          baseURL: '',\n          params: { posthog_properties: options } as any,\n          httpStatus: 500,\n          usage: {\n            input_tokens: 0,\n            output_tokens: 0,\n          },\n        })\n        throw error\n      }\n    },\n  }\n\n  return middleware\n}\n\nexport const wrapVercelLanguageModel = (\n  model: LanguageModelV1,\n  phClient: PostHog,\n  options: CreateInstrumentationMiddlewareOptions\n): LanguageModelV1 => {\n  const traceId = options.posthogTraceId ?? uuidv4()\n  const middleware = createInstrumentationMiddleware(phClient, model, {\n    ...options,\n    posthogTraceId: traceId,\n    posthogDistinctId: options.posthogDistinctId ?? traceId,\n  })\n\n  const wrappedModel = wrapLanguageModel({\n    model,\n    middleware,\n  })\n\n  return wrappedModel\n}\n"],"names":["getModelParams","params","modelParams","paramKeys","key","undefined","mergeSystemPrompt","provider","messages","system","systemMessage","role","content","withPrivacyMode","client","privacyMode","input","privacy_mode","sendEventToPosthog","distinctId","traceId","model","output","latency","baseURL","httpStatus","usage","capture","event","properties","$ai_provider","$ai_model","$ai_model_parameters","$ai_input","posthog_privacy_mode","$ai_output_choices","$ai_http_status","$ai_input_tokens","input_tokens","$ai_output_tokens","output_tokens","$ai_latency","$ai_trace_id","$ai_base_url","posthog_properties","$process_person_profile","groups","posthog_groups","PostHogOpenAI","OpenAIOrignal","constructor","config","posthog","openAIConfig","phClient","chat","WrappedChat","Chat","parentClient","completions","WrappedCompletions","Completions","create","body","options","posthog_distinct_id","posthog_trace_id","openAIParams","uuidv4","startTime","Date","now","parentPromise","stream","then","value","passThroughStream","PassThrough","objectMode","accumulatedContent","openAIStream","chunk","delta","choices","prompt_tokens","completion_tokens","write","end","error","emit","wrappedPromise","result","message","createInstrumentationMiddleware","middleware","wrapGenerate","doGenerate","posthogDistinctId","posthogTraceId","modelId","posthogPrivacyMode","prompt","text","promptTokens","completionTokens","wrapStream","doStream","generatedText","rest","transformStream","TransformStream","transform","controller","type","textDelta","enqueue","flush","pipeThrough","wrapVercelLanguageModel","wrappedModel","wrapLanguageModel"],"mappings":";;;;;;;;;;;;;AAaO,MAAMA,cAAc,GAAIC,MAAyD,IAAyB;EAC/G,MAAMC,WAAW,GAAwB,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,MAAc,CAACG,GAAG,CAAC,KAAKC,SAAS,EAAE;AACvDH,MAAAA,WAAW,CAACE,GAAG,CAAC,GAAIH,MAAc,CAACG,GAAG,CAAC,CAAA;AACxC,KAAA;AACF,GAAA;AACD,EAAA,OAAOF,WAAW,CAAA;AACpB,CAAC,CAAA;AAgEM,MAAMI,iBAAiB,GAAGA,CAACL,MAAyD,EAAEM,QAAgB,KAAS;EACpH,IAAIA,QAAQ,KAAK,WAAW,EAAE;IAC5B,OAAON,MAAM,CAACO,QAAQ,CAAA;AACvB,GAAA;AACD,EAAA,MAAMA,QAAQ,GAAGP,MAAM,CAACO,QAAQ,IAAI,EAAE,CAAA;AACtC,EAAA,IAAI,CAAEP,MAAc,CAACQ,MAAM,EAAE;AAC3B,IAAA,OAAOD,QAAQ,CAAA;AAChB,GAAA;AACD,EAAA,MAAME,aAAa,GAAIT,MAAc,CAACQ,MAAM,CAAA;AAC5C,EAAA,OAAO,CAAC;AAAEE,IAAAA,IAAI,EAAE,QAAQ;AAAEC,IAAAA,OAAO,EAAEF,aAAAA;GAAe,EAAE,GAAGF,QAAQ,CAAC,CAAA;AAClE,CAAC,CAAA;AAEM,MAAMK,eAAe,GAAGA,CAACC,MAAe,EAAEC,WAAoB,EAAEC,KAAU,KAAS;EACxF,OAAQF,MAAc,CAACG,YAAY,IAAIF,WAAW,GAAG,IAAI,GAAGC,KAAK,CAAA;AACnE,CAAC,CAAA;AAiBM,MAAME,kBAAkB,GAAGA,CAAC;EACjCJ,MAAM;EACNK,UAAU;EACVC,OAAO;EACPC,KAAK;EACLd,QAAQ;EACRS,KAAK;EACLM,MAAM;EACNC,OAAO;EACPC,OAAO;EACPvB,MAAM;AACNwB,EAAAA,UAAU,GAAG,GAAG;AAChBC,EAAAA,KAAK,GAAG,EAAE;AAAA,CACe,KAAU;EACnC,IAAIZ,MAAM,CAACa,OAAO,EAAE;IAClBb,MAAM,CAACa,OAAO,CAAC;MACbR,UAAU,EAAEA,UAAU,IAAIC,OAAO;AACjCQ,MAAAA,KAAK,EAAE,gBAAgB;AACvBC,MAAAA,UAAU,EAAE;AACVC,QAAAA,YAAY,EAAEvB,QAAQ;AACtBwB,QAAAA,SAAS,EAAEV,KAAK;AAChBW,QAAAA,oBAAoB,EAAEhC,cAAc,CAACC,MAAM,CAAC;AAC5CgC,QAAAA,SAAS,EAAEpB,eAAe,CAACC,MAAM,EAAEb,MAAM,CAACiC,oBAAoB,IAAI,KAAK,EAAElB,KAAK,CAAC;AAC/EmB,QAAAA,kBAAkB,EAAEtB,eAAe,CAACC,MAAM,EAAEb,MAAM,CAACiC,oBAAoB,IAAI,KAAK,EAAEZ,MAAM,CAAC;AACzFc,QAAAA,eAAe,EAAEX,UAAU;AAC3BY,QAAAA,gBAAgB,EAAEX,KAAK,CAACY,YAAY,IAAI,CAAC;AACzCC,QAAAA,iBAAiB,EAAEb,KAAK,CAACc,aAAa,IAAI,CAAC;AAC3CC,QAAAA,WAAW,EAAElB,OAAO;AACpBmB,QAAAA,YAAY,EAAEtB,OAAO;AACrBuB,QAAAA,YAAY,EAAEnB,OAAO;QACrB,GAAGvB,MAAM,CAAC2C,kBAAkB;AAC5B,QAAA,IAAIzB,UAAU,GAAG,EAAE,GAAG;AAAE0B,UAAAA,uBAAuB,EAAE,KAAA;SAAO,CAAA;OACzD;MACDC,MAAM,EAAE7C,MAAM,CAAC8C,cAAAA;AAChB,KAAA,CAAC,CAAA;AACH,GAAA;AACH,CAAC;;ACjJK,MAAOC,aAAc,SAAQC,iCAAa,CAAA;EAG9CC,WAAAA,CAAYC,MAA8B,EAAA;IACxC,MAAM;MAAEC,OAAO;MAAE,GAAGC,YAAAA;AAAc,KAAA,GAAGF,MAAM,CAAA;IAC3C,KAAK,CAACE,YAAY,CAAC,CAAA;IACnB,IAAI,CAACC,QAAQ,GAAGF,OAAO,CAAA;IACvB,IAAI,CAACG,IAAI,GAAG,IAAIC,WAAW,CAAC,IAAI,EAAE,IAAI,CAACF,QAAQ,CAAC,CAAA;AAClD,GAAA;AAGD,CAAA;AAEY,MAAAE,WAAY,SAAQP,iCAAa,CAACQ,IAAI,CAAA;AACjDP,EAAAA,WAAYA,CAAAQ,YAA2B,EAAEJ,QAAiB,EAAA;IACxD,KAAK,CAACI,YAAY,CAAC,CAAA;IACnB,IAAI,CAACC,WAAW,GAAG,IAAIC,kBAAkB,CAACF,YAAY,EAAEJ,QAAQ,CAAC,CAAA;AACnE,GAAA;AAGD,CAAA;MAEYM,kBAAmB,SAAQX,iCAAa,CAACQ,IAAI,CAACI,WAAW,CAAA;AAGpEX,EAAAA,WAAYA,CAAApC,MAAqB,EAAEwC,QAAiB,EAAA;IAClD,KAAK,CAACxC,MAAM,CAAC,CAAA;IACb,IAAI,CAACwC,QAAQ,GAAGA,QAAQ,CAAA;AAC1B,GAAA;AAoBA;AACOQ,EAAAA,MAAMA,CACXC,IAAuD,EACvDC,OAAwB,EAAA;IAExB,MAAM;MACJC,mBAAmB;MACnBC,gBAAgB;MAChBtB,kBAAkB;AAClBV,MAAAA,oBAAoB,GAAG,KAAK;MAC5Ba,cAAc;MACd,GAAGoB,YAAAA;AACJ,KAAA,GAAGJ,IAAI,CAAA;AAER,IAAA,MAAM3C,OAAO,GAAG8C,gBAAgB,IAAIE,OAAM,EAAE,CAAA;AAC5C,IAAA,MAAMC,SAAS,GAAGC,IAAI,CAACC,GAAG,EAAE,CAAA;IAE5B,MAAMC,aAAa,GAAG,KAAK,CAACV,MAAM,CAACK,YAAY,EAAEH,OAAO,CAAC,CAAA;IAEzD,IAAIG,YAAY,CAACM,MAAM,EAAE;AACvB,MAAA,OAAOD,aAAa,CAACE,IAAI,CAAEC,KAAK,IAAI;AAClC,QAAA,MAAMC,iBAAiB,GAAG,IAAIC,kBAAW,CAAC;AAAEC,UAAAA,UAAU,EAAE,IAAA;AAAM,SAAA,CAAC,CAAA;QAC/D,IAAIC,kBAAkB,GAAG,EAAE,CAAA;AAC3B,QAAA,IAAIrD,KAAK,GAAoD;AAC3DY,UAAAA,YAAY,EAAE,CAAC;AACfE,UAAAA,aAAa,EAAE,CAAA;SAChB,CAAA;QACD,IAAI,KAAK,IAAImC,KAAK,EAAE;UAClB,MAAMK,YAAY,GAAGL,KAAK,CAAA;AACzB,UAAA,CAAC,YAAW;YACX,IAAI;AACF,cAAA,WAAW,MAAMM,KAAK,IAAID,YAAY,EAAE;AACtC,gBAAA,MAAME,KAAK,GAAGD,KAAK,EAAEE,OAAO,GAAG,CAAC,CAAC,EAAED,KAAK,EAAEtE,OAAO,IAAI,EAAE,CAAA;AACvDmE,gBAAAA,kBAAkB,IAAIG,KAAK,CAAA;gBAC3B,IAAID,KAAK,CAACvD,KAAK,EAAE;AACfA,kBAAAA,KAAK,GAAG;AACNY,oBAAAA,YAAY,EAAE2C,KAAK,CAACvD,KAAK,CAAC0D,aAAa,IAAI,CAAC;AAC5C5C,oBAAAA,aAAa,EAAEyC,KAAK,CAACvD,KAAK,CAAC2D,iBAAiB,IAAI,CAAA;mBACjD,CAAA;AACF,iBAAA;AACDT,gBAAAA,iBAAiB,CAACU,KAAK,CAACL,KAAK,CAAC,CAAA;AAC/B,eAAA;cACD,MAAM1D,OAAO,GAAG,CAAC+C,IAAI,CAACC,GAAG,EAAE,GAAGF,SAAS,IAAI,IAAI,CAAA;AAC/CnD,cAAAA,kBAAkB,CAAC;gBACjBJ,MAAM,EAAE,IAAI,CAACwC,QAAQ;gBACrBnC,UAAU,EAAE8C,mBAAmB,IAAI7C,OAAO;gBAC1CA,OAAO;gBACPC,KAAK,EAAE8C,YAAY,CAAC9C,KAAK;AACzBd,gBAAAA,QAAQ,EAAE,QAAQ;gBAClBS,KAAK,EAAEkB,oBAAoB,GAAG,EAAE,GAAG5B,iBAAiB,CAAC6D,YAAY,EAAE,QAAQ,CAAC;AAC5E7C,gBAAAA,MAAM,EAAE,CAAC;AAAEV,kBAAAA,OAAO,EAAEmE,kBAAkB;AAAEpE,kBAAAA,IAAI,EAAE,WAAA;AAAW,iBAAE,CAAC;gBAC5DY,OAAO;AACPC,gBAAAA,OAAO,EAAG,IAAY,CAACA,OAAO,IAAI,EAAE;AACpCvB,gBAAAA,MAAM,EAAE8D,IAAI;AACZtC,gBAAAA,UAAU,EAAE,GAAG;AACfC,gBAAAA,KAAAA;AACD,eAAA,CAAC,CAAA;cACFkD,iBAAiB,CAACW,GAAG,EAAE,CAAA;aACxB,CAAC,OAAOC,KAAK,EAAE;AACd;AACAtE,cAAAA,kBAAkB,CAAC;gBACjBJ,MAAM,EAAE,IAAI,CAACwC,QAAQ;gBACrBnC,UAAU,EAAE8C,mBAAmB,IAAI7C,OAAO;gBAC1CA,OAAO;gBACPC,KAAK,EAAE8C,YAAY,CAAC9C,KAAK;AACzBd,gBAAAA,QAAQ,EAAE,QAAQ;gBAClBS,KAAK,EAAEkB,oBAAoB,GAAG,EAAE,GAAG5B,iBAAiB,CAAC6D,YAAY,EAAE,QAAQ,CAAC;AAC5E7C,gBAAAA,MAAM,EAAE,EAAE;AACVC,gBAAAA,OAAO,EAAE,CAAC;AACVC,gBAAAA,OAAO,EAAG,IAAY,CAACA,OAAO,IAAI,EAAE;AACpCvB,gBAAAA,MAAM,EAAE8D,IAAI;AACZtC,gBAAAA,UAAU,EAAE,GAAG;AACfC,gBAAAA,KAAK,EAAE;AACLY,kBAAAA,YAAY,EAAE,CAAC;AACfE,kBAAAA,aAAa,EAAE,CAAA;AAChB,iBAAA;AACF,eAAA,CAAC,CAAA;AACFoC,cAAAA,iBAAiB,CAACa,IAAI,CAAC,OAAO,EAAED,KAAK,CAAC,CAAA;AACvC,aAAA;AACH,WAAC,GAAG,CAAA;AACL,SAAA;AACD,QAAA,OAAOZ,iBAA2D,CAAA;AACpE,OAAC,CAA4C,CAAA;AAC9C,KAAA,MAAM;AACL,MAAA,MAAMc,cAAc,GAAGlB,aAAa,CAACE,IAAI,CACtCiB,MAAM,IAAI;QACT,IAAI,SAAS,IAAIA,MAAM,EAAE;UACvB,MAAMpE,OAAO,GAAG,CAAC+C,IAAI,CAACC,GAAG,EAAE,GAAGF,SAAS,IAAI,IAAI,CAAA;AAC/CnD,UAAAA,kBAAkB,CAAC;YACjBJ,MAAM,EAAE,IAAI,CAACwC,QAAQ;YACrBnC,UAAU,EAAE8C,mBAAmB,IAAI7C,OAAO;YAC1CA,OAAO;YACPC,KAAK,EAAE8C,YAAY,CAAC9C,KAAK;AACzBd,YAAAA,QAAQ,EAAE,QAAQ;YAClBS,KAAK,EAAEkB,oBAAoB,GAAG,EAAE,GAAG5B,iBAAiB,CAAC6D,YAAY,EAAE,QAAQ,CAAC;AAC5E7C,YAAAA,MAAM,EAAE,CAAC;cAAEV,OAAO,EAAE+E,MAAM,CAACR,OAAO,CAAC,CAAC,CAAC,CAACS,OAAO,CAAChF,OAAO;AAAED,cAAAA,IAAI,EAAE,WAAA;aAAa,CAAC;YAC3EY,OAAO;AACPC,YAAAA,OAAO,EAAG,IAAY,CAACA,OAAO,IAAI,EAAE;AACpCvB,YAAAA,MAAM,EAAE8D,IAAI;AACZtC,YAAAA,UAAU,EAAE,GAAG;AACfC,YAAAA,KAAK,EAAE;AACLY,cAAAA,YAAY,EAAEqD,MAAM,CAACjE,KAAK,EAAE0D,aAAa,IAAI,CAAC;AAC9C5C,cAAAA,aAAa,EAAEmD,MAAM,CAACjE,KAAK,EAAE2D,iBAAiB,IAAI,CAAA;AACnD,aAAA;AACF,WAAA,CAAC,CAAA;AACH,SAAA;AACD,QAAA,OAAOM,MAAM,CAAA;OACd,EACAH,KAAK,IAAI;AACRtE,QAAAA,kBAAkB,CAAC;UACjBJ,MAAM,EAAE,IAAI,CAACwC,QAAQ;UACrBnC,UAAU,EAAE8C,mBAAmB,IAAI7C,OAAO;UAC1CA,OAAO;UACPC,KAAK,EAAE8C,YAAY,CAAC9C,KAAK;AACzBd,UAAAA,QAAQ,EAAE,QAAQ;UAClBS,KAAK,EAAEkB,oBAAoB,GAAG,EAAE,GAAG5B,iBAAiB,CAAC6D,YAAY,EAAE,QAAQ,CAAC;AAC5E7C,UAAAA,MAAM,EAAE,EAAE;AACVC,UAAAA,OAAO,EAAE,CAAC;AACVC,UAAAA,OAAO,EAAG,IAAY,CAACA,OAAO,IAAI,EAAE;AACpCvB,UAAAA,MAAM,EAAE8D,IAAI;AACZtC,UAAAA,UAAU,EAAE,GAAG;AACfC,UAAAA,KAAK,EAAE;AACLY,YAAAA,YAAY,EAAE,CAAC;AACfE,YAAAA,aAAa,EAAE,CAAA;AAChB,WAAA;AACF,SAAA,CAAC,CAAA;AACF,QAAA,MAAMgD,KAAK,CAAA;AACb,OAAC,CAC4B,CAAA;AAE/B,MAAA,OAAOE,cAAc,CAAA;AACtB,KAAA;AACH,GAAA;AACD;;ACvLM,MAAMG,+BAA+B,GAAGA,CAC7CvC,QAAiB,EACjBjC,KAAsB,EACtB2C,OAA+C,KAClB;AAC7B,EAAA,MAAM8B,UAAU,GAA8B;IAC5CC,YAAY,EAAE,OAAO;MAAEC,UAAU;AAAE/F,MAAAA,MAAAA;AAAQ,KAAA,KAAI;AAC7C,MAAA,MAAMoE,SAAS,GAAGC,IAAI,CAACC,GAAG,EAAE,CAAA;MAE5B,IAAI;AACF,QAAA,MAAMoB,MAAM,GAAG,MAAMK,UAAU,EAAE,CAAA;QACjC,MAAMzE,OAAO,GAAG,CAAC+C,IAAI,CAACC,GAAG,EAAE,GAAGF,SAAS,IAAI,IAAI,CAAA;AAE/CnD,QAAAA,kBAAkB,CAAC;AACjBJ,UAAAA,MAAM,EAAEwC,QAAQ;UAChBnC,UAAU,EAAE6C,OAAO,CAACiC,iBAAiB;UACrC7E,OAAO,EAAE4C,OAAO,CAACkC,cAAc;UAC/B7E,KAAK,EAAEA,KAAK,CAAC8E,OAAO;AACpB5F,UAAAA,QAAQ,EAAE,QAAQ;UAClBS,KAAK,EAAEgD,OAAO,CAACoC,kBAAkB,GAAG,EAAE,GAAGnG,MAAM,CAACoG,MAAM;AACtD/E,UAAAA,MAAM,EAAE,CAAC;YAAEV,OAAO,EAAE+E,MAAM,CAACW,IAAI;AAAE3F,YAAAA,IAAI,EAAE,WAAA;WAAa,CAAC;UACrDY,OAAO;AACPC,UAAAA,OAAO,EAAE,EAAE;AACXvB,UAAAA,MAAM,EAAE;AAAE2C,YAAAA,kBAAkB,EAAEoB,OAAAA;WAAgB;AAC9CvC,UAAAA,UAAU,EAAE,GAAG;AACfC,UAAAA,KAAK,EAAE;AACLY,YAAAA,YAAY,EAAEqD,MAAM,CAACjE,KAAK,CAAC6E,YAAY;AACvC/D,YAAAA,aAAa,EAAEmD,MAAM,CAACjE,KAAK,CAAC8E,gBAAAA;AAC7B,WAAA;AACF,SAAA,CAAC,CAAA;AAEF,QAAA,OAAOb,MAAM,CAAA;OACd,CAAC,OAAOH,KAAK,EAAE;AACdtE,QAAAA,kBAAkB,CAAC;AACjBJ,UAAAA,MAAM,EAAEwC,QAAQ;UAChBnC,UAAU,EAAE6C,OAAO,CAACiC,iBAAiB;UACrC7E,OAAO,EAAE4C,OAAO,CAACkC,cAAc;UAC/B7E,KAAK,EAAEA,KAAK,CAAC8E,OAAO;AACpB5F,UAAAA,QAAQ,EAAE,QAAQ;UAClBS,KAAK,EAAEgD,OAAO,CAACoC,kBAAkB,GAAG,EAAE,GAAGnG,MAAM,CAACoG,MAAM;AACtD/E,UAAAA,MAAM,EAAE,EAAE;AACVC,UAAAA,OAAO,EAAE,CAAC;AACVC,UAAAA,OAAO,EAAE,EAAE;AACXvB,UAAAA,MAAM,EAAE;AAAE2C,YAAAA,kBAAkB,EAAEoB,OAAAA;WAAgB;AAC9CvC,UAAAA,UAAU,EAAE,GAAG;AACfC,UAAAA,KAAK,EAAE;AACLY,YAAAA,YAAY,EAAE,CAAC;AACfE,YAAAA,aAAa,EAAE,CAAA;AAChB,WAAA;AACF,SAAA,CAAC,CAAA;AACF,QAAA,MAAMgD,KAAK,CAAA;AACZ,OAAA;KACF;IAEDiB,UAAU,EAAE,OAAO;MAAEC,QAAQ;AAAEzG,MAAAA,MAAAA;AAAQ,KAAA,KAAI;AACzC,MAAA,MAAMoE,SAAS,GAAGC,IAAI,CAACC,GAAG,EAAE,CAAA;MAC5B,IAAIoC,aAAa,GAAG,EAAE,CAAA;MACtB,IAAIjF,KAAK,GAAsD,EAAE,CAAA;MAEjE,IAAI;QACF,MAAM;UAAE+C,MAAM;UAAE,GAAGmC,IAAAA;SAAM,GAAG,MAAMF,QAAQ,EAAE,CAAA;AAE5C,QAAA,MAAMG,eAAe,GAAG,IAAIC,eAAe,CAAuD;AAChGC,UAAAA,SAASA,CAAC9B,KAAK,EAAE+B,UAAU,EAAA;AACzB,YAAA,IAAI/B,KAAK,CAACgC,IAAI,KAAK,YAAY,EAAE;cAC/BN,aAAa,IAAI1B,KAAK,CAACiC,SAAS,CAAA;AACjC,aAAA;AACD,YAAA,IAAIjC,KAAK,CAACgC,IAAI,KAAK,QAAQ,EAAE;AAC3BvF,cAAAA,KAAK,GAAG;AACNY,gBAAAA,YAAY,EAAE2C,KAAK,CAACvD,KAAK,EAAE6E,YAAY;AACvC/D,gBAAAA,aAAa,EAAEyC,KAAK,CAACvD,KAAK,EAAE8E,gBAAAA;eAC7B,CAAA;AACF,aAAA;AACDQ,YAAAA,UAAU,CAACG,OAAO,CAAClC,KAAK,CAAC,CAAA;WAC1B;AAEDmC,UAAAA,KAAKA,GAAA;YACH,MAAM7F,OAAO,GAAG,CAAC+C,IAAI,CAACC,GAAG,EAAE,GAAGF,SAAS,IAAI,IAAI,CAAA;AAC/CnD,YAAAA,kBAAkB,CAAC;AACjBJ,cAAAA,MAAM,EAAEwC,QAAQ;cAChBnC,UAAU,EAAE6C,OAAO,CAACiC,iBAAiB;cACrC7E,OAAO,EAAE4C,OAAO,CAACkC,cAAc;cAC/B7E,KAAK,EAAEA,KAAK,CAAC8E,OAAO;AACpB5F,cAAAA,QAAQ,EAAE,QAAQ;cAClBS,KAAK,EAAEgD,OAAO,CAACoC,kBAAkB,GAAG,EAAE,GAAGnG,MAAM,CAACoG,MAAM;AACtD/E,cAAAA,MAAM,EAAE,CAAC;AAAEV,gBAAAA,OAAO,EAAE+F,aAAa;AAAEhG,gBAAAA,IAAI,EAAE,WAAA;AAAW,eAAE,CAAC;cACvDY,OAAO;AACPC,cAAAA,OAAO,EAAE,EAAE;AACXvB,cAAAA,MAAM,EAAE;AAAE2C,gBAAAA,kBAAkB,EAAEoB,OAAAA;eAAgB;AAC9CvC,cAAAA,UAAU,EAAE,GAAG;AACfC,cAAAA,KAAAA;AACD,aAAA,CAAC,CAAA;AACJ,WAAA;AACD,SAAA,CAAC,CAAA;QAEF,OAAO;AACL+C,UAAAA,MAAM,EAAEA,MAAM,CAAC4C,WAAW,CAACR,eAAe,CAAC;UAC3C,GAAGD,IAAAA;SACJ,CAAA;OACF,CAAC,OAAOpB,KAAK,EAAE;AACdtE,QAAAA,kBAAkB,CAAC;AACjBJ,UAAAA,MAAM,EAAEwC,QAAQ;UAChBnC,UAAU,EAAE6C,OAAO,CAACiC,iBAAiB;UACrC7E,OAAO,EAAE4C,OAAO,CAACkC,cAAc;UAC/B7E,KAAK,EAAEA,KAAK,CAAC8E,OAAO;AACpB5F,UAAAA,QAAQ,EAAE,QAAQ;UAClBS,KAAK,EAAEgD,OAAO,CAACoC,kBAAkB,GAAG,EAAE,GAAGnG,MAAM,CAACoG,MAAM;AACtD/E,UAAAA,MAAM,EAAE,EAAE;AACVC,UAAAA,OAAO,EAAE,CAAC;AACVC,UAAAA,OAAO,EAAE,EAAE;AACXvB,UAAAA,MAAM,EAAE;AAAE2C,YAAAA,kBAAkB,EAAEoB,OAAAA;WAAgB;AAC9CvC,UAAAA,UAAU,EAAE,GAAG;AACfC,UAAAA,KAAK,EAAE;AACLY,YAAAA,YAAY,EAAE,CAAC;AACfE,YAAAA,aAAa,EAAE,CAAA;AAChB,WAAA;AACF,SAAA,CAAC,CAAA;AACF,QAAA,MAAMgD,KAAK,CAAA;AACZ,OAAA;AACH,KAAA;GACD,CAAA;AAED,EAAA,OAAOM,UAAU,CAAA;AACnB,CAAC,CAAA;AAEM,MAAMwB,uBAAuB,GAAGA,CACrCjG,KAAsB,EACtBiC,QAAiB,EACjBU,OAA+C,KAC5B;EACnB,MAAM5C,OAAO,GAAG4C,OAAO,CAACkC,cAAc,IAAI9B,OAAM,EAAE,CAAA;AAClD,EAAA,MAAM0B,UAAU,GAAGD,+BAA+B,CAACvC,QAAQ,EAAEjC,KAAK,EAAE;AAClE,IAAA,GAAG2C,OAAO;AACVkC,IAAAA,cAAc,EAAE9E,OAAO;AACvB6E,IAAAA,iBAAiB,EAAEjC,OAAO,CAACiC,iBAAiB,IAAI7E,OAAAA;AACjD,GAAA,CAAC,CAAA;EAEF,MAAMmG,YAAY,GAAGC,iCAAiB,CAAC;IACrCnG,KAAK;AACLyE,IAAAA,UAAAA;AACD,GAAA,CAAC,CAAA;AAEF,EAAA,OAAOyB,YAAY,CAAA;AACrB;;;;;"}

package/lib/index.d.ts

@@ -0,0 +1,52 @@
+import OpenAIOrignal from 'openai';
+import { PostHog } from 'posthog-node';
+import { RequestOptions, APIPromise } from 'openai/core';
+import { Stream } from 'openai/streaming';
+import { LanguageModelV1 } from 'ai';
+
+interface MonitoringParams {
+    posthog_distinct_id?: string;
+    posthog_trace_id?: string;
+    posthog_properties?: Record<string, any>;
+    posthog_privacy_mode?: boolean;
+    posthog_groups?: Record<string, any>;
+}
+
+type ChatCompletion = OpenAIOrignal.ChatCompletion;
+type ChatCompletionChunk = OpenAIOrignal.ChatCompletionChunk;
+type ChatCompletionCreateParamsBase = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParams;
+type ChatCompletionCreateParamsNonStreaming = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParamsNonStreaming;
+type ChatCompletionCreateParamsStreaming = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParamsStreaming;
+
+interface MonitoringOpenAIConfig {
+    apiKey: string;
+    posthog: PostHog;
+    baseURL?: string;
+}
+declare class PostHogOpenAI extends OpenAIOrignal {
+    private readonly phClient;
+    constructor(config: MonitoringOpenAIConfig);
+    chat: WrappedChat;
+}
+declare class WrappedChat extends OpenAIOrignal.Chat {
+    constructor(parentClient: PostHogOpenAI, phClient: PostHog);
+    completions: WrappedCompletions;
+}
+declare class WrappedCompletions extends OpenAIOrignal.Chat.Completions {
+    private readonly phClient;
+    constructor(client: OpenAIOrignal, phClient: PostHog);
+    create(body: ChatCompletionCreateParamsNonStreaming & MonitoringParams, options?: RequestOptions): APIPromise<ChatCompletion>;
+    create(body: ChatCompletionCreateParamsStreaming & MonitoringParams, options?: RequestOptions): APIPromise<Stream<ChatCompletionChunk>>;
+    create(body: ChatCompletionCreateParamsBase & MonitoringParams, options?: RequestOptions): APIPromise<ChatCompletion | Stream<ChatCompletionChunk>>;
+}
+
+interface CreateInstrumentationMiddlewareOptions {
+    posthogDistinctId?: string;
+    posthogTraceId: string;
+    posthogProperties?: Record<string, any>;
+    posthogPrivacyMode?: boolean;
+    posthogGroups?: string[];
+}
+declare const wrapVercelLanguageModel: (model: LanguageModelV1, phClient: PostHog, options: CreateInstrumentationMiddlewareOptions) => LanguageModelV1;
+
+export { PostHogOpenAI as OpenAI, wrapVercelLanguageModel as withTracing };

package/lib/index.esm.js

@@ -0,0 +1,377 @@
+import OpenAIOrignal from 'openai';
+import { v4 } from 'uuid';
+import { PassThrough } from 'stream';
+import { experimental_wrapLanguageModel } from 'ai';
+
+const getModelParams = params => {
+  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 mergeSystemPrompt = (params, provider) => {
+  if (provider !== 'anthropic') {
+    return params.messages;
+  }
+  const messages = params.messages || [];
+  if (!params.system) {
+    return messages;
+  }
+  const systemMessage = params.system;
+  return [{
+    role: 'system',
+    content: systemMessage
+  }, ...messages];
+};
+const withPrivacyMode = (client, privacyMode, input) => {
+  return client.privacy_mode || privacyMode ? null : input;
+};
+const sendEventToPosthog = ({
+  client,
+  distinctId,
+  traceId,
+  model,
+  provider,
+  input,
+  output,
+  latency,
+  baseURL,
+  params,
+  httpStatus = 200,
+  usage = {}
+}) => {
+  if (client.capture) {
+    client.capture({
+      distinctId: distinctId ?? traceId,
+      event: '$ai_generation',
+      properties: {
+        $ai_provider: provider,
+        $ai_model: model,
+        $ai_model_parameters: getModelParams(params),
+        $ai_input: withPrivacyMode(client, params.posthog_privacy_mode ?? false, input),
+        $ai_output_choices: withPrivacyMode(client, params.posthog_privacy_mode ?? false, output),
+        $ai_http_status: httpStatus,
+        $ai_input_tokens: usage.input_tokens ?? 0,
+        $ai_output_tokens: usage.output_tokens ?? 0,
+        $ai_latency: latency,
+        $ai_trace_id: traceId,
+        $ai_base_url: baseURL,
+        ...params.posthog_properties,
+        ...(distinctId ? {} : {
+          $process_person_profile: false
+        })
+      },
+      groups: params.posthog_groups
+    });
+  }
+};
+
+class PostHogOpenAI extends OpenAIOrignal {
+  constructor(config) {
+    const {
+      posthog,
+      ...openAIConfig
+    } = config;
+    super(openAIConfig);
+    this.phClient = posthog;
+    this.chat = new WrappedChat(this, this.phClient);
+  }
+}
+class WrappedChat extends OpenAIOrignal.Chat {
+  constructor(parentClient, phClient) {
+    super(parentClient);
+    this.completions = new WrappedCompletions(parentClient, phClient);
+  }
+}
+class WrappedCompletions extends OpenAIOrignal.Chat.Completions {
+  constructor(client, phClient) {
+    super(client);
+    this.phClient = phClient;
+  }
+  // --- Implementation Signature
+  create(body, options) {
+    const {
+      posthog_distinct_id,
+      posthog_trace_id,
+      posthog_properties,
+      posthog_privacy_mode = false,
+      posthog_groups,
+      ...openAIParams
+    } = body;
+    const traceId = posthog_trace_id ?? v4();
+    const startTime = Date.now();
+    const parentPromise = super.create(openAIParams, options);
+    if (openAIParams.stream) {
+      return parentPromise.then(value => {
+        const passThroughStream = new PassThrough({
+          objectMode: true
+        });
+        let accumulatedContent = '';
+        let usage = {
+          input_tokens: 0,
+          output_tokens: 0
+        };
+        if ('tee' in value) {
+          const openAIStream = value;
+          (async () => {
+            try {
+              for await (const chunk of openAIStream) {
+                const delta = chunk?.choices?.[0]?.delta?.content ?? '';
+                accumulatedContent += delta;
+                if (chunk.usage) {
+                  usage = {
+                    input_tokens: chunk.usage.prompt_tokens ?? 0,
+                    output_tokens: chunk.usage.completion_tokens ?? 0
+                  };
+                }
+                passThroughStream.write(chunk);
+              }
+              const latency = (Date.now() - startTime) / 1000;
+              sendEventToPosthog({
+                client: this.phClient,
+                distinctId: posthog_distinct_id ?? traceId,
+                traceId,
+                model: openAIParams.model,
+                provider: 'openai',
+                input: posthog_privacy_mode ? '' : mergeSystemPrompt(openAIParams, 'openai'),
+                output: [{
+                  content: accumulatedContent,
+                  role: 'assistant'
+                }],
+                latency,
+                baseURL: this.baseURL ?? '',
+                params: body,
+                httpStatus: 200,
+                usage
+              });
+              passThroughStream.end();
+            } catch (error) {
+              // error handling
+              sendEventToPosthog({
+                client: this.phClient,
+                distinctId: posthog_distinct_id ?? traceId,
+                traceId,
+                model: openAIParams.model,
+                provider: 'openai',
+                input: posthog_privacy_mode ? '' : mergeSystemPrompt(openAIParams, 'openai'),
+                output: [],
+                latency: 0,
+                baseURL: this.baseURL ?? '',
+                params: body,
+                httpStatus: 500,
+                usage: {
+                  input_tokens: 0,
+                  output_tokens: 0
+                }
+              });
+              passThroughStream.emit('error', error);
+            }
+          })();
+        }
+        return passThroughStream;
+      });
+    } else {
+      const wrappedPromise = parentPromise.then(result => {
+        if ('choices' in result) {
+          const latency = (Date.now() - startTime) / 1000;
+          sendEventToPosthog({
+            client: this.phClient,
+            distinctId: posthog_distinct_id ?? traceId,
+            traceId,
+            model: openAIParams.model,
+            provider: 'openai',
+            input: posthog_privacy_mode ? '' : mergeSystemPrompt(openAIParams, 'openai'),
+            output: [{
+              content: result.choices[0].message.content,
+              role: 'assistant'
+            }],
+            latency,
+            baseURL: this.baseURL ?? '',
+            params: body,
+            httpStatus: 200,
+            usage: {
+              input_tokens: result.usage?.prompt_tokens ?? 0,
+              output_tokens: result.usage?.completion_tokens ?? 0
+            }
+          });
+        }
+        return result;
+      }, error => {
+        sendEventToPosthog({
+          client: this.phClient,
+          distinctId: posthog_distinct_id ?? traceId,
+          traceId,
+          model: openAIParams.model,
+          provider: 'openai',
+          input: posthog_privacy_mode ? '' : mergeSystemPrompt(openAIParams, 'openai'),
+          output: [],
+          latency: 0,
+          baseURL: this.baseURL ?? '',
+          params: body,
+          httpStatus: 500,
+          usage: {
+            input_tokens: 0,
+            output_tokens: 0
+          }
+        });
+        throw error;
+      });
+      return wrappedPromise;
+    }
+  }
+}
+
+const createInstrumentationMiddleware = (phClient, model, options) => {
+  const middleware = {
+    wrapGenerate: async ({
+      doGenerate,
+      params
+    }) => {
+      const startTime = Date.now();
+      try {
+        const result = await doGenerate();
+        const latency = (Date.now() - startTime) / 1000;
+        sendEventToPosthog({
+          client: phClient,
+          distinctId: options.posthogDistinctId,
+          traceId: options.posthogTraceId,
+          model: model.modelId,
+          provider: 'vercel',
+          input: options.posthogPrivacyMode ? '' : params.prompt,
+          output: [{
+            content: result.text,
+            role: 'assistant'
+          }],
+          latency,
+          baseURL: '',
+          params: {
+            posthog_properties: options
+          },
+          httpStatus: 200,
+          usage: {
+            input_tokens: result.usage.promptTokens,
+            output_tokens: result.usage.completionTokens
+          }
+        });
+        return result;
+      } catch (error) {
+        sendEventToPosthog({
+          client: phClient,
+          distinctId: options.posthogDistinctId,
+          traceId: options.posthogTraceId,
+          model: model.modelId,
+          provider: 'vercel',
+          input: options.posthogPrivacyMode ? '' : params.prompt,
+          output: [],
+          latency: 0,
+          baseURL: '',
+          params: {
+            posthog_properties: options
+          },
+          httpStatus: 500,
+          usage: {
+            input_tokens: 0,
+            output_tokens: 0
+          }
+        });
+        throw error;
+      }
+    },
+    wrapStream: async ({
+      doStream,
+      params
+    }) => {
+      const startTime = Date.now();
+      let generatedText = '';
+      let usage = {};
+      try {
+        const {
+          stream,
+          ...rest
+        } = await doStream();
+        const transformStream = new TransformStream({
+          transform(chunk, controller) {
+            if (chunk.type === 'text-delta') {
+              generatedText += chunk.textDelta;
+            }
+            if (chunk.type === 'finish') {
+              usage = {
+                input_tokens: chunk.usage?.promptTokens,
+                output_tokens: chunk.usage?.completionTokens
+              };
+            }
+            controller.enqueue(chunk);
+          },
+          flush() {
+            const latency = (Date.now() - startTime) / 1000;
+            sendEventToPosthog({
+              client: phClient,
+              distinctId: options.posthogDistinctId,
+              traceId: options.posthogTraceId,
+              model: model.modelId,
+              provider: 'vercel',
+              input: options.posthogPrivacyMode ? '' : params.prompt,
+              output: [{
+                content: generatedText,
+                role: 'assistant'
+              }],
+              latency,
+              baseURL: '',
+              params: {
+                posthog_properties: options
+              },
+              httpStatus: 200,
+              usage
+            });
+          }
+        });
+        return {
+          stream: stream.pipeThrough(transformStream),
+          ...rest
+        };
+      } catch (error) {
+        sendEventToPosthog({
+          client: phClient,
+          distinctId: options.posthogDistinctId,
+          traceId: options.posthogTraceId,
+          model: model.modelId,
+          provider: 'vercel',
+          input: options.posthogPrivacyMode ? '' : params.prompt,
+          output: [],
+          latency: 0,
+          baseURL: '',
+          params: {
+            posthog_properties: options
+          },
+          httpStatus: 500,
+          usage: {
+            input_tokens: 0,
+            output_tokens: 0
+          }
+        });
+        throw error;
+      }
+    }
+  };
+  return middleware;
+};
+const wrapVercelLanguageModel = (model, phClient, options) => {
+  const traceId = options.posthogTraceId ?? v4();
+  const middleware = createInstrumentationMiddleware(phClient, model, {
+    ...options,
+    posthogTraceId: traceId,
+    posthogDistinctId: options.posthogDistinctId ?? traceId
+  });
+  const wrappedModel = experimental_wrapLanguageModel({
+    model,
+    middleware
+  });
+  return wrappedModel;
+};
+
+export { PostHogOpenAI as OpenAI, wrapVercelLanguageModel as withTracing };
+//# sourceMappingURL=index.esm.js.map

package/lib/index.esm.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.esm.js","sources":["../src/utils.ts","../src/openai/index.ts","../src/vercel/middleware.ts"],"sourcesContent":["import { PostHog } from 'posthog-node'\nimport OpenAIOrignal from 'openai'\n\ntype ChatCompletionCreateParamsBase = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParams\n\nexport interface MonitoringParams {\n  posthog_distinct_id?: string\n  posthog_trace_id?: string\n  posthog_properties?: Record<string, any>\n  posthog_privacy_mode?: boolean\n  posthog_groups?: Record<string, any>\n}\n\nexport const getModelParams = (params: ChatCompletionCreateParamsBase & MonitoringParams): Record<string, any> => {\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\nexport const getUsage = (response: any, provider: string): { input_tokens: number; output_tokens: number } => {\n  if (!response?.usage) {\n    return { input_tokens: 0, output_tokens: 0 }\n  }\n\n  if (provider === 'anthropic') {\n    return {\n      input_tokens: response.usage.input_tokens ?? 0,\n      output_tokens: response.usage.output_tokens ?? 0,\n    }\n  } else if (provider === 'openai') {\n    return {\n      input_tokens: response.usage.prompt_tokens ?? 0,\n      output_tokens: response.usage.completion_tokens ?? 0,\n    }\n  }\n\n  return { input_tokens: 0, output_tokens: 0 }\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  }\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 mergeSystemPrompt = (params: ChatCompletionCreateParamsBase & MonitoringParams, provider: string): any => {\n  if (provider !== 'anthropic') {\n    return params.messages\n  }\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\nexport const withPrivacyMode = (client: PostHog, privacyMode: boolean, input: any): any => {\n  return (client as any).privacy_mode || privacyMode ? null : input\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?: { input_tokens?: number; output_tokens?: number }\n  params: ChatCompletionCreateParamsBase & MonitoringParams\n}\n\nexport const sendEventToPosthog = ({\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}: SendEventToPosthogParams): void => {\n  if (client.capture) {\n    client.capture({\n      distinctId: distinctId ?? traceId,\n      event: '$ai_generation',\n      properties: {\n        $ai_provider: provider,\n        $ai_model: model,\n        $ai_model_parameters: getModelParams(params),\n        $ai_input: withPrivacyMode(client, params.posthog_privacy_mode ?? false, input),\n        $ai_output_choices: withPrivacyMode(client, params.posthog_privacy_mode ?? false, output),\n        $ai_http_status: httpStatus,\n        $ai_input_tokens: usage.input_tokens ?? 0,\n        $ai_output_tokens: usage.output_tokens ?? 0,\n        $ai_latency: latency,\n        $ai_trace_id: traceId,\n        $ai_base_url: baseURL,\n        ...params.posthog_properties,\n        ...(distinctId ? {} : { $process_person_profile: false }),\n      },\n      groups: params.posthog_groups,\n    })\n  }\n}\n","import OpenAIOrignal from 'openai'\nimport { PostHog } from 'posthog-node'\nimport { v4 as uuidv4 } from 'uuid'\nimport { PassThrough } from 'stream'\nimport { mergeSystemPrompt, MonitoringParams, sendEventToPosthog } from '../utils'\n\ntype ChatCompletion = OpenAIOrignal.ChatCompletion\ntype ChatCompletionChunk = OpenAIOrignal.ChatCompletionChunk\ntype ChatCompletionCreateParamsBase = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParams\ntype ChatCompletionCreateParamsNonStreaming = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParamsNonStreaming\ntype ChatCompletionCreateParamsStreaming = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParamsStreaming\nimport type { APIPromise, RequestOptions } from 'openai/core'\nimport type { Stream } from 'openai/streaming'\n\ninterface MonitoringOpenAIConfig {\n  apiKey: string\n  posthog: PostHog\n  baseURL?: string\n}\n\nexport class PostHogOpenAI extends OpenAIOrignal {\n  private readonly phClient: PostHog\n\n  constructor(config: MonitoringOpenAIConfig) {\n    const { posthog, ...openAIConfig } = config\n    super(openAIConfig)\n    this.phClient = posthog\n    this.chat = new WrappedChat(this, this.phClient)\n  }\n\n  public chat: WrappedChat\n}\n\nexport class WrappedChat extends OpenAIOrignal.Chat {\n  constructor(parentClient: PostHogOpenAI, phClient: PostHog) {\n    super(parentClient)\n    this.completions = new WrappedCompletions(parentClient, phClient)\n  }\n\n  public completions: WrappedCompletions\n}\n\nexport class WrappedCompletions extends OpenAIOrignal.Chat.Completions {\n  private readonly phClient: PostHog\n\n  constructor(client: OpenAIOrignal, phClient: PostHog) {\n    super(client)\n    this.phClient = phClient\n  }\n\n  // --- Overload #1: Non-streaming\n  public create(\n    body: ChatCompletionCreateParamsNonStreaming & MonitoringParams,\n    options?: RequestOptions\n  ): APIPromise<ChatCompletion>\n\n  // --- Overload #2: Streaming\n  public create(\n    body: ChatCompletionCreateParamsStreaming & MonitoringParams,\n    options?: RequestOptions\n  ): APIPromise<Stream<ChatCompletionChunk>>\n\n  // --- Overload #3: Generic base\n  public create(\n    body: ChatCompletionCreateParamsBase & MonitoringParams,\n    options?: RequestOptions\n  ): APIPromise<ChatCompletion | Stream<ChatCompletionChunk>>\n\n  // --- Implementation Signature\n  public create(\n    body: ChatCompletionCreateParamsBase & MonitoringParams,\n    options?: RequestOptions\n  ): APIPromise<ChatCompletion | Stream<ChatCompletionChunk>> {\n    const {\n      posthog_distinct_id,\n      posthog_trace_id,\n      posthog_properties,\n      posthog_privacy_mode = false,\n      posthog_groups,\n      ...openAIParams\n    } = body\n\n    const traceId = posthog_trace_id ?? uuidv4()\n    const startTime = Date.now()\n\n    const parentPromise = super.create(openAIParams, options)\n\n    if (openAIParams.stream) {\n      return parentPromise.then((value) => {\n        const passThroughStream = new PassThrough({ objectMode: true })\n        let accumulatedContent = ''\n        let usage: { input_tokens: number; output_tokens: number } = {\n          input_tokens: 0,\n          output_tokens: 0,\n        }\n        if ('tee' in value) {\n          const openAIStream = value\n          ;(async () => {\n            try {\n              for await (const chunk of openAIStream) {\n                const delta = chunk?.choices?.[0]?.delta?.content ?? ''\n                accumulatedContent += delta\n                if (chunk.usage) {\n                  usage = {\n                    input_tokens: chunk.usage.prompt_tokens ?? 0,\n                    output_tokens: chunk.usage.completion_tokens ?? 0,\n                  }\n                }\n                passThroughStream.write(chunk)\n              }\n              const latency = (Date.now() - startTime) / 1000\n              sendEventToPosthog({\n                client: this.phClient,\n                distinctId: posthog_distinct_id ?? traceId,\n                traceId,\n                model: openAIParams.model,\n                provider: 'openai',\n                input: posthog_privacy_mode ? '' : mergeSystemPrompt(openAIParams, 'openai'),\n                output: [{ content: accumulatedContent, role: 'assistant' }],\n                latency,\n                baseURL: (this as any).baseURL ?? '',\n                params: body,\n                httpStatus: 200,\n                usage,\n              })\n              passThroughStream.end()\n            } catch (error) {\n              // error handling\n              sendEventToPosthog({\n                client: this.phClient,\n                distinctId: posthog_distinct_id ?? traceId,\n                traceId,\n                model: openAIParams.model,\n                provider: 'openai',\n                input: posthog_privacy_mode ? '' : mergeSystemPrompt(openAIParams, 'openai'),\n                output: [],\n                latency: 0,\n                baseURL: (this as any).baseURL ?? '',\n                params: body,\n                httpStatus: 500,\n                usage: {\n                  input_tokens: 0,\n                  output_tokens: 0,\n                },\n              })\n              passThroughStream.emit('error', error)\n            }\n          })()\n        }\n        return passThroughStream as unknown as Stream<ChatCompletionChunk>\n      }) as APIPromise<Stream<ChatCompletionChunk>>\n    } else {\n      const wrappedPromise = parentPromise.then(\n        (result) => {\n          if ('choices' in result) {\n            const latency = (Date.now() - startTime) / 1000\n            sendEventToPosthog({\n              client: this.phClient,\n              distinctId: posthog_distinct_id ?? traceId,\n              traceId,\n              model: openAIParams.model,\n              provider: 'openai',\n              input: posthog_privacy_mode ? '' : mergeSystemPrompt(openAIParams, 'openai'),\n              output: [{ content: result.choices[0].message.content, role: 'assistant' }],\n              latency,\n              baseURL: (this as any).baseURL ?? '',\n              params: body,\n              httpStatus: 200,\n              usage: {\n                input_tokens: result.usage?.prompt_tokens ?? 0,\n                output_tokens: result.usage?.completion_tokens ?? 0,\n              },\n            })\n          }\n          return result\n        },\n        (error) => {\n          sendEventToPosthog({\n            client: this.phClient,\n            distinctId: posthog_distinct_id ?? traceId,\n            traceId,\n            model: openAIParams.model,\n            provider: 'openai',\n            input: posthog_privacy_mode ? '' : mergeSystemPrompt(openAIParams, 'openai'),\n            output: [],\n            latency: 0,\n            baseURL: (this as any).baseURL ?? '',\n            params: body,\n            httpStatus: 500,\n            usage: {\n              input_tokens: 0,\n              output_tokens: 0,\n            },\n          })\n          throw error\n        }\n      ) as APIPromise<ChatCompletion>\n\n      return wrappedPromise\n    }\n  }\n}\n\nexport default PostHogOpenAI\n","import { experimental_wrapLanguageModel as wrapLanguageModel } from 'ai'\nimport type {\n  LanguageModelV1,\n  Experimental_LanguageModelV1Middleware as LanguageModelV1Middleware,\n  LanguageModelV1StreamPart,\n} from 'ai'\nimport { v4 as uuidv4 } from 'uuid'\nimport type { PostHog } from 'posthog-node'\nimport { sendEventToPosthog } from '../utils'\n\ninterface CreateInstrumentationMiddlewareOptions {\n  posthogDistinctId?: string\n  posthogTraceId: string\n  posthogProperties?: Record<string, any>\n  posthogPrivacyMode?: boolean\n  posthogGroups?: string[]\n}\n\nexport const createInstrumentationMiddleware = (\n  phClient: PostHog,\n  model: LanguageModelV1,\n  options: CreateInstrumentationMiddlewareOptions\n): LanguageModelV1Middleware => {\n  const middleware: LanguageModelV1Middleware = {\n    wrapGenerate: async ({ doGenerate, params }) => {\n      const startTime = Date.now()\n\n      try {\n        const result = await doGenerate()\n        const latency = (Date.now() - startTime) / 1000\n\n        sendEventToPosthog({\n          client: phClient,\n          distinctId: options.posthogDistinctId,\n          traceId: options.posthogTraceId,\n          model: model.modelId,\n          provider: 'vercel',\n          input: options.posthogPrivacyMode ? '' : params.prompt,\n          output: [{ content: result.text, role: 'assistant' }],\n          latency,\n          baseURL: '',\n          params: { posthog_properties: options } as any,\n          httpStatus: 200,\n          usage: {\n            input_tokens: result.usage.promptTokens,\n            output_tokens: result.usage.completionTokens,\n          },\n        })\n\n        return result\n      } catch (error) {\n        sendEventToPosthog({\n          client: phClient,\n          distinctId: options.posthogDistinctId,\n          traceId: options.posthogTraceId,\n          model: model.modelId,\n          provider: 'vercel',\n          input: options.posthogPrivacyMode ? '' : params.prompt,\n          output: [],\n          latency: 0,\n          baseURL: '',\n          params: { posthog_properties: options } as any,\n          httpStatus: 500,\n          usage: {\n            input_tokens: 0,\n            output_tokens: 0,\n          },\n        })\n        throw error\n      }\n    },\n\n    wrapStream: async ({ doStream, params }) => {\n      const startTime = Date.now()\n      let generatedText = ''\n      let usage: { input_tokens?: number; output_tokens?: number } = {}\n\n      try {\n        const { stream, ...rest } = await doStream()\n\n        const transformStream = new TransformStream<LanguageModelV1StreamPart, LanguageModelV1StreamPart>({\n          transform(chunk, controller) {\n            if (chunk.type === 'text-delta') {\n              generatedText += chunk.textDelta\n            }\n            if (chunk.type === 'finish') {\n              usage = {\n                input_tokens: chunk.usage?.promptTokens,\n                output_tokens: chunk.usage?.completionTokens,\n              }\n            }\n            controller.enqueue(chunk)\n          },\n\n          flush() {\n            const latency = (Date.now() - startTime) / 1000\n            sendEventToPosthog({\n              client: phClient,\n              distinctId: options.posthogDistinctId,\n              traceId: options.posthogTraceId,\n              model: model.modelId,\n              provider: 'vercel',\n              input: options.posthogPrivacyMode ? '' : params.prompt,\n              output: [{ content: generatedText, role: 'assistant' }],\n              latency,\n              baseURL: '',\n              params: { posthog_properties: options } as any,\n              httpStatus: 200,\n              usage,\n            })\n          },\n        })\n\n        return {\n          stream: stream.pipeThrough(transformStream),\n          ...rest,\n        }\n      } catch (error) {\n        sendEventToPosthog({\n          client: phClient,\n          distinctId: options.posthogDistinctId,\n          traceId: options.posthogTraceId,\n          model: model.modelId,\n          provider: 'vercel',\n          input: options.posthogPrivacyMode ? '' : params.prompt,\n          output: [],\n          latency: 0,\n          baseURL: '',\n          params: { posthog_properties: options } as any,\n          httpStatus: 500,\n          usage: {\n            input_tokens: 0,\n            output_tokens: 0,\n          },\n        })\n        throw error\n      }\n    },\n  }\n\n  return middleware\n}\n\nexport const wrapVercelLanguageModel = (\n  model: LanguageModelV1,\n  phClient: PostHog,\n  options: CreateInstrumentationMiddlewareOptions\n): LanguageModelV1 => {\n  const traceId = options.posthogTraceId ?? uuidv4()\n  const middleware = createInstrumentationMiddleware(phClient, model, {\n    ...options,\n    posthogTraceId: traceId,\n    posthogDistinctId: options.posthogDistinctId ?? traceId,\n  })\n\n  const wrappedModel = wrapLanguageModel({\n    model,\n    middleware,\n  })\n\n  return wrappedModel\n}\n"],"names":["getModelParams","params","modelParams","paramKeys","key","undefined","mergeSystemPrompt","provider","messages","system","systemMessage","role","content","withPrivacyMode","client","privacyMode","input","privacy_mode","sendEventToPosthog","distinctId","traceId","model","output","latency","baseURL","httpStatus","usage","capture","event","properties","$ai_provider","$ai_model","$ai_model_parameters","$ai_input","posthog_privacy_mode","$ai_output_choices","$ai_http_status","$ai_input_tokens","input_tokens","$ai_output_tokens","output_tokens","$ai_latency","$ai_trace_id","$ai_base_url","posthog_properties","$process_person_profile","groups","posthog_groups","PostHogOpenAI","OpenAIOrignal","constructor","config","posthog","openAIConfig","phClient","chat","WrappedChat","Chat","parentClient","completions","WrappedCompletions","Completions","create","body","options","posthog_distinct_id","posthog_trace_id","openAIParams","uuidv4","startTime","Date","now","parentPromise","stream","then","value","passThroughStream","PassThrough","objectMode","accumulatedContent","openAIStream","chunk","delta","choices","prompt_tokens","completion_tokens","write","end","error","emit","wrappedPromise","result","message","createInstrumentationMiddleware","middleware","wrapGenerate","doGenerate","posthogDistinctId","posthogTraceId","modelId","posthogPrivacyMode","prompt","text","promptTokens","completionTokens","wrapStream","doStream","generatedText","rest","transformStream","TransformStream","transform","controller","type","textDelta","enqueue","flush","pipeThrough","wrapVercelLanguageModel","wrappedModel","wrapLanguageModel"],"mappings":";;;;;AAaO,MAAMA,cAAc,GAAIC,MAAyD,IAAyB;EAC/G,MAAMC,WAAW,GAAwB,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,MAAc,CAACG,GAAG,CAAC,KAAKC,SAAS,EAAE;AACvDH,MAAAA,WAAW,CAACE,GAAG,CAAC,GAAIH,MAAc,CAACG,GAAG,CAAC,CAAA;AACxC,KAAA;AACF,GAAA;AACD,EAAA,OAAOF,WAAW,CAAA;AACpB,CAAC,CAAA;AAgEM,MAAMI,iBAAiB,GAAGA,CAACL,MAAyD,EAAEM,QAAgB,KAAS;EACpH,IAAIA,QAAQ,KAAK,WAAW,EAAE;IAC5B,OAAON,MAAM,CAACO,QAAQ,CAAA;AACvB,GAAA;AACD,EAAA,MAAMA,QAAQ,GAAGP,MAAM,CAACO,QAAQ,IAAI,EAAE,CAAA;AACtC,EAAA,IAAI,CAAEP,MAAc,CAACQ,MAAM,EAAE;AAC3B,IAAA,OAAOD,QAAQ,CAAA;AAChB,GAAA;AACD,EAAA,MAAME,aAAa,GAAIT,MAAc,CAACQ,MAAM,CAAA;AAC5C,EAAA,OAAO,CAAC;AAAEE,IAAAA,IAAI,EAAE,QAAQ;AAAEC,IAAAA,OAAO,EAAEF,aAAAA;GAAe,EAAE,GAAGF,QAAQ,CAAC,CAAA;AAClE,CAAC,CAAA;AAEM,MAAMK,eAAe,GAAGA,CAACC,MAAe,EAAEC,WAAoB,EAAEC,KAAU,KAAS;EACxF,OAAQF,MAAc,CAACG,YAAY,IAAIF,WAAW,GAAG,IAAI,GAAGC,KAAK,CAAA;AACnE,CAAC,CAAA;AAiBM,MAAME,kBAAkB,GAAGA,CAAC;EACjCJ,MAAM;EACNK,UAAU;EACVC,OAAO;EACPC,KAAK;EACLd,QAAQ;EACRS,KAAK;EACLM,MAAM;EACNC,OAAO;EACPC,OAAO;EACPvB,MAAM;AACNwB,EAAAA,UAAU,GAAG,GAAG;AAChBC,EAAAA,KAAK,GAAG,EAAE;AAAA,CACe,KAAU;EACnC,IAAIZ,MAAM,CAACa,OAAO,EAAE;IAClBb,MAAM,CAACa,OAAO,CAAC;MACbR,UAAU,EAAEA,UAAU,IAAIC,OAAO;AACjCQ,MAAAA,KAAK,EAAE,gBAAgB;AACvBC,MAAAA,UAAU,EAAE;AACVC,QAAAA,YAAY,EAAEvB,QAAQ;AACtBwB,QAAAA,SAAS,EAAEV,KAAK;AAChBW,QAAAA,oBAAoB,EAAEhC,cAAc,CAACC,MAAM,CAAC;AAC5CgC,QAAAA,SAAS,EAAEpB,eAAe,CAACC,MAAM,EAAEb,MAAM,CAACiC,oBAAoB,IAAI,KAAK,EAAElB,KAAK,CAAC;AAC/EmB,QAAAA,kBAAkB,EAAEtB,eAAe,CAACC,MAAM,EAAEb,MAAM,CAACiC,oBAAoB,IAAI,KAAK,EAAEZ,MAAM,CAAC;AACzFc,QAAAA,eAAe,EAAEX,UAAU;AAC3BY,QAAAA,gBAAgB,EAAEX,KAAK,CAACY,YAAY,IAAI,CAAC;AACzCC,QAAAA,iBAAiB,EAAEb,KAAK,CAACc,aAAa,IAAI,CAAC;AAC3CC,QAAAA,WAAW,EAAElB,OAAO;AACpBmB,QAAAA,YAAY,EAAEtB,OAAO;AACrBuB,QAAAA,YAAY,EAAEnB,OAAO;QACrB,GAAGvB,MAAM,CAAC2C,kBAAkB;AAC5B,QAAA,IAAIzB,UAAU,GAAG,EAAE,GAAG;AAAE0B,UAAAA,uBAAuB,EAAE,KAAA;SAAO,CAAA;OACzD;MACDC,MAAM,EAAE7C,MAAM,CAAC8C,cAAAA;AAChB,KAAA,CAAC,CAAA;AACH,GAAA;AACH,CAAC;;ACjJK,MAAOC,aAAc,SAAQC,aAAa,CAAA;EAG9CC,WAAAA,CAAYC,MAA8B,EAAA;IACxC,MAAM;MAAEC,OAAO;MAAE,GAAGC,YAAAA;AAAc,KAAA,GAAGF,MAAM,CAAA;IAC3C,KAAK,CAACE,YAAY,CAAC,CAAA;IACnB,IAAI,CAACC,QAAQ,GAAGF,OAAO,CAAA;IACvB,IAAI,CAACG,IAAI,GAAG,IAAIC,WAAW,CAAC,IAAI,EAAE,IAAI,CAACF,QAAQ,CAAC,CAAA;AAClD,GAAA;AAGD,CAAA;AAEY,MAAAE,WAAY,SAAQP,aAAa,CAACQ,IAAI,CAAA;AACjDP,EAAAA,WAAYA,CAAAQ,YAA2B,EAAEJ,QAAiB,EAAA;IACxD,KAAK,CAACI,YAAY,CAAC,CAAA;IACnB,IAAI,CAACC,WAAW,GAAG,IAAIC,kBAAkB,CAACF,YAAY,EAAEJ,QAAQ,CAAC,CAAA;AACnE,GAAA;AAGD,CAAA;MAEYM,kBAAmB,SAAQX,aAAa,CAACQ,IAAI,CAACI,WAAW,CAAA;AAGpEX,EAAAA,WAAYA,CAAApC,MAAqB,EAAEwC,QAAiB,EAAA;IAClD,KAAK,CAACxC,MAAM,CAAC,CAAA;IACb,IAAI,CAACwC,QAAQ,GAAGA,QAAQ,CAAA;AAC1B,GAAA;AAoBA;AACOQ,EAAAA,MAAMA,CACXC,IAAuD,EACvDC,OAAwB,EAAA;IAExB,MAAM;MACJC,mBAAmB;MACnBC,gBAAgB;MAChBtB,kBAAkB;AAClBV,MAAAA,oBAAoB,GAAG,KAAK;MAC5Ba,cAAc;MACd,GAAGoB,YAAAA;AACJ,KAAA,GAAGJ,IAAI,CAAA;AAER,IAAA,MAAM3C,OAAO,GAAG8C,gBAAgB,IAAIE,EAAM,EAAE,CAAA;AAC5C,IAAA,MAAMC,SAAS,GAAGC,IAAI,CAACC,GAAG,EAAE,CAAA;IAE5B,MAAMC,aAAa,GAAG,KAAK,CAACV,MAAM,CAACK,YAAY,EAAEH,OAAO,CAAC,CAAA;IAEzD,IAAIG,YAAY,CAACM,MAAM,EAAE;AACvB,MAAA,OAAOD,aAAa,CAACE,IAAI,CAAEC,KAAK,IAAI;AAClC,QAAA,MAAMC,iBAAiB,GAAG,IAAIC,WAAW,CAAC;AAAEC,UAAAA,UAAU,EAAE,IAAA;AAAM,SAAA,CAAC,CAAA;QAC/D,IAAIC,kBAAkB,GAAG,EAAE,CAAA;AAC3B,QAAA,IAAIrD,KAAK,GAAoD;AAC3DY,UAAAA,YAAY,EAAE,CAAC;AACfE,UAAAA,aAAa,EAAE,CAAA;SAChB,CAAA;QACD,IAAI,KAAK,IAAImC,KAAK,EAAE;UAClB,MAAMK,YAAY,GAAGL,KAAK,CAAA;AACzB,UAAA,CAAC,YAAW;YACX,IAAI;AACF,cAAA,WAAW,MAAMM,KAAK,IAAID,YAAY,EAAE;AACtC,gBAAA,MAAME,KAAK,GAAGD,KAAK,EAAEE,OAAO,GAAG,CAAC,CAAC,EAAED,KAAK,EAAEtE,OAAO,IAAI,EAAE,CAAA;AACvDmE,gBAAAA,kBAAkB,IAAIG,KAAK,CAAA;gBAC3B,IAAID,KAAK,CAACvD,KAAK,EAAE;AACfA,kBAAAA,KAAK,GAAG;AACNY,oBAAAA,YAAY,EAAE2C,KAAK,CAACvD,KAAK,CAAC0D,aAAa,IAAI,CAAC;AAC5C5C,oBAAAA,aAAa,EAAEyC,KAAK,CAACvD,KAAK,CAAC2D,iBAAiB,IAAI,CAAA;mBACjD,CAAA;AACF,iBAAA;AACDT,gBAAAA,iBAAiB,CAACU,KAAK,CAACL,KAAK,CAAC,CAAA;AAC/B,eAAA;cACD,MAAM1D,OAAO,GAAG,CAAC+C,IAAI,CAACC,GAAG,EAAE,GAAGF,SAAS,IAAI,IAAI,CAAA;AAC/CnD,cAAAA,kBAAkB,CAAC;gBACjBJ,MAAM,EAAE,IAAI,CAACwC,QAAQ;gBACrBnC,UAAU,EAAE8C,mBAAmB,IAAI7C,OAAO;gBAC1CA,OAAO;gBACPC,KAAK,EAAE8C,YAAY,CAAC9C,KAAK;AACzBd,gBAAAA,QAAQ,EAAE,QAAQ;gBAClBS,KAAK,EAAEkB,oBAAoB,GAAG,EAAE,GAAG5B,iBAAiB,CAAC6D,YAAY,EAAE,QAAQ,CAAC;AAC5E7C,gBAAAA,MAAM,EAAE,CAAC;AAAEV,kBAAAA,OAAO,EAAEmE,kBAAkB;AAAEpE,kBAAAA,IAAI,EAAE,WAAA;AAAW,iBAAE,CAAC;gBAC5DY,OAAO;AACPC,gBAAAA,OAAO,EAAG,IAAY,CAACA,OAAO,IAAI,EAAE;AACpCvB,gBAAAA,MAAM,EAAE8D,IAAI;AACZtC,gBAAAA,UAAU,EAAE,GAAG;AACfC,gBAAAA,KAAAA;AACD,eAAA,CAAC,CAAA;cACFkD,iBAAiB,CAACW,GAAG,EAAE,CAAA;aACxB,CAAC,OAAOC,KAAK,EAAE;AACd;AACAtE,cAAAA,kBAAkB,CAAC;gBACjBJ,MAAM,EAAE,IAAI,CAACwC,QAAQ;gBACrBnC,UAAU,EAAE8C,mBAAmB,IAAI7C,OAAO;gBAC1CA,OAAO;gBACPC,KAAK,EAAE8C,YAAY,CAAC9C,KAAK;AACzBd,gBAAAA,QAAQ,EAAE,QAAQ;gBAClBS,KAAK,EAAEkB,oBAAoB,GAAG,EAAE,GAAG5B,iBAAiB,CAAC6D,YAAY,EAAE,QAAQ,CAAC;AAC5E7C,gBAAAA,MAAM,EAAE,EAAE;AACVC,gBAAAA,OAAO,EAAE,CAAC;AACVC,gBAAAA,OAAO,EAAG,IAAY,CAACA,OAAO,IAAI,EAAE;AACpCvB,gBAAAA,MAAM,EAAE8D,IAAI;AACZtC,gBAAAA,UAAU,EAAE,GAAG;AACfC,gBAAAA,KAAK,EAAE;AACLY,kBAAAA,YAAY,EAAE,CAAC;AACfE,kBAAAA,aAAa,EAAE,CAAA;AAChB,iBAAA;AACF,eAAA,CAAC,CAAA;AACFoC,cAAAA,iBAAiB,CAACa,IAAI,CAAC,OAAO,EAAED,KAAK,CAAC,CAAA;AACvC,aAAA;AACH,WAAC,GAAG,CAAA;AACL,SAAA;AACD,QAAA,OAAOZ,iBAA2D,CAAA;AACpE,OAAC,CAA4C,CAAA;AAC9C,KAAA,MAAM;AACL,MAAA,MAAMc,cAAc,GAAGlB,aAAa,CAACE,IAAI,CACtCiB,MAAM,IAAI;QACT,IAAI,SAAS,IAAIA,MAAM,EAAE;UACvB,MAAMpE,OAAO,GAAG,CAAC+C,IAAI,CAACC,GAAG,EAAE,GAAGF,SAAS,IAAI,IAAI,CAAA;AAC/CnD,UAAAA,kBAAkB,CAAC;YACjBJ,MAAM,EAAE,IAAI,CAACwC,QAAQ;YACrBnC,UAAU,EAAE8C,mBAAmB,IAAI7C,OAAO;YAC1CA,OAAO;YACPC,KAAK,EAAE8C,YAAY,CAAC9C,KAAK;AACzBd,YAAAA,QAAQ,EAAE,QAAQ;YAClBS,KAAK,EAAEkB,oBAAoB,GAAG,EAAE,GAAG5B,iBAAiB,CAAC6D,YAAY,EAAE,QAAQ,CAAC;AAC5E7C,YAAAA,MAAM,EAAE,CAAC;cAAEV,OAAO,EAAE+E,MAAM,CAACR,OAAO,CAAC,CAAC,CAAC,CAACS,OAAO,CAAChF,OAAO;AAAED,cAAAA,IAAI,EAAE,WAAA;aAAa,CAAC;YAC3EY,OAAO;AACPC,YAAAA,OAAO,EAAG,IAAY,CAACA,OAAO,IAAI,EAAE;AACpCvB,YAAAA,MAAM,EAAE8D,IAAI;AACZtC,YAAAA,UAAU,EAAE,GAAG;AACfC,YAAAA,KAAK,EAAE;AACLY,cAAAA,YAAY,EAAEqD,MAAM,CAACjE,KAAK,EAAE0D,aAAa,IAAI,CAAC;AAC9C5C,cAAAA,aAAa,EAAEmD,MAAM,CAACjE,KAAK,EAAE2D,iBAAiB,IAAI,CAAA;AACnD,aAAA;AACF,WAAA,CAAC,CAAA;AACH,SAAA;AACD,QAAA,OAAOM,MAAM,CAAA;OACd,EACAH,KAAK,IAAI;AACRtE,QAAAA,kBAAkB,CAAC;UACjBJ,MAAM,EAAE,IAAI,CAACwC,QAAQ;UACrBnC,UAAU,EAAE8C,mBAAmB,IAAI7C,OAAO;UAC1CA,OAAO;UACPC,KAAK,EAAE8C,YAAY,CAAC9C,KAAK;AACzBd,UAAAA,QAAQ,EAAE,QAAQ;UAClBS,KAAK,EAAEkB,oBAAoB,GAAG,EAAE,GAAG5B,iBAAiB,CAAC6D,YAAY,EAAE,QAAQ,CAAC;AAC5E7C,UAAAA,MAAM,EAAE,EAAE;AACVC,UAAAA,OAAO,EAAE,CAAC;AACVC,UAAAA,OAAO,EAAG,IAAY,CAACA,OAAO,IAAI,EAAE;AACpCvB,UAAAA,MAAM,EAAE8D,IAAI;AACZtC,UAAAA,UAAU,EAAE,GAAG;AACfC,UAAAA,KAAK,EAAE;AACLY,YAAAA,YAAY,EAAE,CAAC;AACfE,YAAAA,aAAa,EAAE,CAAA;AAChB,WAAA;AACF,SAAA,CAAC,CAAA;AACF,QAAA,MAAMgD,KAAK,CAAA;AACb,OAAC,CAC4B,CAAA;AAE/B,MAAA,OAAOE,cAAc,CAAA;AACtB,KAAA;AACH,GAAA;AACD;;ACvLM,MAAMG,+BAA+B,GAAGA,CAC7CvC,QAAiB,EACjBjC,KAAsB,EACtB2C,OAA+C,KAClB;AAC7B,EAAA,MAAM8B,UAAU,GAA8B;IAC5CC,YAAY,EAAE,OAAO;MAAEC,UAAU;AAAE/F,MAAAA,MAAAA;AAAQ,KAAA,KAAI;AAC7C,MAAA,MAAMoE,SAAS,GAAGC,IAAI,CAACC,GAAG,EAAE,CAAA;MAE5B,IAAI;AACF,QAAA,MAAMoB,MAAM,GAAG,MAAMK,UAAU,EAAE,CAAA;QACjC,MAAMzE,OAAO,GAAG,CAAC+C,IAAI,CAACC,GAAG,EAAE,GAAGF,SAAS,IAAI,IAAI,CAAA;AAE/CnD,QAAAA,kBAAkB,CAAC;AACjBJ,UAAAA,MAAM,EAAEwC,QAAQ;UAChBnC,UAAU,EAAE6C,OAAO,CAACiC,iBAAiB;UACrC7E,OAAO,EAAE4C,OAAO,CAACkC,cAAc;UAC/B7E,KAAK,EAAEA,KAAK,CAAC8E,OAAO;AACpB5F,UAAAA,QAAQ,EAAE,QAAQ;UAClBS,KAAK,EAAEgD,OAAO,CAACoC,kBAAkB,GAAG,EAAE,GAAGnG,MAAM,CAACoG,MAAM;AACtD/E,UAAAA,MAAM,EAAE,CAAC;YAAEV,OAAO,EAAE+E,MAAM,CAACW,IAAI;AAAE3F,YAAAA,IAAI,EAAE,WAAA;WAAa,CAAC;UACrDY,OAAO;AACPC,UAAAA,OAAO,EAAE,EAAE;AACXvB,UAAAA,MAAM,EAAE;AAAE2C,YAAAA,kBAAkB,EAAEoB,OAAAA;WAAgB;AAC9CvC,UAAAA,UAAU,EAAE,GAAG;AACfC,UAAAA,KAAK,EAAE;AACLY,YAAAA,YAAY,EAAEqD,MAAM,CAACjE,KAAK,CAAC6E,YAAY;AACvC/D,YAAAA,aAAa,EAAEmD,MAAM,CAACjE,KAAK,CAAC8E,gBAAAA;AAC7B,WAAA;AACF,SAAA,CAAC,CAAA;AAEF,QAAA,OAAOb,MAAM,CAAA;OACd,CAAC,OAAOH,KAAK,EAAE;AACdtE,QAAAA,kBAAkB,CAAC;AACjBJ,UAAAA,MAAM,EAAEwC,QAAQ;UAChBnC,UAAU,EAAE6C,OAAO,CAACiC,iBAAiB;UACrC7E,OAAO,EAAE4C,OAAO,CAACkC,cAAc;UAC/B7E,KAAK,EAAEA,KAAK,CAAC8E,OAAO;AACpB5F,UAAAA,QAAQ,EAAE,QAAQ;UAClBS,KAAK,EAAEgD,OAAO,CAACoC,kBAAkB,GAAG,EAAE,GAAGnG,MAAM,CAACoG,MAAM;AACtD/E,UAAAA,MAAM,EAAE,EAAE;AACVC,UAAAA,OAAO,EAAE,CAAC;AACVC,UAAAA,OAAO,EAAE,EAAE;AACXvB,UAAAA,MAAM,EAAE;AAAE2C,YAAAA,kBAAkB,EAAEoB,OAAAA;WAAgB;AAC9CvC,UAAAA,UAAU,EAAE,GAAG;AACfC,UAAAA,KAAK,EAAE;AACLY,YAAAA,YAAY,EAAE,CAAC;AACfE,YAAAA,aAAa,EAAE,CAAA;AAChB,WAAA;AACF,SAAA,CAAC,CAAA;AACF,QAAA,MAAMgD,KAAK,CAAA;AACZ,OAAA;KACF;IAEDiB,UAAU,EAAE,OAAO;MAAEC,QAAQ;AAAEzG,MAAAA,MAAAA;AAAQ,KAAA,KAAI;AACzC,MAAA,MAAMoE,SAAS,GAAGC,IAAI,CAACC,GAAG,EAAE,CAAA;MAC5B,IAAIoC,aAAa,GAAG,EAAE,CAAA;MACtB,IAAIjF,KAAK,GAAsD,EAAE,CAAA;MAEjE,IAAI;QACF,MAAM;UAAE+C,MAAM;UAAE,GAAGmC,IAAAA;SAAM,GAAG,MAAMF,QAAQ,EAAE,CAAA;AAE5C,QAAA,MAAMG,eAAe,GAAG,IAAIC,eAAe,CAAuD;AAChGC,UAAAA,SAASA,CAAC9B,KAAK,EAAE+B,UAAU,EAAA;AACzB,YAAA,IAAI/B,KAAK,CAACgC,IAAI,KAAK,YAAY,EAAE;cAC/BN,aAAa,IAAI1B,KAAK,CAACiC,SAAS,CAAA;AACjC,aAAA;AACD,YAAA,IAAIjC,KAAK,CAACgC,IAAI,KAAK,QAAQ,EAAE;AAC3BvF,cAAAA,KAAK,GAAG;AACNY,gBAAAA,YAAY,EAAE2C,KAAK,CAACvD,KAAK,EAAE6E,YAAY;AACvC/D,gBAAAA,aAAa,EAAEyC,KAAK,CAACvD,KAAK,EAAE8E,gBAAAA;eAC7B,CAAA;AACF,aAAA;AACDQ,YAAAA,UAAU,CAACG,OAAO,CAAClC,KAAK,CAAC,CAAA;WAC1B;AAEDmC,UAAAA,KAAKA,GAAA;YACH,MAAM7F,OAAO,GAAG,CAAC+C,IAAI,CAACC,GAAG,EAAE,GAAGF,SAAS,IAAI,IAAI,CAAA;AAC/CnD,YAAAA,kBAAkB,CAAC;AACjBJ,cAAAA,MAAM,EAAEwC,QAAQ;cAChBnC,UAAU,EAAE6C,OAAO,CAACiC,iBAAiB;cACrC7E,OAAO,EAAE4C,OAAO,CAACkC,cAAc;cAC/B7E,KAAK,EAAEA,KAAK,CAAC8E,OAAO;AACpB5F,cAAAA,QAAQ,EAAE,QAAQ;cAClBS,KAAK,EAAEgD,OAAO,CAACoC,kBAAkB,GAAG,EAAE,GAAGnG,MAAM,CAACoG,MAAM;AACtD/E,cAAAA,MAAM,EAAE,CAAC;AAAEV,gBAAAA,OAAO,EAAE+F,aAAa;AAAEhG,gBAAAA,IAAI,EAAE,WAAA;AAAW,eAAE,CAAC;cACvDY,OAAO;AACPC,cAAAA,OAAO,EAAE,EAAE;AACXvB,cAAAA,MAAM,EAAE;AAAE2C,gBAAAA,kBAAkB,EAAEoB,OAAAA;eAAgB;AAC9CvC,cAAAA,UAAU,EAAE,GAAG;AACfC,cAAAA,KAAAA;AACD,aAAA,CAAC,CAAA;AACJ,WAAA;AACD,SAAA,CAAC,CAAA;QAEF,OAAO;AACL+C,UAAAA,MAAM,EAAEA,MAAM,CAAC4C,WAAW,CAACR,eAAe,CAAC;UAC3C,GAAGD,IAAAA;SACJ,CAAA;OACF,CAAC,OAAOpB,KAAK,EAAE;AACdtE,QAAAA,kBAAkB,CAAC;AACjBJ,UAAAA,MAAM,EAAEwC,QAAQ;UAChBnC,UAAU,EAAE6C,OAAO,CAACiC,iBAAiB;UACrC7E,OAAO,EAAE4C,OAAO,CAACkC,cAAc;UAC/B7E,KAAK,EAAEA,KAAK,CAAC8E,OAAO;AACpB5F,UAAAA,QAAQ,EAAE,QAAQ;UAClBS,KAAK,EAAEgD,OAAO,CAACoC,kBAAkB,GAAG,EAAE,GAAGnG,MAAM,CAACoG,MAAM;AACtD/E,UAAAA,MAAM,EAAE,EAAE;AACVC,UAAAA,OAAO,EAAE,CAAC;AACVC,UAAAA,OAAO,EAAE,EAAE;AACXvB,UAAAA,MAAM,EAAE;AAAE2C,YAAAA,kBAAkB,EAAEoB,OAAAA;WAAgB;AAC9CvC,UAAAA,UAAU,EAAE,GAAG;AACfC,UAAAA,KAAK,EAAE;AACLY,YAAAA,YAAY,EAAE,CAAC;AACfE,YAAAA,aAAa,EAAE,CAAA;AAChB,WAAA;AACF,SAAA,CAAC,CAAA;AACF,QAAA,MAAMgD,KAAK,CAAA;AACZ,OAAA;AACH,KAAA;GACD,CAAA;AAED,EAAA,OAAOM,UAAU,CAAA;AACnB,CAAC,CAAA;AAEM,MAAMwB,uBAAuB,GAAGA,CACrCjG,KAAsB,EACtBiC,QAAiB,EACjBU,OAA+C,KAC5B;EACnB,MAAM5C,OAAO,GAAG4C,OAAO,CAACkC,cAAc,IAAI9B,EAAM,EAAE,CAAA;AAClD,EAAA,MAAM0B,UAAU,GAAGD,+BAA+B,CAACvC,QAAQ,EAAEjC,KAAK,EAAE;AAClE,IAAA,GAAG2C,OAAO;AACVkC,IAAAA,cAAc,EAAE9E,OAAO;AACvB6E,IAAAA,iBAAiB,EAAEjC,OAAO,CAACiC,iBAAiB,IAAI7E,OAAAA;AACjD,GAAA,CAAC,CAAA;EAEF,MAAMmG,YAAY,GAAGC,8BAAiB,CAAC;IACrCnG,KAAK;AACLyE,IAAAA,UAAAA;AACD,GAAA,CAAC,CAAA;AAEF,EAAA,OAAOyB,YAAY,CAAA;AACrB;;;;"}

package/lib/posthog-ai/index.d.ts

@@ -0,0 +1 @@
+export * from './src';

package/lib/posthog-ai/src/index.d.ts

@@ -0,0 +1,4 @@
+import PostHogOpenAI from './openai';
+import { wrapVercelLanguageModel } from './vercel/middleware';
+export { PostHogOpenAI as OpenAI };
+export { wrapVercelLanguageModel as withTracing };

package/lib/posthog-ai/src/openai/index.d.ts

@@ -0,0 +1,32 @@
+import OpenAIOrignal from 'openai';
+import { PostHog } from 'posthog-node';
+import { MonitoringParams } from '../utils';
+type ChatCompletion = OpenAIOrignal.ChatCompletion;
+type ChatCompletionChunk = OpenAIOrignal.ChatCompletionChunk;
+type ChatCompletionCreateParamsBase = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParams;
+type ChatCompletionCreateParamsNonStreaming = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParamsNonStreaming;
+type ChatCompletionCreateParamsStreaming = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParamsStreaming;
+import type { APIPromise, RequestOptions } from 'openai/core';
+import type { Stream } from 'openai/streaming';
+interface MonitoringOpenAIConfig {
+    apiKey: string;
+    posthog: PostHog;
+    baseURL?: string;
+}
+export declare class PostHogOpenAI extends OpenAIOrignal {
+    private readonly phClient;
+    constructor(config: MonitoringOpenAIConfig);
+    chat: WrappedChat;
+}
+export declare class WrappedChat extends OpenAIOrignal.Chat {
+    constructor(parentClient: PostHogOpenAI, phClient: PostHog);
+    completions: WrappedCompletions;
+}
+export declare class WrappedCompletions extends OpenAIOrignal.Chat.Completions {
+    private readonly phClient;
+    constructor(client: OpenAIOrignal, phClient: PostHog);
+    create(body: ChatCompletionCreateParamsNonStreaming & MonitoringParams, options?: RequestOptions): APIPromise<ChatCompletion>;
+    create(body: ChatCompletionCreateParamsStreaming & MonitoringParams, options?: RequestOptions): APIPromise<Stream<ChatCompletionChunk>>;
+    create(body: ChatCompletionCreateParamsBase & MonitoringParams, options?: RequestOptions): APIPromise<ChatCompletion | Stream<ChatCompletionChunk>>;
+}
+export default PostHogOpenAI;

package/lib/posthog-ai/src/utils.d.ts

@@ -0,0 +1,51 @@
+import { PostHog } from 'posthog-node';
+import OpenAIOrignal from 'openai';
+type ChatCompletionCreateParamsBase = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParams;
+export interface MonitoringParams {
+    posthog_distinct_id?: string;
+    posthog_trace_id?: string;
+    posthog_properties?: Record<string, any>;
+    posthog_privacy_mode?: boolean;
+    posthog_groups?: Record<string, any>;
+}
+export declare const getModelParams: (params: ChatCompletionCreateParamsBase & MonitoringParams) => Record<string, any>;
+export declare const getUsage: (response: any, provider: string) => {
+    input_tokens: number;
+    output_tokens: number;
+};
+/**
+ * Helper to format responses (non-streaming) for consumption, mirroring Python's openai vs. anthropic approach.
+ */
+export declare const formatResponse: (response: any, provider: string) => Array<{
+    role: string;
+    content: string;
+}>;
+export declare const formatResponseAnthropic: (response: any) => Array<{
+    role: string;
+    content: string;
+}>;
+export declare const formatResponseOpenAI: (response: any) => Array<{
+    role: string;
+    content: string;
+}>;
+export declare const mergeSystemPrompt: (params: ChatCompletionCreateParamsBase & MonitoringParams, provider: string) => any;
+export declare const withPrivacyMode: (client: PostHog, privacyMode: boolean, input: any) => any;
+export type SendEventToPosthogParams = {
+    client: PostHog;
+    distinctId?: string;
+    traceId: string;
+    model: string;
+    provider: string;
+    input: any;
+    output: any;
+    latency: number;
+    baseURL: string;
+    httpStatus: number;
+    usage?: {
+        input_tokens?: number;
+        output_tokens?: number;
+    };
+    params: ChatCompletionCreateParamsBase & MonitoringParams;
+};
+export declare const sendEventToPosthog: ({ client, distinctId, traceId, model, provider, input, output, latency, baseURL, params, httpStatus, usage, }: SendEventToPosthogParams) => void;
+export {};

package/lib/posthog-ai/src/vercel/middleware.d.ts

@@ -0,0 +1,12 @@
+import type { LanguageModelV1, Experimental_LanguageModelV1Middleware as LanguageModelV1Middleware } from 'ai';
+import type { PostHog } from 'posthog-node';
+interface CreateInstrumentationMiddlewareOptions {
+    posthogDistinctId?: string;
+    posthogTraceId: string;
+    posthogProperties?: Record<string, any>;
+    posthogPrivacyMode?: boolean;
+    posthogGroups?: string[];
+}
+export declare const createInstrumentationMiddleware: (phClient: PostHog, model: LanguageModelV1, options: CreateInstrumentationMiddlewareOptions) => LanguageModelV1Middleware;
+export declare const wrapVercelLanguageModel: (model: LanguageModelV1, phClient: PostHog, options: CreateInstrumentationMiddlewareOptions) => LanguageModelV1;
+export {};

package/lib/posthog-ai/tests/openai.test.d.ts

@@ -0,0 +1 @@
+export {};

package/package.json

@@ -1,15 +1,15 @@
 {
   "name": "@posthog/ai",
-  "version": "1.3.0",
+  "version": "1.5.0",
   "description": "PostHog Node.js AI integrations",
   "repository": {
     "type": "git",
     "url": "git+https://github.com/PostHog/posthog-js-lite.git",
     "directory": "posthog-ai"
   },
-  "main": "./lib/index.js",
-  "module": "./lib/index.mjs",
-  "types": "./lib/index.d.ts",
+  "main": "lib/index.cjs.js",
+  "module": "lib/index.esm.js",
+  "types": "lib/index.d.ts",
   "license": "MIT",
   "devDependencies": {
     "@types/jest": "^28.1.5",