langwatch 0.3.0-prerelease.1 → 0.3.0-prerelease.2

package/examples/vercel-ai/package.json

@@ -1,27 +0,0 @@
-{
-  "name": "vercel-ai-sdk-example",
-  "version": "1.0.0",
-  "description": "",
-  "main": "index.js",
-  "scripts": {
-    "start": "tsc && dotenv -- node dist/index.js",
-    "build": "tsc",
-    "test": "echo \"Error: no test specified\" && exit 1"
-  },
-  "keywords": [],
-  "author": "LangWatch",
-  "type": "module",
-  "license": "MIT",
-  "dependencies": {
-    "@ai-sdk/openai": "^2.0.0",
-    "@opentelemetry/context-async-hooks": "^2.0.1",
-    "@opentelemetry/sdk-node": "^0.203.0",
-    "ai": "^5.0.0",
-    "cli-markdown": "^3.5.1",
-    "dotenv-cli": "^10.0.0",
-    "langwatch": "file:../../"
-  },
-  "devDependencies": {
-    "typescript": "^5.9.2"
-  }
-}

package/examples/vercel-ai/src/cli-markdown.d.ts

@@ -1,137 +0,0 @@
-declare module 'cli-markdown' {
-  export interface CliMarkdownOptions {
-    /**
-     * Custom styles for different markdown elements
-     */
-    styles?: {
-      /**
-       * Style for headings (h1, h2, h3, etc.)
-       */
-      heading?: string;
-      /**
-       * Style for bold text
-       */
-      bold?: string;
-      /**
-       * Style for italic text
-       */
-      italic?: string;
-      /**
-       * Style for code blocks
-       */
-      code?: string;
-      /**
-       * Style for inline code
-       */
-      inlineCode?: string;
-      /**
-       * Style for links
-       */
-      link?: string;
-      /**
-       * Style for lists
-       */
-      list?: string;
-      /**
-       * Style for blockquotes
-       */
-      blockquote?: string;
-      /**
-       * Style for horizontal rules
-       */
-      hr?: string;
-    };
-    /**
-     * Whether to preserve line breaks
-     */
-    preserveLineBreaks?: boolean;
-    /**
-     * Custom renderer functions for different markdown elements
-     */
-    renderers?: {
-      /**
-       * Custom renderer for headings
-       */
-      heading?: (text: string, level: number) => string;
-      /**
-       * Custom renderer for paragraphs
-       */
-      paragraph?: (text: string) => string;
-      /**
-       * Custom renderer for lists
-       */
-      list?: (items: string[], ordered: boolean) => string;
-      /**
-       * Custom renderer for list items
-       */
-      listItem?: (text: string, index: number) => string;
-      /**
-       * Custom renderer for code blocks
-       */
-      code?: (code: string, language?: string) => string;
-      /**
-       * Custom renderer for inline code
-       */
-      inlineCode?: (code: string) => string;
-      /**
-       * Custom renderer for links
-       */
-      link?: (text: string, url: string) => string;
-      /**
-       * Custom renderer for images
-       */
-      image?: (alt: string, url: string) => string;
-      /**
-       * Custom renderer for blockquotes
-       */
-      blockquote?: (text: string) => string;
-      /**
-       * Custom renderer for horizontal rules
-       */
-      hr?: () => string;
-      /**
-       * Custom renderer for emphasis (bold/italic)
-       */
-      emphasis?: (text: string, type: 'bold' | 'italic') => string;
-    };
-    /**
-     * Maximum width for text wrapping
-     */
-    maxWidth?: number;
-    /**
-     * Whether to enable ANSI color codes
-     */
-    colors?: boolean;
-    /**
-     * Custom color theme
-     */
-    theme?: {
-      /**
-       * Color for headings
-       */
-      heading?: string;
-      /**
-       * Color for links
-       */
-      link?: string;
-      /**
-       * Color for code
-       */
-      code?: string;
-      /**
-       * Color for blockquotes
-       */
-      blockquote?: string;
-    };
-  }
-
-  /**
-   * Converts markdown text to formatted CLI output
-   * @param markdown - The markdown text to convert
-   * @param options - Configuration options for the conversion
-   * @returns Formatted CLI text
-   */
-  function cliMarkdown(markdown: string, options?: CliMarkdownOptions): string;
-
-  export = cliMarkdown;
-}

package/examples/vercel-ai/src/index.ts

