opik-langchain 1.0.1 → 1.0.2

package/dist/index.cjs

@@ -0,0 +1 @@
+'use strict';var base=require('@langchain/core/callbacks/base'),opik=require('opik'),messages=require('@langchain/core/messages');var f=r=>typeof r=="object"&&r!==null,j=r=>f(r)&&"content"in r,V=r=>f(r)&&"messages"in r,x=r=>f(r)&&"value"in r,z=r=>f(r)&&"kwargs"in r,k=(...r)=>r.find(e=>e!=null),_=r=>Object.fromEntries(Object.entries(r).filter(([,e])=>!(e==null||Array.isArray(e)&&e.length===0||f(e)&&Object.keys(e).length===0))),M=r=>{try{return JSON.parse(r)}catch{return {value:r}}},R=r=>{var s;let e=(s=r.name)!=null?s:r.getType();r.getType()==="human"?e="user":r.getType()==="ai"?e="assistant":r.getType()==="system"&&(e="system");let n={},t=r;return "tool_calls"in t&&(n.tool_calls=t.tool_calls),"status"in t&&(n.status=t.status),"artifact"in t&&(n.artifact=t.artifact),_({content:r.content,role:e,...n})},$=r=>({messages:r.flatMap(n=>n.map(R))}),S=r=>"message"in r?R(r.message):"text"in r&&r.text?r.text:r,E=r=>({generations:r.flatMap(n=>Array.isArray(n)?n.map(S):S(n))}),A=r=>r instanceof messages.ToolMessage?R(r):void 0,y=(r,e,n)=>{let t=_({model:k(e==null?void 0:e.model,n==null?void 0:n.ls_model_name,r.name),temperature:k(e==null?void 0:e.temperature,n==null?void 0:n.ls_temperature),top_p:k(e==null?void 0:e.top_p,e==null?void 0:e.topP),top_k:k(e==null?void 0:e.top_k,e==null?void 0:e.topK),max_tokens:k(e==null?void 0:e.max_tokens,e==null?void 0:e.maxOutputTokens),frequency_penalty:e==null?void 0:e.frequency_penalty,presence_penalty:e==null?void 0:e.presence_penalty,response_format:e==null?void 0:e.response_format,tool_choice:e==null?void 0:e.tool_choice,function_call:e==null?void 0:e.function_call,n:e==null?void 0:e.n,stop:k(e==null?void 0:e.stop,e==null?void 0:e.stop_sequence)});return Object.keys(t).length?t:e},T=r=>{if(r==null)return {value:r};if(typeof r=="string"||typeof r=="number"||typeof r=="boolean")return {value:r};if(!f(r))return {value:r};if(j(r)&&r.content!==void 0){let e=r.content;return f(e)?e:{value:e}}if(V(r)&&r.messages!==void 0)return {messages:Array.isArray(r.messages)?r.messages.map(T):[T(r.messages)]};if(x(r)&&r.value!==void 0){let e=r.value;return f(e)?e:{value:e}}return z(r)&&r.kwargs!==void 0?T(r.kwargs):Object.fromEntries(Object.entries(r).map(([e,n])=>[e,T(n)]))},L=r=>{try{let e=(Array.isArray(r)?r:[r]).filter(n=>n!=null).flatMap(T);return e.length===1?e[0]:e.length>1?{values:e}:{}}catch(e){return console.warn("Error processing chain outputs:",e),{error:String(e)}}},C=r=>{try{let e=(Array.isArray(r)?r:[r]).filter(n=>n!=null).flatMap(T);return e.length===1?e[0]:e.length>1?{values:e}:{}}catch(e){return console.warn("Error processing chain inputs:",e),{error:String(e)}}};var b=class extends base.BaseCallbackHandler{constructor(n){var t;super();this.name="OpikCallbackHandler";this.spansMap=new Map;this.options={...n},this.client=(t=n==null?void 0:n.client)!=null?t:new opik.Opik,n!=null&&n.projectName&&(this.client.config.projectName=n==null?void 0:n.projectName);}startTracing({runId:n,parentRunId:t,name:s,input:i,tags:o,metadata:l,type:d}){let g=l==null?void 0:l.ls_provider,p=l==null?void 0:l.ls_model_name;if(!t&&n===this.rootTraceId)return;if(t||(this.rootTraceId=n,this.rootTrace=this.client.trace({name:s,input:i,tags:this.options.tags,metadata:l})),!this.rootTrace){opik.logger.debug(`handleChainStart error ${n} has no parent`);return}let c;this.rootTraceId&&this.rootTraceId!==t&&(c=t);let h=this.spansMap.get(n),a;h?a=h.update({input:C(i),tags:o,metadata:l,model:p,provider:g}):a=this.rootTrace.span({type:d||opik.OpikSpanType.General,name:s,input:C(i),tags:o,metadata:l,model:p,provider:g,parentSpanId:c}),this.spansMap.set(n,a);}endTracing({runId:n,output:t,error:s,tags:i,usage:o,metadata:l}){var p,c,h;let d;s&&(opik.logger.debug(`End tracing because of error ${s.message}`),d={message:s.message,exceptionType:s.name,traceback:(p=s.stack)!=null?p:""});let g=this.spansMap.get(n);if(!g){opik.logger.debug(`handleChainEnd span ${n} has not found`);return}g.update({output:t,errorInfo:d,tags:i,usage:o,metadata:l}),g.end(),this.spansMap.delete(n),n===this.rootTraceId&&((c=this.rootTrace)==null||c.update({output:t}),this.rootTraceId=void 0,(h=this.rootTrace)==null||h.end(),this.rootTrace=void 0);}async handleChatModelStart(n,t,s,i,o,l,d,g){var p,c,h;opik.logger.debug(`handleChatModelStart runId - ${s}, parentRunId ${i}`),this.startTracing({runId:s,parentRunId:i,name:(c=g!=null?g:(p=n.id.at(-1))==null?void 0:p.toString())!=null?c:"Chat Model",type:opik.OpikSpanType.Llm,input:$(t),tags:l,metadata:{...d,...y(n,(o==null?void 0:o.invocation_params)||{},d),tools:(h=o==null?void 0:o.invocation_params)==null?void 0:h.tools}});}async handleLLMStart(n,t,s,i,o,l,d,g){var p,c;opik.logger.debug(`handleLLMStart runId - ${s}, parentRunId ${i}`),this.startTracing({runId:s,parentRunId:i,name:(c=g!=null?g:(p=n.id.at(-1))==null?void 0:p.toString())!=null?c:"LLM",type:opik.OpikSpanType.Llm,input:{prompts:t},tags:l,metadata:{...d,...y(n,(o==null?void 0:o.invocation_params)||{},d)}});}async handleLLMError(n,t,s,i){opik.logger.debug(`handleLLMError runId - ${t}, parentRunId ${s}`),this.endTracing({runId:t,error:n,tags:i});}async handleLLMEnd(n,t,s,i){opik.logger.debug(`handleLLMEnd runId - ${t}, parentRunId ${s}`);let{llmOutput:o,generations:l,...d}=n,g=(o==null?void 0:o.tokenUsage)||(o==null?void 0:o.estimatedTokens)||{};this.endTracing({runId:t,output:E(l),usage:{prompt_tokens:g.completionTokens,completion_tokens:g.promptTokens,total_tokens:g.totalTokens},tags:i,metadata:d});}async handleChainStart(n,t,s,i,o,l,d,g){var h,a;if(opik.logger.debug(`handleChainStart runId - ${s}, parentRunId ${i}`),o!=null&&o.includes("langsmith:hidden"))return;let p=(a=g!=null?g:(h=n.id.at(-1))==null?void 0:h.toString())!=null?a:"Chain",c={runId:s,parentRunId:i,name:p,input:C(t),tags:this.options.tags,metadata:{...l,...y(n,{},l)}};i||(c.metadata={...l,...this.options.metadata}),this.startTracing(c);}async handleChainError(n,t,s,i){opik.logger.debug(`handleChainError runId - ${t}, parentRunId ${s}`),this.endTracing({runId:t,tags:i,error:n});}async handleChainEnd(n,t,s,i){opik.logger.debug(`handleChainEnd runId - ${t}, parentRunId ${s}`),this.endTracing({runId:t,output:L(n),tags:i});}async handleToolStart(n,t,s,i,o,l,d){var g,p;opik.logger.debug(`handleToolStart runId - ${s}, parentRunId ${i}`),this.startTracing({runId:s,parentRunId:i,name:(p=d!=null?d:(g=n.id.at(-1))==null?void 0:g.toString())!=null?p:"Tool",input:M(t),tags:o,type:opik.OpikSpanType.Tool,metadata:{...l,...y(n,{},l)}});}async handleToolError(n,t,s,i){opik.logger.debug(`handleToolError runId - ${t}, parentRunId ${s}`),this.endTracing({runId:t,tags:i,error:n});}async handleToolEnd(n,t,s,i){opik.logger.debug(`handleToolEnd runId - ${t}, parentRunId ${s}`),this.endTracing({runId:t,output:A(n),tags:i});}async handleAgentAction(n,t,s,i){opik.logger.debug(`handleAgentAction runId - ${t}, parentRunId ${s}`),this.startTracing({runId:t,parentRunId:s,name:n.tool,input:n,tags:i});}async handleAgentEnd(n,t,s,i){opik.logger.debug(`handleAgentEnd runId - ${t}, parentRunId ${s}`),this.endTracing({runId:t,output:n,tags:i});}async handleRetrieverStart(n,t,s,i,o,l,d){var g,p;opik.logger.debug(`handleRetrieverStart runId - ${s}, parentRunId ${i}`),this.startTracing({runId:s,parentRunId:i,name:(p=d!=null?d:(g=n.id.at(-1))==null?void 0:g.toString())!=null?p:"Retriever",type:opik.OpikSpanType.Tool,input:{query:t},tags:o,metadata:{...l,...y(n,{},l)}});}async handleRetrieverEnd(n,t,s,i){opik.logger.debug(`handleRetrieverEnd runId - ${t}, parentRunId ${s}`),this.endTracing({runId:t,output:{documents:n},tags:i});}async handleRetrieverError(n,t,s,i){opik.logger.debug(`handleRetrieverError runId - ${t}, parentRunId ${s}`),this.endTracing({runId:t,error:n,tags:i});}async flushAsync(){return this.client.flush()}};exports.OpikCallbackHandler=b;

