UNPKG

@posthog/ai

Version:
27 lines (26 loc) 1.17 kB
import type { LanguageModelV1, LanguageModelV1Middleware } from 'ai'; import { PostHog } from 'posthog-node'; import { CostOverride } from '../utils'; interface ClientOptions { posthogDistinctId?: string; posthogTraceId?: string; posthogProperties?: Record<string, any>; posthogPrivacyMode?: boolean; posthogGroups?: Record<string, any>; posthogModelOverride?: string; posthogProviderOverride?: string; posthogCostOverride?: CostOverride; } interface CreateInstrumentationMiddlewareOptions { posthogDistinctId: string; posthogTraceId: string; posthogProperties?: Record<string, any>; posthogPrivacyMode?: boolean; posthogGroups?: Record<string, any>; posthogModelOverride?: string; posthogProviderOverride?: string; posthogCostOverride?: CostOverride; } export declare const createInstrumentationMiddleware: (phClient: PostHog, model: LanguageModelV1, options: CreateInstrumentationMiddlewareOptions) => LanguageModelV1Middleware; export declare const wrapVercelLanguageModel: (model: LanguageModelV1, phClient: PostHog, options: ClientOptions) => LanguageModelV1; export {};