openlit 1.11.0 → 1.13.0

package/dist/instrumentation/langchain/index.js

@@ -13,23 +13,13 @@ class OpenlitLangChainInstrumentation extends instrumentation_1.InstrumentationB
         this._ritmHook = null;
     }
     init() {
-        // We bypass OTel's RequireInTheMiddleSingleton here because it uses a null-whitelist
-        // RITM instance, which causes RITM to report `name` as the `fullModuleName` (including
-        // the .cjs extension), e.g. '@langchain/core/language_models/chat_models.cjs'.
-        // The trie search then fails to match our registered bare name (without .cjs).
-        // Instead, we set up a direct require-in-the-middle Hook in enable() with a proper
-        // whitelist, which causes RITM to preserve the original specifier as `name`.
         return [];
     }
-    /** Override enable() to install a direct RITM Hook rather than going through the singleton. */
     enable() {
-        super.enable(); // sets this._enabled = true, initialises tracer
+        super.enable();
         if (this._ritmHook)
             return;
         try {
-            // Require RITM directly — it is a transitive dep of @opentelemetry/instrumentation.
-            // Using a whitelist ['@langchain/core/language_models/chat_models'] causes RITM to
-            // preserve the original module specifier as `name` (avoiding the .cjs issue).
             // eslint-disable-next-line @typescript-eslint/no-var-requires
             const { Hook } = require('require-in-the-middle');
             this._ritmHook = new Hook(['@langchain/core/language_models/chat_models'], { internals: true }, (exports) => {
@@ -37,16 +27,14 @@ class OpenlitLangChainInstrumentation extends instrumentation_1.InstrumentationB
                 return exports;
             });
         }
-        catch { /* require-in-the-middle not available — no-op */ }
+        catch { /* require-in-the-middle not available */ }
     }
-    /** Override disable() to remove our direct RITM Hook. */
     disable() {
         super.disable();
         this._ritmHook?.unhook?.();
         this._ritmHook = null;
         this._unpatch();
     }
-    /** Called from tests / manual usage — pass the @langchain/core/callbacks/manager exports. */
     manualPatch(callbacksManagerModule) {
         const CallbackManager = callbacksManagerModule?.CallbackManager;
         if (CallbackManager) {
@@ -54,10 +42,6 @@ class OpenlitLangChainInstrumentation extends instrumentation_1.InstrumentationB
             this._applyPatch(CallbackManager);
         }
     }