package/dist/index.d.cts

@@ -0,0 +1,55 @@
+import { BaseCallbackHandler, BaseCallbackHandlerInput } from '@langchain/core/callbacks/base';
+import { Serialized } from '@langchain/core/load/serializable';
+import { ChainValues } from '@langchain/core/utils/types';
+import { Opik } from 'opik';
+import { RunnableConfig } from '@langchain/core/runnables';
+import { LLMResult, ChatResult } from '@langchain/core/outputs';
+import { BaseMessage } from '@langchain/core/messages';
+import { AgentAction, AgentFinish } from '@langchain/core/agents';
+
+type JsonNode = Record<string, unknown>;
+interface OpikCallbackHandlerOptions {
+    tags?: [];
+    metadata?: JsonNode;
+    projectName?: string;
+    client?: Opik;
+}
+declare class OpikCallbackHandler extends BaseCallbackHandler implements BaseCallbackHandlerInput {
+    name: string;
+    private options;
+    private client;
+    private rootTraceId?;
+    private rootTrace?;
+    private spansMap;
+    constructor(options?: Partial<OpikCallbackHandlerOptions>);
+    private startTracing;
+    private endTracing;
+    handleChatModelStart(llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string, extraParams?: {
+        options: RunnableConfig;
+        invocation_params?: Record<string, unknown>;
+        batch_size: number;
+        cache?: boolean;
+    }, tags?: string[], metadata?: Record<string, unknown>, runName?: string): Promise<void>;
+    handleLLMStart(llm: Serialized, prompts: string[], runId: string, parentRunId?: string, extraParams?: {
+        options: RunnableConfig;
+        invocation_params?: Record<string, unknown>;
+        batch_size: number;
+        cache?: boolean;
+    }, tags?: string[], metadata?: Record<string, unknown>, runName?: string): Promise<void>;
+    handleLLMError(error: Error, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleLLMEnd(output: LLMResult | ChatResult, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleChainStart(chain: Serialized, input: ChainValues, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, runType?: string, runName?: string): Promise<void>;
+    handleChainError(error: Error, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleChainEnd(output: ChainValues, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleToolStart(tool: Serialized, input: string, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, runName?: string): Promise<void>;
+    handleToolError(error: Error, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleToolEnd(output: unknown, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleAgentAction(action: AgentAction, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleAgentEnd(action: AgentFinish, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleRetrieverStart(retriever: Serialized, query: string, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, name?: string): Promise<void>;
+    handleRetrieverEnd(documents: unknown[], runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleRetrieverError(error: Error, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    flushAsync(): Promise<void>;
+}
+
+export { OpikCallbackHandler };

package/dist/index.d.ts

@@ -0,0 +1,55 @@
+import { BaseCallbackHandler, BaseCallbackHandlerInput } from '@langchain/core/callbacks/base';
+import { Serialized } from '@langchain/core/load/serializable';
+import { ChainValues } from '@langchain/core/utils/types';
+import { Opik } from 'opik';
+import { RunnableConfig } from '@langchain/core/runnables';
+import { LLMResult, ChatResult } from '@langchain/core/outputs';
+import { BaseMessage } from '@langchain/core/messages';
+import { AgentAction, AgentFinish } from '@langchain/core/agents';
+
+type JsonNode = Record<string, unknown>;
+interface OpikCallbackHandlerOptions {
+    tags?: [];
+    metadata?: JsonNode;
+    projectName?: string;
+    client?: Opik;
+}
+declare class OpikCallbackHandler extends BaseCallbackHandler implements BaseCallbackHandlerInput {
+    name: string;
+    private options;
+    private client;
+    private rootTraceId?;
+    private rootTrace?;
+    private spansMap;
+    constructor(options?: Partial<OpikCallbackHandlerOptions>);
+    private startTracing;
+    private endTracing;
+    handleChatModelStart(llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string, extraParams?: {
+        options: RunnableConfig;
+        invocation_params?: Record<string, unknown>;
+        batch_size: number;
+        cache?: boolean;
+    }, tags?: string[], metadata?: Record<string, unknown>, runName?: string): Promise<void>;
+    handleLLMStart(llm: Serialized, prompts: string[], runId: string, parentRunId?: string, extraParams?: {
+        options: RunnableConfig;
+        invocation_params?: Record<string, unknown>;
+        batch_size: number;
+        cache?: boolean;
+    }, tags?: string[], metadata?: Record<string, unknown>, runName?: string): Promise<void>;
+    handleLLMError(error: Error, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleLLMEnd(output: LLMResult | ChatResult, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleChainStart(chain: Serialized, input: ChainValues, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, runType?: string, runName?: string): Promise<void>;
+    handleChainError(error: Error, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleChainEnd(output: ChainValues, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleToolStart(tool: Serialized, input: string, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, runName?: string): Promise<void>;
+    handleToolError(error: Error, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleToolEnd(output: unknown, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleAgentAction(action: AgentAction, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleAgentEnd(action: AgentFinish, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleRetrieverStart(retriever: Serialized, query: string, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, name?: string): Promise<void>;
+    handleRetrieverEnd(documents: unknown[], runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleRetrieverError(error: Error, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    flushAsync(): Promise<void>;
+}
+
+export { OpikCallbackHandler };

package/dist/index.js

@@ -0,0 +1 @@
+import {BaseCallbackHandler}from'@langchain/core/callbacks/base';import {Opik,logger,OpikSpanType}from'opik';import {ToolMessage}from'@langchain/core/messages';var f=r=>typeof r=="object"&&r!==null,j=r=>f(r)&&"content"in r,V=r=>f(r)&&"messages"in r,x=r=>f(r)&&"value"in r,z=r=>f(r)&&"kwargs"in r,k=(...r)=>r.find(e=>e!=null),_=r=>Object.fromEntries(Object.entries(r).filter(([,e])=>!(e==null||Array.isArray(e)&&e.length===0||f(e)&&Object.keys(e).length===0))),M=r=>{try{return JSON.parse(r)}catch{return {value:r}}},R=r=>{var s;let e=(s=r.name)!=null?s:r.getType();r.getType()==="human"?e="user":r.getType()==="ai"?e="assistant":r.getType()==="system"&&(e="system");let n={},t=r;return "tool_calls"in t&&(n.tool_calls=t.tool_calls),"status"in t&&(n.status=t.status),"artifact"in t&&(n.artifact=t.artifact),_({content:r.content,role:e,...n})},$=r=>({messages:r.flatMap(n=>n.map(R))}),S=r=>"message"in r?R(r.message):"text"in r&&r.text?r.text:r,E=r=>({generations:r.flatMap(n=>Array.isArray(n)?n.map(S):S(n))}),A=r=>r instanceof ToolMessage?R(r):void 0,y=(r,e,n)=>{let t=_({model:k(e==null?void 0:e.model,n==null?void 0:n.ls_model_name,r.name),temperature:k(e==null?void 0:e.temperature,n==null?void 0:n.ls_temperature),top_p:k(e==null?void 0:e.top_p,e==null?void 0:e.topP),top_k:k(e==null?void 0:e.top_k,e==null?void 0:e.topK),max_tokens:k(e==null?void 0:e.max_tokens,e==null?void 0:e.maxOutputTokens),frequency_penalty:e==null?void 0:e.frequency_penalty,presence_penalty:e==null?void 0:e.presence_penalty,response_format:e==null?void 0:e.response_format,tool_choice:e==null?void 0:e.tool_choice,function_call:e==null?void 0:e.function_call,n:e==null?void 0:e.n,stop:k(e==null?void 0:e.stop,e==null?void 0:e.stop_sequence)});return Object.keys(t).length?t:e},T=r=>{if(r==null)return {value:r};if(typeof r=="string"||typeof r=="number"||typeof r=="boolean")return {value:r};if(!f(r))return {value:r};if(j(r)&&r.content!==void 0){let e=r.content;return f(e)?e:{value:e}}if(V(r)&&r.messages!==void 0)return {messages:Array.isArray(r.messages)?r.messages.map(T):[T(r.messages)]};if(x(r)&&r.value!==void 0){let e=r.value;return f(e)?e:{value:e}}return z(r)&&r.kwargs!==void 0?T(r.kwargs):Object.fromEntries(Object.entries(r).map(([e,n])=>[e,T(n)]))},L=r=>{try{let e=(Array.isArray(r)?r:[r]).filter(n=>n!=null).flatMap(T);return e.length===1?e[0]:e.length>1?{values:e}:{}}catch(e){return console.warn("Error processing chain outputs:",e),{error:String(e)}}},C=r=>{try{let e=(Array.isArray(r)?r:[r]).filter(n=>n!=null).flatMap(T);return e.length===1?e[0]:e.length>1?{values:e}:{}}catch(e){return console.warn("Error processing chain inputs:",e),{error:String(e)}}};var b=class extends BaseCallbackHandler{constructor(n){var t;super();this.name="OpikCallbackHandler";this.spansMap=new Map;this.options={...n},this.client=(t=n==null?void 0:n.client)!=null?t:new Opik,n!=null&&n.projectName&&(this.client.config.projectName=n==null?void 0:n.projectName);}startTracing({runId:n,parentRunId:t,name:s,input:i,tags:o,metadata:l,type:d}){let g=l==null?void 0:l.ls_provider,p=l==null?void 0:l.ls_model_name;if(!t&&n===this.rootTraceId)return;if(t||(this.rootTraceId=n,this.rootTrace=this.client.trace({name:s,input:i,tags:this.options.tags,metadata:l})),!this.rootTrace){logger.debug(`handleChainStart error ${n} has no parent`);return}let c;this.rootTraceId&&this.rootTraceId!==t&&(c=t);let h=this.spansMap.get(n),a;h?a=h.update({input:C(i),tags:o,metadata:l,model:p,provider:g}):a=this.rootTrace.span({type:d||OpikSpanType.General,name:s,input:C(i),tags:o,metadata:l,model:p,provider:g,parentSpanId:c}),this.spansMap.set(n,a);}endTracing({runId:n,output:t,error:s,tags:i,usage:o,metadata:l}){var p,c,h;let d;s&&(logger.debug(`End tracing because of error ${s.message}`),d={message:s.message,exceptionType:s.name,traceback:(p=s.stack)!=null?p:""});let g=this.spansMap.get(n);if(!g){logger.debug(`handleChainEnd span ${n} has not found`);return}g.update({output:t,errorInfo:d,tags:i,usage:o,metadata:l}),g.end(),this.spansMap.delete(n),n===this.rootTraceId&&((c=this.rootTrace)==null||c.update({output:t}),this.rootTraceId=void 0,(h=this.rootTrace)==null||h.end(),this.rootTrace=void 0);}async handleChatModelStart(n,t,s,i,o,l,d,g){var p,c,h;logger.debug(`handleChatModelStart runId - ${s}, parentRunId ${i}`),this.startTracing({runId:s,parentRunId:i,name:(c=g!=null?g:(p=n.id.at(-1))==null?void 0:p.toString())!=null?c:"Chat Model",type:OpikSpanType.Llm,input:$(t),tags:l,metadata:{...d,...y(n,(o==null?void 0:o.invocation_params)||{},d),tools:(h=o==null?void 0:o.invocation_params)==null?void 0:h.tools}});}async handleLLMStart(n,t,s,i,o,l,d,g){var p,c;logger.debug(`handleLLMStart runId - ${s}, parentRunId ${i}`),this.startTracing({runId:s,parentRunId:i,name:(c=g!=null?g:(p=n.id.at(-1))==null?void 0:p.toString())!=null?c:"LLM",type:OpikSpanType.Llm,input:{prompts:t},tags:l,metadata:{...d,...y(n,(o==null?void 0:o.invocation_params)||{},d)}});}async handleLLMError(n,t,s,i){logger.debug(`handleLLMError runId - ${t}, parentRunId ${s}`),this.endTracing({runId:t,error:n,tags:i});}async handleLLMEnd(n,t,s,i){logger.debug(`handleLLMEnd runId - ${t}, parentRunId ${s}`);let{llmOutput:o,generations:l,...d}=n,g=(o==null?void 0:o.tokenUsage)||(o==null?void 0:o.estimatedTokens)||{};this.endTracing({runId:t,output:E(l),usage:{prompt_tokens:g.completionTokens,completion_tokens:g.promptTokens,total_tokens:g.totalTokens},tags:i,metadata:d});}async handleChainStart(n,t,s,i,o,l,d,g){var h,a;if(logger.debug(`handleChainStart runId - ${s}, parentRunId ${i}`),o!=null&&o.includes("langsmith:hidden"))return;let p=(a=g!=null?g:(h=n.id.at(-1))==null?void 0:h.toString())!=null?a:"Chain",c={runId:s,parentRunId:i,name:p,input:C(t),tags:this.options.tags,metadata:{...l,...y(n,{},l)}};i||(c.metadata={...l,...this.options.metadata}),this.startTracing(c);}async handleChainError(n,t,s,i){logger.debug(`handleChainError runId - ${t}, parentRunId ${s}`),this.endTracing({runId:t,tags:i,error:n});}async handleChainEnd(n,t,s,i){logger.debug(`handleChainEnd runId - ${t}, parentRunId ${s}`),this.endTracing({runId:t,output:L(n),tags:i});}async handleToolStart(n,t,s,i,o,l,d){var g,p;logger.debug(`handleToolStart runId - ${s}, parentRunId ${i}`),this.startTracing({runId:s,parentRunId:i,name:(p=d!=null?d:(g=n.id.at(-1))==null?void 0:g.toString())!=null?p:"Tool",input:M(t),tags:o,type:OpikSpanType.Tool,metadata:{...l,...y(n,{},l)}});}async handleToolError(n,t,s,i){logger.debug(`handleToolError runId - ${t}, parentRunId ${s}`),this.endTracing({runId:t,tags:i,error:n});}async handleToolEnd(n,t,s,i){logger.debug(`handleToolEnd runId - ${t}, parentRunId ${s}`),this.endTracing({runId:t,output:A(n),tags:i});}async handleAgentAction(n,t,s,i){logger.debug(`handleAgentAction runId - ${t}, parentRunId ${s}`),this.startTracing({runId:t,parentRunId:s,name:n.tool,input:n,tags:i});}async handleAgentEnd(n,t,s,i){logger.debug(`handleAgentEnd runId - ${t}, parentRunId ${s}`),this.endTracing({runId:t,output:n,tags:i});}async handleRetrieverStart(n,t,s,i,o,l,d){var g,p;logger.debug(`handleRetrieverStart runId - ${s}, parentRunId ${i}`),this.startTracing({runId:s,parentRunId:i,name:(p=d!=null?d:(g=n.id.at(-1))==null?void 0:g.toString())!=null?p:"Retriever",type:OpikSpanType.Tool,input:{query:t},tags:o,metadata:{...l,...y(n,{},l)}});}async handleRetrieverEnd(n,t,s,i){logger.debug(`handleRetrieverEnd runId - ${t}, parentRunId ${s}`),this.endTracing({runId:t,output:{documents:n},tags:i});}async handleRetrieverError(n,t,s,i){logger.debug(`handleRetrieverError runId - ${t}, parentRunId ${s}`),this.endTracing({runId:t,error:n,tags:i});}async flushAsync(){return this.client.flush()}};export{b as OpikCallbackHandler};

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "opik-langchain",
   "description": "Opik TypeScript and JavaScript SDK integration with LangChain",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "engines": {
     "node": ">=18"
   },