@traccia2/sdk 0.0.4 → 0.0.6

package/README.md

@@ -332,11 +332,11 @@ await stopTracing();
 
 #### LangChain with Callback Handler
 
-The SDK provides a `TraciaCallbackHandler` for seamless integration with LangChain:
+The SDK provides a `TracciaCallbackHandler` for seamless integration with LangChain. It extends LangChain's `BaseCallbackHandler` for proper type safety and interface compliance:
 
 ```typescript
-import { startTracing, getTracer } from '@traccia/sdk';
-import { TraciaCallbackHandler } from '@traccia/sdk/integrations';
+import { startTracing, getTracerProvider } from '@traccia/sdk';
+import { TracciaCallbackHandler } from '@traccia/sdk/integrations';
 import { ChatOpenAI } from '@langchain/openai';
 import { AgentExecutor, createOpenAIToolsAgent } from 'langchain/agents';
 
@@ -344,35 +344,170 @@ import { AgentExecutor, createOpenAIToolsAgent } from 'langchain/agents';
 await startTracing({
   apiKey: 'your-api-key',
   enableConsoleExporter: true,
+  enableSpanLogging: true,        // Log spans immediately as they complete
+  maxExportBatchSize: 1,          // Export every span immediately
+  scheduleDelayMs: 100,           // Quick flush interval
 });
 
 // Create callback handler for tracing
-const traciaHandler = new TraciaCallbackHandler();
+const tracciaHandler = new TracciaCallbackHandler();
 
 // Use with LangChain models and agents
 const model = new ChatOpenAI({ 
   modelName: 'gpt-4',
-  callbacks: [traciaHandler],
+  callbacks: [tracciaHandler],
 });
 
 const agent = await createOpenAIToolsAgent({
   llmWithTools: model,
   tools: [yourTools],
-  callbacks: [traciaHandler],
+  callbacks: [tracciaHandler],
 });
 
 const executor = new AgentExecutor({
   agent,
   tools: [yourTools],
-  callbacks: [traciaHandler],
+  callbacks: [tracciaHandler],
 });
 
 // Run with automatic tracing
 const result = await executor.invoke({
   input: 'Your prompt here',
 });
+
+// Flush pending spans to ensure they're exported/logged
+const provider = getTracerProvider();
+await provider.forceFlush();
 ```
 
+**Using with ChatOllama (Local Models):**
+
+If you're using ChatOllama instead of ChatOpenAI for local LLM inference:
+
+```typescript
+import { startTracing, getTracerProvider } from '@traccia/sdk';
+import { TracciaCallbackHandler } from '@traccia/sdk/integrations';
+import { ChatOllama } from '@langchain/ollama';
+
+// Initialize tracing
+await startTracing({
+  enableConsoleExporter: true,
+  enableSpanLogging: true,
+  maxExportBatchSize: 1,
+  scheduleDelayMs: 100,
+});
+
+// Create callback handler
+const tracciaHandler = new TracciaCallbackHandler();
+
+// Use with ChatOllama (local model)
+const model = new ChatOllama({
+  model: 'mistral',                    // or 'llama2', 'neural-chat', etc.
+  baseUrl: 'http://localhost:11434',   // Ollama server
+  callbacks: [tracciaHandler],          // Enable tracing
+});
+
+// Use the model - fully traced!
+const response = await model.invoke({ input: 'Hello, how are you?' });
+console.log(response);
+
+// Flush spans
+const provider = getTracerProvider();
+await provider.forceFlush();
+```
+
+**Seeing Traces in Console:**
+
+To see traces printed to the console immediately, use these configuration options:
+
+```typescript
+await startTracing({
+  enableConsoleExporter: true,    // Export to console
+  enableSpanLogging: true,        // Log each completed span
+  maxExportBatchSize: 1,          // Export immediately (don't batch)
+  scheduleDelayMs: 100,           // Flush frequently
+});
+```
+
+Without these options, spans are batched and exported on a delay (default 5 seconds).
+
+**Troubleshooting - No Traces in Console:**
+
+If you don't see traces in the console, check:
+
+1. **LangChain Version Compatibility:**
+   - TraciaCallbackHandler is compatible with LangChain 0.0.x, 0.1.x, 0.2.x, and 1.x
+   - Verified to work with LangChain 0.2.20
+   - Works with both `ChatOpenAI` and `ChatOllama`
+   - Check your installed version: `npm list @langchain/core`
+   
+   **For ChatOllama (Ollama local models):**
+   ```typescript
+   // Install Ollama integration
+   npm install @langchain/ollama
+   
+   // Use with ChatOllama
+   import { ChatOllama } from '@langchain/ollama';
+   
+   const handler = new TraciaCallbackHandler();
+   const model = new ChatOllama({ 
+     model: 'mistral',
+     baseUrl: 'http://localhost:11434',
+     callbacks: [handler],  // Enable tracing
+   });
+   
+   const response = await model.invoke({ input: 'Hello!' });
+   
+   // Don't forget to flush!
+   const provider = getTracerProvider();
+   await provider.forceFlush();
+   ```
+
+2. **Callback Handler Configuration:**
+   ```typescript
+   // Make sure the handler is passed to ALL components
+   const handler = new TraciaCallbackHandler();
+   
+   const model = new ChatOpenAI({ 
+     callbacks: [handler],  // Required for LLM
+   });
+   
+   const executor = new AgentExecutor({
+     callbacks: [handler],  // Required for executor
+     // ... other options
+   });
+   ```
+
+3. **Force Flush is Critical:**
+   ```typescript
+   // Without this, spans may not be exported!
+   const provider = getTracerProvider();
+   await provider.forceFlush();  // Wait for all spans to flush
+   ```
+
+4. **Check Configuration:**
+   ```typescript
+   // Verify these are set correctly
+   await startTracing({
+     enableSpanLogging: true,      // Logs each span immediately
+     enableConsoleExporter: true,  // Exports to console
+     maxExportBatchSize: 1,        // Export immediately (don't batch)
+     scheduleDelayMs: 100,         // Quick flush
+   });
+   ```
+
+5. **Verify Callbacks Are Being Called:**
+   - Add a console.log in your code to confirm model is being invoked
+   - Check that the handler instance is the same object passed to all components
+   - Ensure no errors are being silently caught
+
+6. **Check Node.js Event Loop:**
+   - Spans are processed asynchronously
+   - Make sure you're not exiting the process immediately
+   - Use `await provider.forceFlush()` before process.exit()
+
+**What Gets Traced:**
+
 The `TraciaCallbackHandler` automatically traces:
 
 - **LLM Calls**: Model name, temperature, max tokens, top-p/k, token counts, latency

package/dist/integrations/auto-langchain.d.ts

@@ -4,19 +4,19 @@
  * Instead of manually passing callbacks to every component, use these
  * convenience functions for automatic instrumentation with zero boilerplate.
  */
-import { TraciaCallbackHandler } from './langchain-callback';
+import { TracciaCallbackHandler } from './langchain-callback';
 /**
  * Get or create the global Traccia callback handler.
  *
  * @example
  * // Instead of:
- * const handler = new TraciaCallbackHandler();
+ * const handler = new TracciaCallbackHandler();
  * const model = new ChatOpenAI({ callbacks: [handler] });
  *
  * // Just do:
  * const model = new ChatOpenAI({ callbacks: [getTraciaHandler()] });
  */
-export declare function getTraciaHandler(): TraciaCallbackHandler;
+export declare function getTraciaHandler(): TracciaCallbackHandler;
 /**
  * Wrap any LangChain model/chain/agent with automatic tracing.
  *
@@ -103,6 +103,6 @@ export declare function setupLangChainWithTracing(options: {
 }): Promise<{
     model: any;
     executor: any;
-    handler: TraciaCallbackHandler;
+    handler: TracciaCallbackHandler;
 }>;
 //# sourceMappingURL=auto-langchain.d.ts.map

package/dist/integrations/auto-langchain.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"auto-langchain.d.ts","sourceRoot":"","sources":["../../src/integrations/auto-langchain.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAO7D;;;;;;;;;;GAUG;AACH,wBAAgB,gBAAgB,IAAI,qBAAqB,CAKxD;AAED;;;;;;;;;GASG;AACH,wBAAgB,WAAW,CAAC,CAAC,SAAS,GAAG,EAAE,SAAS,EAAE,CAAC,GAAG,CAAC,CAkB1D;AAED;;;;;;GAMG;AACH,wBAAsB,kBAAkB,CAAC,MAAM,EAAE,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,CAWlE;AAED;;;;;;;;;;;GAWG;AACH,wBAAsB,yBAAyB,CAAC,OAAO,EAAE;IACvD,KAAK,EAAE,GAAG,CAAC;IACX,KAAK,EAAE,GAAG,EAAE,CAAC;IACb,oBAAoB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC5C,GAAG,OAAO,CAAC,GAAG,CAAC,CAmBf;AAED;;;;;;;;;;GAUG;AACH,wBAAsB,oBAAoB,CAAC,OAAO,EAAE;IAClD,GAAG,EAAE,GAAG,CAAC;IACT,MAAM,EAAE,GAAG,CAAC;CACb,GAAG,OAAO,CAAC,GAAG,CAAC,CAkBf;AAED;;;;;;;;;;;;;GAaG;AACH,wBAAgB,MAAM,CAAC,QAAQ,EAAE,MAAM,IAEnC,SAAS,GAAG,EACZ,aAAa,MAAM,EACnB,YAAY,kBAAkB,wBA0BjC;AAED;;;;;;;;;;;GAWG;AACH,wBAAsB,yBAAyB,CAAC,OAAO,EAAE;IACvD,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAClC,KAAK,CAAC,EAAE,GAAG,EAAE,CAAC;IACd,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,GAAG,OAAO,CAAC;IACV,KAAK,EAAE,GAAG,CAAC;IACX,QAAQ,EAAE,GAAG,CAAC;IACd,OAAO,EAAE,qBAAqB,CAAC;CAChC,CAAC,CA8DD"}
+{"version":3,"file":"auto-langchain.d.ts","sourceRoot":"","sources":["../../src/integrations/auto-langchain.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAO9D;;;;;;;;;;GAUG;AACH,wBAAgB,gBAAgB,IAAI,sBAAsB,CAKzD;AAED;;;;;;;;;GASG;AACH,wBAAgB,WAAW,CAAC,CAAC,SAAS,GAAG,EAAE,SAAS,EAAE,CAAC,GAAG,CAAC,CAkB1D;AAED;;;;;;GAMG;AACH,wBAAsB,kBAAkB,CAAC,MAAM,EAAE,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,CAWlE;AAED;;;;;;;;;;;GAWG;AACH,wBAAsB,yBAAyB,CAAC,OAAO,EAAE;IACvD,KAAK,EAAE,GAAG,CAAC;IACX,KAAK,EAAE,GAAG,EAAE,CAAC;IACb,oBAAoB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC5C,GAAG,OAAO,CAAC,GAAG,CAAC,CAmBf;AAED;;;;;;;;;;GAUG;AACH,wBAAsB,oBAAoB,CAAC,OAAO,EAAE;IAClD,GAAG,EAAE,GAAG,CAAC;IACT,MAAM,EAAE,GAAG,CAAC;CACb,GAAG,OAAO,CAAC,GAAG,CAAC,CAkBf;AAED;;;;;;;;;;;;;GAaG;AACH,wBAAgB,MAAM,CAAC,QAAQ,EAAE,MAAM,IAEnC,SAAS,GAAG,EACZ,aAAa,MAAM,EACnB,YAAY,kBAAkB,wBA0BjC;AAED;;;;;;;;;;;GAWG;AACH,wBAAsB,yBAAyB,CAAC,OAAO,EAAE;IACvD,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAClC,KAAK,CAAC,EAAE,GAAG,EAAE,CAAC;IACd,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,GAAG,OAAO,CAAC;IACV,KAAK,EAAE,GAAG,CAAC;IACX,QAAQ,EAAE,GAAG,CAAC;IACd,OAAO,EAAE,sBAAsB,CAAC;CACjC,CAAC,CA8DD"}

package/dist/integrations/auto-langchain.js

@@ -56,7 +56,7 @@ let globalTraciaHandler = null;
  *
  * @example
  * // Instead of:
- * const handler = new TraciaCallbackHandler();
+ * const handler = new TracciaCallbackHandler();
  * const model = new ChatOpenAI({ callbacks: [handler] });
  *
  * // Just do:
@@ -64,7 +64,7 @@ let globalTraciaHandler = null;
  */
 function getTraciaHandler() {
     if (!globalTraciaHandler) {
-        globalTraciaHandler = new langchain_callback_1.TraciaCallbackHandler();
+        globalTraciaHandler = new langchain_callback_1.TracciaCallbackHandler();
     }
     return globalTraciaHandler;
 }