-    /**
-     * Scan require.cache for @langchain/core's dist/callbacks/manager.cjs.
-     * It is always loaded as a relative dep before the chat_models hook fires.
-     */
     _patchFromCache() {
         try {
             const cache = require.cache;
@@ -82,8 +66,6 @@ class OpenlitLangChainInstrumentation extends instrumentation_1.InstrumentationB
     }
     _applyPatch(CallbackManager) {
         try {
-            // Patch _configureSync — the single low-level entry point called by both
-            // CallbackManager.configure and getCallbackManagerForConfig (Runnables/chains).
             if ((0, instrumentation_1.isWrapped)(CallbackManager._configureSync)) {
                 this._unwrap(CallbackManager, '_configureSync');
             }

package/dist/instrumentation/langchain/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/langchain/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAIwC;AAExC,6CAAwD;AACxD,wDAAyC;AAIzC,MAAqB,+BAAgC,SAAQ,qCAAmB;IAI9E,YAAY,SAAyC,EAAE;QACrD,KAAK,CAAC,GAAG,iCAAsB,4BAA4B,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;QAJxE,qBAAgB,GAAQ,IAAI,CAAC;QAC7B,cAAS,GAAQ,IAAI,CAAC;IAI9B,CAAC;IAES,IAAI;QACZ,qFAAqF;QACrF,uFAAuF;QACvF,+EAA+E;QAC/E,+EAA+E;QAC/E,mFAAmF;QACnF,6EAA6E;QAC7E,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,+FAA+F;IACxF,MAAM;QACX,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,gDAAgD;QAChE,IAAI,IAAI,CAAC,SAAS;YAAE,OAAO;QAC3B,IAAI,CAAC;YACH,oFAAoF;YACpF,mFAAmF;YACnF,8EAA8E;YAC9E,8DAA8D;YAC9D,MAAM,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,uBAAuB,CAAC,CAAC;YAClD,IAAI,CAAC,SAAS,GAAG,IAAI,IAAI,CACvB,CAAC,6CAA6C,CAAC,EAC/C,EAAE,SAAS,EAAE,IAAI,EAAE,EACnB,CAAC,OAAY,EAAE,EAAE;gBACf,IAAI,CAAC,eAAe,EAAE,CAAC;gBACvB,OAAO,OAAO,CAAC;YACjB,CAAC,CACF,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC,CAAC,iDAAiD,CAAC,CAAC;IAC/D,CAAC;IAED,yDAAyD;IAClD,OAAO;QACZ,KAAK,CAAC,OAAO,EAAE,CAAC;QAChB,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,IAAI,CAAC,QAAQ,EAAE,CAAC;IAClB,CAAC;IAED,6FAA6F;IACtF,WAAW,CAAC,sBAA2B;QAC5C,MAAM,eAAe,GAAG,sBAAsB,EAAE,eAAe,CAAC;QAChE,IAAI,eAAe,EAAE,CAAC;YACpB,IAAI,CAAC,gBAAgB,GAAG,eAAe,CAAC;YACxC,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,eAAe;QACrB,IAAI,CAAC;YACH,MAAM,KAAK,GAAI,OAA2D,CAAC,KAAK,CAAC;YACjF,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC1C,IACE,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC;oBAC/B,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC;oBAC9B,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC;oBAC5B,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,EAC1B,CAAC;oBACD,MAAM,GAAG,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAC;oBAC5B,MAAM,eAAe,GAAG,GAAG,EAAE,OAAO,EAAE,eAAe,CAAC;oBACtD,IAAI,eAAe,IAAI,OAAO,eAAe,CAAC,cAAc,KAAK,UAAU,EAAE,CAAC;wBAC5E,IAAI,CAAC,gBAAgB,GAAG,eAAe,CAAC;wBACxC,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;wBAClC,OAAO;oBACT,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,8DAA8D,EAAE,CAAC,CAAC,CAAC;QACnF,CAAC;IACH,CAAC;IAEO,WAAW,CAAC,eAAoB;QACtC,IAAI,CAAC;YACH,yEAAyE;YACzE,gFAAgF;YAChF,IAAI,IAAA,2BAAS,EAAC,eAAe,CAAC,cAAc,CAAC,EAAE,CAAC;gBAC9C,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,gBAAgB,CAAC,CAAC;YAClD,CAAC;YACD,IAAI,CAAC,KAAK,CACR,eAAe,EACf,gBAAgB,EAChB,iBAAgB,CAAC,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,CAC9C,CAAC;QACJ,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,CAAC,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IAEO,QAAQ;QACd,IAAI,CAAC,IAAI,CAAC,gBAAgB;YAAE,OAAO;QACnC,IAAI,CAAC;YACH,IAAI,IAAA,2BAAS,EAAC,IAAI,CAAC,gBAAgB,CAAC,cAAc,CAAC,EAAE,CAAC;gBACpD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,gBAAgB,EAAE,gBAAgB,CAAC,CAAC;YACxD,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;QACxB,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;IAC/B,CAAC;CACF;AA9GD,kDA8GC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/langchain/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAIwC;AAExC,6CAAwD;AACxD,wDAAyC;AAIzC,MAAqB,+BAAgC,SAAQ,qCAAmB;IAI9E,YAAY,SAAyC,EAAE;QACrD,KAAK,CAAC,GAAG,iCAAsB,4BAA4B,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;QAJxE,qBAAgB,GAAQ,IAAI,CAAC;QAC7B,cAAS,GAAQ,IAAI,CAAC;IAI9B,CAAC;IAES,IAAI;QACZ,OAAO,EAAE,CAAC;IACZ,CAAC;IAEM,MAAM;QACX,KAAK,CAAC,MAAM,EAAE,CAAC;QACf,IAAI,IAAI,CAAC,SAAS;YAAE,OAAO;QAC3B,IAAI,CAAC;YACH,8DAA8D;YAC9D,MAAM,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,uBAAuB,CAAC,CAAC;YAClD,IAAI,CAAC,SAAS,GAAG,IAAI,IAAI,CACvB,CAAC,6CAA6C,CAAC,EAC/C,EAAE,SAAS,EAAE,IAAI,EAAE,EACnB,CAAC,OAAY,EAAE,EAAE;gBACf,IAAI,CAAC,eAAe,EAAE,CAAC;gBACvB,OAAO,OAAO,CAAC;YACjB,CAAC,CACF,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC,CAAC,yCAAyC,CAAC,CAAC;IACvD,CAAC;IAEM,OAAO;QACZ,KAAK,CAAC,OAAO,EAAE,CAAC;QAChB,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,IAAI,CAAC,QAAQ,EAAE,CAAC;IAClB,CAAC;IAEM,WAAW,CAAC,sBAA2B;QAC5C,MAAM,eAAe,GAAG,sBAAsB,EAAE,eAAe,CAAC;QAChE,IAAI,eAAe,EAAE,CAAC;YACpB,IAAI,CAAC,gBAAgB,GAAG,eAAe,CAAC;YACxC,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAEO,eAAe;QACrB,IAAI,CAAC;YACH,MAAM,KAAK,GAAI,OAA2D,CAAC,KAAK,CAAC;YACjF,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC1C,IACE,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC;oBAC/B,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC;oBAC9B,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC;oBAC5B,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,EAC1B,CAAC;oBACD,MAAM,GAAG,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAC;oBAC5B,MAAM,eAAe,GAAG,GAAG,EAAE,OAAO,EAAE,eAAe,CAAC;oBACtD,IAAI,eAAe,IAAI,OAAO,eAAe,CAAC,cAAc,KAAK,UAAU,EAAE,CAAC;wBAC5E,IAAI,CAAC,gBAAgB,GAAG,eAAe,CAAC;wBACxC,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;wBAClC,OAAO;oBACT,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,8DAA8D,EAAE,CAAC,CAAC,CAAC;QACnF,CAAC;IACH,CAAC;IAEO,WAAW,CAAC,eAAoB;QACtC,IAAI,CAAC;YACH,IAAI,IAAA,2BAAS,EAAC,eAAe,CAAC,cAAc,CAAC,EAAE,CAAC;gBAC9C,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,gBAAgB,CAAC,CAAC;YAClD,CAAC;YACD,IAAI,CAAC,KAAK,CACR,eAAe,EACf,gBAAgB,EAChB,iBAAgB,CAAC,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,CAC9C,CAAC;QACJ,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,CAAC,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IAEO,QAAQ;QACd,IAAI,CAAC,IAAI,CAAC,gBAAgB;YAAE,OAAO;QACnC,IAAI,CAAC;YACH,IAAI,IAAA,2BAAS,EAAC,IAAI,CAAC,gBAAgB,CAAC,cAAc,CAAC,EAAE,CAAC;gBACpD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,gBAAgB,EAAE,gBAAgB,CAAC,CAAC;YACxD,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;QACxB,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;IAC/B,CAAC;CACF;AA5FD,kDA4FC"}

package/dist/instrumentation/langchain/wrapper.d.ts

@@ -1,6 +1,41 @@
 import { Tracer } from '@opentelemetry/api';
+import { runWithFrameworkLlm } from '../../helpers';
 import BaseWrapper from '../base-wrapper';
+declare class OpenLITCallbackHandler {
+    name: string;
+    lc_serializable: boolean;
+    awaitHandlers: boolean;
+    private tracer;
+    private spans;
+    private skippedRuns;
+    constructor(tracer: Tracer);
+    private _getNameFromCallback;
+    private _resolveParentRunId;
+    private _getParentContext;
+    private _createSpan;
+    private _newHolder;
+    private _setCommonAttributes;
+    private _setModelParameters;
+    private _endSpan;
+    handleChainStart(chain: any, inputs: any, runId: string, parentRunId?: string, _tags?: string[], metadata?: any, _runType?: string, name?: string): void;
+    handleChainEnd(outputs: any, runId: string): void;
+    handleChainError(error: any, runId: string): void;
+    handleChatModelStart(llm: any, messages: any[][], runId: string, parentRunId?: string, _extraParams?: any, _tags?: string[], metadata?: any, kwargs?: any): void;
+    handleLLMStart(serialized: any, prompts: string[], runId: string, parentRunId?: string, _extraParams?: any, _tags?: string[], metadata?: any, kwargs?: any): void;
+    handleLLMNewToken(token: string, _idx: any, runId: string, _chunk?: any): void;
+    handleLLMEnd(output: any, runId: string): void;
+    handleLLMError(error: any, runId: string): void;
+    handleToolStart(tool: any, input: string, runId: string, parentRunId?: string, _tags?: string[], metadata?: any, kwargs?: any): void;
+    handleToolEnd(output: any, runId: string): void;
+    handleToolError(error: any, runId: string): void;
+    handleRetrieverStart(retriever: any, query: string, runId: string, parentRunId?: string, _tags?: string[], metadata?: any, kwargs?: any): void;
+    handleRetrieverEnd(documents: any[], runId: string): void;
+    handleRetrieverError(error: any, runId: string): void;
+    handleAgentAction(action: any, runId: string): void;
+    handleAgentFinish(finish: any, runId: string): void;
+}
 declare class LangChainWrapper extends BaseWrapper {
     static _patchConfigure(tracer: Tracer): any;
 }
+export { OpenLITCallbackHandler, runWithFrameworkLlm };
 export default LangChainWrapper;