@databuddy/sdk 2.3.23 → 2.3.25

package/dist/ai/vercel/index.d.mts

@@ -1,5 +1,58 @@
-import { LanguageModelV2 } from '@ai-sdk/provider';
+import { LanguageModelV2, LanguageModelV3 } from '@ai-sdk/provider';
 
+/**
+ * Type guards and union types for Vercel AI SDK V2/V3 support
+ *
+ * Adapted from PostHog's AI SDK implementation:
+ * https://github.com/PostHog/posthog-js/tree/main/packages/ai
+ */
+
+type LanguageModel = LanguageModelV2 | LanguageModelV3;
+
+/**
+ * Content types for input/output arrays
+ */
+type MessageContent = {
+    type: "text";
+    text: string;
+} | {
+    type: "reasoning";
+    text: string;
+} | {
+    type: "tool-call";
+    id: string;
+    function: {
+        name: string;
+        arguments: string;
+    };
+} | {
+    type: "tool-result";
+    toolCallId: string;
+    toolName: string;
+    output: unknown;
+    isError?: boolean;
+} | {
+    type: "file";
+    file: string;
+    mediaType: string;
+} | {
+    type: "image";
+    image: string;
+    mediaType: string;
+} | {
+    type: "source";
+    sourceType: string;
+    id: string;
+    url: string;
+    title: string;
+};
+/**
+ * Message format for input/output
+ */
+interface AIMessage {
+    role: string;
+    content: string | MessageContent[];
+}
 /**
  * Token usage from AI model calls
  */
@@ -8,6 +61,9 @@ interface TokenUsage {
     outputTokens: number;
     totalTokens: number;
     cachedInputTokens?: number;
+    cacheCreationInputTokens?: number;
+    reasoningTokens?: number;
+    webSearchCount?: number;
 }
 /**
  * Cost breakdown from TokenLens
@@ -24,6 +80,7 @@ interface ToolCallInfo {
     toolCallCount: number;
     toolResultCount: number;
     toolCallNames: string[];
+    availableTools?: string[];
 }
 /**
  * Error information for failed AI calls
@@ -38,15 +95,20 @@ interface AIError {
  */
 interface AICall {
     timestamp: Date;
+    traceId: string;
     type: "generate" | "stream";
     model: string;
     provider: string;
     finishReason?: string;
+    input: AIMessage[];
+    output: AIMessage[];
     usage: TokenUsage;
     cost: TokenCost;
     tools: ToolCallInfo;
     error?: AIError;
     durationMs: number;
+    httpStatus?: number;
+    params?: Record<string, unknown>;
 }
 /**
  * Transport function for sending log entries
@@ -81,6 +143,16 @@ interface DatabuddyLLMOptions {
      * @default true
      */
     computeCosts?: boolean;
+    /**
+     * Privacy mode - when true, input/output content is not captured
+     * @default false
+     */
+    privacyMode?: boolean;
+    /**
+     * Maximum size for input/output content in bytes
+     * @default 1048576 (1MB)
+     */
+    maxContentSize?: number;
     /**
      * Called on successful AI calls
      */
