@traccia2/sdk 0.0.3 → 0.0.5

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 `TraciaCallbackHandler` 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/langchain-callback';
+import { startTracing, getTracerProvider } from '@traccia/sdk';
+import { TraciaCallbackHandler } from '@traccia/sdk/integrations';
 import { ChatOpenAI } from '@langchain/openai';
 import { AgentExecutor, createOpenAIToolsAgent } from 'langchain/agents';
 
@@ -344,6 +344,9 @@ 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
@@ -371,32 +374,199 @@ const executor = new AgentExecutor({
 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 { TraciaCallbackHandler } 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 traciaHandler = new TraciaCallbackHandler();
+
+// Use with ChatOllama (local model)
+const model = new ChatOllama({
+  model: 'mistral',                    // or 'llama2', 'neural-chat', etc.
+  baseUrl: 'http://localhost:11434',   // Ollama server
+  callbacks: [traciaHandler],          // 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, token counts, latency
+- **LLM Calls**: Model name, temperature, max tokens, top-p/k, token counts, latency
 - **Agent Actions**: Tool selection and execution
 - **Chain Steps**: Multi-step reasoning and intermediate results
 - **Tool Usage**: Tool names, inputs, and outputs
 - **Errors & Exceptions**: Automatic error recording
 
+**Auto-Captured LLM Configuration:**
+
+The handler automatically extracts and logs LLM defaults:
+
+```typescript
+// For any LLM, these are automatically captured:
+{
+  'model': 'gpt-4',                    // Auto-detected from modelName/model/name
+  'temperature': 0.7,                  // Auto-detected from llm.temperature
+  'max_tokens': 2000,                  // Auto-detected from llm.maxTokens
+  'top_p': 0.9,                        // Auto-detected if available
+  'top_k': 40,                         // Auto-detected if available
+  'base_url': 'http://localhost:11434' // For local models (Ollama)
+}
+```
+
 **Traced Metrics:**
 
 ```typescript
 // Automatically captured attributes:
 {
-  'llm.model': 'gpt-4',
+  // LLM Configuration (auto-extracted from defaults)
+  'model': 'gpt-4',                    // Auto-detected model name
+  'temperature': 0.7,                  // If configured
+  'max_tokens': 2000,                  // If configured
+  'top_p': 0.9,                        // If available
+  'top_k': 40,                         // If available
+  'base_url': 'http://localhost:11434' // For local models
+
+  // LLM Runtime Metrics
   'llm.tokens.prompt': 150,
   'llm.tokens.completion': 75,
   'llm.tokens.total': 225,
+  'output_length': 450,
+  'finish_reason': 'stop',             // If provided by model
+  
+  // Agent & Tool Usage
   'agent.action': 'tool_use',
   'agent.tool': 'calculator',
   'agent.tool.input': '{"expression": "2 + 2"}',
   'agent.tool.output': '4',
+  
+  // Chain Execution
   'chain.steps': 3,
-  'error.type': 'ValidationError', // if applicable
-  'error.message': 'Invalid input', // if applicable
+  'chain_type': 'agent_executor',
+  'input_length': 1250,
+  'output_length': 450,
+  
+  // Error Handling
+  'error.type': 'ValidationError',     // If applicable
+  'error.message': 'Invalid input',    // If applicable
 }
 ```
 

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

@@ -3,27 +3,32 @@
  * Integrates with LangChain's callback system to automatically instrument
  * LLM calls, chains, agents, and tools.
  */
+declare let BaseCallbackHandler: any;
 /**
  * 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 { TraciaCallbackHandler } from '@traccia/sdk/integrations';
  *
  * const handler = new TraciaCallbackHandler();
- * const chain = new LLMChain({
- *   llm: new ChatOpenAI(),
- *   callbacks: [handler],
- * });
+ * 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 TraciaCallbackHandler extends BaseCallbackHandler {
     private tracer;
     private spanStack;
+    /**
+     * Extract model name from LLM instance, checking multiple property locations.
+     * Different LLM implementations store the model name in different properties.
+     */
+    private extractModelName;
     /**
      * Handle LLM start - called when an LLM begins execution.
      */
@@ -68,5 +73,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>;
+    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>;
 }
+export {};
 //# 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;;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;IAahB;;OAEG;IACU,YAAY,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IA0BpE;;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;IAchB;;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;IAchB;;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;AAMH,QAAA,IAAI,mBAAmB,EAAE,GAMxB,CAAC;AAkBF;;;;;;;;;;;;;;;;GAgBG;AACH,qBAAa,qBAAsB,SAAQ,mBAAmB;IAC5D,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;IAuDpE;;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;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,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

@@ -7,41 +7,109 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.TraciaCallbackHandler = void 0;
 const auto_1 = require("../auto");
+// Try to import BaseCallbackHandler from different LangChain versions
+let BaseCallbackHandler = class {
+    constructor() {
+        this.ignoreAgent = false;
+        this.ignoreChain = false;
+        this.ignoreLLM = false;
+        this.ignoreRetriever = false;
+        this.ignoreToolUse = false;
+    }
+};
+try {
+    // Try LangChain 0.2.x+
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
+    const { BaseCallbackHandler: Handler } = require('@langchain/core/callbacks/base');
+    BaseCallbackHandler = Handler;
+}
+catch {
+    try {
+        // Try older LangChain versions
+        // eslint-disable-next-line @typescript-eslint/no-var-requires
+        const { BaseCallbackHandler: Handler } = require('langchain/callbacks');
+        BaseCallbackHandler = Handler;
+    }
+    catch {
+        // Fallback to basic implementation - user likely has LangChain installed
+    }
+}
 /**
  * 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 { TraciaCallbackHandler } from '@traccia/sdk/integrations';
  *
  * const handler = new TraciaCallbackHandler();
- * const chain = new LLMChain({
- *   llm: new ChatOpenAI(),
- *   callbacks: [handler],
- * });
+ * 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 TraciaCallbackHandler extends BaseCallbackHandler {
     constructor() {
+        super(...arguments);
         this.tracer = (0, auto_1.getTracer)('langchain');
         this.spanStack = new Map();
     }
+    /**
+     * Extract model name from LLM instance, checking multiple property locations.
+     * Different LLM implementations store the model name in different properties.
+     */
+    extractModelName(llm) {
+        // Check common model name properties
+        if (llm.modelName)
+            return llm.modelName; // ChatOpenAI, ChatAnthropic, etc.
+        if (llm.model)
+            return llm.model; // Ollama, etc.
+        if (llm.name && !llm.name.startsWith('langchain'))
+            return llm.name; // Generic name
+        if (llm._modelType)
+            return llm._modelType; // Fallback to type
+        if (llm.client?.model)
+            return llm.client.model; // Nested model property
+        return 'unknown';
+    }
     /**
      * Handle LLM start - called when an LLM begins execution.
      */
     async handleLLMStart(llm, prompts, runId, _parentRunId) {
-        const span = this.tracer.startSpan('llm', {
-            attributes: {
-                type: 'llm',
-                model: llm.name || llm._modelType || 'unknown',
-                prompt_count: prompts.length,
-                first_prompt_length: prompts[0]?.length || 0,
-            },
-        });
+        const modelName = this.extractModelName(llm);
+        const attributes = {
+            type: 'llm',
+            model: modelName,
+            prompt_count: prompts.length,
+            first_prompt_length: prompts[0]?.length || 0,
+        };
+        // Capture temperature if available
+        if (llm.temperature !== undefined) {
+            attributes.temperature = llm.temperature;
+        }
+        // Capture max tokens if available
+        if (llm.maxTokens !== undefined) {
+            attributes.max_tokens = llm.maxTokens;
+        }
+        if (llm.max_tokens !== undefined) {
+            attributes.max_tokens = llm.max_tokens;
+        }
+        // Capture top_p if available
+        if (llm.topP !== undefined) {
+            attributes.top_p = llm.topP;
+        }
+        // Capture top_k if available
+        if (llm.topK !== undefined) {
+            attributes.top_k = llm.topK;
+        }
+        // Capture base URL for local models (Ollama, etc.)
+        if (llm.baseUrl) {
+            attributes.base_url = llm.baseUrl;
+        }
+        const span = this.tracer.startSpan('llm', { attributes });
         this.spanStack.set(runId, span);
     }
     /**
@@ -51,15 +119,47 @@ class TraciaCallbackHandler {
         const span = this.spanStack.get(runId);
         if (span) {
             try {
-                const tokenUsage = output?.llmOutput?.token_usage || output?.token_usage;
+                // Try multiple ways to get token usage
+                // OpenAI format
+                const tokenUsage = output?.llmOutput?.token_usage ||
+                    output?.token_usage ||
+                    output?.metadata?.token_usage;
                 if (tokenUsage) {
-                    span.setAttribute('prompt_tokens', tokenUsage.prompt_tokens);
-                    span.setAttribute('completion_tokens', tokenUsage.completion_tokens);
-                    span.setAttribute('total_tokens', (tokenUsage.prompt_tokens || 0) + (tokenUsage.completion_tokens || 0));
+                    if (tokenUsage.prompt_tokens !== undefined) {
+                        span.setAttribute('llm.tokens.prompt', tokenUsage.prompt_tokens);
+                    }
+                    if (tokenUsage.completion_tokens !== undefined) {
+                        span.setAttribute('llm.tokens.completion', tokenUsage.completion_tokens);
+                    }
+                    const totalTokens = (tokenUsage.prompt_tokens || 0) + (tokenUsage.completion_tokens || 0);
+                    if (totalTokens > 0) {
+                        span.setAttribute('llm.tokens.total', totalTokens);
+                    }
                 }
+                // Capture output text length
                 if (output?.text) {
                     span.setAttribute('output_length', output.text.length);
                 }
+                else if (output?.generations && Array.isArray(output.generations)) {
+                    const firstGeneration = output.generations[0];
+                    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
+                if (output?.llmOutput?.finish_reason) {
+                    span.setAttribute('finish_reason', output.llmOutput.finish_reason);
+                }
             }
             catch (error) {
                 // Silently fail on attribute setting
@@ -84,14 +184,22 @@ class TraciaCallbackHandler {
      */
     async handleChainStart(chain, inputs, runId, _parentRunId) {
         const chainName = chain.name || chain._chainType || 'chain';
-        const span = this.tracer.startSpan(`chain:${chainName}`, {
-            attributes: {
-                type: 'chain',
-                chain_name: chainName,
-                chain_type: chain._chainType,
-                input_keys: Object.keys(inputs).join(','),
-            },
-        });
+        const attributes = {
+            type: 'chain',
+            chain_name: chainName,
+            chain_type: chain._chainType,
+            input_keys: Object.keys(inputs || {}).join(','),
+            input_count: Object.keys(inputs || {}).length,
+        };
+        // Capture total input length
+        try {
+            const inputStr = JSON.stringify(inputs);
+            attributes.input_length = inputStr.length;
+        }
+        catch (error) {
+            // Silently fail
+        }
+        const span = this.tracer.startSpan(`chain:${chainName}`, { attributes });
         this.spanStack.set(runId, span);
     }
     /**
@@ -129,14 +237,22 @@ class TraciaCallbackHandler {
      */
     async handleToolStart(tool, input, runId, _parentRunId) {
         const toolName = tool.name || 'unknown-tool';
-        const span = this.tracer.startSpan(`tool:${toolName}`, {
-            attributes: {
-                type: 'tool',
-                tool_name: toolName,
-                tool_description: tool.description,
-                input_length: typeof input === 'string' ? input.length : 0,
-            },
-        });
+        const attributes = {
+            type: 'tool',
+            tool_name: toolName,
+            tool_description: tool.description || '',
+            input_length: typeof input === 'string' ? input.length : (typeof input === 'object' ? JSON.stringify(input).length : 0),
+        };
+        // Try to capture structured input
+        try {
+            if (typeof input === 'object') {
+                attributes.input_keys = Object.keys(input).join(',');
+            }
+        }
+        catch (error) {
+            // Silently fail
+        }
+        const span = this.tracer.startSpan(`tool:${toolName}`, { attributes });
         this.spanStack.set(runId, span);
     }
     /**
@@ -196,6 +312,41 @@ 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 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;
 //# 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;IAuMpD,CAAC;IArMC;;OAEG;IACI,KAAK,CAAC,cAAc,CACzB,GAAQ,EACR,OAAiB,EACjB,KAAa,EACb,YAAqB;QAErB,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,EAAE;YACxC,UAAU,EAAE;gBACV,IAAI,EAAE,KAAK;gBACX,KAAK,EAAE,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,UAAU,IAAI,SAAS;gBAC9C,YAAY,EAAE,OAAO,CAAC,MAAM;gBAC5B,mBAAmB,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,IAAI,CAAC;aAC7C;SACF,CAAC,CAAC;QAEH,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,MAAM,UAAU,GAAG,MAAM,EAAE,SAAS,EAAE,WAAW,IAAI,MAAM,EAAE,WAAW,CAAC;gBACzE,IAAI,UAAU,EAAE,CAAC;oBACf,IAAI,CAAC,YAAY,CAAC,eAAe,EAAE,UAAU,CAAC,aAAa,CAAC,CAAC;oBAC7D,IAAI,CAAC,YAAY,CAAC,mBAAmB,EAAE,UAAU,CAAC,iBAAiB,CAAC,CAAC;oBACrE,IAAI,CAAC,YAAY,CACf,cAAc,EACd,CAAC,UAAU,CAAC,aAAa,IAAI,CAAC,CAAC,GAAG,CAAC,UAAU,CAAC,iBAAiB,IAAI,CAAC,CAAC,CACtE,CAAC;gBACJ,CAAC;gBAED,IAAI,MAAM,EAAE,IAAI,EAAE,CAAC;oBACjB,IAAI,CAAC,YAAY,CAAC,eAAe,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACzD,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;QAC5D,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,SAAS,EAAE,EAAE;YACvD,UAAU,EAAE;gBACV,IAAI,EAAE,OAAO;gBACb,UAAU,EAAE,SAAS;gBACrB,UAAU,EAAE,KAAK,CAAC,UAAU;gBAC5B,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;aAC1C;SACF,CAAC,CAAC;QAEH,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;QAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,QAAQ,QAAQ,EAAE,EAAE;YACrD,UAAU,EAAE;gBACV,IAAI,EAAE,MAAM;gBACZ,SAAS,EAAE,QAAQ;gBACnB,gBAAgB,EAAE,IAAI,CAAC,WAAW;gBAClC,YAAY,EAAE,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;aAC3D;SACF,CAAC,CAAC;QAEH,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;AAzMD,sDAyMC"}
+{"version":3,"file":"langchain-callback.js","sourceRoot":"","sources":["../../src/integrations/langchain-callback.ts"],"names":[],"mappings":";AAAA;;;;GAIG;;;AAGH,kCAAoC;AAEpC,sEAAsE;AACtE,IAAI,mBAAmB,GAAQ;IAAA;QAC7B,gBAAW,GAAG,KAAK,CAAC;QACpB,gBAAW,GAAG,KAAK,CAAC;QACpB,cAAS,GAAG,KAAK,CAAC;QAClB,oBAAe,GAAG,KAAK,CAAC;QACxB,kBAAa,GAAG,KAAK,CAAC;IACxB,CAAC;CAAA,CAAC;AAEF,IAAI,CAAC;IACH,uBAAuB;IACvB,8DAA8D;IAC9D,MAAM,EAAE,mBAAmB,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,gCAAgC,CAAC,CAAC;IACnF,mBAAmB,GAAG,OAAO,CAAC;AAChC,CAAC;AAAC,MAAM,CAAC;IACP,IAAI,CAAC;QACH,+BAA+B;QAC/B,8DAA8D;QAC9D,MAAM,EAAE,mBAAmB,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,qBAAqB,CAAC,CAAC;QACxE,mBAAmB,GAAG,OAAO,CAAC;IAChC,CAAC;IAAC,MAAM,CAAC;QACP,yEAAyE;IAC3E,CAAC;AACH,CAAC;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,MAAa,qBAAsB,SAAQ,mBAAmB;IAA9D;;QACU,WAAM,GAAG,IAAA,gBAAS,EAAC,WAAW,CAAC,CAAC;QAChC,cAAS,GAAuB,IAAI,GAAG,EAAE,CAAC;IA8VpD,CAAC;IA5VC;;;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,gBAAgB;gBAChB,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;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;;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,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;AAhWD,sDAgWC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@traccia2/sdk",
-  "version": "0.0.3",
+  "version": "0.0.5",
   "description": "Tracing SDK for agent applications",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/integrations/langchain-callback.ts

@@ -7,28 +7,66 @@
 import { ISpan } from '../types';
 import { getTracer } from '../auto';
 
+// Try to import BaseCallbackHandler from different LangChain versions
+let BaseCallbackHandler: any = class {
+  ignoreAgent = false;
+  ignoreChain = false;
+  ignoreLLM = false;
+  ignoreRetriever = false;
+  ignoreToolUse = false;
+};
+
+try {
+  // Try LangChain 0.2.x+
+  // eslint-disable-next-line @typescript-eslint/no-var-requires
+  const { BaseCallbackHandler: Handler } = require('@langchain/core/callbacks/base');
+  BaseCallbackHandler = Handler;
+} catch {
+  try {
+    // Try older LangChain versions
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
+    const { BaseCallbackHandler: Handler } = require('langchain/callbacks');
+    BaseCallbackHandler = Handler;
+  } catch {
+    // Fallback to basic implementation - user likely has LangChain installed
+  }
+}
+
 /**
  * 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 { TraciaCallbackHandler } from '@traccia/sdk/integrations';
  *
  * const handler = new TraciaCallbackHandler();
- * const chain = new LLMChain({
- *   llm: new ChatOpenAI(),
- *   callbacks: [handler],
- * });
+ * 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 TraciaCallbackHandler extends BaseCallbackHandler {
   private tracer = getTracer('langchain');
   private spanStack: Map<string, ISpan> = new Map();
 
+  /**
+   * Extract model name from LLM instance, checking multiple property locations.
+   * Different LLM implementations store the model name in different properties.
+   */
+  private extractModelName(llm: any): string {
+    // Check common model name properties
+    if (llm.modelName) return llm.modelName;           // ChatOpenAI, ChatAnthropic, etc.
+    if (llm.model) return llm.model;                   // Ollama, etc.
+    if (llm.name && !llm.name.startsWith('langchain')) return llm.name;  // Generic name
+    if (llm._modelType) return llm._modelType;         // Fallback to type
+    if (llm.client?.model) return llm.client.model;    // Nested model property
+    return 'unknown';
+  }
+
   /**
    * Handle LLM start - called when an LLM begins execution.
    */
@@ -38,15 +76,44 @@ export class TraciaCallbackHandler {
     runId: string,
     _parentRunId?: string
   ): Promise<void> {
-    const span = this.tracer.startSpan('llm', {
-      attributes: {
-        type: 'llm',
-        model: llm.name || llm._modelType || 'unknown',
-        prompt_count: prompts.length,
-        first_prompt_length: prompts[0]?.length || 0,
-      },
-    });
+    const modelName = this.extractModelName(llm);
+    
+    const attributes: Record<string, any> = {
+      type: 'llm',
+      model: modelName,
+      prompt_count: prompts.length,
+      first_prompt_length: prompts[0]?.length || 0,
+    };
+
+    // Capture temperature if available
+    if (llm.temperature !== undefined) {
+      attributes.temperature = llm.temperature;
+    }
+
+    // Capture max tokens if available
+    if (llm.maxTokens !== undefined) {
+      attributes.max_tokens = llm.maxTokens;
+    }
+    if (llm.max_tokens !== undefined) {
+      attributes.max_tokens = llm.max_tokens;
+    }
+
+    // Capture top_p if available
+    if (llm.topP !== undefined) {
+      attributes.top_p = llm.topP;
+    }
 
+    // Capture top_k if available
+    if (llm.topK !== undefined) {
+      attributes.top_k = llm.topK;
+    }
+
+    // Capture base URL for local models (Ollama, etc.)
+    if (llm.baseUrl) {
+      attributes.base_url = llm.baseUrl;
+    }
+
+    const span = this.tracer.startSpan('llm', { attributes });
     this.spanStack.set(runId, span);
   }
 
@@ -57,18 +124,47 @@ export class TraciaCallbackHandler {
     const span = this.spanStack.get(runId);
     if (span) {
       try {
-        const tokenUsage = output?.llmOutput?.token_usage || output?.token_usage;
+        // Try multiple ways to get token usage
+        // OpenAI format
+        const tokenUsage = 
+          output?.llmOutput?.token_usage || 
+          output?.token_usage || 
+          output?.metadata?.token_usage;
+
         if (tokenUsage) {
-          span.setAttribute('prompt_tokens', tokenUsage.prompt_tokens);
-          span.setAttribute('completion_tokens', tokenUsage.completion_tokens);
-          span.setAttribute(
-            'total_tokens',
-            (tokenUsage.prompt_tokens || 0) + (tokenUsage.completion_tokens || 0)
-          );
+          if (tokenUsage.prompt_tokens !== undefined) {
+            span.setAttribute('llm.tokens.prompt', tokenUsage.prompt_tokens);
+          }
+          if (tokenUsage.completion_tokens !== undefined) {
+            span.setAttribute('llm.tokens.completion', tokenUsage.completion_tokens);
+          }
+          const totalTokens = (tokenUsage.prompt_tokens || 0) + (tokenUsage.completion_tokens || 0);
+          if (totalTokens > 0) {
+            span.setAttribute('llm.tokens.total', totalTokens);
+          }
         }
 
+        // Capture output text length
         if (output?.text) {
           span.setAttribute('output_length', output.text.length);
+        } else if (output?.generations && Array.isArray(output.generations)) {
+          const firstGeneration = output.generations[0];
+          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
+        if (output?.llmOutput?.finish_reason) {
+          span.setAttribute('finish_reason', output.llmOutput.finish_reason);
         }
       } catch (error) {
         // Silently fail on attribute setting
@@ -101,15 +197,24 @@ export class TraciaCallbackHandler {
     _parentRunId?: string
   ): Promise<void> {
     const chainName = chain.name || chain._chainType || 'chain';
-    const span = this.tracer.startSpan(`chain:${chainName}`, {
-      attributes: {
-        type: 'chain',
-        chain_name: chainName,
-        chain_type: chain._chainType,
-        input_keys: Object.keys(inputs).join(','),
-      },
-    });
+    
+    const attributes: Record<string, any> = {
+      type: 'chain',
+      chain_name: chainName,
+      chain_type: chain._chainType,
+      input_keys: Object.keys(inputs || {}).join(','),
+      input_count: Object.keys(inputs || {}).length,
+    };
+
+    // Capture total input length
+    try {
+      const inputStr = JSON.stringify(inputs);
+      attributes.input_length = inputStr.length;
+    } catch (error) {
+      // Silently fail
+    }
 
+    const span = this.tracer.startSpan(`chain:${chainName}`, { attributes });
     this.spanStack.set(runId, span);
   }
 
@@ -155,15 +260,24 @@ export class TraciaCallbackHandler {
     _parentRunId?: string
   ): Promise<void> {
     const toolName = tool.name || 'unknown-tool';
-    const span = this.tracer.startSpan(`tool:${toolName}`, {
-      attributes: {
-        type: 'tool',
-        tool_name: toolName,
-        tool_description: tool.description,
-        input_length: typeof input === 'string' ? input.length : 0,
-      },
-    });
+    
+    const attributes: Record<string, any> = {
+      type: 'tool',
+      tool_name: toolName,
+      tool_description: tool.description || '',
+      input_length: typeof input === 'string' ? input.length : (typeof input === 'object' ? JSON.stringify(input).length : 0),
+    };
+
+    // Try to capture structured input
+    try {
+      if (typeof input === 'object') {
+        attributes.input_keys = Object.keys(input).join(',');
+      }
+    } catch (error) {
+      // Silently fail
+    }
 
+    const span = this.tracer.startSpan(`tool:${toolName}`, { attributes });
     this.spanStack.set(runId, span);
   }
 
@@ -226,4 +340,66 @@ 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 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);
+  }
 }
+