@@ -1,110 +0,0 @@
-import { setupLangWatch } from "langwatch/node";
-import { getLangWatchTracer } from "langwatch";
-import { semconv } from "langwatch/observability";
-import { openai } from "@ai-sdk/openai";
-import { generateText } from "ai";
-import * as readline from "readline";
-import cliMarkdown from "cli-markdown";
-
-await setupLangWatch();
-
-const tracer = getLangWatchTracer("vercel-ai-sdk-example");
-
-async function main() {
-  const threadId = crypto.randomUUID();
-
-  const rl = readline.createInterface({
-    input: process.stdin,
-    output: process.stdout,
-  });
-
-  console.log('🤖 AI Chatbot started! Type "quit" to exit.');
-  console.log("---");
-
-  const conversationHistory: Array<{
-    role: "user" | "assistant" | "system";
-    content: string;
-  }> = [
-    {
-      role: "system",
-      content:
-        "You are a helpful assistant that can answer questions and help with tasks. You may use markdown to format your responses.",
-    },
-  ];
-
-  while (true) {
-    let finish = false;
-
-    await tracer.withActiveSpan("iteration", {
-      attributes: {
-        [semconv.ATTR_LANGWATCH_THREAD_ID]: threadId,
-      },
-    }, async (span) => {
-      try {
-        // Get user input
-        const userInput = await new Promise<string>((resolve) => {
-          rl.question("You: ", resolve);
-        });
-
-        // Check for exit command
-        if (
-          userInput.toLowerCase() === "quit" ||
-          userInput.toLowerCase() === "exit"
-        ) {
-          console.log("👋 Goodbye!");
-          finish = true;
-          return;
-        }
-
-        // Skip empty input
-        if (!userInput.trim()) {
-          return;
-        }
-
-        // Add user message to conversation history
-        conversationHistory.push({ role: "user", content: userInput });
-
-        // Generate AI response
-        console.log("🤖 Thinking...");
-
-        const result = await generateText({
-          model: openai("gpt-4.1-mini"),
-          messages: conversationHistory,
-          experimental_telemetry: { isEnabled: true },
-        });
-
-        const aiResponse = result.text;
-
-        // Add AI response to conversation history
-        conversationHistory.push({ role: "assistant", content: aiResponse });
-
-        // Display AI response with markdown formatting
-        console.log("AI:");
-        console.log(
-          `${cliMarkdown(aiResponse, {
-            colors: true,
-            maxWidth: 80,
-            theme: {
-              heading: "cyan",
-              link: "blue",
-              code: "green",
-              blockquote: "yellow",
-            },
-          })}---`,
-        );
-      } catch (error) {
-        console.error("❌ Error:", error);
-        console.log("Please try again.");
-      }
-    });
-
-    if (finish) {
-      break;
-    }
-  }
-
-  rl.close();
-}
-
-// Run the chatbot
-main().catch(console.error);

package/examples/vercel-ai/src/instrumentation.ts

@@ -1,9 +0,0 @@
-import { registerOTel } from '@vercel/otel';
-
-export function register() {
-  registerOTel({
-    serviceName: 'vercel-ai-sdk-example',
-    // Note: LangWatch SDK automatically sets up the trace exporter,
-    // so no manual configuration is needed here
-  });
-}

package/examples/vercel-ai/tsconfig.json

@@ -1,25 +0,0 @@
-{
-  "compilerOptions": {
-    "target": "ES2022",
-    "module": "node16",
-    "moduleResolution": "node16",
-    "outDir": "./dist",
-    "rootDir": "./src",
-    "strict": true,
-    "esModuleInterop": true,
-    "skipLibCheck": true,
-    "forceConsistentCasingInFileNames": true,
-    "allowSyntheticDefaultImports": true,
-    "resolveJsonModule": true,
-    "declaration": true,
-    "declarationMap": true,
-    "sourceMap": true
-  },
-  "include": [
-    "src/**/*"
-  ],
-  "exclude": [
-    "node_modules",
-    "dist"
-  ]
-}

package/src/__tests__/client-browser.test.ts

