@uselemma/tracing 4.0.1 → 4.1.0

package/README.md

@@ -1,6 +1,6 @@
 # @uselemma/tracing
 
-HTTP tracing SDK for AI agents. No OpenTelemetry setup is required: the SDK sends completed trace payloads directly to the Lemma API.
+HTTP tracing SDK for AI agents. No OpenTelemetry setup is required: the SDK sends trace payloads directly to the Lemma API.
 
 ## Installation
 
@@ -74,6 +74,7 @@ trace.recordGeneration({
   input: messages,
   output: text,
   model: "gpt-4o",
+  durationMs: measuredModelMs,
   llmInputMessages: [{ role: "user", content: userMessage }],
   llmInvocationParameters: { temperature: 0.2 },
 });
@@ -89,7 +90,7 @@ Use `startSpan()`, `startTool()`, or `startGeneration()` when you want the SDK t
 const span = trace.startSpan({ name: "retrieve-context", input: query });
 try {
   const docs = await retrieve(query);
-  span.end({ output: docs, durationMs: 250 });
+  span.end({ output: docs });
 } catch (error) {
   span.end({ error });
   throw error;
@@ -99,7 +100,7 @@ try {
 ```typescript
 const tool = trace.startTool({ name: "search_docs", input: { query } });
 const docs = await searchDocs(query);
-tool.end({ output: docs, durationMs: 25 });
+tool.end({ output: docs });
 
 const generation = trace.startGeneration({
   name: "answer",
@@ -107,10 +108,10 @@ const generation = trace.startGeneration({
   model: "gpt-4o",
 });
 const response = await callModel(messages);
-generation.end({ output: response.text, durationMs: response.durationMs });
+generation.end({ output: response.text, usage: response.usage });
 ```
 
-The SDK measures trace durations from timestamps. Pass `durationMs` on callback traces, spans, generations, tools, `span.end({ durationMs })`, or `trace.end({ durationMs })` when you already measured the work yourself. When child spans, generations, or tools omit `durationMs`, Lemma splits the parent's remaining unclaimed duration equally across siblings that also omitted duration.
+The SDK measures live handle durations from start and end timestamps. Pass `durationMs` only when replaying historical work or overriding the measured duration with a value from another timer. When child spans, generations, or tools omit `durationMs`, Lemma derives timing from timestamps during ingest.
 
 You can also create a trace handle first and record work on it over time:
 
@@ -127,7 +128,6 @@ span.recordTool({
 });
 span.end({
   output: { count: docs.length },
-  durationMs: 250,
   retrievalDocuments: docs.map((doc) => ({
     id: doc.id,
     content: doc.text,
@@ -135,9 +135,11 @@ span.end({
   })),
 });
 
-await trace.end({ output: "final answer", durationMs: 1234 });
+await trace.end({ output: "final answer" });
 ```
 
+For live trace handles, `trace.end({ output })` is usually enough. Pass `durationMs` to `trace.end()` only when you need to override the measured trace duration.
+
 ## Vercel AI SDK
 
 Pass `vercelAI()` to the AI SDK telemetry integrations option while the call runs inside a Lemma trace. AI SDK v7 uses `telemetry`; AI SDK v6 uses `experimental_telemetry`.
@@ -180,6 +182,35 @@ The integration records model calls as generations and tool executions as tool c
 
 When you pass a trace handle with `vercelAI({ trace })`, the integration ends it from the AI SDK terminal callback: `onEnd` in AI SDK v7 and `onFinish` in AI SDK v6. When you use the callback form of `lemma.trace()`, the callback still owns trace closure.
 
+## OpenAI Agents SDK
+
+Register the Lemma processor with the OpenAI Agents SDK tracing provider:
+
+```typescript
+import { addTraceProcessor } from "@openai/agents";
+import { openAIAgents } from "@uselemma/tracing";
+
+addTraceProcessor(openAIAgents());
+```
+
+The processor creates one Lemma trace for each OpenAI Agents trace. OpenAI
+generation spans become Lemma generations, function spans become Lemma tool
+spans, and other OpenAI Agents spans are preserved as regular spans with the
+original OpenAI trace/span fields in attributes.
+
+Function spans stay nested under their OpenAI parent span. To verify nesting
+locally, enable debug mode and check that the tool span log includes the
+generation span ID as `parentId`:
+
+```typescript
+import { enableDebugMode } from "@uselemma/tracing";
+
+enableDebugMode();
+```
+
+Use `openAIAgents({ recordInputs: false, recordOutputs: false })` to avoid
+sending prompts, tool inputs, tool outputs, or model output text.
+
 When a helper only has IDs, use the client-level methods:
 
 ```typescript
@@ -220,6 +251,49 @@ export function recordSearch(docs: unknown[]) {
 | `projectId` | `LEMMA_PROJECT_ID`   | Required                  |
 | `baseUrl`   | none                 | `https://api.uselemma.ai` |
 
+The SDK sends to `${baseUrl}/traces/ingest`.
+
+You can pass configuration directly to the constructor instead of using
+environment variables:
+
+```typescript
+const lemma = new Lemma({
+  apiKey: "sk_...",
+  projectId: "proj_...",
+  baseUrl: "https://api.uselemma.ai",
+});
+```
+
+## Debug Mode
+
+Debug mode logs trace starts, span starts, span completions, send attempts, and
+send results as they happen:
+
+```typescript
+import { enableDebugMode } from "@uselemma/tracing";
+
+enableDebugMode();
+```
+
+You can also set `LEMMA_DEBUG=true`. Use this when validating that spans arrive
+in the expected order, parent IDs are attached, and the SDK is sending to the
+right URL.
+
+## Supported Contract Fields
+
+Use native SDK props for OpenInference-style fields:
+
+- LLM: `llmModelName`, `llmProvider`, `llmSystem`,
+  `llmInvocationParameters`, `llmInputMessages`, `llmOutputMessages`,
+  `llmTools`, token counts, and prompt template fields
+- tools: `toolName`, `toolDescription`, `toolParameters`
+- retrieval: `retrievalDocuments`
+- embeddings and rerankers: `embeddingModelName`,
+  `embeddingInvocationParameters`, `embeddingEmbeddings`,
+  `rerankerModelName`, `rerankerInputDocuments`, `rerankerOutputDocuments`
+
+Use `attributes` for raw attributes that do not yet have a native SDK prop.
+
 ## Documentation
 
 - [Quickstart](https://docs.uselemma.ai/getting-started/quickstart)
@@ -227,7 +301,8 @@ export function recordSearch(docs: unknown[]) {
 
 ## Examples
 
-See [`examples/`](./examples) for complete callback tracing, trace handle, record-by-ID, and Vercel AI SDK v6/v7 examples.
+See [`examples/`](./examples) for complete callback tracing, trace handle,
+record-by-ID, Vercel AI SDK v6/v7, and OpenAI Agents SDK examples.
 
 ## License
 

package/dist/index.d.ts

@@ -1,4 +1,5 @@
 export { Lemma, NoopSpanHandle, SpanHandle, TraceContext, TraceHandle, active, type DetachedGenerationOptions, type DetachedSpanOptions, type DetachedToolOptions, type GenerationOptions, type JsonValue, type LemmaClientOptions, type SpanOptions, type ToolOptions, type TraceEndOptions, type TraceOptions, } from "./client";
 export { disableDebugMode, enableDebugMode, isDebugModeEnabled, lemmaDebug, } from "./debug-mode";
+export { openAIAgents, type OpenAIAgentsIntegrationOptions, type OpenAIAgentsSpan, type OpenAIAgentsSpanData, type OpenAIAgentsTrace, type OpenAIAgentsTracingProcessor, } from "./openai-agents";
 export { vercelAI, type LemmaVercelAIIntegrationOptions, type VercelAIIntegrationOptions, type VercelAITelemetryIntegration, } from "./vercel-ai";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,EACL,cAAc,EACd,UAAU,EACV,YAAY,EACZ,WAAW,EACX,MAAM,EACN,KAAK,yBAAyB,EAC9B,KAAK,mBAAmB,EACxB,KAAK,mBAAmB,EACxB,KAAK,iBAAiB,EACtB,KAAK,SAAS,EACd,KAAK,kBAAkB,EACvB,KAAK,WAAW,EAChB,KAAK,WAAW,EAChB,KAAK,eAAe,EACpB,KAAK,YAAY,GAClB,MAAM,UAAU,CAAC;AAClB,OAAO,EACL,gBAAgB,EAChB,eAAe,EACf,kBAAkB,EAClB,UAAU,GACX,MAAM,cAAc,CAAC;AACtB,OAAO,EACL,QAAQ,EACR,KAAK,+BAA+B,EACpC,KAAK,0BAA0B,EAC/B,KAAK,4BAA4B,GAClC,MAAM,aAAa,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,EACL,cAAc,EACd,UAAU,EACV,YAAY,EACZ,WAAW,EACX,MAAM,EACN,KAAK,yBAAyB,EAC9B,KAAK,mBAAmB,EACxB,KAAK,mBAAmB,EACxB,KAAK,iBAAiB,EACtB,KAAK,SAAS,EACd,KAAK,kBAAkB,EACvB,KAAK,WAAW,EAChB,KAAK,WAAW,EAChB,KAAK,eAAe,EACpB,KAAK,YAAY,GAClB,MAAM,UAAU,CAAC;AAClB,OAAO,EACL,gBAAgB,EAChB,eAAe,EACf,kBAAkB,EAClB,UAAU,GACX,MAAM,cAAc,CAAC;AACtB,OAAO,EACL,YAAY,EACZ,KAAK,8BAA8B,EACnC,KAAK,gBAAgB,EACrB,KAAK,oBAAoB,EACzB,KAAK,iBAAiB,EACtB,KAAK,4BAA4B,GAClC,MAAM,iBAAiB,CAAC;AACzB,OAAO,EACL,QAAQ,EACR,KAAK,+BAA+B,EACpC,KAAK,0BAA0B,EAC/B,KAAK,4BAA4B,GAClC,MAAM,aAAa,CAAC"}

package/dist/index.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.vercelAI = exports.lemmaDebug = exports.isDebugModeEnabled = exports.enableDebugMode = exports.disableDebugMode = exports.active = exports.TraceHandle = exports.TraceContext = exports.SpanHandle = exports.NoopSpanHandle = exports.Lemma = void 0;
+exports.vercelAI = exports.openAIAgents = exports.lemmaDebug = exports.isDebugModeEnabled = exports.enableDebugMode = exports.disableDebugMode = exports.active = exports.TraceHandle = exports.TraceContext = exports.SpanHandle = exports.NoopSpanHandle = exports.Lemma = void 0;
 var client_1 = require("./client");
 Object.defineProperty(exports, "Lemma", { enumerable: true, get: function () { return client_1.Lemma; } });
 Object.defineProperty(exports, "NoopSpanHandle", { enumerable: true, get: function () { return client_1.NoopSpanHandle; } });
@@ -13,6 +13,8 @@ Object.defineProperty(exports, "disableDebugMode", { enumerable: true, get: func
 Object.defineProperty(exports, "enableDebugMode", { enumerable: true, get: function () { return debug_mode_1.enableDebugMode; } });
 Object.defineProperty(exports, "isDebugModeEnabled", { enumerable: true, get: function () { return debug_mode_1.isDebugModeEnabled; } });
 Object.defineProperty(exports, "lemmaDebug", { enumerable: true, get: function () { return debug_mode_1.lemmaDebug; } });
+var openai_agents_1 = require("./openai-agents");
+Object.defineProperty(exports, "openAIAgents", { enumerable: true, get: function () { return openai_agents_1.openAIAgents; } });
 var vercel_ai_1 = require("./vercel-ai");
 Object.defineProperty(exports, "vercelAI", { enumerable: true, get: function () { return vercel_ai_1.vercelAI; } });
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,mCAiBkB;AAhBhB,+FAAA,KAAK,OAAA;AACL,wGAAA,cAAc,OAAA;AACd,oGAAA,UAAU,OAAA;AACV,sGAAA,YAAY,OAAA;AACZ,qGAAA,WAAW,OAAA;AACX,gGAAA,MAAM,OAAA;AAYR,2CAKsB;AAJpB,8GAAA,gBAAgB,OAAA;AAChB,6GAAA,eAAe,OAAA;AACf,gHAAA,kBAAkB,OAAA;AAClB,wGAAA,UAAU,OAAA;AAEZ,yCAKqB;AAJnB,qGAAA,QAAQ,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,mCAiBkB;AAhBhB,+FAAA,KAAK,OAAA;AACL,wGAAA,cAAc,OAAA;AACd,oGAAA,UAAU,OAAA;AACV,sGAAA,YAAY,OAAA;AACZ,qGAAA,WAAW,OAAA;AACX,gGAAA,MAAM,OAAA;AAYR,2CAKsB;AAJpB,8GAAA,gBAAgB,OAAA;AAChB,6GAAA,eAAe,OAAA;AACf,gHAAA,kBAAkB,OAAA;AAClB,wGAAA,UAAU,OAAA;AAEZ,iDAOyB;AANvB,6GAAA,YAAY,OAAA;AAOd,yCAKqB;AAJnB,qGAAA,QAAQ,OAAA"}

package/dist/openai-agents.d.ts

@@ -0,0 +1,45 @@
+import { Lemma } from "./client";
+export type OpenAIAgentsTrace = {
+    traceId: string;
+    name: string;
+    groupId?: string | null;
+    metadata?: Record<string, unknown>;
+};
+export type OpenAIAgentsSpanData = {
+    type: string;
+    [key: string]: unknown;
+};
+export type OpenAIAgentsSpan = {
+    traceId: string;
+    spanId: string;
+    parentId?: string | null;
+    spanData: OpenAIAgentsSpanData;
+    traceMetadata?: Record<string, unknown>;
+    startedAt?: string | null;
+    endedAt?: string | null;
+    error?: {
+        message?: string;
+        data?: Record<string, unknown>;
+    } | null;
+};
+export type OpenAIAgentsTracingProcessor = {
+    start?: () => void;
+    onTraceStart: (trace: OpenAIAgentsTrace) => Promise<void>;
+    onTraceEnd: (trace: OpenAIAgentsTrace) => Promise<void>;
+    onSpanStart: (span: OpenAIAgentsSpan) => Promise<void>;
+    onSpanEnd: (span: OpenAIAgentsSpan) => Promise<void>;
+    shutdown: (timeout?: number) => Promise<void>;
+    forceFlush: () => Promise<void>;
+};
+export type OpenAIAgentsIntegrationOptions = {
+    lemma?: Lemma;
+    apiKey?: string;
+    projectId?: string;
+    baseUrl?: string;
+    fetch?: typeof fetch;
+    metadata?: Record<string, unknown>;
+    recordInputs?: boolean;
+    recordOutputs?: boolean;
+};
+export declare function openAIAgents(options?: OpenAIAgentsIntegrationOptions): OpenAIAgentsTracingProcessor;
+//# sourceMappingURL=openai-agents.d.ts.map

package/dist/openai-agents.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-agents.d.ts","sourceRoot":"","sources":["../src/openai-agents.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAqC,MAAM,UAAU,CAAC;AAIpE,MAAM,MAAM,iBAAiB,GAAG;IAC9B,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC,CAAC;AAEF,MAAM,MAAM,oBAAoB,GAAG;IACjC,IAAI,EAAE,MAAM,CAAC;IACb,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB,CAAC;AAEF,MAAM,MAAM,gBAAgB,GAAG;IAC7B,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,QAAQ,EAAE,oBAAoB,CAAC;IAC/B,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACxC,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,KAAK,CAAC,EAAE;QACN,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KAChC,GAAG,IAAI,CAAC;CACV,CAAC;AAEF,MAAM,MAAM,4BAA4B,GAAG;IACzC,KAAK,CAAC,EAAE,MAAM,IAAI,CAAC;IACnB,YAAY,EAAE,CAAC,KAAK,EAAE,iBAAiB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1D,UAAU,EAAE,CAAC,KAAK,EAAE,iBAAiB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACxD,WAAW,EAAE,CAAC,IAAI,EAAE,gBAAgB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACvD,SAAS,EAAE,CAAC,IAAI,EAAE,gBAAgB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACrD,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC9C,UAAU,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;CACjC,CAAC;AAEF,MAAM,MAAM,8BAA8B,GAAG;IAC3C,KAAK,CAAC,EAAE,KAAK,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,OAAO,KAAK,CAAC;IACrB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnC,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,aAAa,CAAC,EAAE,OAAO,CAAC;CACzB,CAAC;AAoFF,wBAAgB,YAAY,CAC1B,OAAO,GAAE,8BAAmC,GAC3C,4BAA4B,CA8H9B"}

package/dist/openai-agents.js

@@ -0,0 +1,195 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.openAIAgents = openAIAgents;
+const client_1 = require("./client");
+function parseMaybeJson(value) {
+    if (typeof value !== "string")
+        return value;
+    try {
+        return JSON.parse(value);
+    }
+    catch {
+        return value;
+    }
+}
+function textFromGenerationOutput(output) {
+    if (!Array.isArray(output))
+        return output;
+    const text = output
+        .map((item) => {
+        if (!item || typeof item !== "object")
+            return "";
+        const record = item;
+        if (typeof record["text"] === "string")
+            return record["text"];
+        if (typeof record["content"] === "string")
+            return record["content"];
+        return "";
+    })
+        .join("");
+    return text || output;
+}
+function spanName(data) {
+    if (typeof data["name"] === "string" && data["name"]) {
+        return data["name"];
+    }
+    if (data.type === "generation")
+        return "openai-agents-generation";
+    if (data.type === "response")
+        return "openai-agents-response";
+    if (data.type === "handoff") {
+        const from = typeof data["from_agent"] === "string" ? data["from_agent"] : "";
+        const to = typeof data["to_agent"] === "string" ? data["to_agent"] : "";
+        return from && to ? `${from} to ${to}` : "openai-agents-handoff";
+    }
+    return `openai-agents-${data.type || "span"}`;
+}
+function openAIAttributes(span) {
+    return Object.fromEntries(Object.entries({
+        "openai.agents.trace_id": span.traceId,
+        "openai.agents.span_id": span.spanId,
+        "openai.agents.parent_id": span.parentId,
+        "openai.agents.span_type": span.spanData.type,
+        "openai.agents.trace_metadata": span.traceMetadata
+            ? JSON.stringify(span.traceMetadata)
+            : undefined,
+        "openai.agents.span_data": JSON.stringify(span.spanData),
+    }).filter(([, value]) => value !== undefined && value !== null));
+}
+function usage(data) {
+    const raw = data["usage"];
+    if (!raw || typeof raw !== "object")
+        return undefined;
+    const usageData = raw;
+    return {
+        inputTokens: typeof usageData["input_tokens"] === "number"
+            ? usageData["input_tokens"]
+            : undefined,
+        outputTokens: typeof usageData["output_tokens"] === "number"
+            ? usageData["output_tokens"]
+            : undefined,
+    };
+}
+function startedAt(span) {
+    return span.startedAt ?? new Date();
+}
+function endedAt(span) {
+    return span.endedAt ?? new Date();
+}
+function openAIAgents(options = {}) {
+    const lemma = options.lemma ??
+        new client_1.Lemma({
+            apiKey: options.apiKey,
+            projectId: options.projectId,
+            baseUrl: options.baseUrl,
+            fetch: options.fetch,
+        });
+    const traces = new Map();
+    const spans = new Map();
+    function ensureTrace(trace) {
+        const existing = traces.get(trace.traceId);
+        if (existing)
+            return existing;
+        const handle = lemma.trace({
+            name: trace.name || "openai-agents-trace",
+            metadata: {
+                ...options.metadata,
+                ...(trace.metadata ?? {}),
+                openaiAgentsTraceId: trace.traceId,
+                openaiAgentsGroupId: trace.groupId ?? undefined,
+            },
+            threadId: trace.groupId ?? undefined,
+        });
+        const stored = { handle, ended: false };
+        traces.set(trace.traceId, stored);
+        return stored;
+    }
+    function startSpan(span) {
+        const trace = traces.get(span.traceId);
+        if (!trace)
+            return undefined;
+        const data = span.spanData;
+        const base = {
+            id: span.spanId,
+            parentId: span.parentId ?? null,
+            name: spanName(data),
+            input: options.recordInputs === false
+                ? undefined
+                : parseMaybeJson(data["input"] ?? data["_input"]),
+            metadata: options.metadata,
+            attributes: openAIAttributes(span),
+            startedAt: startedAt(span),
+        };
+        if (data.type === "generation" || data.type === "response") {
+            return trace.handle.startGeneration({
+                ...base,
+                model: typeof data["model"] === "string" ? data["model"] : undefined,
+                llmProvider: "openai",
+                llmInputMessages: options.recordInputs === false || !Array.isArray(data["input"])
+                    ? undefined
+                    : data["input"],
+                llmInvocationParameters: data["model_config"],
+            });
+        }
+        if (data.type === "function") {
+            return trace.handle.startTool({
+                ...base,
+                toolName: typeof data["name"] === "string" ? data["name"] : undefined,
+            });
+        }
+        return trace.handle.startSpan(base);
+    }
+    function endSpan(span) {
+        const handle = spans.get(span.spanId) ?? startSpan(span);
+        if (!handle)
+            return;
+        spans.delete(span.spanId);
+        const data = span.spanData;
+        const output = options.recordOutputs === false
+            ? undefined
+            : data.type === "generation"
+                ? textFromGenerationOutput(data["output"])
+                : parseMaybeJson(data["output"] ?? data["_response"]);
+        handle.end({
+            output,
+            error: span.error?.message,
+            status: span.error ? "ERROR" : undefined,
+            model: typeof data["model"] === "string" ? data["model"] : undefined,
+            usage: usage(data),
+            endedAt: endedAt(span),
+            llmOutputMessages: options.recordOutputs === false || !Array.isArray(data["output"])
+                ? undefined
+                : data["output"],
+        });
+    }
+    async function forEachTrace(fn) {
+        await Promise.all(Array.from(traces.values(), fn));
+    }
+    return {
+        onTraceStart: async (trace) => {
+            ensureTrace(trace);
+        },
+        onTraceEnd: async (trace) => {
+            const stored = ensureTrace(trace);
+            if (stored.ended)
+                return;
+            stored.ended = true;
+            await stored.handle.end();
+        },
+        onSpanStart: async (span) => {
+            const handle = startSpan(span);
+            if (handle)
+                spans.set(span.spanId, handle);
+        },
+        onSpanEnd: async (span) => {
+            endSpan(span);
+        },
+        shutdown: async () => {
+            await forEachTrace((trace) => trace.handle.flush());
+        },
+        forceFlush: async () => {
+            await forEachTrace((trace) => trace.handle.flush());
+        },
+    };
+}
+//# sourceMappingURL=openai-agents.js.map

package/dist/openai-agents.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-agents.js","sourceRoot":"","sources":["../src/openai-agents.ts"],"names":[],"mappings":";;AAqIA,oCAgIC;AArQD,qCAAoE;AAwDpE,SAAS,cAAc,CAAC,KAAc;IACpC,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IAC5C,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAC3B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,SAAS,wBAAwB,CAAC,MAAe;IAC/C,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;QAAE,OAAO,MAAM,CAAC;IAC1C,MAAM,IAAI,GAAG,MAAM;SAChB,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACZ,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ;YAAE,OAAO,EAAE,CAAC;QACjD,MAAM,MAAM,GAAG,IAA+B,CAAC;QAC/C,IAAI,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,QAAQ;YAAE,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC;QAC9D,IAAI,OAAO,MAAM,CAAC,SAAS,CAAC,KAAK,QAAQ;YAAE,OAAO,MAAM,CAAC,SAAS,CAAC,CAAC;QACpE,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC;SACD,IAAI,CAAC,EAAE,CAAC,CAAC;IACZ,OAAO,IAAI,IAAI,MAAM,CAAC;AACxB,CAAC;AAED,SAAS,QAAQ,CAAC,IAA0B;IAC1C,IAAI,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;QACrD,OAAO,IAAI,CAAC,MAAM,CAAC,CAAC;IACtB,CAAC;IACD,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY;QAAE,OAAO,0BAA0B,CAAC;IAClE,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU;QAAE,OAAO,wBAAwB,CAAC;IAC9D,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;QAC5B,MAAM,IAAI,GACR,OAAO,IAAI,CAAC,YAAY,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACnE,MAAM,EAAE,GAAG,OAAO,IAAI,CAAC,UAAU,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACxE,OAAO,IAAI,IAAI,EAAE,CAAC,CAAC,CAAC,GAAG,IAAI,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC,uBAAuB,CAAC;IACnE,CAAC;IACD,OAAO,iBAAiB,IAAI,CAAC,IAAI,IAAI,MAAM,EAAE,CAAC;AAChD,CAAC;AAED,SAAS,gBAAgB,CAAC,IAAsB;IAC9C,OAAO,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC;QACb,wBAAwB,EAAE,IAAI,CAAC,OAAO;QACtC,uBAAuB,EAAE,IAAI,CAAC,MAAM;QACpC,yBAAyB,EAAE,IAAI,CAAC,QAAQ;QACxC,yBAAyB,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI;QAC7C,8BAA8B,EAAE,IAAI,CAAC,aAAa;YAChD,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC;YACpC,CAAC,CAAC,SAAS;QACb,yBAAyB,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC;KACzD,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,CAAC,CAChE,CAAC;AACJ,CAAC;AAED,SAAS,KAAK,CAAC,IAA0B;IACvC,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1B,IAAI,CAAC,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ;QAAE,OAAO,SAAS,CAAC;IACtD,MAAM,SAAS,GAAG,GAA8B,CAAC;IACjD,OAAO;QACL,WAAW,EACT,OAAO,SAAS,CAAC,cAAc,CAAC,KAAK,QAAQ;YAC3C,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC;YAC3B,CAAC,CAAC,SAAS;QACf,YAAY,EACV,OAAO,SAAS,CAAC,eAAe,CAAC,KAAK,QAAQ;YAC5C,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC;YAC5B,CAAC,CAAC,SAAS;KAChB,CAAC;AACJ,CAAC;AAED,SAAS,SAAS,CAAC,IAAsB;IACvC,OAAO,IAAI,CAAC,SAAS,IAAI,IAAI,IAAI,EAAE,CAAC;AACtC,CAAC;AAED,SAAS,OAAO,CAAC,IAAsB;IACrC,OAAO,IAAI,CAAC,OAAO,IAAI,IAAI,IAAI,EAAE,CAAC;AACpC,CAAC;AAED,SAAgB,YAAY,CAC1B,UAA0C,EAAE;IAE5C,MAAM,KAAK,GACT,OAAO,CAAC,KAAK;QACb,IAAI,cAAK,CAAC;YACR,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,KAAK,EAAE,OAAO,CAAC,KAAK;SACrB,CAAC,CAAC;IACL,MAAM,MAAM,GAAG,IAAI,GAAG,EAAuB,CAAC;IAC9C,MAAM,KAAK,GAAG,IAAI,GAAG,EAAsB,CAAC;IAE5C,SAAS,WAAW,CAAC,KAAwB;QAC3C,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC3C,IAAI,QAAQ;YAAE,OAAO,QAAQ,CAAC;QAE9B,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC;YACzB,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,qBAAqB;YACzC,QAAQ,EAAE;gBACR,GAAG,OAAO,CAAC,QAAQ;gBACnB,GAAG,CAAC,KAAK,CAAC,QAAQ,IAAI,EAAE,CAAC;gBACzB,mBAAmB,EAAE,KAAK,CAAC,OAAO;gBAClC,mBAAmB,EAAE,KAAK,CAAC,OAAO,IAAI,SAAS;aAChD;YACD,QAAQ,EAAE,KAAK,CAAC,OAAO,IAAI,SAAS;SACrC,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;QACxC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAClC,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,SAAS,SAAS,CAAC,IAAsB;QACvC,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvC,IAAI,CAAC,KAAK;YAAE,OAAO,SAAS,CAAC;QAE7B,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC3B,MAAM,IAAI,GAAG;YACX,EAAE,EAAE,IAAI,CAAC,MAAM;YACf,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,IAAI;YAC/B,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC;YACpB,KAAK,EACH,OAAO,CAAC,YAAY,KAAK,KAAK;gBAC5B,CAAC,CAAC,SAAS;gBACX,CAAC,CAAC,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,CAAC;YACrD,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,UAAU,EAAE,gBAAgB,CAAC,IAAI,CAAC;YAClC,SAAS,EAAE,SAAS,CAAC,IAAI,CAAC;SAC3B,CAAC;QAEF,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YAC3D,OAAO,KAAK,CAAC,MAAM,CAAC,eAAe,CAAC;gBAClC,GAAG,IAAI;gBACP,KAAK,EAAE,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS;gBACpE,WAAW,EAAE,QAAQ;gBACrB,gBAAgB,EACd,OAAO,CAAC,YAAY,KAAK,KAAK,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;oBAC7D,CAAC,CAAC,SAAS;oBACX,CAAC,CAAE,IAAI,CAAC,OAAO,CAAe;gBAClC,uBAAuB,EAAE,IAAI,CAAC,cAAc,CAAC;aAC9C,CAAC,CAAC;QACL,CAAC;QAED,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YAC7B,OAAO,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC;gBAC5B,GAAG,IAAI;gBACP,QAAQ,EAAE,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS;aACtE,CAAC,CAAC;QACL,CAAC;QAED,OAAO,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IACtC,CAAC;IAED,SAAS,OAAO,CAAC,IAAsB;QACrC,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC;QACzD,IAAI,CAAC,MAAM;YAAE,OAAO;QACpB,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAE1B,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC3B,MAAM,MAAM,GACV,OAAO,CAAC,aAAa,KAAK,KAAK;YAC7B,CAAC,CAAC,SAAS;YACX,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,YAAY;gBAC1B,CAAC,CAAC,wBAAwB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAC1C,CAAC,CAAC,cAAc,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC;QAC5D,MAAM,CAAC,GAAG,CAAC;YACT,MAAM;YACN,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,OAAO;YAC1B,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;YACxC,KAAK,EAAE,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS;YACpE,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC;YAClB,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC;YACtB,iBAAiB,EACf,OAAO,CAAC,aAAa,KAAK,KAAK,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAC/D,CAAC,CAAC,SAAS;gBACX,CAAC,CAAE,IAAI,CAAC,QAAQ,CAAe;SACpC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,UAAU,YAAY,CAAC,EAA8C;QACxE,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;IACrD,CAAC;IAED,OAAO;QACL,YAAY,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YAC5B,WAAW,CAAC,KAAK,CAAC,CAAC;QACrB,CAAC;QACD,UAAU,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YAC1B,MAAM,MAAM,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC;YAClC,IAAI,MAAM,CAAC,KAAK;gBAAE,OAAO;YACzB,MAAM,CAAC,KAAK,GAAG,IAAI,CAAC;YACpB,MAAM,MAAM,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC;QAC5B,CAAC;QACD,WAAW,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YAC1B,MAAM,MAAM,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC;YAC/B,IAAI,MAAM;gBAAE,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC7C,CAAC;QACD,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACxB,OAAO,CAAC,IAAI,CAAC,CAAC;QAChB,CAAC;QACD,QAAQ,EAAE,KAAK,IAAI,EAAE;YACnB,MAAM,YAAY,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;QACtD,CAAC;QACD,UAAU,EAAE,KAAK,IAAI,EAAE;YACrB,MAAM,YAAY,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;QACtD,CAAC;KACF,CAAC;AACJ,CAAC"}

package/examples/openai-agents.ts

@@ -0,0 +1,13 @@
+import { Agent, addTraceProcessor, run } from "@openai/agents";
+import { openAIAgents } from "@uselemma/tracing";
+
+addTraceProcessor(openAIAgents());
+
+const agent = new Agent({
+  name: "support-agent",
+  instructions: "Answer customer questions clearly and concisely.",
+});
+
+const result = await run(agent, "Where is my order?");
+
+console.log(result.finalOutput);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@uselemma/tracing",
-  "version": "4.0.1",
+  "version": "4.1.0",
   "description": "HTTP tracing SDK for Lemma",
   "license": "MIT",
   "author": "Lemma",