@posthog/ai 6.1.2 → 6.3.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -27,17 +27,17 @@ declare class LangChainCallbackHandler extends BaseCallbackHandler {
27
27
  groups?: Record<string, any>;
28
28
  debug?: boolean;
29
29
  });
30
- handleChainStart(chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, runType?: string, runName?: string): void;
31
- handleChainEnd(outputs: ChainValues, runId: string, parentRunId?: string, tags?: string[], kwargs?: {
30
+ handleChainStart(chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, _runType?: string, runName?: string): void;
31
+ handleChainEnd(outputs: ChainValues, runId: string, parentRunId?: string, tags?: string[], _kwargs?: {
32
32
  inputs?: Record<string, unknown>;
33
33
  }): void;
34
- handleChainError(error: Error, runId: string, parentRunId?: string, tags?: string[], kwargs?: {
34
+ handleChainError(error: Error, runId: string, parentRunId?: string, tags?: string[], _kwargs?: {
35
35
  inputs?: Record<string, unknown>;
36
36
  }): void;
37
37
  handleChatModelStart(serialized: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string, extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>, runName?: string): void;
38
38
  handleLLMStart(serialized: Serialized, prompts: string[], runId: string, parentRunId?: string, extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>, runName?: string): void;
39
- handleLLMEnd(output: LLMResult, runId: string, parentRunId?: string, tags?: string[], extraParams?: Record<string, unknown>): void;
40
- handleLLMError(err: Error, runId: string, parentRunId?: string, tags?: string[], extraParams?: Record<string, unknown>): void;
39
+ handleLLMEnd(output: LLMResult, runId: string, parentRunId?: string, tags?: string[], _extraParams?: Record<string, unknown>): void;
40
+ handleLLMError(err: Error, runId: string, parentRunId?: string, tags?: string[], _extraParams?: Record<string, unknown>): void;
41
41
  handleToolStart(tool: Serialized, input: string, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, runName?: string): void;
42
42
  handleToolEnd(output: any, runId: string, parentRunId?: string, tags?: string[]): void;
43
43
  handleToolError(err: Error, runId: string, parentRunId?: string, tags?: string[]): void;
@@ -1,6 +1,8 @@
1
1
  import 'buffer';
2
2
  import * as uuid from 'uuid';
3
3
 
4
+ var version = "6.3.0";
5
+
4
6
  const getModelParams = params => {
5
7
  if (!params) {
6
8
  return {};
@@ -687,7 +689,7 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
687
689
 
688
690
  // ===== CALLBACK METHODS =====
689
691
 
690
- handleChainStart(chain, inputs, runId, parentRunId, tags, metadata, runType, runName) {
692
+ handleChainStart(chain, inputs, runId, parentRunId, tags, metadata, _runType, runName) {
691
693
  this._logDebugEvent('on_chain_start', runId, parentRunId, {
692
694
  inputs,
693
695
  tags
@@ -695,18 +697,14 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
695
697
  this._setParentOfRun(runId, parentRunId);
696
698
  this._setTraceOrSpanMetadata(chain, inputs, runId, parentRunId, metadata, tags, runName);
697
699
  }
698
- handleChainEnd(outputs, runId, parentRunId, tags,
699
- // eslint-disable-next-line @typescript-eslint/no-unused-vars
700
- kwargs) {
700
+ handleChainEnd(outputs, runId, parentRunId, tags, _kwargs) {
701
701
  this._logDebugEvent('on_chain_end', runId, parentRunId, {
702
702
  outputs,
703
703
  tags
704
704
  });
705
705
  this._popRunAndCaptureTraceOrSpan(runId, parentRunId, outputs);
706
706
  }
707
- handleChainError(error, runId, parentRunId, tags,
708
- // eslint-disable-next-line @typescript-eslint/no-unused-vars
709
- kwargs) {
707
+ handleChainError(error, runId, parentRunId, tags, _kwargs) {
710
708
  this._logDebugEvent('on_chain_error', runId, parentRunId, {
711
709
  error,
712
710
  tags
@@ -731,18 +729,14 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
731
729
  this._setParentOfRun(runId, parentRunId);
732
730
  this._setLLMMetadata(serialized, runId, prompts, metadata, extraParams, runName);
733
731
  }
734
- handleLLMEnd(output, runId, parentRunId, tags,
735
- // eslint-disable-next-line @typescript-eslint/no-unused-vars
736
- extraParams) {
732
+ handleLLMEnd(output, runId, parentRunId, tags, _extraParams) {
737
733
  this._logDebugEvent('on_llm_end', runId, parentRunId, {
738
734
  output,
739
735
  tags
740
736
  });
741
737
  this._popRunAndCaptureGeneration(runId, parentRunId, output);
742
738
  }
743
- handleLLMError(err, runId, parentRunId, tags,
744
- // eslint-disable-next-line @typescript-eslint/no-unused-vars
745
- extraParams) {
739
+ handleLLMError(err, runId, parentRunId, tags, _extraParams) {
746
740
  this._logDebugEvent('on_llm_error', runId, parentRunId, {
747
741
  err,
748
742
  tags
@@ -879,7 +873,7 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
879
873
  _getTraceId(runId) {
880
874
  return this.traceId ? String(this.traceId) : this._findRootRun(runId);
881
875
  }
882
- _getParentRunId(traceId, runId, parentRunId) {
876
+ _getParentRunId(traceId, _runId, parentRunId) {
883
877
  // Replace the parent-run if not found in our stored parent tree.
884
878
  if (parentRunId && !this.parentTree[parentRunId]) {
885
879
  return traceId;
@@ -904,6 +898,8 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
904
898
  const eventName = parentRunId ? '$ai_span' : '$ai_trace';
905
899
  const latency = run.endTime ? (run.endTime - run.startTime) / 1000 : 0;
906
900
  const eventProperties = {
901
+ $ai_lib: 'posthog-ai',
902
+ $ai_lib_version: version,
907
903
  $ai_trace_id: traceId,
908
904
  $ai_input_state: withPrivacyMode(this.client, this.privacyMode, run.input),
909
905
  $ai_latency: latency,
@@ -944,6 +940,8 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
944
940
  _captureGeneration(traceId, runId, run, output, parentRunId) {
945
941
  const latency = run.endTime ? (run.endTime - run.startTime) / 1000 : 0;
946
942
  const eventProperties = {
943
+ $ai_lib: 'posthog-ai',
944
+ $ai_lib_version: version,
947
945
  $ai_trace_id: traceId,
948
946
  $ai_span_id: runId,
949
947
  $ai_span_name: run.name,