@@ -1,92 +0,0 @@
-import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
-
-// Mock window object for Node.js test environment
-const mockWindow = {
-  addEventListener: vi.fn(),
-  removeEventListener: vi.fn(),
-};
-global.window = mockWindow as any;
-
-vi.mock('@opentelemetry/sdk-trace-web', () => ({
-  WebTracerProvider: vi.fn().mockImplementation(function (this: any, opts: any) {
-    this.opts = opts;
-    this.register = vi.fn();
-    this.shutdown = vi.fn();
-  }),
-  BatchSpanProcessor: vi.fn().mockImplementation((exporter) => ({ exporter }))
-}));
-vi.mock('@opentelemetry/exporter-trace-otlp-http', () => ({
-  OTLPTraceExporter: vi.fn().mockImplementation((opts) => ({
-    url: opts.url,
-    headers: opts.headers
-  }))
-}));
-vi.mock('@opentelemetry/context-zone', () => ({
-  ZoneContextManager: vi.fn().mockImplementation(() => ({}))
-}));
-vi.mock('@opentelemetry/propagator-b3', () => ({
-  B3Propagator: vi.fn().mockImplementation(() => ({}))
-}));
-vi.mock('../observability/processors', () => ({
-  FilterableBatchSpanProcessor: vi.fn().mockImplementation((exporter) => ({ exporter }))
-}));
-
-describe('client-browser setup', () => {
-  let clientBrowser: any;
-  let client: any;
-  let WebTracerProvider: any;
-  let FilterableBatchSpanProcessor: any;
-  let OTLPTraceExporter: any;
-
-  beforeEach(async () => {
-    vi.clearAllMocks();
-    vi.resetModules();
-
-    mockWindow.addEventListener.mockClear();
-
-    // Import fresh modules for each test to get clean state
-    clientBrowser = await import('../client-browser.js');
-    client = await import('../client.js');
-
-    // Set config after importing modules
-    client.setConfig({ apiKey: undefined, endpoint: undefined });
-
-    // Import the mocked modules before each test to get fresh spies
-    const sdkTraceWeb = await vi.importMock<any>('@opentelemetry/sdk-trace-web');
-    WebTracerProvider = sdkTraceWeb.WebTracerProvider;
-    const processors = await vi.importMock<any>('../observability/processors');
-    FilterableBatchSpanProcessor = processors.FilterableBatchSpanProcessor;
-    const otlpExporter = await vi.importMock<any>('@opentelemetry/exporter-trace-otlp-http');
-    OTLPTraceExporter = otlpExporter.OTLPTraceExporter;
-  });
-
-  afterEach(() => {
-    // Clean up global state by clearing the module cache
-    vi.resetModules();
-  });
-
-  it('calls setConfig and sets up WebTracerProvider with correct options', async () => {
-    await clientBrowser.setupLangWatch({ apiKey: 'abc', endpoint: 'https://foo', skipOpenTelemetrySetup: false });
-    expect(WebTracerProvider).toHaveBeenCalledTimes(1);
-    const opts = WebTracerProvider.mock.instances[0].opts;
-    expect(opts.spanProcessors[0].exporter.headers.Authorization).toBe('Bearer abc');
-    expect(opts.spanProcessors[0].exporter.url).toContain('/api/otel/v1/traces');
-    expect(mockWindow.addEventListener).toHaveBeenCalledWith('beforeunload', expect.any(Function));
-  });
-
-  it('calls shutdown on existing provider if setup is called again', async () => {
-    await clientBrowser.setupLangWatch({ apiKey: 'abc', endpoint: 'https://foo', skipOpenTelemetrySetup: false });
-    const providerInstance = WebTracerProvider.mock.instances[0];
-
-    // The current implementation prevents multiple setup calls, so we test the shutdown behavior differently
-    // We'll test that the provider instance has a shutdown method that can be called
-    expect(providerInstance.shutdown).toBeDefined();
-    expect(typeof providerInstance.shutdown).toBe('function');
-  });
-
-  it('does not set up WebTracerProvider if skipOpenTelemetrySetup is true', async () => {
-    await clientBrowser.setupLangWatch({ apiKey: 'abc', endpoint: 'https://foo', skipOpenTelemetrySetup: true });
-    expect(WebTracerProvider).not.toHaveBeenCalled();
-    expect(mockWindow.addEventListener).not.toHaveBeenCalled();
-  });
-});

package/src/__tests__/client-node.test.ts

