@posthog/ai 8.8.0 → 8.9.0

package/dist/langchain/index.d.ts

@@ -7,6 +7,15 @@ import { AgentAction, AgentFinish } from '@langchain/core/agents';
 import { DocumentInterface } from '@langchain/core/documents';
 import { BaseMessage } from '@langchain/core/messages';
 
+interface LangChainCallbackHandlerOptions {
+    client: PostHog;
+    distinctId?: string | number;
+    traceId?: string | number;
+    properties?: Record<string, any>;
+    privacyMode?: boolean;
+    groups?: Record<string, any>;
+    debug?: boolean;
+}
 declare class LangChainCallbackHandler extends BaseCallbackHandler {
     name: string;
     private client;
@@ -18,16 +27,8 @@ declare class LangChainCallbackHandler extends BaseCallbackHandler {
     private debug;
     private runs;
     private parentTree;
-    constructor(options: {
-        client: PostHog;
-        distinctId?: string | number;
-        traceId?: string | number;
-        properties?: Record<string, any>;
-        privacyMode?: boolean;
-        groups?: Record<string, any>;
-        debug?: boolean;
-    });
-    handleChainStart(chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, _runType?: string, runName?: string): void;
+    constructor(options: LangChainCallbackHandlerOptions);
+    handleChainStart(chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, _runType?: string, runName?: string, extra?: Record<string, unknown>): void;
     handleChainEnd(outputs: ChainValues, runId: string, parentRunId?: string, tags?: string[], _kwargs?: {
         inputs?: Record<string, unknown>;
     }): void;
@@ -56,7 +57,7 @@ declare class LangChainCallbackHandler extends BaseCallbackHandler {
     private _setLLMMetadata;
     private _popRunMetadata;
     private _getTraceId;
-    private _getParentRunId;
+    protected _getParentRunId(traceId: string, _runId: string, parentRunId?: string): string | undefined;
     private _safeCapture;
     private _popRunAndCaptureTraceOrSpan;
     private _captureTraceOrSpan;
@@ -75,3 +76,4 @@ declare class LangChainCallbackHandler extends BaseCallbackHandler {
 }
 
 export { LangChainCallbackHandler };
+export type { LangChainCallbackHandlerOptions };

package/dist/langchain/index.mjs

@@ -245,7 +245,7 @@ function sanitizeValues(obj) {
   return jsonSafe;
 }
 
-var version = "8.8.0";
+var version = "8.9.0";
 
 const DEFAULT_MAX_DEPTH = 3;
 const MAX_STACK_LINES = 20;
@@ -367,13 +367,16 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
 
   // ===== CALLBACK METHODS =====
 
-  handleChainStart(chain, inputs, runId, parentRunId, tags, metadata, _runType, runName) {
+  handleChainStart(chain, inputs, runId, parentRunId, tags, metadata, _runType, runName, extra) {
     this._logDebugEvent('on_chain_start', runId, parentRunId, {
       inputs,
       tags
     });
     this._setParentOfRun(runId, parentRunId);
     this._setTraceOrSpanMetadata(chain, inputs, runId, parentRunId, metadata, tags, runName);
+    if (typeof extra?.posthogStartTime === 'number' && Number.isFinite(extra.posthogStartTime)) {
+      this.runs[runId].startTime = extra.posthogStartTime;
+    }
   }
   handleChainEnd(outputs, runId, parentRunId, tags, _kwargs) {
     this._logAndPopTraceOrSpan('on_chain_end', runId, parentRunId, {
@@ -568,6 +571,7 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
   }
   _popRunAndCaptureTraceOrSpan(runId, parentRunId, outputs) {
     const traceId = this._getTraceId(runId);
+    const isSpan = Boolean(parentRunId || this.parentTree[runId]);
     this._popParentOfRun(runId);
     const run = this._popRunMetadata(runId);
     if (!run) {
@@ -578,10 +582,10 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
       return;
     }
     const actualParentRunId = this._getParentRunId(traceId, runId, parentRunId);
-    this._captureTraceOrSpan(traceId, runId, run, outputs, actualParentRunId);
+    this._captureTraceOrSpan(traceId, runId, run, outputs, isSpan, actualParentRunId);
   }
-  _captureTraceOrSpan(traceId, runId, run, outputs, parentRunId) {
-    const eventName = parentRunId ? '$ai_span' : '$ai_trace';
+  _captureTraceOrSpan(traceId, runId, run, outputs, isSpan, parentRunId) {
+    const eventName = isSpan ? '$ai_span' : '$ai_trace';
     const latency = run.endTime ? (run.endTime - run.startTime) / 1000 : 0;
     const eventProperties = {
       $ai_lib: 'posthog-ai',
@@ -645,7 +649,6 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
       $ai_trace_id: traceId,
       $ai_span_id: runId,
       $ai_span_name: run.name,
-      $ai_parent_id: parentRunId,
       $ai_provider: run.provider,
       $ai_model: run.model,
       $ai_model_parameters: run.modelParams,
@@ -655,6 +658,9 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
       $ai_base_url: run.baseUrl,
       $ai_framework: 'langchain'
     };
+    if (parentRunId) {
+      eventProperties['$ai_parent_id'] = parentRunId;
+    }
     if (run.tools) {
       eventProperties['$ai_tools'] = run.tools;
     }