@traccia2/sdk 0.0.6 → 0.0.7

package/README.md

@@ -449,7 +449,7 @@ If you don't see traces in the console, check:
    // Use with ChatOllama
    import { ChatOllama } from '@langchain/ollama';
    
-   const handler = new TraciaCallbackHandler();
+   const handler = new TracciaCallbackHandler();
    const model = new ChatOllama({ 
      model: 'mistral',
      baseUrl: 'http://localhost:11434',
@@ -466,7 +466,7 @@ If you don't see traces in the console, check:
 2. **Callback Handler Configuration:**
    ```typescript
    // Make sure the handler is passed to ALL components
-   const handler = new TraciaCallbackHandler();
+   const handler = new TracciaCallbackHandler();
    
    const model = new ChatOpenAI({ 
      callbacks: [handler],  // Required for LLM

package/dist/integrations/index.d.ts

@@ -5,6 +5,7 @@
  * @module integrations
  */
 export { TracciaCallbackHandler } from './langchain-callback';
+export { TracciaCallbackHandlerNew } from './langchain-callback.new';
 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,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"}
+{"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,EAAE,yBAAyB,EAAE,MAAM,0BAA0B,CAAC;AACrE,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,12 @@
  * @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.TracciaCallbackHandler = 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.TracciaCallbackHandlerNew = exports.TracciaCallbackHandler = void 0;
 // LangChain integrations
 var langchain_callback_1 = require("./langchain-callback");
 Object.defineProperty(exports, "TracciaCallbackHandler", { enumerable: true, get: function () { return langchain_callback_1.TracciaCallbackHandler; } });
+var langchain_callback_new_1 = require("./langchain-callback.new");
+Object.defineProperty(exports, "TracciaCallbackHandlerNew", { enumerable: true, get: function () { return langchain_callback_new_1.TracciaCallbackHandlerNew; } });
 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,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"}
+{"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,mEAAqE;AAA5D,mIAAA,yBAAyB,OAAA;AAClC,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.new.d.ts

@@ -0,0 +1,62 @@
+import type { AgentAction, AgentFinish } from "@langchain/core/agents";
+import { BaseCallbackHandler } from "@langchain/core/callbacks/base";
+import type { Document } from "@langchain/core/documents";
+import type { Serialized } from "@langchain/core/load/serializable";
+import { BaseMessage, type BaseMessageFields, type MessageContent } from "@langchain/core/messages";
+import type { LLMResult } from "@langchain/core/outputs";
+import type { ChainValues } from "@langchain/core/utils/types";
+export type LlmMessage = {
+    role: string;
+    content: BaseMessageFields["content"];
+    additional_kwargs?: BaseMessageFields["additional_kwargs"];
+};
+export type AnonymousLlmMessage = {
+    content: BaseMessageFields["content"];
+    additional_kwargs?: BaseMessageFields["additional_kwargs"];
+};
+type ConstructorParams = {
+    userId?: string;
+    sessionId?: string;
+    tags?: string[];
+    version?: string;
+    traceMetadata?: Record<string, unknown>;
+};
+export declare class TracciaCallbackHandlerNew extends BaseCallbackHandler {
+    name: string;
+    private tracer;
+    private tags;
+    private completionStartTimes;
+    private promptToParentRunMap;
+    private runMap;
+    last_trace_id: string | null;
+    constructor(params?: ConstructorParams);
+    handleLLMNewToken(_token: string, _idx: any, runId: string, _parentRunId?: string, _tags?: string[], _fields?: any): Promise<void>;
+    handleChainStart(chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, _runType?: string, name?: string): Promise<void>;
+    handleAgentAction(action: AgentAction, runId: string, parentRunId?: string): Promise<void>;
+    handleAgentEnd?(action: AgentFinish, runId: string, _parentRunId?: string): Promise<void>;
+    handleChainError(err: any, runId: string, _parentRunId?: string | undefined): Promise<void>;
+    handleGenerationStart(llm: Serialized, messages: (LlmMessage | MessageContent | AnonymousLlmMessage)[], runId: string, parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string): Promise<void>;
+    handleChatModelStart(llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string): Promise<void>;
+    handleChainEnd(outputs: ChainValues, runId: string, _parentRunId?: string | undefined): Promise<void>;
+    handleLLMStart(llm: Serialized, prompts: string[], runId: string, parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string): Promise<void>;
+    handleToolStart(tool: Serialized, input: string, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string): Promise<void>;
+    handleRetrieverStart(retriever: Serialized, query: string, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string): Promise<void>;
+    handleRetrieverEnd(documents: Document<Record<string, any>>[], runId: string, _parentRunId?: string | undefined): Promise<void>;
+    handleRetrieverError(err: any, runId: string, _parentRunId?: string | undefined): Promise<void>;
+    handleToolEnd(output: string, runId: string, _parentRunId?: string | undefined): Promise<void>;
+    handleToolError(err: any, runId: string, _parentRunId?: string | undefined): Promise<void>;
+    handleLLMEnd(output: LLMResult, runId: string, _parentRunId?: string | undefined): Promise<void>;
+    handleLLMError(err: any, runId: string, _parentRunId?: string | undefined): Promise<void>;
+    private registerLangfusePrompt;
+    private deregisterLangfusePrompt;
+    private startAndRegisterSpan;
+    private handleSpanEnd;
+    private parseAzureRefusalError;
+    private joinTagsAndMetaData;
+    private stripLangfuseKeysFromMetadata;
+    private extractUsageMetadata;
+    private extractModelNameFromMetadata;
+    private extractChatMessageContent;
+}
+export {};
+//# sourceMappingURL=langchain-callback.new.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"langchain-callback.new.d.ts","sourceRoot":"","sources":["../../src/integrations/langchain-callback.new.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AACvE,OAAO,EAAE,mBAAmB,EAAE,MAAM,gCAAgC,CAAC;AACrE,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAC;AAC1D,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mCAAmC,CAAC;AACpE,OAAO,EAGL,WAAW,EAEX,KAAK,iBAAiB,EACtB,KAAK,cAAc,EACpB,MAAM,0BAA0B,CAAC;AAClC,OAAO,KAAK,EAAc,SAAS,EAAE,MAAM,yBAAyB,CAAC;AACrE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,6BAA6B,CAAC;AAU/D,MAAM,MAAM,UAAU,GAAG;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,iBAAiB,CAAC,SAAS,CAAC,CAAC;IACtC,iBAAiB,CAAC,EAAE,iBAAiB,CAAC,mBAAmB,CAAC,CAAC;CAC5D,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG;IAChC,OAAO,EAAE,iBAAiB,CAAC,SAAS,CAAC,CAAC;IACtC,iBAAiB,CAAC,EAAE,iBAAiB,CAAC,mBAAmB,CAAC,CAAC;CAC5D,CAAC;AAEF,KAAK,iBAAiB,GAAG;IACvB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACzC,CAAC;AAEF,qBAAa,yBAA0B,SAAQ,mBAAmB;IAChE,IAAI,SAA4B;IAChC,OAAO,CAAC,MAAM,CAA0B;IAExC,OAAO,CAAC,IAAI,CAA4B;IAExC,OAAO,CAAC,oBAAoB,CAA4B;IACxD,OAAO,CAAC,oBAAoB,CAAC;IAC7B,OAAO,CAAC,MAAM,CAAiC;IAExC,aAAa,EAAE,MAAM,GAAG,IAAI,CAAQ;gBAE/B,MAAM,CAAC,EAAE,iBAAiB;IAQhC,iBAAiB,CACrB,MAAM,EAAE,MAAM,EACd,IAAI,EAAE,GAAG,EACT,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,EACrB,KAAK,CAAC,EAAE,MAAM,EAAE,EAChB,OAAO,CAAC,EAAE,GAAG,GACZ,OAAO,CAAC,IAAI,CAAC;IAQV,gBAAgB,CACpB,KAAK,EAAE,UAAU,EACjB,MAAM,EAAE,WAAW,EACnB,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,EAChC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,SAAS,EAC3B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EAC9C,QAAQ,CAAC,EAAE,MAAM,EACjB,IAAI,CAAC,EAAE,MAAM,GACZ,OAAO,CAAC,IAAI,CAAC;IAwDV,iBAAiB,CACrB,MAAM,EAAE,WAAW,EACnB,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,GACnB,OAAO,CAAC,IAAI,CAAC;IAgBV,cAAc,CAAC,CACnB,MAAM,EAAE,WAAW,EACnB,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GACpB,OAAO,CAAC,IAAI,CAAC;IAaV,gBAAgB,CACpB,GAAG,EAAE,GAAG,EACR,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,GAChC,OAAO,CAAC,IAAI,CAAC;IAkBV,qBAAqB,CACzB,GAAG,EAAE,UAAU,EACf,QAAQ,EAAE,CAAC,UAAU,GAAG,cAAc,GAAG,mBAAmB,CAAC,EAAE,EAC/D,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,EAChC,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EACjD,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,SAAS,EAC3B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EAC9C,IAAI,CAAC,EAAE,MAAM,GACZ,OAAO,CAAC,IAAI,CAAC;IAiEV,oBAAoB,CACxB,GAAG,EAAE,UAAU,EACf,QAAQ,EAAE,WAAW,EAAE,EAAE,EACzB,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,EAChC,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EACjD,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,SAAS,EAC3B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EAC9C,IAAI,CAAC,EAAE,MAAM,GACZ,OAAO,CAAC,IAAI,CAAC;IAuBV,cAAc,CAClB,OAAO,EAAE,WAAW,EACpB,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,GAChC,OAAO,CAAC,IAAI,CAAC;IAoCV,cAAc,CAClB,GAAG,EAAE,UAAU,EACf,OAAO,EAAE,MAAM,EAAE,EACjB,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,EAChC,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EACjD,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,SAAS,EAC3B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EAC9C,IAAI,CAAC,EAAE,MAAM,GACZ,OAAO,CAAC,IAAI,CAAC;IAmBV,eAAe,CACnB,IAAI,EAAE,UAAU,EAChB,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,EAChC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,SAAS,EAC3B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EAC9C,IAAI,CAAC,EAAE,MAAM,GACZ,OAAO,CAAC,IAAI,CAAC;IAmBV,oBAAoB,CACxB,SAAS,EAAE,UAAU,EACrB,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,EAChC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,SAAS,EAC3B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EAC9C,IAAI,CAAC,EAAE,MAAM,GACZ,OAAO,CAAC,IAAI,CAAC;IAmBV,kBAAkB,CACtB,SAAS,EAAE,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,EAAE,EAC1C,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,GAChC,OAAO,CAAC,IAAI,CAAC;IAeV,oBAAoB,CACxB,GAAG,EAAE,GAAG,EACR,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,GAChC,OAAO,CAAC,IAAI,CAAC;IAcV,aAAa,CACjB,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,GAChC,OAAO,CAAC,IAAI,CAAC;IAaV,eAAe,CACnB,GAAG,EAAE,GAAG,EACR,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,GAChC,OAAO,CAAC,IAAI,CAAC;IAgBV,YAAY,CAChB,MAAM,EAAE,SAAS,EACjB,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,GAChC,OAAO,CAAC,IAAI,CAAC;IA+EV,cAAc,CAClB,GAAG,EAAE,GAAG,EACR,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,GAChC,OAAO,CAAC,IAAI,CAAC;IAkBhB,OAAO,CAAC,sBAAsB;IAY9B,OAAO,CAAC,wBAAwB;IAIhC,OAAO,CAAC,oBAAoB;IAsB5B,OAAO,CAAC,aAAa;IAoBrB,OAAO,CAAC,sBAAsB;IAY9B,OAAO,CAAC,mBAAmB;IAkB3B,OAAO,CAAC,6BAA6B;IAoBrC,OAAO,CAAC,oBAAoB;IAmB5B,OAAO,CAAC,4BAA4B;IAYpC,OAAO,CAAC,yBAAyB;CA8DlC"}

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

@@ -0,0 +1,519 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TracciaCallbackHandlerNew = void 0;
+const base_1 = require("@langchain/core/callbacks/base");
+const messages_1 = require("@langchain/core/messages");
+const auto_1 = require("../auto");
+class TracciaCallbackHandlerNew extends base_1.BaseCallbackHandler {
+    constructor(params) {
+        super();
+        this.name = "TracciaCallbackHandler";
+        this.tracer = (0, auto_1.getTracer)('langchain');
+        this.tags = [];
+        this.completionStartTimes = {};
+        this.runMap = new Map();
+        this.last_trace_id = null;
+        this.tags = params?.tags ?? [];
+        this.promptToParentRunMap = new Map();
+    }
+    async handleLLMNewToken(_token, _idx, runId, _parentRunId, _tags, _fields) {
+        // if this is the first token, add it to completionStartTimes
+        if (runId && !(runId in this.completionStartTimes)) {
+            console.debug(`LLM first streaming token: ${runId}`);
+            this.completionStartTimes[runId] = new Date();
+        }
+    }
+    async handleChainStart(chain, inputs, runId, parentRunId, tags, metadata, _runType, name) {
+        try {
+            console.debug(`Chain start with Id: ${runId}`);
+            this.tags;
+            const runName = name ?? chain.id.at(-1)?.toString() ?? "Langchain Run";
+            this.registerLangfusePrompt(parentRunId, metadata);
+            // In chains, inputs can be a string or an array of BaseMessage
+            let finalInput = inputs;
+            if (typeof inputs === "object" &&
+                "input" in inputs &&
+                Array.isArray(inputs["input"]) &&
+                inputs["input"].every((m) => m instanceof messages_1.BaseMessage)) {
+                finalInput = inputs["input"].map((m) => this.extractChatMessageContent(m));
+            }
+            else if (typeof inputs === "object" &&
+                "messages" in inputs &&
+                Array.isArray(inputs["messages"]) &&
+                inputs["messages"].every((m) => m instanceof messages_1.BaseMessage)) {
+                finalInput = inputs["messages"].map((m) => this.extractChatMessageContent(m));
+            }
+            else if (typeof inputs === "object" &&
+                "content" in inputs &&
+                typeof inputs["content"] === "string") {
+                finalInput = inputs["content"];
+            }
+            this.startAndRegisterSpan({
+                runName,
+                parentRunId,
+                runId,
+                tags,
+                metadata,
+                attributes: {
+                    input: finalInput,
+                },
+            });
+            if (!parentRunId) {
+                // Here we would update the trace, but traccia sdk doesn't support it yet
+            }
+        }
+        catch (e) {
+            console.debug(e instanceof Error ? e.message : String(e));
+        }
+    }
+    async handleAgentAction(action, runId, parentRunId) {
+        try {
+            console.debug(`Agent action ${action.tool} with ID: ${runId}`);
+            this.startAndRegisterSpan({
+                runId,
+                parentRunId,
+                runName: action.tool,
+                attributes: {
+                    input: action,
+                },
+            });
+        }
+        catch (e) {
+            console.debug(e instanceof Error ? e.message : String(e));
+        }
+    }
+    async handleAgentEnd(action, runId, _parentRunId) {
+        try {
+            console.debug(`Agent finish with ID: ${runId}`);
+            this.handleSpanEnd({
+                runId,
+                attributes: { output: action },
+            });
+        }
+        catch (e) {
+            console.debug(e instanceof Error ? e.message : String(e));
+        }
+    }
+    async handleChainError(err, runId, _parentRunId) {
+        try {
+            console.debug(`Chain error: ${err} with ID: ${runId}`);
+            const azureRefusalError = this.parseAzureRefusalError(err);
+            this.handleSpanEnd({
+                runId,
+                attributes: {
+                    level: "ERROR",
+                    statusMessage: err.toString() + azureRefusalError,
+                },
+            });
+        }
+        catch (e) {
+            console.debug(e instanceof Error ? e.message : String(e));
+        }
+    }
+    async handleGenerationStart(llm, messages, runId, parentRunId, extraParams, tags, metadata, name) {
+        console.debug(`Generation start with ID: ${runId} and parentRunId ${parentRunId}`);
+        const runName = name ?? llm.id.at(-1)?.toString() ?? "Langchain Generation";
+        const modelParameters = {};
+        const invocationParams = extraParams?.["invocation_params"];
+        for (const [key, value] of Object.entries({
+            temperature: invocationParams?.temperature,
+            max_tokens: invocationParams?.max_tokens,
+            top_p: invocationParams?.top_p,
+            frequency_penalty: invocationParams?.frequency_penalty,
+            presence_penalty: invocationParams?.presence_penalty,
+            request_timeout: invocationParams?.request_timeout,
+        })) {
+            if (value !== undefined && value !== null) {
+                modelParameters[key] = value;
+            }
+        }
+        let extractedModelName;
+        if (extraParams) {
+            const invocationParamsModelName = extraParams.invocation_params.model;
+            const metadataModelName = metadata && "ls_model_name" in metadata
+                ? metadata["ls_model_name"]
+                : undefined;
+            extractedModelName = invocationParamsModelName ?? metadataModelName;
+        }
+        const registeredPrompt = this.promptToParentRunMap.get(parentRunId ?? "root");
+        if (registeredPrompt && parentRunId) {
+            this.deregisterLangfusePrompt(parentRunId);
+        }
+        this.startAndRegisterSpan({
+            runId,
+            parentRunId,
+            metadata,
+            tags,
+            runName,
+            attributes: {
+                input: messages,
+                model: extractedModelName,
+                modelParameters: modelParameters,
+                prompt: registeredPrompt,
+            },
+        });
+    }
+    async handleChatModelStart(llm, messages, runId, parentRunId, extraParams, tags, metadata, name) {
+        try {
+            console.debug(`Chat model start with ID: ${runId}`);
+            const prompts = messages.flatMap((message) => message.map((m) => this.extractChatMessageContent(m)));
+            this.handleGenerationStart(llm, prompts, runId, parentRunId, extraParams, tags, metadata, name);
+        }
+        catch (e) {
+            console.debug(e instanceof Error ? e.message : String(e));
+        }
+    }
+    async handleChainEnd(outputs, runId, _parentRunId) {
+        try {
+            console.debug(`Chain end with ID: ${runId}`);
+            let finalOutput = outputs;
+            if (typeof outputs === "object" &&
+                "output" in outputs &&
+                typeof outputs["output"] === "string") {
+                finalOutput = outputs["output"];
+            }
+            else if (typeof outputs === "object" &&
+                "messages" in outputs &&
+                Array.isArray(outputs["messages"]) &&
+                outputs["messages"].every((m) => m instanceof messages_1.BaseMessage)) {
+                finalOutput = {
+                    messages: outputs.messages.map((message) => this.extractChatMessageContent(message)),
+                };
+            }
+            this.handleSpanEnd({
+                runId,
+                attributes: {
+                    output: finalOutput,
+                },
+            });
+            this.deregisterLangfusePrompt(runId);
+        }
+        catch (e) {
+            console.debug(e instanceof Error ? e.message : String(e));
+        }
+    }
+    async handleLLMStart(llm, prompts, runId, parentRunId, extraParams, tags, metadata, name) {
+        try {
+            console.debug(`LLM start with ID: ${runId}`);
+            this.handleGenerationStart(llm, prompts, runId, parentRunId, extraParams, tags, metadata, name);
+        }
+        catch (e) {
+            console.debug(e instanceof Error ? e.message : String(e));
+        }
+    }
+    async handleToolStart(tool, input, runId, parentRunId, tags, metadata, name) {
+        try {
+            console.debug(`Tool start with ID: ${runId}`);
+            this.startAndRegisterSpan({
+                runId,
+                parentRunId,
+                runName: name ?? tool.id.at(-1)?.toString() ?? "Tool execution",
+                attributes: {
+                    input,
+                },
+                metadata,
+                tags,
+            });
+        }
+        catch (e) {
+            console.debug(e instanceof Error ? e.message : String(e));
+        }
+    }
+    async handleRetrieverStart(retriever, query, runId, parentRunId, tags, metadata, name) {
+        try {
+            console.debug(`Retriever start with ID: ${runId}`);
+            this.startAndRegisterSpan({
+                runId,
+                parentRunId,
+                runName: name ?? retriever.id.at(-1)?.toString() ?? "Retriever",
+                attributes: {
+                    input: query,
+                },
+                tags,
+                metadata,
+            });
+        }
+        catch (e) {
+            console.debug(e instanceof Error ? e.message : String(e));
+        }
+    }
+    async handleRetrieverEnd(documents, runId, _parentRunId) {
+        try {
+            console.debug(`Retriever end with ID: ${runId}`);
+            this.handleSpanEnd({
+                runId,
+                attributes: {
+                    output: documents,
+                },
+            });
+        }
+        catch (e) {
+            console.debug(e instanceof Error ? e.message : String(e));
+        }
+    }
+    async handleRetrieverError(err, runId, _parentRunId) {
+        try {
+            console.debug(`Retriever error: ${err} with ID: ${runId}`);
+            this.handleSpanEnd({
+                runId,
+                attributes: {
+                    level: "ERROR",
+                    statusMessage: err.toString(),
+                },
+            });
+        }
+        catch (e) {
+            console.debug(e instanceof Error ? e.message : String(e));
+        }
+    }
+    async handleToolEnd(output, runId, _parentRunId) {
+        try {
+            console.debug(`Tool end with ID: ${runId}`);
+            this.handleSpanEnd({
+                runId,
+                attributes: { output },
+            });
+        }
+        catch (e) {
+            console.debug(e instanceof Error ? e.message : String(e));
+        }
+    }
+    async handleToolError(err, runId, _parentRunId) {
+        try {
+            console.debug(`Tool error ${err} with ID: ${runId}`);
+            this.handleSpanEnd({
+                runId,
+                attributes: {
+                    level: "ERROR",
+                    statusMessage: err.toString(),
+                },
+            });
+        }
+        catch (e) {
+            console.debug(e instanceof Error ? e.message : String(e));
+        }
+    }
+    async handleLLMEnd(output, runId, _parentRunId) {
+        try {
+            console.debug(`LLM end with ID: ${runId}`);
+            const lastResponse = output.generations[output.generations.length - 1][output.generations[output.generations.length - 1].length - 1];
+            const llmUsage = this.extractUsageMetadata(lastResponse) ??
+                output.llmOutput?.["tokenUsage"];
+            const modelName = this.extractModelNameFromMetadata(lastResponse);
+            const usageDetails = {
+                input: llmUsage?.input_tokens ??
+                    ("promptTokens" in llmUsage ? llmUsage?.promptTokens : undefined),
+                output: llmUsage?.output_tokens ??
+                    ("completionTokens" in llmUsage
+                        ? llmUsage?.completionTokens
+                        : undefined),
+                total: llmUsage?.total_tokens ??
+                    ("totalTokens" in llmUsage ? llmUsage?.totalTokens : undefined),
+            };
+            if (llmUsage && "input_token_details" in llmUsage) {
+                for (const [key, val] of Object.entries(llmUsage["input_token_details"] ?? {})) {
+                    usageDetails[`input_${key}`] = val;
+                    if ("input" in usageDetails && typeof val === "number") {
+                        usageDetails["input"] = Math.max(0, usageDetails["input"] - val);
+                    }
+                }
+            }
+            if (llmUsage && "output_token_details" in llmUsage) {
+                for (const [key, val] of Object.entries(llmUsage["output_token_details"] ?? {})) {
+                    usageDetails[`output_${key}`] = val;
+                    if ("output" in usageDetails && typeof val === "number") {
+                        usageDetails["output"] = Math.max(0, usageDetails["output"] - val);
+                    }
+                }
+            }
+            const extractedOutput = "message" in lastResponse
+                ? this.extractChatMessageContent(lastResponse["message"])
+                : lastResponse.text;
+            this.handleSpanEnd({
+                runId,
+                attributes: {
+                    model: modelName,
+                    output: extractedOutput,
+                    completionStartTime: runId in this.completionStartTimes
+                        ? this.completionStartTimes[runId]
+                        : undefined,
+                    usageDetails: usageDetails,
+                },
+            });
+            if (runId in this.completionStartTimes) {
+                delete this.completionStartTimes[runId];
+            }
+        }
+        catch (e) {
+            console.debug(e instanceof Error ? e.message : String(e));
+        }
+    }
+    async handleLLMError(err, runId, _parentRunId) {
+        try {
+            console.debug(`LLM error ${err} with ID: ${runId}`);
+            const azureRefusalError = this.parseAzureRefusalError(err);
+            this.handleSpanEnd({
+                runId,
+                attributes: {
+                    level: "ERROR",
+                    statusMessage: err.toString() + azureRefusalError,
+                },
+            });
+        }
+        catch (e) {
+            console.debug(e instanceof Error ? e.message : String(e));
+        }
+    }
+    registerLangfusePrompt(parentRunId, metadata) {
+        if (metadata && "langfusePrompt" in metadata && parentRunId) {
+            this.promptToParentRunMap.set(parentRunId, metadata.langfusePrompt);
+        }
+    }
+    deregisterLangfusePrompt(runId) {
+        this.promptToParentRunMap.delete(runId);
+    }
+    startAndRegisterSpan(params) {
+        const { runName, runId, attributes, metadata, tags } = params;
+        const span = this.tracer.startSpan(runName, {
+            attributes: {
+                ...attributes,
+                ...this.joinTagsAndMetaData(tags, metadata)
+            }
+        });
+        this.runMap.set(runId, span);
+        return span;
+    }
+    handleSpanEnd(params) {
+        const { runId, attributes = {} } = params;
+        const span = this.runMap.get(runId);
+        if (!span) {
+            console.warn("Span not found in runMap. Skipping operation");
+            return;
+        }
+        for (const [key, value] of Object.entries(attributes)) {
+            span.setAttribute(key, value);
+        }
+        span.end();
+        this.runMap.delete(runId);
+    }
+    parseAzureRefusalError(err) {
+        let azureRefusalError = "";
+        if (typeof err == "object" && "error" in err) {
+            try {
+                azureRefusalError =
+                    "\n\nError details:\n" + JSON.stringify(err["error"], null, 2);
+            }
+            catch { }
+        }
+        return azureRefusalError;
+    }
+    joinTagsAndMetaData(tags, metadata1, metadata2) {
+        const finalDict = {};
+        if (tags && tags.length > 0) {
+            finalDict.tags = tags;
+        }
+        if (metadata1) {
+            Object.assign(finalDict, metadata1);
+        }
+        if (metadata2) {
+            Object.assign(finalDict, metadata2);
+        }
+        return this.stripLangfuseKeysFromMetadata(finalDict);
+    }
+    stripLangfuseKeysFromMetadata(metadata) {
+        if (!metadata) {
+            return;
+        }
+        const langfuseKeys = [
+            "langfusePrompt",
+            "langfuseUserId",
+            "langfuseSessionId",
+        ];
+        return Object.fromEntries(Object.entries(metadata).filter(([key, _]) => !langfuseKeys.includes(key)));
+    }
+    extractUsageMetadata(generation) {
+        try {
+            const usageMetadata = "message" in generation &&
+                (generation["message"] instanceof messages_1.AIMessage ||
+                    generation["message"] instanceof messages_1.AIMessageChunk)
+                ? generation["message"].usage_metadata
+                : undefined;
+            return usageMetadata;
+        }
+        catch (err) {
+            console.debug(`Error extracting usage metadata: ${err}`);
+            return;
+        }
+    }
+    extractModelNameFromMetadata(generation) {
+        try {
+            return "message" in generation &&
+                (generation["message"] instanceof messages_1.AIMessage ||
+                    generation["message"] instanceof messages_1.AIMessageChunk)
+                ? generation["message"].response_metadata.model_name
+                : undefined;
+        }
+        catch {
+            return undefined;
+        }
+    }
+    extractChatMessageContent(message) {
+        let response = undefined;
+        if (message.getType() === "human") {
+            response = { content: message.content, role: "user" };
+        }
+        else if (message.getType() === "generic") {
+            response = {
+                content: message.content,
+                role: "human",
+            };
+        }
+        else if (message.getType() === "ai") {
+            response = { content: message.content, role: "assistant" };
+            if ("tool_calls" in message &&
+                Array.isArray(message.tool_calls) &&
+                (message.tool_calls?.length ?? 0) > 0) {
+                response[`tool_calls`] = message[`tool_calls`];
+            }
+            if ("additional_kwargs" in message &&
+                "tool_calls" in message["additional_kwargs"]) {
+                response[`tool_calls`] =
+                    message["additional_kwargs"]["tool_calls"];
+            }
+        }
+        else if (message.getType() === "system") {
+            response = { content: message.content, role: "system" };
+        }
+        else if (message.getType() === "function") {
+            response = {
+                content: message.content,
+                additional_kwargs: message.additional_kwargs,
+                role: message.name,
+            };
+        }
+        else if (message.getType() === "tool") {
+            response = {
+                content: message.content,
+                additional_kwargs: message.additional_kwargs,
+                role: message.name,
+            };
+        }
+        else if (!message.name) {
+            response = { content: message.content };
+        }
+        else {
+            response = {
+                role: message.name,
+                content: message.content,
+            };
+        }
+        if ((message.additional_kwargs.function_call ||
+            message.additional_kwargs.tool_calls) &&
+            response[`tool_calls`] === undefined) {
+            return { ...response, additional_kwargs: message.additional_kwargs };
+        }
+        return response;
+    }
+}
+exports.TracciaCallbackHandlerNew = TracciaCallbackHandlerNew;
+//# sourceMappingURL=langchain-callback.new.js.map