@@ -1,76 +0,0 @@
-import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
-
-// Mock the modules before importing
-vi.mock('@opentelemetry/sdk-node', () => ({
-  NodeSDK: vi.fn().mockImplementation(function (this: any, opts: any) {
-    this.opts = opts;
-    this.start = vi.fn();
-    this.shutdown = vi.fn();
-  })
-}));
-vi.mock('@opentelemetry/sdk-trace-base', () => ({
-  BatchSpanProcessor: vi.fn().mockImplementation((exporter) => ({ exporter }))
-}));
-vi.mock('@opentelemetry/exporter-trace-otlp-http', () => ({
-  OTLPTraceExporter: vi.fn().mockImplementation((opts) => opts)
-}));
-vi.mock('../observability', () => ({
-  FilterableBatchSpanProcessor: vi.fn().mockImplementation((exporter) => ({ exporter }))
-}));
-
-describe('client-node setup', () => {
-  let clientNode: any;
-  let client: any;
-  let NodeSDK: any;
-  let BatchSpanProcessor: any;
-  let OTLPTraceExporter: any;
-
-  beforeEach(async () => {
-    vi.clearAllMocks();
-
-    // Clear module cache to get fresh instances
-    vi.resetModules();
-
-    // Import fresh modules for each test to get clean state
-    clientNode = await import('../client-node.js');
-    client = await import('../client.js');
-
-    client.setConfig({ apiKey: undefined, endpoint: undefined });
-
-    const sdkNode = await vi.importMock<any>('@opentelemetry/sdk-node');
-    NodeSDK = sdkNode.NodeSDK;
-    const sdkTraceBase = await vi.importMock<any>('@opentelemetry/sdk-trace-base');
-    BatchSpanProcessor = sdkTraceBase.BatchSpanProcessor;
-    const otlpExporter = await vi.importMock<any>('@opentelemetry/exporter-trace-otlp-http');
-    OTLPTraceExporter = otlpExporter.OTLPTraceExporter;
-  });
-
-  afterEach(() => {
-    // Clean up global state by clearing the module cache
-    vi.resetModules();
-  });
-
-  it('calls setConfig and sets up NodeSDK with correct options', async () => {
-    await clientNode.setupLangWatch({ apiKey: 'abc', endpoint: 'https://foo', disableOpenTelemetryAutomaticSetup: false });
-    expect(NodeSDK).toHaveBeenCalledTimes(1);
-    const opts = NodeSDK.mock.instances[0].opts;
-    expect(opts.spanProcessors[0].exporter.headers.Authorization).toBe('Bearer abc');
-    expect(opts.spanProcessors[0].exporter.url).toContain('/api/otel/v1/traces');
-  });
-
-  it('calls shutdown on existing sdk if setup is called again', async () => {
-    // First setup
-    await clientNode.setupLangWatch({ apiKey: 'abc', endpoint: 'https://foo', disableOpenTelemetryAutomaticSetup: false });
-    const sdkInstance = NodeSDK.mock.instances[0];
-
-    // The current implementation prevents multiple setup calls, so we test the shutdown behavior differently
-    // We'll test that the SDK instance has a shutdown method that can be called
-    expect(sdkInstance.shutdown).toBeDefined();
-    expect(typeof sdkInstance.shutdown).toBe('function');
-  });
-
-  it('does not set up NodeSDK if skipOpenTelemetrySetup is true', async () => {
-    await clientNode.setupLangWatch({ apiKey: 'abc', endpoint: 'https://foo', skipOpenTelemetrySetup: true });
-    expect(NodeSDK).not.toHaveBeenCalled();
-  });
-});

package/src/__tests__/client.test.ts

@@ -1,71 +0,0 @@
-import { describe, it, expect, beforeEach, afterAll, vi } from "vitest";
-
-describe("client config", () => {
-  let client: any;
-
-  beforeEach(async () => {
-    // Mock environment variables before importing the client module
-    vi.stubEnv('LANGWATCH_API_KEY', undefined);
-    vi.stubEnv('LANGWATCH_ENDPOINT', undefined);
-
-    // Reset modules to ensure fresh import after env stubbing
-    vi.resetModules();
-
-    // Dynamically import the client module
-    client = await import("../client.js");
-
-    // Reset config before each test
-    client.setConfig({
-      apiKey: "",
-      endpoint: void 0,
-      disableOpenTelemetryAutomaticSetup: false,
-      disableAutomaticInputCapture: false,
-      disableAutomaticOutputCapture: false,
-    });
-  });
-
-  afterAll(() => {
-    // Restore original environment
-    vi.unstubAllEnvs();
-  });
-
-  it("should use default values if nothing is set", () => {
-    expect(client.getApiKey()).toBe("");
-    expect(client.getEndpoint()).toBe("https://app.langwatch.ai");
-    expect(client.canAutomaticallyCaptureInput()).toBe(true);
-    expect(client.canAutomaticallyCaptureOutput()).toBe(true);
-  });
-
-  it("should use env vars if set", () => {
-    process.env.LANGWATCH_API_KEY = "env-key";
-    process.env.LANGWATCH_ENDPOINT = "https://env.endpoint";
-
-    client.setConfig({});
-
-    expect(client.getApiKey()).toBe("env-key");
-    expect(client.getEndpoint()).toBe("https://env.endpoint");
-  });
-
-  it("should update config with setConfig", () => {
-    client.setConfig({
-      apiKey: "test-key",
-      endpoint: "https://test.endpoint",
-      disableAutomaticInputCapture: true,
-      disableAutomaticOutputCapture: true,
-    });
-    expect(client.getApiKey()).toBe("test-key");
-    expect(client.getEndpoint()).toBe("https://test.endpoint");
-    expect(client.canAutomaticallyCaptureInput()).toBe(false);
-    expect(client.canAutomaticallyCaptureOutput()).toBe(false);
-  });
-
-  it("should not override values if undefined is passed", () => {
-    client.setConfig({
-      apiKey: "first-key",
-      endpoint: "https://first.endpoint",
-    });
-    client.setConfig({ apiKey: undefined, endpoint: undefined });
-    expect(client.getApiKey()).toBe("first-key");
-    expect(client.getEndpoint()).toBe("https://first.endpoint");
-  });
-});