@@ -99,11 +171,24 @@ interface TrackOptions {
      * If not provided, uses the transport from DatabuddyLLM instance
      */
     transport?: Transport;
+    /**
+     * Trace ID to link related calls together
+     */
+    traceId?: string;
+    /**
+     * Client ID for this specific call (overrides instance-level clientId)
+     */
+    clientId?: string;
     /**
      * Whether to compute costs using TokenLens
      * @default true
      */
     computeCosts?: boolean;
+    /**
+     * Privacy mode - when true, input/output content is not captured
+     * @default false
+     */
+    privacyMode?: boolean;
     /**
      * Called on successful AI calls
      */
@@ -115,48 +200,40 @@ interface TrackOptions {
 }
 
 /**
- * Create a Databuddy LLM tracking instance
+ * Create an HTTP transport that sends logs to an API endpoint
  *
  * @example
  * ```ts
- * import { databuddyLLM } from "@databuddy/sdk/ai/vercel";
- *
- * // Use default endpoint (basket.databuddy.cc/llm)
- * const { track } = databuddyLLM({
- *   apiKey: "your-api-key",
- * });
- *
- * // Or override with custom endpoint
- * const { track } = databuddyLLM({
- *   apiUrl: "https://custom.example.com/llm",
- *   apiKey: "your-api-key",
- * });
- *
- * // Track a model
- * const model = track(openai("gpt-4"));
+ * import { databuddyLLM, httpTransport } from "@databuddy/sdk/ai/vercel";
  *
- * // Or with custom transport
  * const { track } = databuddyLLM({
- *   transport: async (call) => console.log(call),
+ *   transport: httpTransport("https://api.example.com/ai-logs", "client-id", "api-key"),
  * });
  * ```
  */
-declare const databuddyLLM: (options?: DatabuddyLLMOptions) => {
-    track: (model: LanguageModelV2, trackOptions?: TrackOptions) => LanguageModelV2;
-};
+declare const httpTransport: (url: string, clientId?: string, apiKey?: string) => Transport;
+
 /**
- * Create an HTTP transport that sends logs to an API endpoint
+ * Vercel AI SDK middleware for Databuddy
+ *
+ * Inspired by and adapted from PostHog's AI SDK implementation:
+ * https://github.com/PostHog/posthog-js/tree/main/packages/ai
+ */
+
+/**
+ * Create a Databuddy LLM tracking instance
  *
  * @example
  * ```ts
- * import { databuddyLLM, httpTransport } from "@databuddy/sdk/ai/vercel";
+ * import { databuddyLLM } from "@databuddy/sdk/ai/vercel";
  *
- * const { track } = databuddyLLM({
- *   transport: httpTransport("https://api.example.com/ai-logs", "client-id", "api-key"),
- * });
+ * const { track } = databuddyLLM({ apiKey: "your-api-key" });
+ * const model = track(openai("gpt-4"));
  * ```
  */
-declare const httpTransport: (url: string, clientId?: string, apiKey?: string) => Transport;
+declare const databuddyLLM: (options?: DatabuddyLLMOptions) => {
+    track: <T extends LanguageModel>(model: T, trackOptions?: TrackOptions) => T;
+};
 
 export { databuddyLLM, httpTransport };
 export type { AICall, AIError, DatabuddyLLMOptions, TokenCost, TokenUsage, ToolCallInfo, TrackOptions, Transport };

package/dist/ai/vercel/index.d.ts

@@ -1,5 +1,58 @@
-import { LanguageModelV2 } from '@ai-sdk/provider';
+import { LanguageModelV2, LanguageModelV3 } from '@ai-sdk/provider';
 
+/**
+ * Type guards and union types for Vercel AI SDK V2/V3 support
+ *
+ * Adapted from PostHog's AI SDK implementation:
+ * https://github.com/PostHog/posthog-js/tree/main/packages/ai
+ */
+
+type LanguageModel = LanguageModelV2 | LanguageModelV3;
+
+/**
+ * Content types for input/output arrays
+ */
+type MessageContent = {
+    type: "text";
+    text: string;
+} | {
+    type: "reasoning";
+    text: string;
+} | {
+    type: "tool-call";
+    id: string;
+    function: {
+        name: string;
+        arguments: string;
+    };
+} | {
+    type: "tool-result";
+    toolCallId: string;
+    toolName: string;
+    output: unknown;
+    isError?: boolean;
+} | {
+    type: "file";
+    file: string;
+    mediaType: string;
+} | {
+    type: "image";
+    image: string;
+    mediaType: string;
+} | {
+    type: "source";
+    sourceType: string;
+    id: string;
+    url: string;
+    title: string;
+};
+/**
+ * Message format for input/output
+ */
+interface AIMessage {
+    role: string;
+    content: string | MessageContent[];
+}
 /**
  * Token usage from AI model calls
  */
@@ -8,6 +61,9 @@ interface TokenUsage {
     outputTokens: number;
     totalTokens: number;
     cachedInputTokens?: number;
+    cacheCreationInputTokens?: number;
+    reasoningTokens?: number;
+    webSearchCount?: number;
 }
 /**
  * Cost breakdown from TokenLens
@@ -24,6 +80,7 @@ interface ToolCallInfo {
     toolCallCount: number;
     toolResultCount: number;
     toolCallNames: string[];
+    availableTools?: string[];
 }
 /**
  * Error information for failed AI calls
@@ -38,15 +95,20 @@ interface AIError {
  */
 interface AICall {
     timestamp: Date;
+    traceId: string;
     type: "generate" | "stream";
     model: string;
     provider: string;
     finishReason?: string;
+    input: AIMessage[];
+    output: AIMessage[];
     usage: TokenUsage;
     cost: TokenCost;
     tools: ToolCallInfo;
     error?: AIError;
     durationMs: number;
+    httpStatus?: number;
+    params?: Record<string, unknown>;
 }
 /**
  * Transport function for sending log entries
@@ -81,6 +143,16 @@ interface DatabuddyLLMOptions {
      * @default true
      */
     computeCosts?: boolean;
+    /**
+     * Privacy mode - when true, input/output content is not captured
+     * @default false
+     */
+    privacyMode?: boolean;
+    /**
+     * Maximum size for input/output content in bytes
+     * @default 1048576 (1MB)
+     */
+    maxContentSize?: number;
     /**
      * Called on successful AI calls
      */
@@ -99,11 +171,24 @@ interface TrackOptions {
      * If not provided, uses the transport from DatabuddyLLM instance
      */
     transport?: Transport;
+    /**
+     * Trace ID to link related calls together
+     */
+    traceId?: string;
+    /**
+     * Client ID for this specific call (overrides instance-level clientId)
+     */
+    clientId?: string;
     /**
      * Whether to compute costs using TokenLens
      * @default true
      */
     computeCosts?: boolean;
+    /**
+     * Privacy mode - when true, input/output content is not captured
+     * @default false
+     */
+    privacyMode?: boolean;
     /**
      * Called on successful AI calls
      */
@@ -115,48 +200,40 @@ interface TrackOptions {
 }
 
 /**
- * Create a Databuddy LLM tracking instance
+ * Create an HTTP transport that sends logs to an API endpoint
  *
  * @example
  * ```ts
- * import { databuddyLLM } from "@databuddy/sdk/ai/vercel";
- *
- * // Use default endpoint (basket.databuddy.cc/llm)
- * const { track } = databuddyLLM({
- *   apiKey: "your-api-key",
- * });
- *
- * // Or override with custom endpoint
- * const { track } = databuddyLLM({
- *   apiUrl: "https://custom.example.com/llm",
- *   apiKey: "your-api-key",
- * });
- *
- * // Track a model
- * const model = track(openai("gpt-4"));
+ * import { databuddyLLM, httpTransport } from "@databuddy/sdk/ai/vercel";
  *
- * // Or with custom transport
  * const { track } = databuddyLLM({
- *   transport: async (call) => console.log(call),
+ *   transport: httpTransport("https://api.example.com/ai-logs", "client-id", "api-key"),
  * });
  * ```
  */
-declare const databuddyLLM: (options?: DatabuddyLLMOptions) => {
-    track: (model: LanguageModelV2, trackOptions?: TrackOptions) => LanguageModelV2;
-};
+declare const httpTransport: (url: string, clientId?: string, apiKey?: string) => Transport;
+
 /**
- * Create an HTTP transport that sends logs to an API endpoint
+ * Vercel AI SDK middleware for Databuddy
+ *
+ * Inspired by and adapted from PostHog's AI SDK implementation:
+ * https://github.com/PostHog/posthog-js/tree/main/packages/ai
+ */
+
+/**
+ * Create a Databuddy LLM tracking instance
  *
  * @example
  * ```ts
- * import { databuddyLLM, httpTransport } from "@databuddy/sdk/ai/vercel";
+ * import { databuddyLLM } from "@databuddy/sdk/ai/vercel";
  *
- * const { track } = databuddyLLM({
- *   transport: httpTransport("https://api.example.com/ai-logs", "client-id", "api-key"),
- * });
+ * const { track } = databuddyLLM({ apiKey: "your-api-key" });
+ * const model = track(openai("gpt-4"));
  * ```
  */
-declare const httpTransport: (url: string, clientId?: string, apiKey?: string) => Transport;
+declare const databuddyLLM: (options?: DatabuddyLLMOptions) => {
+    track: <T extends LanguageModel>(model: T, trackOptions?: TrackOptions) => T;
+};
 
 export { databuddyLLM, httpTransport };
 export type { AICall, AIError, DatabuddyLLMOptions, TokenCost, TokenUsage, ToolCallInfo, TrackOptions, Transport };