@posthog/ai 8.1.0 → 8.1.2

package/dist/langchain/index.mjs

@@ -215,7 +215,7 @@ function sanitizeValues(obj) {
   return jsonSafe;
 }
 
-var version = "8.1.0";
+var version = "8.1.2";
 
 const DEFAULT_MAX_DEPTH = 3;
 const MAX_STACK_LINES = 20;
@@ -301,18 +301,16 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
     this._setTraceOrSpanMetadata(chain, inputs, runId, parentRunId, metadata, tags, runName);
   }
   handleChainEnd(outputs, runId, parentRunId, tags, _kwargs) {
-    this._logDebugEvent('on_chain_end', runId, parentRunId, {
+    this._logAndPopTraceOrSpan('on_chain_end', runId, parentRunId, {
       outputs,
       tags
-    });
-    this._popRunAndCaptureTraceOrSpan(runId, parentRunId, outputs);
+    }, outputs);
   }
   handleChainError(error, runId, parentRunId, tags, _kwargs) {
-    this._logDebugEvent('on_chain_error', runId, parentRunId, {
+    this._logAndPopTraceOrSpan('on_chain_error', runId, parentRunId, {
       error,
       tags
-    });
-    this._popRunAndCaptureTraceOrSpan(runId, parentRunId, error);
+    }, error);
   }
   handleChatModelStart(serialized, messages, runId, parentRunId, extraParams, tags, metadata, runName) {
     this._logDebugEvent('on_chat_model_start', runId, parentRunId, {
@@ -333,62 +331,52 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
     this._setLLMMetadata(serialized, runId, prompts, metadata, extraParams, runName);
   }
   handleLLMEnd(output, runId, parentRunId, tags, _extraParams) {
-    this._logDebugEvent('on_llm_end', runId, parentRunId, {
+    this._logAndPopGeneration('on_llm_end', runId, parentRunId, {
       output,
       tags
-    });
-    this._popRunAndCaptureGeneration(runId, parentRunId, output);
+    }, output);
   }
   handleLLMError(err, runId, parentRunId, tags, _extraParams) {
-    this._logDebugEvent('on_llm_error', runId, parentRunId, {
+    this._logAndPopGeneration('on_llm_error', runId, parentRunId, {
       err,
       tags
-    });
-    this._popRunAndCaptureGeneration(runId, parentRunId, err);
+    }, err);
   }
   handleToolStart(tool, input, runId, parentRunId, tags, metadata, runName) {
-    this._logDebugEvent('on_tool_start', runId, parentRunId, {
+    this._logAndSetTraceOrSpan('on_tool_start', tool, input, runId, parentRunId, {
       input,
       tags
-    });
-    this._setParentOfRun(runId, parentRunId);
-    this._setTraceOrSpanMetadata(tool, input, runId, parentRunId, metadata, tags, runName);
+    }, tags, metadata, runName);
   }
   handleToolEnd(output, runId, parentRunId, tags) {
-    this._logDebugEvent('on_tool_end', runId, parentRunId, {
+    this._logAndPopTraceOrSpan('on_tool_end', runId, parentRunId, {
       output,
       tags
-    });
-    this._popRunAndCaptureTraceOrSpan(runId, parentRunId, output);
+    }, output);
   }
   handleToolError(err, runId, parentRunId, tags) {
-    this._logDebugEvent('on_tool_error', runId, parentRunId, {
+    this._logAndPopTraceOrSpan('on_tool_error', runId, parentRunId, {
       err,
       tags
-    });
-    this._popRunAndCaptureTraceOrSpan(runId, parentRunId, err);
+    }, err);
   }
   handleRetrieverStart(retriever, query, runId, parentRunId, tags, metadata, name) {
-    this._logDebugEvent('on_retriever_start', runId, parentRunId, {
+    this._logAndSetTraceOrSpan('on_retriever_start', retriever, query, runId, parentRunId, {
       query,
       tags
-    });
-    this._setParentOfRun(runId, parentRunId);
-    this._setTraceOrSpanMetadata(retriever, query, runId, parentRunId, metadata, tags, name);
+    }, tags, metadata, name);
   }
   handleRetrieverEnd(documents, runId, parentRunId, tags) {
-    this._logDebugEvent('on_retriever_end', runId, parentRunId, {
+    this._logAndPopTraceOrSpan('on_retriever_end', runId, parentRunId, {
       documents,
       tags
-    });
-    this._popRunAndCaptureTraceOrSpan(runId, parentRunId, documents);
+    }, documents);
   }
   handleRetrieverError(err, runId, parentRunId, tags) {
-    this._logDebugEvent('on_retriever_error', runId, parentRunId, {
+    this._logAndPopTraceOrSpan('on_retriever_error', runId, parentRunId, {
       err,
       tags
-    });
-    this._popRunAndCaptureTraceOrSpan(runId, parentRunId, err);
+    }, err);
   }
   handleAgentAction(action, runId, parentRunId, tags) {
     this._logDebugEvent('on_agent_action', runId, parentRunId, {
@@ -408,6 +396,19 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
 
   // ===== PRIVATE HELPERS =====
 
+  _logAndSetTraceOrSpan(eventName, serialized, input, runId, parentRunId, debugPayload, tags, metadata, runName) {
+    this._logDebugEvent(eventName, runId, parentRunId, debugPayload);
+    this._setParentOfRun(runId, parentRunId);
+    this._setTraceOrSpanMetadata(serialized, input, runId, parentRunId, metadata, tags, runName);
+  }
+  _logAndPopTraceOrSpan(eventName, runId, parentRunId, debugPayload, result) {
+    this._logDebugEvent(eventName, runId, parentRunId, debugPayload);
+    this._popRunAndCaptureTraceOrSpan(runId, parentRunId, result);
+  }
+  _logAndPopGeneration(eventName, runId, parentRunId, debugPayload, result) {
+    this._logDebugEvent(eventName, runId, parentRunId, debugPayload);
+    this._popRunAndCaptureGeneration(runId, parentRunId, result);
+  }
   _setParentOfRun(runId, parentRunId) {
     if (parentRunId) {
       this.parentTree[runId] = parentRunId;