package/src/__tests__/integration/client-browser.test.ts

@@ -1,46 +0,0 @@
-import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
-import { setupLangWatch as setupBrowser } from '../../client-browser';
-
-// Mock window object for Node.js test environment
-const mockWindow = {
-  addEventListener: vi.fn(),
-  removeEventListener: vi.fn(),
-};
-global.window = mockWindow as any;
-
-describe('client-browser integration', () => {
-  beforeEach(() => {
-    vi.clearAllMocks();
-    mockWindow.addEventListener.mockClear();
-  });
-
-  afterEach(() => {
-    // Clean up global state by clearing the module cache
-    vi.resetModules();
-  });
-
-  it('should start and shut down the WebTracerProvider without error', async () => {
-    // First setup should complete without error
-    await expect(setupBrowser({
-      apiKey: 'integration-key',
-      endpoint: 'http://localhost:9999',
-      skipOpenTelemetrySetup: false,
-    })).resolves.not.toThrow();
-
-    // Second setup should also complete without error (triggers shutdown of previous)
-    await expect(setupBrowser({
-      apiKey: 'integration-key2',
-      endpoint: 'http://localhost:9999',
-      skipOpenTelemetrySetup: false,
-    })).resolves.not.toThrow();
-  });
-
-  it('should not set up WebTracerProvider if skipOpenTelemetrySetup is true', async () => {
-    // Should complete without error and without setting up WebTracerProvider
-    await expect(setupBrowser({
-      apiKey: 'integration-key',
-      endpoint: 'http://localhost:9999',
-      skipOpenTelemetrySetup: true,
-    })).resolves.not.toThrow();
-  });
-});

package/src/__tests__/integration/client-node.test.ts

@@ -1,46 +0,0 @@
-import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
-
-describe('client-node integration', () => {
-  let clientNode: any;
-
-  beforeEach(async () => {
-    vi.clearAllMocks();
-    vi.resetModules();
-    // Import fresh modules for each test to get clean state
-    clientNode = await import('../../client-node.js');
-  });
-
-  afterEach(() => {
-    // Clean up global state by clearing the module cache
-    vi.resetModules();
-  });
-
-  it('should start and shut down the NodeSDK without error', async () => {
-    // First setup should complete without error
-    await expect(clientNode.setupLangWatch({
-      apiKey: 'integration-key',
-      endpoint: 'http://localhost:9999',
-      disableOpenTelemetryAutomaticSetup: false,
-    })).resolves.not.toThrow();
-
-    // Reset module to clear setupCalled state for second call
-    vi.resetModules();
-    clientNode = await import('../../client-node.js');
-
-    // Second setup should also complete without error (triggers shutdown of previous)
-    await expect(clientNode.setupLangWatch({
-      apiKey: 'integration-key2',
-      endpoint: 'http://localhost:9999',
-      disableOpenTelemetryAutomaticSetup: false,
-    })).resolves.not.toThrow();
-  });
-
-  it('should not start NodeSDK if disableOpenTelemetryAutomaticSetup is true', async () => {
-    // Should complete without error and without setting up NodeSDK
-    await expect(clientNode.setupLangWatch({
-      apiKey: 'integration-key',
-      endpoint: 'http://localhost:9999',
-      disableOpenTelemetryAutomaticSetup: true,
-    })).resolves.not.toThrow();
-  });
-});