@fallom/trace 0.1.6 → 0.1.10

package/dist/index.d.ts

@@ -183,6 +183,45 @@ declare function wrapAnthropic<T extends {
 declare function wrapGoogleAI<T extends {
     generateContent: (...args: any[]) => Promise<any>;
 }>(model: T): T;
+/**
+ * Wrap the Vercel AI SDK to automatically trace all LLM calls.
+ * Works with generateText, streamText, generateObject, streamObject.
+ *
+ * @param ai - The ai module (import * as ai from "ai")
+ * @returns Object with wrapped generateText, streamText, generateObject, streamObject
+ *
+ * @example
+ * ```typescript
+ * import * as ai from "ai";
+ * import { createOpenAI } from "@ai-sdk/openai";
+ * import { trace } from "@fallom/trace";
+ *
+ * await trace.init({ apiKey: process.env.FALLOM_API_KEY });
+ * const { generateText, streamText } = trace.wrapAISDK(ai);
+ *
+ * const openrouter = createOpenAI({
+ *   apiKey: process.env.OPENROUTER_API_KEY,
+ *   baseURL: "https://openrouter.ai/api/v1",
+ * });
+ *
+ * trace.setSession("my-config", sessionId);
+ * const { text } = await generateText({
+ *   model: openrouter("openai/gpt-4o-mini"),
+ *   prompt: "Hello!",
+ * }); // Automatically traced!
+ * ```
+ */
+declare function wrapAISDK<T extends {
+    generateText: (...args: any[]) => Promise<any>;
+    streamText: (...args: any[]) => any;
+    generateObject?: (...args: any[]) => Promise<any>;
+    streamObject?: (...args: any[]) => any;
+}>(ai: T): {
+    generateText: T["generateText"];
+    streamText: T["streamText"];
+    generateObject: T["generateObject"];
+    streamObject: T["streamObject"];
+};
 
 declare const trace_clearSession: typeof clearSession;
 declare const trace_getSession: typeof getSession;
@@ -190,11 +229,12 @@ declare const trace_runWithSession: typeof runWithSession;
 declare const trace_setSession: typeof setSession;
 declare const trace_shutdown: typeof shutdown;
 declare const trace_span: typeof span;
+declare const trace_wrapAISDK: typeof wrapAISDK;
 declare const trace_wrapAnthropic: typeof wrapAnthropic;
 declare const trace_wrapGoogleAI: typeof wrapGoogleAI;
 declare const trace_wrapOpenAI: typeof wrapOpenAI;
 declare namespace trace {
-  export { trace_clearSession as clearSession, trace_getSession as getSession, init$3 as init, trace_runWithSession as runWithSession, trace_setSession as setSession, trace_shutdown as shutdown, trace_span as span, trace_wrapAnthropic as wrapAnthropic, trace_wrapGoogleAI as wrapGoogleAI, trace_wrapOpenAI as wrapOpenAI };
+  export { trace_clearSession as clearSession, trace_getSession as getSession, init$3 as init, trace_runWithSession as runWithSession, trace_setSession as setSession, trace_shutdown as shutdown, trace_span as span, trace_wrapAISDK as wrapAISDK, trace_wrapAnthropic as wrapAnthropic, trace_wrapGoogleAI as wrapGoogleAI, trace_wrapOpenAI as wrapOpenAI };
 }
 
 /**
@@ -368,16 +408,20 @@ declare namespace prompts {
  */
 interface InitOptions {
     apiKey?: string;
-    baseUrl?: string;
+    tracesUrl?: string;
+    configsUrl?: string;
+    promptsUrl?: string;
     captureContent?: boolean;
     debug?: boolean;
 }
 /**
- * Initialize both trace and models at once.
+ * Initialize trace, models, and prompts at once.
  *
  * @param options - Configuration options
  * @param options.apiKey - Your Fallom API key. Defaults to FALLOM_API_KEY env var.
- * @param options.baseUrl - API base URL. Defaults to FALLOM_BASE_URL or https://spans.fallom.com
+ * @param options.tracesUrl - Traces API URL. Defaults to FALLOM_TRACES_URL or https://traces.fallom.com
+ * @param options.configsUrl - Configs API URL. Defaults to FALLOM_CONFIGS_URL or https://configs.fallom.com
+ * @param options.promptsUrl - Prompts API URL. Defaults to FALLOM_PROMPTS_URL or https://prompts.fallom.com
  * @param options.captureContent - Whether to capture prompt/completion content (default: true)
  *
  * @example
@@ -388,7 +432,11 @@ interface InitOptions {
  * fallom.init({ apiKey: "your-api-key" });
  *
  * // Local development
- * fallom.init({ baseUrl: "http://localhost:8001" });
+ * fallom.init({
+ *   tracesUrl: "http://localhost:3002",
+ *   configsUrl: "http://localhost:3003",
+ *   promptsUrl: "http://localhost:3004"
+ * });
  *
  * // Privacy mode
  * fallom.init({ captureContent: false });