package/dist/integrations/auto-langchain.js.map

@@ -1 +1 @@
-{"version":3,"file":"auto-langchain.js","sourceRoot":"","sources":["../../src/integrations/auto-langchain.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAoBH,4CAKC;AAYD,kCAkBC;AASD,gDAWC;AAcD,8DAuBC;AAaD,oDAqBC;AAgBD,wBA8BC;AAcD,8DAuEC;AAnRD,6DAA6D;AAE7D;;GAEG;AACH,IAAI,mBAAmB,GAAiC,IAAI,CAAC;AAE7D;;;;;;;;;;GAUG;AACH,SAAgB,gBAAgB;IAC9B,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACzB,mBAAmB,GAAG,IAAI,0CAAqB,EAAE,CAAC;IACpD,CAAC;IACD,OAAO,mBAAmB,CAAC;AAC7B,CAAC;AAED;;;;;;;;;GASG;AACH,SAAgB,WAAW,CAAgB,SAAY;IACrD,MAAM,OAAO,GAAG,gBAAgB,EAAE,CAAC;IAEnC,sDAAsD;IACtD,IAAI,SAAS,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;QAC/C,IAAI,WAAW,IAAI,SAAS,EAAE,CAAC;YAC7B,sCAAsC;YACtC,MAAM,QAAQ,GAAI,SAAiB,CAAC,SAAS,IAAI,EAAE,CAAC;YACnD,SAAiB,CAAC,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACpD,CAAC,CAAC,CAAC,GAAG,QAAQ,EAAE,OAAO,CAAC;gBACxB,CAAC,CAAC,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAC1B,CAAC;aAAM,CAAC;YACN,mCAAmC;YAClC,SAAiB,CAAC,SAAS,GAAG,CAAC,OAAO,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;;;;;GAMG;AACI,KAAK,UAAU,kBAAkB,CAAC,MAAW;IAClD,IAAI,CAAC;QACH,8DAA8D;QAC9D,MAAM,eAAe,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAAC;QACrD,MAAM,KAAK,GAAG,IAAI,eAAe,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QACrD,OAAO,WAAW,CAAC,KAAK,CAAC,CAAC;IAC5B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CACb,+EAA+E,CAChF,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;;;;;;;;;GAWG;AACI,KAAK,UAAU,yBAAyB,CAAC,OAI/C;IACC,IAAI,CAAC;QACH,8DAA8D;QAC9D,MAAM,eAAe,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAAC;QACpD,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,oBAAoB,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC;QAE5D,MAAM,QAAQ,GAAG,eAAe,CAAC,aAAa,CAAC,iBAAiB,CAAC;YAC/D,KAAK;YACL,KAAK;YACL,GAAG,oBAAoB;YACvB,SAAS,EAAE,CAAC,gBAAgB,EAAE,EAAE,GAAG,CAAC,oBAAoB,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC;SAC3E,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC;IAClB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CACb,iFAAiF,CAClF,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;;;;;;;;GAUG;AACI,KAAK,UAAU,oBAAoB,CAAC,OAG1C;IACC,IAAI,CAAC;QACH,8DAA8D;QAC9D,MAAM,eAAe,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAAC;QACpD,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAEhC,MAAM,KAAK,GAAG,IAAI,eAAe,CAAC,QAAQ,CAAC;YACzC,GAAG,EAAE,WAAW,CAAC,GAAG,CAAC;YACrB,MAAM;YACN,SAAS,EAAE,CAAC,gBAAgB,EAAE,CAAC;SAChC,CAAC,CAAC;QAEH,OAAO,KAAK,CAAC;IACf,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CACb,4EAA4E,CAC7E,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;;;;;;;;;;;GAaG;AACH,SAAgB,MAAM,CAAC,QAAgB;IACrC,OAAO,UACL,OAAY,EACZ,WAAmB,EACnB,UAA8B;QAE9B,MAAM,cAAc,GAAG,UAAU,CAAC,KAAK,CAAC;QAExC,UAAU,CAAC,KAAK,GAAG,KAAK,WAAW,GAAG,IAAW;YAC/C,MAAM,EAAE,SAAS,EAAE,GAAG,wDAAa,SAAS,GAAC,CAAC;YAC9C,MAAM,MAAM,GAAG,SAAS,CAAC,kBAAkB,CAAC,CAAC;YAE7C,OAAO,MAAM,CAAC,eAAe,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;gBACrD,IAAI,CAAC;oBACH,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;oBACzC,IAAI,CAAC,YAAY,CAAC,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;oBAC7C,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;oBACtD,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;oBACnC,OAAO,MAAM,CAAC;gBAChB,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;wBAC3B,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;oBAC9B,CAAC;oBACD,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC;QAEF,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;GAWG;AACI,KAAK,UAAU,yBAAyB,CAAC,OAK/C;IAKC,IAAI,CAAC;QACH,MAAM,EACJ,SAAS,GAAG,OAAO,EACnB,WAAW,GAAG,EAAE,EAChB,KAAK,GAAG,EAAE,EACV,YAAY,GACb,GAAG,OAAO,CAAC;QAEZ,sBAAsB;QACtB,MAAM,KAAK,GAAG,MAAM,kBAAkB,CAAC;YACrC,SAAS;YACT,GAAG,WAAW;SACf,CAAC,CAAC;QAEH,iCAAiC;QACjC,IAAI,QAAQ,GAAG,IAAI,CAAC;QACpB,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrB,IAAI,CAAC;gBACH,iCAAiC;gBACjC,MAAM,eAAe,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAAC;gBACpD,MAAM,aAAa,GAAG,OAAO,CAAC,yBAAyB,CAAC,CAAC;gBAEzD,gBAAgB;gBAChB,MAAM,MAAM,GAAG,aAAa,CAAC,kBAAkB,CAAC,YAAY,CAAC;oBAC3D,GAAG,CAAC,YAAY;wBACd,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;wBAC5B,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,8BAA8B,CAAC,CAAC,CAAC;oBACjD,CAAC,OAAO,EAAE,SAAS,CAAC;oBACpB,IAAI,aAAa,CAAC,mBAAmB,CAAC,kBAAkB,CAAC;iBAC1D,CAAC,CAAC;gBAEH,eAAe;gBACf,MAAM,KAAK,GAAG,MAAM,eAAe,CAAC,sBAAsB,CAAC;oBACzD,YAAY,EAAE,KAAK;oBACnB,KAAK;oBACL,MAAM;oBACN,SAAS,EAAE,CAAC,gBAAgB,EAAE,CAAC;iBAChC,CAAC,CAAC;gBAEH,kBAAkB;gBAClB,QAAQ,GAAG,MAAM,yBAAyB,CAAC;oBACzC,KAAK;oBACL,KAAK;iBACN,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,8DAA8D;gBAC9D,OAAO,CAAC,IAAI,CAAC,kCAAkC,EAAG,KAAe,CAAC,OAAO,CAAC,CAAC;YAC7E,CAAC;QACH,CAAC;QAED,OAAO;YACL,KAAK;YACL,QAAQ;YACR,OAAO,EAAE,gBAAgB,EAAE;SAC5B,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,GAAG,GAAG,KAAc,CAAC;QAC3B,MAAM,IAAI,KAAK,CACb,2CAA2C,GAAG,CAAC,OAAO,EAAE,CACzD,CAAC;IACJ,CAAC;AACH,CAAC"}
+{"version":3,"file":"auto-langchain.js","sourceRoot":"","sources":["../../src/integrations/auto-langchain.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAoBH,4CAKC;AAYD,kCAkBC;AASD,gDAWC;AAcD,8DAuBC;AAaD,oDAqBC;AAgBD,wBA8BC;AAcD,8DAuEC;AAnRD,6DAA8D;AAE9D;;GAEG;AACH,IAAI,mBAAmB,GAAkC,IAAI,CAAC;AAE9D;;;;;;;;;;GAUG;AACH,SAAgB,gBAAgB;IAC9B,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACzB,mBAAmB,GAAG,IAAI,2CAAsB,EAAE,CAAC;IACrD,CAAC;IACD,OAAO,mBAAmB,CAAC;AAC7B,CAAC;AAED;;;;;;;;;GASG;AACH,SAAgB,WAAW,CAAgB,SAAY;IACrD,MAAM,OAAO,GAAG,gBAAgB,EAAE,CAAC;IAEnC,sDAAsD;IACtD,IAAI,SAAS,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;QAC/C,IAAI,WAAW,IAAI,SAAS,EAAE,CAAC;YAC7B,sCAAsC;YACtC,MAAM,QAAQ,GAAI,SAAiB,CAAC,SAAS,IAAI,EAAE,CAAC;YACnD,SAAiB,CAAC,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACpD,CAAC,CAAC,CAAC,GAAG,QAAQ,EAAE,OAAO,CAAC;gBACxB,CAAC,CAAC,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAC1B,CAAC;aAAM,CAAC;YACN,mCAAmC;YAClC,SAAiB,CAAC,SAAS,GAAG,CAAC,OAAO,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;;;;;GAMG;AACI,KAAK,UAAU,kBAAkB,CAAC,MAAW;IAClD,IAAI,CAAC;QACH,8DAA8D;QAC9D,MAAM,eAAe,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAAC;QACrD,MAAM,KAAK,GAAG,IAAI,eAAe,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QACrD,OAAO,WAAW,CAAC,KAAK,CAAC,CAAC;IAC5B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CACb,+EAA+E,CAChF,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;;;;;;;;;GAWG;AACI,KAAK,UAAU,yBAAyB,CAAC,OAI/C;IACC,IAAI,CAAC;QACH,8DAA8D;QAC9D,MAAM,eAAe,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAAC;QACpD,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,oBAAoB,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC;QAE5D,MAAM,QAAQ,GAAG,eAAe,CAAC,aAAa,CAAC,iBAAiB,CAAC;YAC/D,KAAK;YACL,KAAK;YACL,GAAG,oBAAoB;YACvB,SAAS,EAAE,CAAC,gBAAgB,EAAE,EAAE,GAAG,CAAC,oBAAoB,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC;SAC3E,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC;IAClB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CACb,iFAAiF,CAClF,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;;;;;;;;GAUG;AACI,KAAK,UAAU,oBAAoB,CAAC,OAG1C;IACC,IAAI,CAAC;QACH,8DAA8D;QAC9D,MAAM,eAAe,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAAC;QACpD,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAEhC,MAAM,KAAK,GAAG,IAAI,eAAe,CAAC,QAAQ,CAAC;YACzC,GAAG,EAAE,WAAW,CAAC,GAAG,CAAC;YACrB,MAAM;YACN,SAAS,EAAE,CAAC,gBAAgB,EAAE,CAAC;SAChC,CAAC,CAAC;QAEH,OAAO,KAAK,CAAC;IACf,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CACb,4EAA4E,CAC7E,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;;;;;;;;;;;GAaG;AACH,SAAgB,MAAM,CAAC,QAAgB;IACrC,OAAO,UACL,OAAY,EACZ,WAAmB,EACnB,UAA8B;QAE9B,MAAM,cAAc,GAAG,UAAU,CAAC,KAAK,CAAC;QAExC,UAAU,CAAC,KAAK,GAAG,KAAK,WAAW,GAAG,IAAW;YAC/C,MAAM,EAAE,SAAS,EAAE,GAAG,wDAAa,SAAS,GAAC,CAAC;YAC9C,MAAM,MAAM,GAAG,SAAS,CAAC,kBAAkB,CAAC,CAAC;YAE7C,OAAO,MAAM,CAAC,eAAe,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;gBACrD,IAAI,CAAC;oBACH,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;oBACzC,IAAI,CAAC,YAAY,CAAC,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;oBAC7C,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;oBACtD,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;oBACnC,OAAO,MAAM,CAAC;gBAChB,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;wBAC3B,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;oBAC9B,CAAC;oBACD,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC;QAEF,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;GAWG;AACI,KAAK,UAAU,yBAAyB,CAAC,OAK/C;IAKC,IAAI,CAAC;QACH,MAAM,EACJ,SAAS,GAAG,OAAO,EACnB,WAAW,GAAG,EAAE,EAChB,KAAK,GAAG,EAAE,EACV,YAAY,GACb,GAAG,OAAO,CAAC;QAEZ,sBAAsB;QACtB,MAAM,KAAK,GAAG,MAAM,kBAAkB,CAAC;YACrC,SAAS;YACT,GAAG,WAAW;SACf,CAAC,CAAC;QAEH,iCAAiC;QACjC,IAAI,QAAQ,GAAG,IAAI,CAAC;QACpB,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrB,IAAI,CAAC;gBACH,iCAAiC;gBACjC,MAAM,eAAe,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAAC;gBACpD,MAAM,aAAa,GAAG,OAAO,CAAC,yBAAyB,CAAC,CAAC;gBAEzD,gBAAgB;gBAChB,MAAM,MAAM,GAAG,aAAa,CAAC,kBAAkB,CAAC,YAAY,CAAC;oBAC3D,GAAG,CAAC,YAAY;wBACd,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;wBAC5B,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,8BAA8B,CAAC,CAAC,CAAC;oBACjD,CAAC,OAAO,EAAE,SAAS,CAAC;oBACpB,IAAI,aAAa,CAAC,mBAAmB,CAAC,kBAAkB,CAAC;iBAC1D,CAAC,CAAC;gBAEH,eAAe;gBACf,MAAM,KAAK,GAAG,MAAM,eAAe,CAAC,sBAAsB,CAAC;oBACzD,YAAY,EAAE,KAAK;oBACnB,KAAK;oBACL,MAAM;oBACN,SAAS,EAAE,CAAC,gBAAgB,EAAE,CAAC;iBAChC,CAAC,CAAC;gBAEH,kBAAkB;gBAClB,QAAQ,GAAG,MAAM,yBAAyB,CAAC;oBACzC,KAAK;oBACL,KAAK;iBACN,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,8DAA8D;gBAC9D,OAAO,CAAC,IAAI,CAAC,kCAAkC,EAAG,KAAe,CAAC,OAAO,CAAC,CAAC;YAC7E,CAAC;QACH,CAAC;QAED,OAAO;YACL,KAAK;YACL,QAAQ;YACR,OAAO,EAAE,gBAAgB,EAAE;SAC5B,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,GAAG,GAAG,KAAc,CAAC;QAC3B,MAAM,IAAI,KAAK,CACb,2CAA2C,GAAG,CAAC,OAAO,EAAE,CACzD,CAAC;IACJ,CAAC;AACH,CAAC"}

package/dist/integrations/index.d.ts

@@ -4,7 +4,7 @@
  *
  * @module integrations
  */
-export { TraciaCallbackHandler } from './langchain-callback';
+export { TracciaCallbackHandler } from './langchain-callback';
 export { getTraciaHandler, withTracing, createTracedOpenAI, createTracedAgentExecutor, createTracedLLMChain, traced, setupLangChainWithTracing, } from './auto-langchain';
 export { instrumentLangGraph, createTracedNode, createTracedConditional, } from './langgraph-instrumentation';
 export { wrapGraphWithTracing, tracedNode, tracedConditional, createSimpleTracedGraph, traceableFunction, createAgentWorkflow, } from './auto-langgraph';

package/dist/integrations/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/integrations/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAC7D,OAAO,EACL,gBAAgB,EAChB,WAAW,EACX,kBAAkB,EAClB,yBAAyB,EACzB,oBAAoB,EACpB,MAAM,EACN,yBAAyB,GAC1B,MAAM,kBAAkB,CAAC;AAG1B,OAAO,EACL,mBAAmB,EACnB,gBAAgB,EAChB,uBAAuB,GACxB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EACL,oBAAoB,EACpB,UAAU,EACV,iBAAiB,EACjB,uBAAuB,EACvB,iBAAiB,EACjB,mBAAmB,GACpB,MAAM,kBAAkB,CAAC;AAG1B,OAAO,EACL,uBAAuB,EACvB,sBAAsB,EACtB,mBAAmB,EACnB,4BAA4B,EAC5B,0BAA0B,EAC1B,qBAAqB,GACtB,MAAM,sBAAsB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/integrations/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EACL,gBAAgB,EAChB,WAAW,EACX,kBAAkB,EAClB,yBAAyB,EACzB,oBAAoB,EACpB,MAAM,EACN,yBAAyB,GAC1B,MAAM,kBAAkB,CAAC;AAG1B,OAAO,EACL,mBAAmB,EACnB,gBAAgB,EAChB,uBAAuB,GACxB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EACL,oBAAoB,EACpB,UAAU,EACV,iBAAiB,EACjB,uBAAuB,EACvB,iBAAiB,EACjB,mBAAmB,GACpB,MAAM,kBAAkB,CAAC;AAG1B,OAAO,EACL,uBAAuB,EACvB,sBAAsB,EACtB,mBAAmB,EACnB,4BAA4B,EAC5B,0BAA0B,EAC1B,qBAAqB,GACtB,MAAM,sBAAsB,CAAC"}

package/dist/integrations/index.js

@@ -6,10 +6,10 @@
  * @module integrations
  */
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.POPULAR_OLLAMA_MODELS = exports.getOllamaSetupInstructions = exports.createOllamaStreamingChatbot = exports.createOllamaChatbot = exports.setupOllamaWithTracing = exports.createOllamaWithTracing = exports.createAgentWorkflow = exports.traceableFunction = exports.createSimpleTracedGraph = exports.tracedConditional = exports.tracedNode = exports.wrapGraphWithTracing = exports.createTracedConditional = exports.createTracedNode = exports.instrumentLangGraph = exports.setupLangChainWithTracing = exports.traced = exports.createTracedLLMChain = exports.createTracedAgentExecutor = exports.createTracedOpenAI = exports.withTracing = exports.getTraciaHandler = exports.TraciaCallbackHandler = void 0;
+exports.POPULAR_OLLAMA_MODELS = exports.getOllamaSetupInstructions = exports.createOllamaStreamingChatbot = exports.createOllamaChatbot = exports.setupOllamaWithTracing = exports.createOllamaWithTracing = exports.createAgentWorkflow = exports.traceableFunction = exports.createSimpleTracedGraph = exports.tracedConditional = exports.tracedNode = exports.wrapGraphWithTracing = exports.createTracedConditional = exports.createTracedNode = exports.instrumentLangGraph = exports.setupLangChainWithTracing = exports.traced = exports.createTracedLLMChain = exports.createTracedAgentExecutor = exports.createTracedOpenAI = exports.withTracing = exports.getTraciaHandler = exports.TracciaCallbackHandler = void 0;
 // LangChain integrations
 var langchain_callback_1 = require("./langchain-callback");
-Object.defineProperty(exports, "TraciaCallbackHandler", { enumerable: true, get: function () { return langchain_callback_1.TraciaCallbackHandler; } });
+Object.defineProperty(exports, "TracciaCallbackHandler", { enumerable: true, get: function () { return langchain_callback_1.TracciaCallbackHandler; } });
 var auto_langchain_1 = require("./auto-langchain");
 Object.defineProperty(exports, "getTraciaHandler", { enumerable: true, get: function () { return auto_langchain_1.getTraciaHandler; } });
 Object.defineProperty(exports, "withTracing", { enumerable: true, get: function () { return auto_langchain_1.withTracing; } });

package/dist/integrations/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/integrations/index.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAEH,yBAAyB;AACzB,2DAA6D;AAApD,2HAAA,qBAAqB,OAAA;AAC9B,mDAQ0B;AAPxB,kHAAA,gBAAgB,OAAA;AAChB,6GAAA,WAAW,OAAA;AACX,oHAAA,kBAAkB,OAAA;AAClB,2HAAA,yBAAyB,OAAA;AACzB,sHAAA,oBAAoB,OAAA;AACpB,wGAAA,MAAM,OAAA;AACN,2HAAA,yBAAyB,OAAA;AAG3B,yBAAyB;AACzB,yEAIqC;AAHnC,gIAAA,mBAAmB,OAAA;AACnB,6HAAA,gBAAgB,OAAA;AAChB,oIAAA,uBAAuB,OAAA;AAEzB,mDAO0B;AANxB,sHAAA,oBAAoB,OAAA;AACpB,4GAAA,UAAU,OAAA;AACV,mHAAA,iBAAiB,OAAA;AACjB,yHAAA,uBAAuB,OAAA;AACvB,mHAAA,iBAAiB,OAAA;AACjB,qHAAA,mBAAmB,OAAA;AAGrB,sBAAsB;AACtB,2DAO8B;AAN5B,6HAAA,uBAAuB,OAAA;AACvB,4HAAA,sBAAsB,OAAA;AACtB,yHAAA,mBAAmB,OAAA;AACnB,kIAAA,4BAA4B,OAAA;AAC5B,gIAAA,0BAA0B,OAAA;AAC1B,2HAAA,qBAAqB,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/integrations/index.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAEH,yBAAyB;AACzB,2DAA8D;AAArD,4HAAA,sBAAsB,OAAA;AAC/B,mDAQ0B;AAPxB,kHAAA,gBAAgB,OAAA;AAChB,6GAAA,WAAW,OAAA;AACX,oHAAA,kBAAkB,OAAA;AAClB,2HAAA,yBAAyB,OAAA;AACzB,sHAAA,oBAAoB,OAAA;AACpB,wGAAA,MAAM,OAAA;AACN,2HAAA,yBAAyB,OAAA;AAG3B,yBAAyB;AACzB,yEAIqC;AAHnC,gIAAA,mBAAmB,OAAA;AACnB,6HAAA,gBAAgB,OAAA;AAChB,oIAAA,uBAAuB,OAAA;AAEzB,mDAO0B;AANxB,sHAAA,oBAAoB,OAAA;AACpB,4GAAA,UAAU,OAAA;AACV,mHAAA,iBAAiB,OAAA;AACjB,yHAAA,uBAAuB,OAAA;AACvB,mHAAA,iBAAiB,OAAA;AACjB,qHAAA,mBAAmB,OAAA;AAGrB,sBAAsB;AACtB,2DAO8B;AAN5B,6HAAA,uBAAuB,OAAA;AACvB,4HAAA,sBAAsB,OAAA;AACtB,yHAAA,mBAAmB,OAAA;AACnB,kIAAA,4BAA4B,OAAA;AAC5B,gIAAA,0BAA0B,OAAA;AAC1B,2HAAA,qBAAqB,OAAA"}

package/dist/integrations/langchain-callback.d.ts

@@ -3,27 +3,29 @@
  * Integrates with LangChain's callback system to automatically instrument
  * LLM calls, chains, agents, and tools.
  */
+import { BaseCallbackHandler } from '@langchain/core/callbacks/base';
 /**
  * LangChain Callback Handler for Traccia SDK.
  * Automatically traces LLM calls, chains, agents, and tools.
  *
+ * Extends LangChain's BaseCallbackHandler for proper interface compliance.
+ * Compatible with LangChain 0.0.x, 0.1.x, 0.2.x, and 1.x versions.
+ *
  * @example
- * import { ChatOpenAI } from 'langchain/chat_models/openai';
- * import { LLMChain } from 'langchain/chains';
- * import { TraciaCallbackHandler } from '@traccia/sdk/integrations/langchain';
+ * import { ChatOpenAI } from '@langchain/openai';
+ * import { TracciaCallbackHandler } from '@traccia/sdk/integrations';
  *
- * const handler = new TraciaCallbackHandler();
- * const chain = new LLMChain({
- *   llm: new ChatOpenAI(),
- *   callbacks: [handler],
- * });
+ * const handler = new TracciaCallbackHandler();
+ * const model = new ChatOpenAI({ callbacks: [handler] });
  *
- * await chain.run({ input: 'What is 2+2?' });
- * // Automatically traced with spans for LLM, chain execution, tokens, etc.
+ * const response = await model.invoke({ input: 'Hello!' });
+ * // Automatically traced with spans for LLM calls, tokens, latency, etc.
  */
-export declare class TraciaCallbackHandler {
+export declare class TracciaCallbackHandler extends BaseCallbackHandler {
+    name: string;
     private tracer;
     private spanStack;
+    private streamingStartTimes;
     /**
      * Extract model name from LLM instance, checking multiple property locations.
      * Different LLM implementations store the model name in different properties.
@@ -41,6 +43,11 @@ export declare class TraciaCallbackHandler {
      * Handle LLM error.
      */
     handleLLMError(error: Error, runId: string): Promise<void>;
+    /**
+     * Handle LLM new token - called when a new token is generated during streaming.
+     * Tracks first token latency and token count for streaming scenarios.
+     */
+    handleLLMNewToken(_token: string, _idx?: any, runId?: string): Promise<void>;
     /**
      * Handle chain start - called when a chain begins execution.
      */
@@ -73,5 +80,17 @@ export declare class TraciaCallbackHandler {
      * Handle agent finish.
      */
     handleAgentFinish(finish: any, runId: string): Promise<void>;
+    onLLMStart(llm: any, prompts: string[], runId: string, parentRunId?: string): Promise<void>;
+    onLLMEnd(output: any, runId: string): Promise<void>;
+    onLLMError(error: Error, runId: string): Promise<void>;
+    onLLMNewToken(_token: string, idx?: any, runId?: string): Promise<void>;
+    onChainStart(chain: any, inputs: any, runId: string, parentRunId?: string): Promise<void>;
+    onChainEnd(output: any, runId: string): Promise<void>;
+    onChainError(error: Error, runId: string): Promise<void>;
+    onToolStart(tool: any, input: string, runId: string, parentRunId?: string): Promise<void>;
+    onToolEnd(output: string, runId: string): Promise<void>;
+    onToolError(error: Error, runId: string): Promise<void>;
+    onAgentAction(action: any, runId: string): Promise<void>;
+    onAgentFinish(finish: any, runId: string): Promise<void>;
 }
 //# sourceMappingURL=langchain-callback.d.ts.map

package/dist/integrations/langchain-callback.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"langchain-callback.d.ts","sourceRoot":"","sources":["../../src/integrations/langchain-callback.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAKH;;;;;;;;;;;;;;;;;GAiBG;AACH,qBAAa,qBAAqB;IAChC,OAAO,CAAC,MAAM,CAA0B;IACxC,OAAO,CAAC,SAAS,CAAiC;IAElD;;;OAGG;IACH,OAAO,CAAC,gBAAgB;IAUxB;;OAEG;IACU,cAAc,CACzB,GAAG,EAAE,GAAG,EACR,OAAO,EAAE,MAAM,EAAE,EACjB,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GACpB,OAAO,CAAC,IAAI,CAAC;IA0ChB;;OAEG;IACU,YAAY,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IA8CpE;;OAEG;IACU,cAAc,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IASvE;;OAEG;IACU,gBAAgB,CAC3B,KAAK,EAAE,GAAG,EACV,MAAM,EAAE,GAAG,EACX,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GACpB,OAAO,CAAC,IAAI,CAAC;IAuBhB;;OAEG;IACU,cAAc,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAiBtE;;OAEG;IACU,gBAAgB,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IASzE;;OAEG;IACU,eAAe,CAC1B,IAAI,EAAE,GAAG,EACT,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GACpB,OAAO,CAAC,IAAI,CAAC;IAuBhB;;OAEG;IACU,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAcxE;;OAEG;IACU,eAAe,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IASxE;;OAEG;IACU,iBAAiB,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAWzE;;OAEG;IACU,iBAAiB,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAa1E"}
+{"version":3,"file":"langchain-callback.d.ts","sourceRoot":"","sources":["../../src/integrations/langchain-callback.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,OAAO,EAAE,mBAAmB,EAAE,MAAM,gCAAgC,CAAC;AAErE;;;;;;;;;;;;;;;;GAgBG;AACH,qBAAa,sBAAuB,SAAQ,mBAAmB;IAC7D,IAAI,SAA4B;IAChC,OAAO,CAAC,MAAM,CAA0B;IACxC,OAAO,CAAC,SAAS,CAAiC;IAClD,OAAO,CAAC,mBAAmB,CAA4B;IAEvD;;;OAGG;IACH,OAAO,CAAC,gBAAgB;IAUxB;;OAEG;IACU,cAAc,CACzB,GAAG,EAAE,GAAG,EACR,OAAO,EAAE,MAAM,EAAE,EACjB,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GACpB,OAAO,CAAC,IAAI,CAAC;IA0ChB;;OAEG;IACU,YAAY,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IA+EpE;;OAEG;IACU,cAAc,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IASvE;;;OAGG;IACU,iBAAiB,CAC5B,MAAM,EAAE,MAAM,EACd,IAAI,CAAC,EAAE,GAAG,EACV,KAAK,CAAC,EAAE,MAAM,GACb,OAAO,CAAC,IAAI,CAAC;IAehB;;OAEG;IACU,gBAAgB,CAC3B,KAAK,EAAE,GAAG,EACV,MAAM,EAAE,GAAG,EACX,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GACpB,OAAO,CAAC,IAAI,CAAC;IAuBhB;;OAEG;IACU,cAAc,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAiBtE;;OAEG;IACU,gBAAgB,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IASzE;;OAEG;IACU,eAAe,CAC1B,IAAI,EAAE,GAAG,EACT,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GACpB,OAAO,CAAC,IAAI,CAAC;IAuBhB;;OAEG;IACU,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAcxE;;OAEG;IACU,eAAe,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IASxE;;OAEG;IACU,iBAAiB,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAWzE;;OAEG;IACU,iBAAiB,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAgB5D,UAAU,CACrB,GAAG,EAAE,GAAG,EACR,OAAO,EAAE,MAAM,EAAE,EACjB,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,GACnB,OAAO,CAAC,IAAI,CAAC;IAIH,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAInD,UAAU,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAItD,aAAa,CACxB,MAAM,EAAE,MAAM,EACd,GAAG,CAAC,EAAE,GAAG,EACT,KAAK,CAAC,EAAE,MAAM,GACb,OAAO,CAAC,IAAI,CAAC;IAIH,YAAY,CACvB,KAAK,EAAE,GAAG,EACV,MAAM,EAAE,GAAG,EACX,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,GACnB,OAAO,CAAC,IAAI,CAAC;IAIH,UAAU,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIrD,YAAY,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIxD,WAAW,CACtB,IAAI,EAAE,GAAG,EACT,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,GACnB,OAAO,CAAC,IAAI,CAAC;IAIH,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIvD,WAAW,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIvD,aAAa,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIxD,aAAa,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAGtE"}

package/dist/integrations/langchain-callback.js

@@ -5,30 +5,33 @@
  * LLM calls, chains, agents, and tools.
  */
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.TraciaCallbackHandler = void 0;
+exports.TracciaCallbackHandler = void 0;
 const auto_1 = require("../auto");
+const base_1 = require("@langchain/core/callbacks/base");
 /**
  * LangChain Callback Handler for Traccia SDK.
  * Automatically traces LLM calls, chains, agents, and tools.
  *
+ * Extends LangChain's BaseCallbackHandler for proper interface compliance.
+ * Compatible with LangChain 0.0.x, 0.1.x, 0.2.x, and 1.x versions.
+ *
  * @example
- * import { ChatOpenAI } from 'langchain/chat_models/openai';
- * import { LLMChain } from 'langchain/chains';
- * import { TraciaCallbackHandler } from '@traccia/sdk/integrations/langchain';
+ * import { ChatOpenAI } from '@langchain/openai';
+ * import { TracciaCallbackHandler } from '@traccia/sdk/integrations';
  *
- * const handler = new TraciaCallbackHandler();
- * const chain = new LLMChain({
- *   llm: new ChatOpenAI(),
- *   callbacks: [handler],
- * });
+ * const handler = new TracciaCallbackHandler();
+ * const model = new ChatOpenAI({ callbacks: [handler] });
  *
- * await chain.run({ input: 'What is 2+2?' });
- * // Automatically traced with spans for LLM, chain execution, tokens, etc.
+ * const response = await model.invoke({ input: 'Hello!' });
+ * // Automatically traced with spans for LLM calls, tokens, latency, etc.
  */
-class TraciaCallbackHandler {
+class TracciaCallbackHandler extends base_1.BaseCallbackHandler {
     constructor() {
+        super(...arguments);
+        this.name = 'TracciaCallbackHandler';
         this.tracer = (0, auto_1.getTracer)('langchain');
         this.spanStack = new Map();
+        this.streamingStartTimes = {};
     }
     /**
      * Extract model name from LLM instance, checking multiple property locations.
@@ -93,20 +96,42 @@ class TraciaCallbackHandler {
         if (span) {
             try {
                 // Try multiple ways to get token usage
+                // OpenAI format and new @langchain/core format
                 const tokenUsage = output?.llmOutput?.token_usage ||
+                    output?.llmOutput?.tokenUsage ||
                     output?.token_usage ||
                     output?.metadata?.token_usage;
                 if (tokenUsage) {
-                    if (tokenUsage.prompt_tokens !== undefined) {
-                        span.setAttribute('llm.tokens.prompt', tokenUsage.prompt_tokens);
+                    // Handle standard token counts
+                    const promptTokens = tokenUsage.prompt_tokens ?? tokenUsage.promptTokens;
+                    const completionTokens = tokenUsage.completion_tokens ?? tokenUsage.completionTokens;
+                    const totalTokens = tokenUsage.total_tokens ?? tokenUsage.totalTokens;
+                    if (promptTokens !== undefined) {
+                        span.setAttribute('llm.tokens.prompt', promptTokens);
                     }
-                    if (tokenUsage.completion_tokens !== undefined) {
-                        span.setAttribute('llm.tokens.completion', tokenUsage.completion_tokens);
+                    if (completionTokens !== undefined) {
+                        span.setAttribute('llm.tokens.completion', completionTokens);
                     }
-                    const totalTokens = (tokenUsage.prompt_tokens || 0) + (tokenUsage.completion_tokens || 0);
-                    if (totalTokens > 0) {
+                    if (totalTokens !== undefined) {
                         span.setAttribute('llm.tokens.total', totalTokens);
                     }
+                    // Handle detailed token breakdown for models like GPT-4o vision
+                    // input_token_details contains breakdown of prompt token usage
+                    if (tokenUsage.input_token_details && typeof tokenUsage.input_token_details === 'object') {
+                        for (const [key, value] of Object.entries(tokenUsage.input_token_details)) {
+                            if (typeof value === 'number') {
+                                span.setAttribute(`llm.tokens.input_${key}`, value);
+                            }
+                        }
+                    }
+                    // output_token_details contains breakdown of completion token usage
+                    if (tokenUsage.output_token_details && typeof tokenUsage.output_token_details === 'object') {
+                        for (const [key, value] of Object.entries(tokenUsage.output_token_details)) {
+                            if (typeof value === 'number') {
+                                span.setAttribute(`llm.tokens.output_${key}`, value);
+                            }
+                        }
+                    }
                 }
                 // Capture output text length
                 if (output?.text) {
@@ -118,6 +143,16 @@ class TraciaCallbackHandler {
                         span.setAttribute('output_length', firstGeneration[0].text.length);
                     }
                 }
+                else if (output?.message?.content) {
+                    // Ollama format (uses message.content)
+                    const content = output.message.content;
+                    const contentStr = typeof content === 'string' ? content : JSON.stringify(content);
+                    span.setAttribute('output_length', contentStr.length);
+                }
+                else if (typeof output === 'string') {
+                    // Direct string output
+                    span.setAttribute('output_length', output.length);
+                }
                 // Capture finish reason if available
                 if (output?.llmOutput?.finish_reason) {
                     span.setAttribute('finish_reason', output.llmOutput.finish_reason);
@@ -141,6 +176,25 @@ class TraciaCallbackHandler {
             this.spanStack.delete(runId);
         }
     }
+    /**
+     * Handle LLM new token - called when a new token is generated during streaming.
+     * Tracks first token latency and token count for streaming scenarios.
+     */
+    async handleLLMNewToken(_token, _idx, runId) {
+        if (runId && !(runId in this.streamingStartTimes)) {
+            // Record the time of the first streaming token
+            this.streamingStartTimes[runId] = new Date();
+            const span = this.spanStack.get(runId);
+            if (span) {
+                try {
+                    span.setAttribute('stream.first_token_generated', true);
+                }
+                catch (error) {
+                    // Silently fail
+                }
+            }
+        }
+    }
     /**
      * Handle chain start - called when a chain begins execution.
      */
@@ -274,6 +328,44 @@ class TraciaCallbackHandler {
             this.spanStack.delete(runId);
         }
     }
+    // LangChain uses 'on*' prefix for callback methods
+    // Provide aliases for compatibility
+    async onLLMStart(llm, prompts, runId, parentRunId) {
+        return this.handleLLMStart(llm, prompts, runId, parentRunId);
+    }
+    async onLLMEnd(output, runId) {
+        return this.handleLLMEnd(output, runId);
+    }
+    async onLLMError(error, runId) {
+        return this.handleLLMError(error, runId);
+    }
+    async onLLMNewToken(_token, idx, runId) {
+        return this.handleLLMNewToken(_token, idx, runId);
+    }
+    async onChainStart(chain, inputs, runId, parentRunId) {
+        return this.handleChainStart(chain, inputs, runId, parentRunId);
+    }
+    async onChainEnd(output, runId) {
+        return this.handleChainEnd(output, runId);
+    }
+    async onChainError(error, runId) {
+        return this.handleChainError(error, runId);
+    }
+    async onToolStart(tool, input, runId, parentRunId) {
+        return this.handleToolStart(tool, input, runId, parentRunId);
+    }
+    async onToolEnd(output, runId) {
+        return this.handleToolEnd(output, runId);
+    }
+    async onToolError(error, runId) {
+        return this.handleToolError(error, runId);
+    }
+    async onAgentAction(action, runId) {
+        return this.handleAgentAction(action, runId);
+    }
+    async onAgentFinish(finish, runId) {
+        return this.handleAgentFinish(finish, runId);
+    }
 }
-exports.TraciaCallbackHandler = TraciaCallbackHandler;
+exports.TracciaCallbackHandler = TracciaCallbackHandler;
 //# sourceMappingURL=langchain-callback.js.map

package/dist/integrations/langchain-callback.js.map

@@ -1 +1 @@
-{"version":3,"file":"langchain-callback.js","sourceRoot":"","sources":["../../src/integrations/langchain-callback.ts"],"names":[],"mappings":";AAAA;;;;GAIG;;;AAGH,kCAAoC;AAEpC;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAa,qBAAqB;IAAlC;QACU,WAAM,GAAG,IAAA,gBAAS,EAAC,WAAW,CAAC,CAAC;QAChC,cAAS,GAAuB,IAAI,GAAG,EAAE,CAAC;IAwRpD,CAAC;IAtRC;;;OAGG;IACK,gBAAgB,CAAC,GAAQ;QAC/B,qCAAqC;QACrC,IAAI,GAAG,CAAC,SAAS;YAAE,OAAO,GAAG,CAAC,SAAS,CAAC,CAAW,kCAAkC;QACrF,IAAI,GAAG,CAAC,KAAK;YAAE,OAAO,GAAG,CAAC,KAAK,CAAC,CAAmB,eAAe;QAClE,IAAI,GAAG,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC;YAAE,OAAO,GAAG,CAAC,IAAI,CAAC,CAAE,eAAe;QACpF,IAAI,GAAG,CAAC,UAAU;YAAE,OAAO,GAAG,CAAC,UAAU,CAAC,CAAS,mBAAmB;QACtE,IAAI,GAAG,CAAC,MAAM,EAAE,KAAK;YAAE,OAAO,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAI,wBAAwB;QAC3E,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,cAAc,CACzB,GAAQ,EACR,OAAiB,EACjB,KAAa,EACb,YAAqB;QAErB,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC;QAE7C,MAAM,UAAU,GAAwB;YACtC,IAAI,EAAE,KAAK;YACX,KAAK,EAAE,SAAS;YAChB,YAAY,EAAE,OAAO,CAAC,MAAM;YAC5B,mBAAmB,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,IAAI,CAAC;SAC7C,CAAC;QAEF,mCAAmC;QACnC,IAAI,GAAG,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;YAClC,UAAU,CAAC,WAAW,GAAG,GAAG,CAAC,WAAW,CAAC;QAC3C,CAAC;QAED,kCAAkC;QAClC,IAAI,GAAG,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YAChC,UAAU,CAAC,UAAU,GAAG,GAAG,CAAC,SAAS,CAAC;QACxC,CAAC;QACD,IAAI,GAAG,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;YACjC,UAAU,CAAC,UAAU,GAAG,GAAG,CAAC,UAAU,CAAC;QACzC,CAAC;QAED,6BAA6B;QAC7B,IAAI,GAAG,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YAC3B,UAAU,CAAC,KAAK,GAAG,GAAG,CAAC,IAAI,CAAC;QAC9B,CAAC;QAED,6BAA6B;QAC7B,IAAI,GAAG,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YAC3B,UAAU,CAAC,KAAK,GAAG,GAAG,CAAC,IAAI,CAAC;QAC9B,CAAC;QAED,mDAAmD;QACnD,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;YAChB,UAAU,CAAC,QAAQ,GAAG,GAAG,CAAC,OAAO,CAAC;QACpC,CAAC;QAED,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;QAC1D,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAClC,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,YAAY,CAAC,MAAW,EAAE,KAAa;QAClD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC;gBACH,uCAAuC;gBACvC,MAAM,UAAU,GACd,MAAM,EAAE,SAAS,EAAE,WAAW;oBAC9B,MAAM,EAAE,WAAW;oBACnB,MAAM,EAAE,QAAQ,EAAE,WAAW,CAAC;gBAEhC,IAAI,UAAU,EAAE,CAAC;oBACf,IAAI,UAAU,CAAC,aAAa,KAAK,SAAS,EAAE,CAAC;wBAC3C,IAAI,CAAC,YAAY,CAAC,mBAAmB,EAAE,UAAU,CAAC,aAAa,CAAC,CAAC;oBACnE,CAAC;oBACD,IAAI,UAAU,CAAC,iBAAiB,KAAK,SAAS,EAAE,CAAC;wBAC/C,IAAI,CAAC,YAAY,CAAC,uBAAuB,EAAE,UAAU,CAAC,iBAAiB,CAAC,CAAC;oBAC3E,CAAC;oBACD,MAAM,WAAW,GAAG,CAAC,UAAU,CAAC,aAAa,IAAI,CAAC,CAAC,GAAG,CAAC,UAAU,CAAC,iBAAiB,IAAI,CAAC,CAAC,CAAC;oBAC1F,IAAI,WAAW,GAAG,CAAC,EAAE,CAAC;wBACpB,IAAI,CAAC,YAAY,CAAC,kBAAkB,EAAE,WAAW,CAAC,CAAC;oBACrD,CAAC;gBACH,CAAC;gBAED,6BAA6B;gBAC7B,IAAI,MAAM,EAAE,IAAI,EAAE,CAAC;oBACjB,IAAI,CAAC,YAAY,CAAC,eAAe,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACzD,CAAC;qBAAM,IAAI,MAAM,EAAE,WAAW,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC;oBACpE,MAAM,eAAe,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;oBAC9C,IAAI,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC;wBAC/B,IAAI,CAAC,YAAY,CAAC,eAAe,EAAE,eAAe,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;oBACrE,CAAC;gBACH,CAAC;gBAED,qCAAqC;gBACrC,IAAI,MAAM,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;oBACrC,IAAI,CAAC,YAAY,CAAC,eAAe,EAAE,MAAM,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;gBACrE,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,qCAAqC;YACvC,CAAC;YAED,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,cAAc,CAAC,KAAY,EAAE,KAAa;QACrD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,eAAe,EAAE,CAAC,CAAC;YACzD,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,gBAAgB,CAC3B,KAAU,EACV,MAAW,EACX,KAAa,EACb,YAAqB;QAErB,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,UAAU,IAAI,OAAO,CAAC;QAE5D,MAAM,UAAU,GAAwB;YACtC,IAAI,EAAE,OAAO;YACb,UAAU,EAAE,SAAS;YACrB,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;YAC/C,WAAW,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,MAAM;SAC9C,CAAC;QAEF,6BAA6B;QAC7B,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACxC,UAAU,CAAC,YAAY,GAAG,QAAQ,CAAC,MAAM,CAAC;QAC5C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,gBAAgB;QAClB,CAAC;QAED,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,SAAS,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;QACzE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAClC,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,cAAc,CAAC,MAAW,EAAE,KAAa;QACpD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC;gBACH,IAAI,MAAM,EAAE,CAAC;oBACX,MAAM,SAAS,GAAG,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;oBAC/E,IAAI,CAAC,YAAY,CAAC,eAAe,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;gBACvD,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,gBAAgB;YAClB,CAAC;YAED,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,gBAAgB,CAAC,KAAY,EAAE,KAAa;QACvD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC,CAAC;YAC3D,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,eAAe,CAC1B,IAAS,EACT,KAAa,EACb,KAAa,EACb,YAAqB;QAErB,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,IAAI,cAAc,CAAC;QAE7C,MAAM,UAAU,GAAwB;YACtC,IAAI,EAAE,MAAM;YACZ,SAAS,EAAE,QAAQ;YACnB,gBAAgB,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;YACxC,YAAY,EAAE,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;SACxH,CAAC;QAEF,kCAAkC;QAClC,IAAI,CAAC;YACH,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;gBAC9B,UAAU,CAAC,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,gBAAgB;QAClB,CAAC;QAED,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,QAAQ,QAAQ,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;QACvE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAClC,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,aAAa,CAAC,MAAc,EAAE,KAAa;QACtD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC;gBACH,IAAI,CAAC,YAAY,CAAC,eAAe,EAAE,MAAM,EAAE,MAAM,IAAI,CAAC,CAAC,CAAC;YAC1D,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,gBAAgB;YAClB,CAAC;YAED,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,eAAe,CAAC,KAAY,EAAE,KAAa;QACtD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,gBAAgB,EAAE,CAAC,CAAC;YAC1D,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,iBAAiB,CAAC,MAAW,EAAE,KAAa;QACvD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC;gBACH,IAAI,CAAC,YAAY,CAAC,cAAc,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;YACjD,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,gBAAgB;YAClB,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,iBAAiB,CAAC,MAAW,EAAE,KAAa;QACvD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC;gBACH,IAAI,CAAC,YAAY,CAAC,qBAAqB,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC;YAChF,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,gBAAgB;YAClB,CAAC;YAED,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;CACF;AA1RD,sDA0RC"}
+{"version":3,"file":"langchain-callback.js","sourceRoot":"","sources":["../../src/integrations/langchain-callback.ts"],"names":[],"mappings":";AAAA;;;;GAIG;;;AAGH,kCAAoC;AACpC,yDAAqE;AAErE;;;;;;;;;;;;;;;;GAgBG;AACH,MAAa,sBAAuB,SAAQ,0BAAmB;IAA/D;;QACE,SAAI,GAAG,wBAAwB,CAAC;QACxB,WAAM,GAAG,IAAA,gBAAS,EAAC,WAAW,CAAC,CAAC;QAChC,cAAS,GAAuB,IAAI,GAAG,EAAE,CAAC;QAC1C,wBAAmB,GAAyB,EAAE,CAAC;IAqZzD,CAAC;IAnZC;;;OAGG;IACK,gBAAgB,CAAC,GAAQ;QAC/B,qCAAqC;QACrC,IAAI,GAAG,CAAC,SAAS;YAAE,OAAO,GAAG,CAAC,SAAS,CAAC,CAAW,kCAAkC;QACrF,IAAI,GAAG,CAAC,KAAK;YAAE,OAAO,GAAG,CAAC,KAAK,CAAC,CAAmB,eAAe;QAClE,IAAI,GAAG,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC;YAAE,OAAO,GAAG,CAAC,IAAI,CAAC,CAAE,eAAe;QACpF,IAAI,GAAG,CAAC,UAAU;YAAE,OAAO,GAAG,CAAC,UAAU,CAAC,CAAS,mBAAmB;QACtE,IAAI,GAAG,CAAC,MAAM,EAAE,KAAK;YAAE,OAAO,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAI,wBAAwB;QAC3E,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,cAAc,CACzB,GAAQ,EACR,OAAiB,EACjB,KAAa,EACb,YAAqB;QAErB,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC;QAE7C,MAAM,UAAU,GAAwB;YACtC,IAAI,EAAE,KAAK;YACX,KAAK,EAAE,SAAS;YAChB,YAAY,EAAE,OAAO,CAAC,MAAM;YAC5B,mBAAmB,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,IAAI,CAAC;SAC7C,CAAC;QAEF,mCAAmC;QACnC,IAAI,GAAG,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;YAClC,UAAU,CAAC,WAAW,GAAG,GAAG,CAAC,WAAW,CAAC;QAC3C,CAAC;QAED,kCAAkC;QAClC,IAAI,GAAG,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YAChC,UAAU,CAAC,UAAU,GAAG,GAAG,CAAC,SAAS,CAAC;QACxC,CAAC;QACD,IAAI,GAAG,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;YACjC,UAAU,CAAC,UAAU,GAAG,GAAG,CAAC,UAAU,CAAC;QACzC,CAAC;QAED,6BAA6B;QAC7B,IAAI,GAAG,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YAC3B,UAAU,CAAC,KAAK,GAAG,GAAG,CAAC,IAAI,CAAC;QAC9B,CAAC;QAED,6BAA6B;QAC7B,IAAI,GAAG,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YAC3B,UAAU,CAAC,KAAK,GAAG,GAAG,CAAC,IAAI,CAAC;QAC9B,CAAC;QAED,mDAAmD;QACnD,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;YAChB,UAAU,CAAC,QAAQ,GAAG,GAAG,CAAC,OAAO,CAAC;QACpC,CAAC;QAED,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;QAC1D,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAClC,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,YAAY,CAAC,MAAW,EAAE,KAAa;QAClD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC;gBACH,uCAAuC;gBACvC,+CAA+C;gBAC/C,MAAM,UAAU,GACd,MAAM,EAAE,SAAS,EAAE,WAAW;oBAC9B,MAAM,EAAE,SAAS,EAAE,UAAU;oBAC7B,MAAM,EAAE,WAAW;oBACnB,MAAM,EAAE,QAAQ,EAAE,WAAW,CAAC;gBAEhC,IAAI,UAAU,EAAE,CAAC;oBACf,+BAA+B;oBAC/B,MAAM,YAAY,GAAG,UAAU,CAAC,aAAa,IAAI,UAAU,CAAC,YAAY,CAAC;oBACzE,MAAM,gBAAgB,GAAG,UAAU,CAAC,iBAAiB,IAAI,UAAU,CAAC,gBAAgB,CAAC;oBACrF,MAAM,WAAW,GAAG,UAAU,CAAC,YAAY,IAAI,UAAU,CAAC,WAAW,CAAC;oBAEtE,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;wBAC/B,IAAI,CAAC,YAAY,CAAC,mBAAmB,EAAE,YAAY,CAAC,CAAC;oBACvD,CAAC;oBACD,IAAI,gBAAgB,KAAK,SAAS,EAAE,CAAC;wBACnC,IAAI,CAAC,YAAY,CAAC,uBAAuB,EAAE,gBAAgB,CAAC,CAAC;oBAC/D,CAAC;oBACD,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;wBAC9B,IAAI,CAAC,YAAY,CAAC,kBAAkB,EAAE,WAAW,CAAC,CAAC;oBACrD,CAAC;oBAED,gEAAgE;oBAChE,+DAA+D;oBAC/D,IAAI,UAAU,CAAC,mBAAmB,IAAI,OAAO,UAAU,CAAC,mBAAmB,KAAK,QAAQ,EAAE,CAAC;wBACzF,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,mBAAmB,CAAC,EAAE,CAAC;4BAC1E,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;gCAC9B,IAAI,CAAC,YAAY,CAAC,oBAAoB,GAAG,EAAE,EAAE,KAAK,CAAC,CAAC;4BACtD,CAAC;wBACH,CAAC;oBACH,CAAC;oBAED,oEAAoE;oBACpE,IAAI,UAAU,CAAC,oBAAoB,IAAI,OAAO,UAAU,CAAC,oBAAoB,KAAK,QAAQ,EAAE,CAAC;wBAC3F,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,oBAAoB,CAAC,EAAE,CAAC;4BAC3E,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;gCAC9B,IAAI,CAAC,YAAY,CAAC,qBAAqB,GAAG,EAAE,EAAE,KAAK,CAAC,CAAC;4BACvD,CAAC;wBACH,CAAC;oBACH,CAAC;gBACH,CAAC;gBAED,6BAA6B;gBAC7B,IAAI,MAAM,EAAE,IAAI,EAAE,CAAC;oBACjB,IAAI,CAAC,YAAY,CAAC,eAAe,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACzD,CAAC;qBAAM,IAAI,MAAM,EAAE,WAAW,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC;oBACpE,MAAM,eAAe,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;oBAC9C,IAAI,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC;wBAC/B,IAAI,CAAC,YAAY,CAAC,eAAe,EAAE,eAAe,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;oBACrE,CAAC;gBACH,CAAC;qBAAM,IAAI,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;oBACpC,uCAAuC;oBACvC,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC;oBACvC,MAAM,UAAU,GAAG,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;oBACnF,IAAI,CAAC,YAAY,CAAC,eAAe,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC;gBACxD,CAAC;qBAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;oBACtC,uBAAuB;oBACvB,IAAI,CAAC,YAAY,CAAC,eAAe,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;gBACpD,CAAC;gBAED,qCAAqC;gBACrC,IAAI,MAAM,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;oBACrC,IAAI,CAAC,YAAY,CAAC,eAAe,EAAE,MAAM,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;gBACrE,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,qCAAqC;YACvC,CAAC;YAED,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,cAAc,CAAC,KAAY,EAAE,KAAa;QACrD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,eAAe,EAAE,CAAC,CAAC;YACzD,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,iBAAiB,CAC5B,MAAc,EACd,IAAU,EACV,KAAc;QAEd,IAAI,KAAK,IAAI,CAAC,CAAC,KAAK,IAAI,IAAI,CAAC,mBAAmB,CAAC,EAAE,CAAC;YAClD,+CAA+C;YAC/C,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,GAAG,IAAI,IAAI,EAAE,CAAC;YAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACvC,IAAI,IAAI,EAAE,CAAC;gBACT,IAAI,CAAC;oBACH,IAAI,CAAC,YAAY,CAAC,8BAA8B,EAAE,IAAI,CAAC,CAAC;gBAC1D,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,gBAAgB;gBAClB,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,gBAAgB,CAC3B,KAAU,EACV,MAAW,EACX,KAAa,EACb,YAAqB;QAErB,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,UAAU,IAAI,OAAO,CAAC;QAE5D,MAAM,UAAU,GAAwB;YACtC,IAAI,EAAE,OAAO;YACb,UAAU,EAAE,SAAS;YACrB,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;YAC/C,WAAW,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,MAAM;SAC9C,CAAC;QAEF,6BAA6B;QAC7B,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACxC,UAAU,CAAC,YAAY,GAAG,QAAQ,CAAC,MAAM,CAAC;QAC5C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,gBAAgB;QAClB,CAAC;QAED,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,SAAS,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;QACzE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAClC,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,cAAc,CAAC,MAAW,EAAE,KAAa;QACpD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC;gBACH,IAAI,MAAM,EAAE,CAAC;oBACX,MAAM,SAAS,GAAG,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;oBAC/E,IAAI,CAAC,YAAY,CAAC,eAAe,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;gBACvD,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,gBAAgB;YAClB,CAAC;YAED,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,gBAAgB,CAAC,KAAY,EAAE,KAAa;QACvD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC,CAAC;YAC3D,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,eAAe,CAC1B,IAAS,EACT,KAAa,EACb,KAAa,EACb,YAAqB;QAErB,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,IAAI,cAAc,CAAC;QAE7C,MAAM,UAAU,GAAwB;YACtC,IAAI,EAAE,MAAM;YACZ,SAAS,EAAE,QAAQ;YACnB,gBAAgB,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;YACxC,YAAY,EAAE,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;SACxH,CAAC;QAEF,kCAAkC;QAClC,IAAI,CAAC;YACH,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;gBAC9B,UAAU,CAAC,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,gBAAgB;QAClB,CAAC;QAED,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,QAAQ,QAAQ,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;QACvE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAClC,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,aAAa,CAAC,MAAc,EAAE,KAAa;QACtD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC;gBACH,IAAI,CAAC,YAAY,CAAC,eAAe,EAAE,MAAM,EAAE,MAAM,IAAI,CAAC,CAAC,CAAC;YAC1D,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,gBAAgB;YAClB,CAAC;YAED,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,eAAe,CAAC,KAAY,EAAE,KAAa;QACtD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,gBAAgB,EAAE,CAAC,CAAC;YAC1D,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,iBAAiB,CAAC,MAAW,EAAE,KAAa;QACvD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC;gBACH,IAAI,CAAC,YAAY,CAAC,cAAc,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;YACjD,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,gBAAgB;YAClB,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,iBAAiB,CAAC,MAAW,EAAE,KAAa;QACvD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC;gBACH,IAAI,CAAC,YAAY,CAAC,qBAAqB,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC;YAChF,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,gBAAgB;YAClB,CAAC;YAED,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAED,mDAAmD;IACnD,oCAAoC;IAC7B,KAAK,CAAC,UAAU,CACrB,GAAQ,EACR,OAAiB,EACjB,KAAa,EACb,WAAoB;QAEpB,OAAO,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC;IAC/D,CAAC;IAEM,KAAK,CAAC,QAAQ,CAAC,MAAW,EAAE,KAAa;QAC9C,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAC1C,CAAC;IAEM,KAAK,CAAC,UAAU,CAAC,KAAY,EAAE,KAAa;QACjD,OAAO,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IAC3C,CAAC;IAEM,KAAK,CAAC,aAAa,CACxB,MAAc,EACd,GAAS,EACT,KAAc;QAEd,OAAO,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,CAAC,CAAC;IACpD,CAAC;IAEM,KAAK,CAAC,YAAY,CACvB,KAAU,EACV,MAAW,EACX,KAAa,EACb,WAAoB;QAEpB,OAAO,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC;IAClE,CAAC;IAEM,KAAK,CAAC,UAAU,CAAC,MAAW,EAAE,KAAa;QAChD,OAAO,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAC5C,CAAC;IAEM,KAAK,CAAC,YAAY,CAAC,KAAY,EAAE,KAAa;QACnD,OAAO,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IAC7C,CAAC;IAEM,KAAK,CAAC,WAAW,CACtB,IAAS,EACT,KAAa,EACb,KAAa,EACb,WAAoB;QAEpB,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC;IAC/D,CAAC;IAEM,KAAK,CAAC,SAAS,CAAC,MAAc,EAAE,KAAa;QAClD,OAAO,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAC3C,CAAC;IAEM,KAAK,CAAC,WAAW,CAAC,KAAY,EAAE,KAAa;QAClD,OAAO,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IAC5C,CAAC;IAEM,KAAK,CAAC,aAAa,CAAC,MAAW,EAAE,KAAa;QACnD,OAAO,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAC/C,CAAC;IAEM,KAAK,CAAC,aAAa,CAAC,MAAW,EAAE,KAAa;QACnD,OAAO,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAC/C,CAAC;CACF;AAzZD,wDAyZC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@traccia2/sdk",
-  "version": "0.0.4",
+  "version": "0.0.6",
   "description": "Tracing SDK for agent applications",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -67,6 +67,7 @@
     "prepare": "npm run build"
   },
   "devDependencies": {
+    "@langchain/core": ">=0.3.8",
     "@types/jest": "^29.5.11",
     "@types/node": "^20.10.6",
     "@typescript-eslint/eslint-plugin": "^6.17.0",
@@ -80,5 +81,9 @@
   "publishConfig": {
     "access": "public"
   },
+  "peerDependencies": {
+    "@langchain/core": ">=0.3.8",
+    "@opentelemetry/api": "^1.9.0"
+  },
   "dependencies": {}
 }

package/src/__tests__/integrations-langchain.test.ts

@@ -2,10 +2,10 @@
  * Tests for LangChain integration (TraciaCallbackHandler)
  */
 
-import { TraciaCallbackHandler } from '../integrations/langchain-callback';
+import { TracciaCallbackHandler } from '../integrations/langchain-callback';
 import { startTracing, stopTracing } from '../auto';
 
-describe('TraciaCallbackHandler', () => {
+describe('TracciaCallbackHandler', () => {
   beforeEach(() => {
     startTracing({
       enableTokenCounting: false,
@@ -19,7 +19,7 @@ describe('TraciaCallbackHandler', () => {
 
   describe('handleLLMStart', () => {
     it('should create a span for LLM invocation', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       const mockLLM = {
         name: 'gpt-4',
         _modelType: 'openai',
@@ -32,7 +32,7 @@ describe('TraciaCallbackHandler', () => {
     });
 
     it('should capture LLM model name in attributes', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       const mockLLM = {
         name: 'claude-3',
         _modelType: 'anthropic',
@@ -45,7 +45,7 @@ describe('TraciaCallbackHandler', () => {
     });
 
     it('should capture prompt count', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       const mockLLM = { name: 'test-model' };
       const prompts = ['p1', 'p2', 'p3'];
 
@@ -56,7 +56,7 @@ describe('TraciaCallbackHandler', () => {
     });
 
     it('should handle missing model name gracefully', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       const mockLLM = {}; // No name or _modelType
 
       await handler.handleLLMStart(mockLLM, ['prompt'], 'run-4');
@@ -68,7 +68,7 @@ describe('TraciaCallbackHandler', () => {
 
   describe('handleLLMEnd', () => {
     it('should end LLM span and record token counts', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       const mockLLM = { name: 'gpt-4' };
 
       // Start span
@@ -85,7 +85,7 @@ describe('TraciaCallbackHandler', () => {
     });
 
     it('should record output length', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       const mockLLM = { name: 'test' };
 
       await handler.handleLLMStart(mockLLM, ['prompt'], 'run-6');
@@ -102,7 +102,7 @@ describe('TraciaCallbackHandler', () => {
 
   describe('handleChainStart', () => {
     it('should create a span for chain execution', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       const mockChain = {
         name: 'test-chain',
         _chainType: 'stuff',
@@ -114,7 +114,7 @@ describe('TraciaCallbackHandler', () => {
     });
 
     it('should capture chain name and type', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       const mockChain = {
         name: 'qa-chain',
         _chainType: 'retrieval_qa',
@@ -128,7 +128,7 @@ describe('TraciaCallbackHandler', () => {
     });
 
     it('should record input keys', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       const mockChain = { name: 'chain' };
       const inputs = { key1: 'value1', key2: 'value2' };
 
@@ -142,7 +142,7 @@ describe('TraciaCallbackHandler', () => {
 
   describe('handleChainEnd', () => {
     it('should end chain span and record output', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       const mockChain = { name: 'chain' };
 
       await handler.handleChainStart(mockChain, { input: 'test' }, 'chain-4');
@@ -156,7 +156,7 @@ describe('TraciaCallbackHandler', () => {
 
   describe('handleToolStart', () => {
     it('should create a span for tool invocation', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       const mockTool = {
         name: 'search-api',
       };
@@ -167,7 +167,7 @@ describe('TraciaCallbackHandler', () => {
     });
 
     it('should capture tool name', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       const mockTool = { name: 'calculator' };
 
       await handler.handleToolStart(mockTool, '2+2', 'tool-2');
@@ -177,7 +177,7 @@ describe('TraciaCallbackHandler', () => {
     });
 
     it('should record input length', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       const mockTool = { name: 'tool' };
       const input = 'test input with some length';
 
@@ -190,7 +190,7 @@ describe('TraciaCallbackHandler', () => {
 
   describe('handleToolEnd', () => {
     it('should end tool span', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       const mockTool = { name: 'tool' };
 
       await handler.handleToolStart(mockTool, 'input', 'tool-4');
@@ -200,7 +200,7 @@ describe('TraciaCallbackHandler', () => {
     });
 
     it('should record output length', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       const mockTool = { name: 'tool' };
 
       await handler.handleToolStart(mockTool, 'input', 'tool-5');
@@ -213,7 +213,7 @@ describe('TraciaCallbackHandler', () => {
 
   describe('handleAgentAction', () => {
     it('should update existing span on agent action', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       const mockChain = { name: 'chain' };
 
       // Start a span first
@@ -231,7 +231,7 @@ describe('TraciaCallbackHandler', () => {
     });
 
     it('should record agent action details', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       const mockChain = { name: 'chain' };
 
       // Start a span first
@@ -252,7 +252,7 @@ describe('TraciaCallbackHandler', () => {
 
   describe('handleAgentFinish', () => {
     it('should end agent span on finish', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
 
       // Create a span first
       const mockAction = { tool: 'test', toolInput: 'input' };
@@ -266,7 +266,7 @@ describe('TraciaCallbackHandler', () => {
     });
 
     it('should record final output', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       const mockAction = { tool: 'test', toolInput: 'input' };
 
       await handler.handleAgentAction(mockAction, 'agent-4');
@@ -280,7 +280,7 @@ describe('TraciaCallbackHandler', () => {
 
   describe('Error Handling', () => {
     it('should handle LLM errors gracefully', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       const mockLLM = { name: 'llm' };
 
       await handler.handleLLMStart(mockLLM, ['prompt'], 'error-1');
@@ -293,7 +293,7 @@ describe('TraciaCallbackHandler', () => {
     });
 
     it('should handle chain errors gracefully', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       const mockChain = { name: 'chain' };
 
       await handler.handleChainStart(mockChain, {}, 'error-2');
@@ -305,7 +305,7 @@ describe('TraciaCallbackHandler', () => {
     });
 
     it('should handle tool errors gracefully', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       const mockTool = { name: 'tool' };
 
       await handler.handleToolStart(mockTool, 'input', 'error-3');
@@ -319,7 +319,7 @@ describe('TraciaCallbackHandler', () => {
 
   describe('Span Nesting', () => {
     it('should handle nested spans (chain containing LLM)', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
 
       // Start chain
       const chain = { name: 'chain' };
@@ -346,7 +346,7 @@ describe('TraciaCallbackHandler', () => {
     });
 
     it('should handle multiple concurrent spans', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
 
       // Start multiple spans
       const chain1 = { name: 'chain1' };
@@ -369,12 +369,12 @@ describe('TraciaCallbackHandler', () => {
 
   describe('Integration with Tracer', () => {
     it('should use the SDK tracer', () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       expect(handler['tracer']).toBeDefined();
     });
 
     it('should handle unavailable tracer gracefully', async () => {
-      const handler = new TraciaCallbackHandler();
+      const handler = new TracciaCallbackHandler();
       
       // Should not throw even if operations fail
       const mockLLM = { name: 'llm' };

package/src/integrations/auto-langchain.ts

@@ -5,27 +5,27 @@
  * convenience functions for automatic instrumentation with zero boilerplate.
  */
 
-import { TraciaCallbackHandler } from './langchain-callback';
+import { TracciaCallbackHandler } from './langchain-callback';
 
 /**
  * Global handler instance to avoid creating multiple handlers
  */
-let globalTraciaHandler: TraciaCallbackHandler | null = null;
+let globalTraciaHandler: TracciaCallbackHandler | null = null;
 
 /**
  * Get or create the global Traccia callback handler.
  * 
  * @example
  * // Instead of:
- * const handler = new TraciaCallbackHandler();
+ * const handler = new TracciaCallbackHandler();
  * const model = new ChatOpenAI({ callbacks: [handler] });
  * 
  * // Just do:
  * const model = new ChatOpenAI({ callbacks: [getTraciaHandler()] });
  */
-export function getTraciaHandler(): TraciaCallbackHandler {
+export function getTraciaHandler(): TracciaCallbackHandler {
   if (!globalTraciaHandler) {
-    globalTraciaHandler = new TraciaCallbackHandler();
+    globalTraciaHandler = new TracciaCallbackHandler();
   }
   return globalTraciaHandler;
 }
@@ -217,7 +217,7 @@ export async function setupLangChainWithTracing(options: {
 }): Promise<{
   model: any;
   executor: any;
-  handler: TraciaCallbackHandler;
+  handler: TracciaCallbackHandler;
 }> {
   try {
     const {

package/src/integrations/index.ts

@@ -6,7 +6,7 @@
  */
 
 // LangChain integrations
-export { TraciaCallbackHandler } from './langchain-callback';
+export { TracciaCallbackHandler } from './langchain-callback';
 export {
   getTraciaHandler,
   withTracing,

package/src/integrations/langchain-callback.ts

@@ -6,28 +6,30 @@
 
 import { ISpan } from '../types';
 import { getTracer } from '../auto';
+import { BaseCallbackHandler } from '@langchain/core/callbacks/base';
 
 /**
  * LangChain Callback Handler for Traccia SDK.
  * Automatically traces LLM calls, chains, agents, and tools.
  *
+ * Extends LangChain's BaseCallbackHandler for proper interface compliance.
+ * Compatible with LangChain 0.0.x, 0.1.x, 0.2.x, and 1.x versions.
+ *
  * @example
- * import { ChatOpenAI } from 'langchain/chat_models/openai';
- * import { LLMChain } from 'langchain/chains';
- * import { TraciaCallbackHandler } from '@traccia/sdk/integrations/langchain';
+ * import { ChatOpenAI } from '@langchain/openai';
+ * import { TracciaCallbackHandler } from '@traccia/sdk/integrations';
  *
- * const handler = new TraciaCallbackHandler();
- * const chain = new LLMChain({
- *   llm: new ChatOpenAI(),
- *   callbacks: [handler],
- * });
+ * const handler = new TracciaCallbackHandler();
+ * const model = new ChatOpenAI({ callbacks: [handler] });
  *
- * await chain.run({ input: 'What is 2+2?' });
- * // Automatically traced with spans for LLM, chain execution, tokens, etc.
+ * const response = await model.invoke({ input: 'Hello!' });
+ * // Automatically traced with spans for LLM calls, tokens, latency, etc.
  */
-export class TraciaCallbackHandler {
+export class TracciaCallbackHandler extends BaseCallbackHandler {
+  name = 'TracciaCallbackHandler';
   private tracer = getTracer('langchain');
   private spanStack: Map<string, ISpan> = new Map();
+  private streamingStartTimes: Record<string, Date> = {};
 
   /**
    * Extract model name from LLM instance, checking multiple property locations.
@@ -101,22 +103,47 @@ export class TraciaCallbackHandler {
     if (span) {
       try {
         // Try multiple ways to get token usage
+        // OpenAI format and new @langchain/core format
         const tokenUsage = 
           output?.llmOutput?.token_usage || 
+          output?.llmOutput?.tokenUsage ||
           output?.token_usage || 
           output?.metadata?.token_usage;
 
         if (tokenUsage) {
-          if (tokenUsage.prompt_tokens !== undefined) {
-            span.setAttribute('llm.tokens.prompt', tokenUsage.prompt_tokens);
+          // Handle standard token counts
+          const promptTokens = tokenUsage.prompt_tokens ?? tokenUsage.promptTokens;
+          const completionTokens = tokenUsage.completion_tokens ?? tokenUsage.completionTokens;
+          const totalTokens = tokenUsage.total_tokens ?? tokenUsage.totalTokens;
+
+          if (promptTokens !== undefined) {
+            span.setAttribute('llm.tokens.prompt', promptTokens);
           }
-          if (tokenUsage.completion_tokens !== undefined) {
-            span.setAttribute('llm.tokens.completion', tokenUsage.completion_tokens);
+          if (completionTokens !== undefined) {
+            span.setAttribute('llm.tokens.completion', completionTokens);
           }
-          const totalTokens = (tokenUsage.prompt_tokens || 0) + (tokenUsage.completion_tokens || 0);
-          if (totalTokens > 0) {
+          if (totalTokens !== undefined) {
             span.setAttribute('llm.tokens.total', totalTokens);
           }
+
+          // Handle detailed token breakdown for models like GPT-4o vision
+          // input_token_details contains breakdown of prompt token usage
+          if (tokenUsage.input_token_details && typeof tokenUsage.input_token_details === 'object') {
+            for (const [key, value] of Object.entries(tokenUsage.input_token_details)) {
+              if (typeof value === 'number') {
+                span.setAttribute(`llm.tokens.input_${key}`, value);
+              }
+            }
+          }
+
+          // output_token_details contains breakdown of completion token usage
+          if (tokenUsage.output_token_details && typeof tokenUsage.output_token_details === 'object') {
+            for (const [key, value] of Object.entries(tokenUsage.output_token_details)) {
+              if (typeof value === 'number') {
+                span.setAttribute(`llm.tokens.output_${key}`, value);
+              }
+            }
+          }
         }
 
         // Capture output text length
@@ -127,6 +154,14 @@ export class TraciaCallbackHandler {
           if (firstGeneration?.[0]?.text) {
             span.setAttribute('output_length', firstGeneration[0].text.length);
           }
+        } else if (output?.message?.content) {
+          // Ollama format (uses message.content)
+          const content = output.message.content;
+          const contentStr = typeof content === 'string' ? content : JSON.stringify(content);
+          span.setAttribute('output_length', contentStr.length);
+        } else if (typeof output === 'string') {
+          // Direct string output
+          span.setAttribute('output_length', output.length);
         }
 
         // Capture finish reason if available
@@ -154,6 +189,29 @@ export class TraciaCallbackHandler {
     }
   }
 
+  /**
+   * Handle LLM new token - called when a new token is generated during streaming.
+   * Tracks first token latency and token count for streaming scenarios.
+   */
+  public async handleLLMNewToken(
+    _token: string,
+    _idx?: any,
+    runId?: string
+  ): Promise<void> {
+    if (runId && !(runId in this.streamingStartTimes)) {
+      // Record the time of the first streaming token
+      this.streamingStartTimes[runId] = new Date();
+      const span = this.spanStack.get(runId);
+      if (span) {
+        try {
+          span.setAttribute('stream.first_token_generated', true);
+        } catch (error) {
+          // Silently fail
+        }
+      }
+    }
+  }
+
   /**
    * Handle chain start - called when a chain begins execution.
    */
@@ -307,4 +365,74 @@ export class TraciaCallbackHandler {
       this.spanStack.delete(runId);
     }
   }
+
+  // LangChain uses 'on*' prefix for callback methods
+  // Provide aliases for compatibility
+  public async onLLMStart(
+    llm: any,
+    prompts: string[],
+    runId: string,
+    parentRunId?: string
+  ): Promise<void> {
+    return this.handleLLMStart(llm, prompts, runId, parentRunId);
+  }
+
+  public async onLLMEnd(output: any, runId: string): Promise<void> {
+    return this.handleLLMEnd(output, runId);
+  }
+
+  public async onLLMError(error: Error, runId: string): Promise<void> {
+    return this.handleLLMError(error, runId);
+  }
+
+  public async onLLMNewToken(
+    _token: string,
+    idx?: any,
+    runId?: string
+  ): Promise<void> {
+    return this.handleLLMNewToken(_token, idx, runId);
+  }
+
+  public async onChainStart(
+    chain: any,
+    inputs: any,
+    runId: string,
+    parentRunId?: string
+  ): Promise<void> {
+    return this.handleChainStart(chain, inputs, runId, parentRunId);
+  }
+
+  public async onChainEnd(output: any, runId: string): Promise<void> {
+    return this.handleChainEnd(output, runId);
+  }
+
+  public async onChainError(error: Error, runId: string): Promise<void> {
+    return this.handleChainError(error, runId);
+  }
+
+  public async onToolStart(
+    tool: any,
+    input: string,
+    runId: string,
+    parentRunId?: string
+  ): Promise<void> {
+    return this.handleToolStart(tool, input, runId, parentRunId);
+  }
+
+  public async onToolEnd(output: string, runId: string): Promise<void> {
+    return this.handleToolEnd(output, runId);
+  }
+
+  public async onToolError(error: Error, runId: string): Promise<void> {
+    return this.handleToolError(error, runId);
+  }
+
+  public async onAgentAction(action: any, runId: string): Promise<void> {
+    return this.handleAgentAction(action, runId);
+  }
+
+  public async onAgentFinish(finish: any, runId: string): Promise<void> {
+    return this.handleAgentFinish(finish, runId);
+  }
 }
+