braintrust 3.8.0 → 3.10.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.
Files changed (48) hide show
  1. package/dev/dist/index.d.mts +191 -4
  2. package/dev/dist/index.d.ts +191 -4
  3. package/dev/dist/index.js +7269 -2590
  4. package/dev/dist/index.mjs +7281 -2602
  5. package/dist/auto-instrumentations/bundler/esbuild.cjs +741 -2
  6. package/dist/auto-instrumentations/bundler/esbuild.mjs +2 -2
  7. package/dist/auto-instrumentations/bundler/rollup.cjs +741 -2
  8. package/dist/auto-instrumentations/bundler/rollup.mjs +2 -2
  9. package/dist/auto-instrumentations/bundler/vite.cjs +741 -2
  10. package/dist/auto-instrumentations/bundler/vite.mjs +2 -2
  11. package/dist/auto-instrumentations/bundler/webpack-loader.cjs +738 -2
  12. package/dist/auto-instrumentations/bundler/webpack.cjs +741 -2
  13. package/dist/auto-instrumentations/bundler/webpack.mjs +2 -2
  14. package/dist/auto-instrumentations/{chunk-OLBMPZXE.mjs → chunk-GZNXBBPU.mjs} +741 -3
  15. package/dist/auto-instrumentations/{chunk-MD7W27YH.mjs → chunk-XWEQQOQH.mjs} +12 -1
  16. package/dist/auto-instrumentations/hook.mjs +1193 -199
  17. package/dist/auto-instrumentations/index.cjs +1173 -187
  18. package/dist/auto-instrumentations/index.d.mts +22 -1
  19. package/dist/auto-instrumentations/index.d.ts +22 -1
  20. package/dist/auto-instrumentations/index.mjs +256 -1
  21. package/dist/auto-instrumentations/loader/esm-hook.mjs +2 -1
  22. package/dist/browser.d.mts +451 -11
  23. package/dist/browser.d.ts +451 -11
  24. package/dist/browser.js +7659 -2321
  25. package/dist/browser.mjs +7659 -2321
  26. package/dist/cli.js +7143 -2435
  27. package/dist/edge-light.d.mts +1 -1
  28. package/dist/edge-light.d.ts +1 -1
  29. package/dist/edge-light.js +7659 -2321
  30. package/dist/edge-light.mjs +7659 -2321
  31. package/dist/index.d.mts +451 -11
  32. package/dist/index.d.ts +451 -11
  33. package/dist/index.js +8054 -2716
  34. package/dist/index.mjs +7659 -2321
  35. package/dist/instrumentation/index.d.mts +17 -0
  36. package/dist/instrumentation/index.d.ts +17 -0
  37. package/dist/instrumentation/index.js +6420 -2051
  38. package/dist/instrumentation/index.mjs +6420 -2051
  39. package/dist/workerd.d.mts +1 -1
  40. package/dist/workerd.d.ts +1 -1
  41. package/dist/workerd.js +7659 -2321
  42. package/dist/workerd.mjs +7659 -2321
  43. package/package.json +56 -51
  44. package/util/dist/index.d.mts +42 -1
  45. package/util/dist/index.d.ts +42 -1
  46. package/util/dist/index.js +5 -1
  47. package/util/dist/index.mjs +4 -0
  48. package/LICENSE +0 -201
@@ -377,6 +377,10 @@ var anthropicChannels = defineChannels("@anthropic-ai/sdk", {
377
377
  betaMessagesCreate: channel({
378
378
  channelName: "beta.messages.create",
379
379
  kind: "async"
380
+ }),
381
+ betaMessagesToolRunner: channel({
382
+ channelName: "beta.messages.toolRunner",
383
+ kind: "sync-stream"
380
384
  })
381
385
  });
382
386
 
@@ -423,6 +427,20 @@ var anthropicConfigs = [
423
427
  methodName: "create",
424
428
  kind: "Async"
425
429
  }
430
+ },
431
+ // Beta Messages API - toolRunner (sync helper returning async iterable/thenable)
432
+ {
433
+ channelName: anthropicChannels.betaMessagesToolRunner.channelName,
434
+ module: {
435
+ name: "@anthropic-ai/sdk",
436
+ versionRange: ">=0.39.0",
437
+ filePath: "resources/beta/messages/messages.mjs"
438
+ },
439
+ functionQuery: {
440
+ className: "Messages",
441
+ methodName: "toolRunner",
442
+ kind: "Sync"
443
+ }
426
444
  }
427
445
  ];
428
446
 
@@ -462,6 +480,10 @@ var aiSDKChannels = defineChannels("ai", {
462
480
  channelName: "embedMany",
463
481
  kind: "async"
464
482
  }),
483
+ rerank: channel({
484
+ channelName: "rerank",
485
+ kind: "async"
486
+ }),
465
487
  agentGenerate: channel({
466
488
  channelName: "Agent.generate",
467
489
  kind: "async"
@@ -541,7 +563,7 @@ var aiSDKConfigs = [
541
563
  channelName: aiSDKChannels.streamText.channelName,
542
564
  module: {
543
565
  name: "ai",
544
- versionRange: ">=3.0.0",
566
+ versionRange: ">=3.0.0 <4.0.0",
545
567
  filePath: "dist/index.js"
546
568
  },
547
569
  functionQuery: {
@@ -549,6 +571,18 @@ var aiSDKConfigs = [
549
571
  kind: "Async"
550
572
  }
551
573
  },
574
+ {
575
+ channelName: aiSDKChannels.streamTextSync.channelName,
576
+ module: {
577
+ name: "ai",
578
+ versionRange: ">=4.0.0",
579
+ filePath: "dist/index.js"
580
+ },
581
+ functionQuery: {
582
+ functionName: "streamText",
583
+ kind: "Sync"
584
+ }
585
+ },
552
586
  // generateObject - async function
553
587
  {
554
588
  channelName: aiSDKChannels.generateObject.channelName,
@@ -624,6 +658,31 @@ var aiSDKConfigs = [
624
658
  kind: "Async"
625
659
  }
626
660
  },
661
+ // rerank - async function
662
+ {
663
+ channelName: aiSDKChannels.rerank.channelName,
664
+ module: {
665
+ name: "ai",
666
+ versionRange: ">=5.0.0",
667
+ filePath: "dist/index.mjs"
668
+ },
669
+ functionQuery: {
670
+ functionName: "rerank",
671
+ kind: "Async"
672
+ }
673
+ },
674
+ {
675
+ channelName: aiSDKChannels.rerank.channelName,
676
+ module: {
677
+ name: "ai",
678
+ versionRange: ">=5.0.0",
679
+ filePath: "dist/index.js"
680
+ },
681
+ functionQuery: {
682
+ functionName: "rerank",
683
+ kind: "Async"
684
+ }
685
+ },
627
686
  // streamObject - async function (v3 only, before the sync refactor in v4)
628
687
  {
629
688
  channelName: aiSDKChannels.streamObject.channelName,
@@ -654,7 +713,7 @@ var aiSDKConfigs = [
654
713
  channelName: aiSDKChannels.streamObject.channelName,
655
714
  module: {
656
715
  name: "ai",
657
- versionRange: ">=3.0.0",
716
+ versionRange: ">=3.0.0 <4.0.0",
658
717
  filePath: "dist/index.js"
659
718
  },
660
719
  functionQuery: {
@@ -662,6 +721,18 @@ var aiSDKConfigs = [
662
721
  kind: "Async"
663
722
  }
664
723
  },
724
+ {
725
+ channelName: aiSDKChannels.streamObjectSync.channelName,
726
+ module: {
727
+ name: "ai",
728
+ versionRange: ">=4.0.0",
729
+ filePath: "dist/index.js"
730
+ },
731
+ functionQuery: {
732
+ functionName: "streamObject",
733
+ kind: "Sync"
734
+ }
735
+ },
665
736
  // Agent.generate - async method (v5 only)
666
737
  // The compiled AI SDK bundle emits this as an anonymous class method, so we
667
738
  // target the first async `generate` method in the file instead of a class name.
@@ -822,6 +893,71 @@ var claudeAgentSDKConfigs = [
822
893
  }
823
894
  ];
824
895
 
896
+ // src/instrumentation/plugins/cursor-sdk-channels.ts
897
+ var cursorSDKChannels = defineChannels("@cursor/sdk", {
898
+ create: channel({
899
+ channelName: "Agent.create",
900
+ kind: "async"
901
+ }),
902
+ resume: channel({
903
+ channelName: "Agent.resume",
904
+ kind: "async"
905
+ }),
906
+ prompt: channel({
907
+ channelName: "Agent.prompt",
908
+ kind: "async"
909
+ }),
910
+ send: channel({
911
+ channelName: "agent.send",
912
+ kind: "async"
913
+ })
914
+ });
915
+
916
+ // src/auto-instrumentations/configs/cursor-sdk.ts
917
+ var cursorSDKVersionRange = ">=1.0.7 <2.0.0";
918
+ var cursorSDKEntrypoints = ["dist/esm/index.js", "dist/cjs/index.js"];
919
+ var cursorSDKConfigs = cursorSDKEntrypoints.flatMap((filePath) => [
920
+ {
921
+ channelName: cursorSDKChannels.create.channelName,
922
+ module: {
923
+ name: "@cursor/sdk",
924
+ versionRange: cursorSDKVersionRange,
925
+ filePath
926
+ },
927
+ functionQuery: {
928
+ className: "Agent",
929
+ methodName: "create",
930
+ kind: "Async"
931
+ }
932
+ },
933
+ {
934
+ channelName: cursorSDKChannels.resume.channelName,
935
+ module: {
936
+ name: "@cursor/sdk",
937
+ versionRange: cursorSDKVersionRange,
938
+ filePath
939
+ },
940
+ functionQuery: {
941
+ className: "Agent",
942
+ methodName: "resume",
943
+ kind: "Async"
944
+ }
945
+ },
946
+ {
947
+ channelName: cursorSDKChannels.prompt.channelName,
948
+ module: {
949
+ name: "@cursor/sdk",
950
+ versionRange: cursorSDKVersionRange,
951
+ filePath
952
+ },
953
+ functionQuery: {
954
+ className: "Agent",
955
+ methodName: "prompt",
956
+ kind: "Async"
957
+ }
958
+ }
959
+ ]);
960
+
825
961
  // src/instrumentation/plugins/google-genai-channels.ts
826
962
  var googleGenAIChannels = defineChannels("@google/genai", {
827
963
  generateContent: channel({
@@ -831,6 +967,10 @@ var googleGenAIChannels = defineChannels("@google/genai", {
831
967
  generateContentStream: channel({
832
968
  channelName: "models.generateContentStream",
833
969
  kind: "async"
970
+ }),
971
+ embedContent: channel({
972
+ channelName: "models.embedContent",
973
+ kind: "async"
834
974
  })
835
975
  });
836
976
 
@@ -865,204 +1005,520 @@ var googleGenAIConfigs = [
865
1005
  methodName: "generateContentStreamInternal",
866
1006
  kind: "Async"
867
1007
  }
868
- }
869
- ];
870
-
871
- // src/instrumentation/plugins/openrouter-agent-channels.ts
872
- var openRouterAgentChannels = defineChannels("@openrouter/agent", {
873
- callModel: channel({
874
- channelName: "callModel",
875
- kind: "sync-stream"
876
- }),
877
- callModelTurn: channel({
878
- channelName: "callModel.turn",
879
- kind: "async"
880
- }),
881
- toolExecute: channel({
882
- channelName: "tool.execute",
883
- kind: "async"
884
- })
885
- });
886
-
887
- // src/auto-instrumentations/configs/openrouter-agent.ts
888
- var openRouterAgentConfigs = [
1008
+ },
1009
+ // Models.embedContent - class method in older SDK versions
889
1010
  {
890
- channelName: openRouterAgentChannels.callModel.channelName,
1011
+ channelName: googleGenAIChannels.embedContent.channelName,
891
1012
  module: {
892
- name: "@openrouter/agent",
893
- versionRange: ">=0.1.2",
894
- filePath: "esm/inner-loop/call-model.js"
1013
+ name: "@google/genai",
1014
+ versionRange: ">=1.0.0 <1.44.0",
1015
+ filePath: "dist/node/index.mjs"
895
1016
  },
896
1017
  functionQuery: {
897
- functionName: "callModel",
898
- kind: "Sync"
1018
+ className: "Models",
1019
+ methodName: "embedContent",
1020
+ kind: "Async"
1021
+ }
1022
+ },
1023
+ // Models.embedContentInternal - class method in newer SDK versions
1024
+ // Note: embedContent is an arrow function property that calls this method
1025
+ {
1026
+ channelName: googleGenAIChannels.embedContent.channelName,
1027
+ module: {
1028
+ name: "@google/genai",
1029
+ versionRange: ">=1.44.0",
1030
+ filePath: "dist/node/index.mjs"
1031
+ },
1032
+ functionQuery: {
1033
+ className: "Models",
1034
+ methodName: "embedContentInternal",
1035
+ kind: "Async"
899
1036
  }
900
1037
  }
901
1038
  ];
902
1039
 
903
- // src/instrumentation/plugins/openrouter-channels.ts
904
- var openRouterChannels = defineChannels("@openrouter/sdk", {
905
- chatSend: channel({
906
- channelName: "chat.send",
1040
+ // src/instrumentation/plugins/huggingface-channels.ts
1041
+ var huggingFaceChannels = defineChannels("@huggingface/inference", {
1042
+ chatCompletion: channel({
1043
+ channelName: "chatCompletion",
907
1044
  kind: "async"
908
1045
  }),
909
- embeddingsGenerate: channel({
910
- channelName: "embeddings.generate",
911
- kind: "async"
1046
+ chatCompletionStream: channel({
1047
+ channelName: "chatCompletionStream",
1048
+ kind: "sync-stream"
912
1049
  }),
913
- betaResponsesSend: channel({
914
- channelName: "beta.responses.send",
1050
+ textGeneration: channel({
1051
+ channelName: "textGeneration",
915
1052
  kind: "async"
916
1053
  }),
917
- callModel: channel({
918
- channelName: "callModel",
1054
+ textGenerationStream: channel({
1055
+ channelName: "textGenerationStream",
919
1056
  kind: "sync-stream"
920
1057
  }),
921
- callModelTurn: channel({
922
- channelName: "callModel.turn",
923
- kind: "async"
924
- }),
925
- toolExecute: channel({
926
- channelName: "tool.execute",
1058
+ featureExtraction: channel({
1059
+ channelName: "featureExtraction",
927
1060
  kind: "async"
928
1061
  })
929
1062
  });
930
1063
 
931
- // src/auto-instrumentations/configs/openrouter.ts
932
- var openRouterConfigs = [
1064
+ // src/auto-instrumentations/configs/huggingface.ts
1065
+ var huggingFaceConfigs = [
933
1066
  {
934
- channelName: openRouterChannels.chatSend.channelName,
1067
+ channelName: huggingFaceChannels.chatCompletion.channelName,
935
1068
  module: {
936
- name: "@openrouter/sdk",
937
- versionRange: ">=0.9.11 <1.0.0",
938
- filePath: "esm/sdk/chat.js"
1069
+ name: "@huggingface/inference",
1070
+ versionRange: ">=2.0.0 <3.0.0",
1071
+ filePath: "dist/index.js"
939
1072
  },
940
1073
  functionQuery: {
941
- className: "Chat",
942
- methodName: "send",
1074
+ functionName: "chatCompletion",
943
1075
  kind: "Async"
944
1076
  }
945
1077
  },
946
1078
  {
947
- channelName: openRouterChannels.embeddingsGenerate.channelName,
1079
+ channelName: huggingFaceChannels.chatCompletion.channelName,
948
1080
  module: {
949
- name: "@openrouter/sdk",
950
- versionRange: ">=0.9.11 <1.0.0",
951
- filePath: "esm/sdk/embeddings.js"
1081
+ name: "@huggingface/inference",
1082
+ versionRange: ">=2.0.0 <3.0.0",
1083
+ filePath: "dist/index.cjs"
952
1084
  },
953
1085
  functionQuery: {
954
- className: "Embeddings",
955
- methodName: "generate",
1086
+ functionName: "chatCompletion",
956
1087
  kind: "Async"
957
1088
  }
958
1089
  },
959
1090
  {
960
- channelName: openRouterChannels.betaResponsesSend.channelName,
1091
+ channelName: huggingFaceChannels.chatCompletionStream.channelName,
961
1092
  module: {
962
- name: "@openrouter/sdk",
963
- versionRange: ">=0.9.11 <1.0.0",
964
- filePath: "esm/sdk/responses.js"
1093
+ name: "@huggingface/inference",
1094
+ versionRange: ">=2.0.0 <3.0.0",
1095
+ filePath: "dist/index.js"
965
1096
  },
966
1097
  functionQuery: {
967
- className: "Responses",
968
- methodName: "send",
969
- kind: "Async"
1098
+ functionName: "chatCompletionStream",
1099
+ kind: "Sync"
970
1100
  }
971
1101
  },
972
1102
  {
973
- channelName: openRouterChannels.callModel.channelName,
1103
+ channelName: huggingFaceChannels.chatCompletionStream.channelName,
974
1104
  module: {
975
- name: "@openrouter/sdk",
976
- versionRange: ">=0.9.11 <1.0.0",
977
- filePath: "esm/sdk/sdk.js"
1105
+ name: "@huggingface/inference",
1106
+ versionRange: ">=2.0.0 <3.0.0",
1107
+ filePath: "dist/index.cjs"
978
1108
  },
979
1109
  functionQuery: {
980
- className: "OpenRouter",
981
- methodName: "callModel",
1110
+ functionName: "chatCompletionStream",
982
1111
  kind: "Sync"
983
1112
  }
984
- }
985
- ];
986
-
987
- // src/instrumentation/plugins/mistral-channels.ts
988
- var mistralChannels = defineChannels("@mistralai/mistralai", {
989
- chatComplete: channel({
990
- channelName: "chat.complete",
991
- kind: "async"
992
- }),
993
- chatStream: channel({
994
- channelName: "chat.stream",
995
- kind: "async"
996
- }),
997
- embeddingsCreate: channel({
998
- channelName: "embeddings.create",
999
- kind: "async"
1000
- }),
1001
- fimComplete: channel({
1002
- channelName: "fim.complete",
1003
- kind: "async"
1004
- }),
1005
- fimStream: channel({
1006
- channelName: "fim.stream",
1007
- kind: "async"
1008
- }),
1009
- agentsComplete: channel({
1010
- channelName: "agents.complete",
1011
- kind: "async"
1012
- }),
1013
- agentsStream: channel({
1014
- channelName: "agents.stream",
1015
- kind: "async"
1016
- })
1017
- });
1018
-
1019
- // src/auto-instrumentations/configs/mistral.ts
1020
- var mistralConfigs = [
1113
+ },
1021
1114
  {
1022
- channelName: mistralChannels.chatComplete.channelName,
1115
+ channelName: huggingFaceChannels.textGeneration.channelName,
1023
1116
  module: {
1024
- name: "@mistralai/mistralai",
1025
- versionRange: ">=1.0.0 <2.0.0",
1026
- filePath: "sdk/chat.js"
1117
+ name: "@huggingface/inference",
1118
+ versionRange: ">=2.0.0 <3.0.0",
1119
+ filePath: "dist/index.js"
1027
1120
  },
1028
1121
  functionQuery: {
1029
- className: "Chat",
1030
- methodName: "complete",
1122
+ functionName: "textGeneration",
1031
1123
  kind: "Async"
1032
1124
  }
1033
1125
  },
1034
1126
  {
1035
- channelName: mistralChannels.chatComplete.channelName,
1127
+ channelName: huggingFaceChannels.textGeneration.channelName,
1036
1128
  module: {
1037
- name: "@mistralai/mistralai",
1129
+ name: "@huggingface/inference",
1038
1130
  versionRange: ">=2.0.0 <3.0.0",
1039
- filePath: "esm/sdk/chat.js"
1131
+ filePath: "dist/index.cjs"
1040
1132
  },
1041
1133
  functionQuery: {
1042
- className: "Chat",
1043
- methodName: "complete",
1134
+ functionName: "textGeneration",
1044
1135
  kind: "Async"
1045
1136
  }
1046
1137
  },
1047
1138
  {
1048
- channelName: mistralChannels.chatStream.channelName,
1139
+ channelName: huggingFaceChannels.textGenerationStream.channelName,
1049
1140
  module: {
1050
- name: "@mistralai/mistralai",
1051
- versionRange: ">=1.0.0 <2.0.0",
1052
- filePath: "sdk/chat.js"
1141
+ name: "@huggingface/inference",
1142
+ versionRange: ">=2.0.0 <3.0.0",
1143
+ filePath: "dist/index.js"
1053
1144
  },
1054
1145
  functionQuery: {
1055
- className: "Chat",
1056
- methodName: "stream",
1057
- kind: "Async"
1146
+ functionName: "textGenerationStream",
1147
+ kind: "Sync"
1058
1148
  }
1059
1149
  },
1060
1150
  {
1061
- channelName: mistralChannels.chatStream.channelName,
1151
+ channelName: huggingFaceChannels.textGenerationStream.channelName,
1062
1152
  module: {
1063
- name: "@mistralai/mistralai",
1153
+ name: "@huggingface/inference",
1064
1154
  versionRange: ">=2.0.0 <3.0.0",
1065
- filePath: "esm/sdk/chat.js"
1155
+ filePath: "dist/index.cjs"
1156
+ },
1157
+ functionQuery: {
1158
+ functionName: "textGenerationStream",
1159
+ kind: "Sync"
1160
+ }
1161
+ },
1162
+ {
1163
+ channelName: huggingFaceChannels.featureExtraction.channelName,
1164
+ module: {
1165
+ name: "@huggingface/inference",
1166
+ versionRange: ">=2.0.0 <3.0.0",
1167
+ filePath: "dist/index.js"
1168
+ },
1169
+ functionQuery: {
1170
+ functionName: "featureExtraction",
1171
+ kind: "Async"
1172
+ }
1173
+ },
1174
+ {
1175
+ channelName: huggingFaceChannels.featureExtraction.channelName,
1176
+ module: {
1177
+ name: "@huggingface/inference",
1178
+ versionRange: ">=2.0.0 <3.0.0",
1179
+ filePath: "dist/index.cjs"
1180
+ },
1181
+ functionQuery: {
1182
+ functionName: "featureExtraction",
1183
+ kind: "Async"
1184
+ }
1185
+ },
1186
+ {
1187
+ channelName: huggingFaceChannels.chatCompletion.channelName,
1188
+ module: {
1189
+ name: "@huggingface/inference",
1190
+ versionRange: ">=3.0.0 <5.0.0",
1191
+ filePath: "dist/esm/tasks/nlp/chatCompletion.js"
1192
+ },
1193
+ functionQuery: {
1194
+ functionName: "chatCompletion",
1195
+ kind: "Async"
1196
+ }
1197
+ },
1198
+ {
1199
+ channelName: huggingFaceChannels.chatCompletion.channelName,
1200
+ module: {
1201
+ name: "@huggingface/inference",
1202
+ versionRange: ">=3.0.0 <5.0.0",
1203
+ filePath: "dist/commonjs/tasks/nlp/chatCompletion.js"
1204
+ },
1205
+ functionQuery: {
1206
+ functionName: "chatCompletion",
1207
+ kind: "Async"
1208
+ }
1209
+ },
1210
+ {
1211
+ channelName: huggingFaceChannels.chatCompletionStream.channelName,
1212
+ module: {
1213
+ name: "@huggingface/inference",
1214
+ versionRange: ">=3.0.0 <5.0.0",
1215
+ filePath: "dist/esm/tasks/nlp/chatCompletionStream.js"
1216
+ },
1217
+ functionQuery: {
1218
+ functionName: "chatCompletionStream",
1219
+ kind: "Sync"
1220
+ }
1221
+ },
1222
+ {
1223
+ channelName: huggingFaceChannels.chatCompletionStream.channelName,
1224
+ module: {
1225
+ name: "@huggingface/inference",
1226
+ versionRange: ">=3.0.0 <5.0.0",
1227
+ filePath: "dist/commonjs/tasks/nlp/chatCompletionStream.js"
1228
+ },
1229
+ functionQuery: {
1230
+ functionName: "chatCompletionStream",
1231
+ kind: "Sync"
1232
+ }
1233
+ },
1234
+ {
1235
+ channelName: huggingFaceChannels.textGeneration.channelName,
1236
+ module: {
1237
+ name: "@huggingface/inference",
1238
+ versionRange: ">=3.0.0 <5.0.0",
1239
+ filePath: "dist/esm/tasks/nlp/textGeneration.js"
1240
+ },
1241
+ functionQuery: {
1242
+ functionName: "textGeneration",
1243
+ kind: "Async"
1244
+ }
1245
+ },
1246
+ {
1247
+ channelName: huggingFaceChannels.textGeneration.channelName,
1248
+ module: {
1249
+ name: "@huggingface/inference",
1250
+ versionRange: ">=3.0.0 <5.0.0",
1251
+ filePath: "dist/commonjs/tasks/nlp/textGeneration.js"
1252
+ },
1253
+ functionQuery: {
1254
+ functionName: "textGeneration",
1255
+ kind: "Async"
1256
+ }
1257
+ },
1258
+ {
1259
+ channelName: huggingFaceChannels.textGenerationStream.channelName,
1260
+ module: {
1261
+ name: "@huggingface/inference",
1262
+ versionRange: ">=3.0.0 <5.0.0",
1263
+ filePath: "dist/esm/tasks/nlp/textGenerationStream.js"
1264
+ },
1265
+ functionQuery: {
1266
+ functionName: "textGenerationStream",
1267
+ kind: "Sync"
1268
+ }
1269
+ },
1270
+ {
1271
+ channelName: huggingFaceChannels.textGenerationStream.channelName,
1272
+ module: {
1273
+ name: "@huggingface/inference",
1274
+ versionRange: ">=3.0.0 <5.0.0",
1275
+ filePath: "dist/commonjs/tasks/nlp/textGenerationStream.js"
1276
+ },
1277
+ functionQuery: {
1278
+ functionName: "textGenerationStream",
1279
+ kind: "Sync"
1280
+ }
1281
+ },
1282
+ {
1283
+ channelName: huggingFaceChannels.featureExtraction.channelName,
1284
+ module: {
1285
+ name: "@huggingface/inference",
1286
+ versionRange: ">=3.0.0 <5.0.0",
1287
+ filePath: "dist/esm/tasks/nlp/featureExtraction.js"
1288
+ },
1289
+ functionQuery: {
1290
+ functionName: "featureExtraction",
1291
+ kind: "Async"
1292
+ }
1293
+ },
1294
+ {
1295
+ channelName: huggingFaceChannels.featureExtraction.channelName,
1296
+ module: {
1297
+ name: "@huggingface/inference",
1298
+ versionRange: ">=3.0.0 <5.0.0",
1299
+ filePath: "dist/commonjs/tasks/nlp/featureExtraction.js"
1300
+ },
1301
+ functionQuery: {
1302
+ functionName: "featureExtraction",
1303
+ kind: "Async"
1304
+ }
1305
+ }
1306
+ ];
1307
+
1308
+ // src/instrumentation/plugins/openrouter-agent-channels.ts
1309
+ var openRouterAgentChannels = defineChannels("@openrouter/agent", {
1310
+ callModel: channel({
1311
+ channelName: "callModel",
1312
+ kind: "sync-stream"
1313
+ }),
1314
+ callModelTurn: channel({
1315
+ channelName: "callModel.turn",
1316
+ kind: "async"
1317
+ }),
1318
+ toolExecute: channel({
1319
+ channelName: "tool.execute",
1320
+ kind: "async"
1321
+ })
1322
+ });
1323
+
1324
+ // src/auto-instrumentations/configs/openrouter-agent.ts
1325
+ var openRouterAgentConfigs = [
1326
+ {
1327
+ channelName: openRouterAgentChannels.callModel.channelName,
1328
+ module: {
1329
+ name: "@openrouter/agent",
1330
+ versionRange: ">=0.1.2",
1331
+ filePath: "esm/inner-loop/call-model.js"
1332
+ },
1333
+ functionQuery: {
1334
+ functionName: "callModel",
1335
+ kind: "Sync"
1336
+ }
1337
+ }
1338
+ ];
1339
+
1340
+ // src/instrumentation/plugins/openrouter-channels.ts
1341
+ var openRouterChannels = defineChannels("@openrouter/sdk", {
1342
+ chatSend: channel({
1343
+ channelName: "chat.send",
1344
+ kind: "async"
1345
+ }),
1346
+ embeddingsGenerate: channel({
1347
+ channelName: "embeddings.generate",
1348
+ kind: "async"
1349
+ }),
1350
+ rerankRerank: channel(
1351
+ {
1352
+ channelName: "rerank.rerank",
1353
+ kind: "async"
1354
+ }
1355
+ ),
1356
+ betaResponsesSend: channel({
1357
+ channelName: "beta.responses.send",
1358
+ kind: "async"
1359
+ }),
1360
+ callModel: channel({
1361
+ channelName: "callModel",
1362
+ kind: "sync-stream"
1363
+ }),
1364
+ callModelTurn: channel({
1365
+ channelName: "callModel.turn",
1366
+ kind: "async"
1367
+ }),
1368
+ toolExecute: channel({
1369
+ channelName: "tool.execute",
1370
+ kind: "async"
1371
+ })
1372
+ });
1373
+
1374
+ // src/auto-instrumentations/configs/openrouter.ts
1375
+ var openRouterConfigs = [
1376
+ {
1377
+ channelName: openRouterChannels.chatSend.channelName,
1378
+ module: {
1379
+ name: "@openrouter/sdk",
1380
+ versionRange: ">=0.9.11 <1.0.0",
1381
+ filePath: "esm/sdk/chat.js"
1382
+ },
1383
+ functionQuery: {
1384
+ className: "Chat",
1385
+ methodName: "send",
1386
+ kind: "Async"
1387
+ }
1388
+ },
1389
+ {
1390
+ channelName: openRouterChannels.embeddingsGenerate.channelName,
1391
+ module: {
1392
+ name: "@openrouter/sdk",
1393
+ versionRange: ">=0.9.11 <1.0.0",
1394
+ filePath: "esm/sdk/embeddings.js"
1395
+ },
1396
+ functionQuery: {
1397
+ className: "Embeddings",
1398
+ methodName: "generate",
1399
+ kind: "Async"
1400
+ }
1401
+ },
1402
+ {
1403
+ channelName: openRouterChannels.rerankRerank.channelName,
1404
+ module: {
1405
+ name: "@openrouter/sdk",
1406
+ versionRange: ">=0.12.0 <1.0.0",
1407
+ filePath: "esm/sdk/rerank.js"
1408
+ },
1409
+ functionQuery: {
1410
+ className: "Rerank",
1411
+ methodName: "rerank",
1412
+ kind: "Async"
1413
+ }
1414
+ },
1415
+ {
1416
+ channelName: openRouterChannels.betaResponsesSend.channelName,
1417
+ module: {
1418
+ name: "@openrouter/sdk",
1419
+ versionRange: ">=0.9.11 <1.0.0",
1420
+ filePath: "esm/sdk/responses.js"
1421
+ },
1422
+ functionQuery: {
1423
+ className: "Responses",
1424
+ methodName: "send",
1425
+ kind: "Async"
1426
+ }
1427
+ },
1428
+ {
1429
+ channelName: openRouterChannels.callModel.channelName,
1430
+ module: {
1431
+ name: "@openrouter/sdk",
1432
+ versionRange: ">=0.9.11 <1.0.0",
1433
+ filePath: "esm/sdk/sdk.js"
1434
+ },
1435
+ functionQuery: {
1436
+ className: "OpenRouter",
1437
+ methodName: "callModel",
1438
+ kind: "Sync"
1439
+ }
1440
+ }
1441
+ ];
1442
+
1443
+ // src/instrumentation/plugins/mistral-channels.ts
1444
+ var mistralChannels = defineChannels("@mistralai/mistralai", {
1445
+ chatComplete: channel({
1446
+ channelName: "chat.complete",
1447
+ kind: "async"
1448
+ }),
1449
+ chatStream: channel({
1450
+ channelName: "chat.stream",
1451
+ kind: "async"
1452
+ }),
1453
+ embeddingsCreate: channel({
1454
+ channelName: "embeddings.create",
1455
+ kind: "async"
1456
+ }),
1457
+ fimComplete: channel({
1458
+ channelName: "fim.complete",
1459
+ kind: "async"
1460
+ }),
1461
+ fimStream: channel({
1462
+ channelName: "fim.stream",
1463
+ kind: "async"
1464
+ }),
1465
+ agentsComplete: channel({
1466
+ channelName: "agents.complete",
1467
+ kind: "async"
1468
+ }),
1469
+ agentsStream: channel({
1470
+ channelName: "agents.stream",
1471
+ kind: "async"
1472
+ })
1473
+ });
1474
+
1475
+ // src/auto-instrumentations/configs/mistral.ts
1476
+ var mistralConfigs = [
1477
+ {
1478
+ channelName: mistralChannels.chatComplete.channelName,
1479
+ module: {
1480
+ name: "@mistralai/mistralai",
1481
+ versionRange: ">=1.0.0 <2.0.0",
1482
+ filePath: "sdk/chat.js"
1483
+ },
1484
+ functionQuery: {
1485
+ className: "Chat",
1486
+ methodName: "complete",
1487
+ kind: "Async"
1488
+ }
1489
+ },
1490
+ {
1491
+ channelName: mistralChannels.chatComplete.channelName,
1492
+ module: {
1493
+ name: "@mistralai/mistralai",
1494
+ versionRange: ">=2.0.0 <3.0.0",
1495
+ filePath: "esm/sdk/chat.js"
1496
+ },
1497
+ functionQuery: {
1498
+ className: "Chat",
1499
+ methodName: "complete",
1500
+ kind: "Async"
1501
+ }
1502
+ },
1503
+ {
1504
+ channelName: mistralChannels.chatStream.channelName,
1505
+ module: {
1506
+ name: "@mistralai/mistralai",
1507
+ versionRange: ">=1.0.0 <2.0.0",
1508
+ filePath: "sdk/chat.js"
1509
+ },
1510
+ functionQuery: {
1511
+ className: "Chat",
1512
+ methodName: "stream",
1513
+ kind: "Async"
1514
+ }
1515
+ },
1516
+ {
1517
+ channelName: mistralChannels.chatStream.channelName,
1518
+ module: {
1519
+ name: "@mistralai/mistralai",
1520
+ versionRange: ">=2.0.0 <3.0.0",
1521
+ filePath: "esm/sdk/chat.js"
1066
1522
  },
1067
1523
  functionQuery: {
1068
1524
  className: "Chat",
@@ -1071,132 +1527,652 @@ var mistralConfigs = [
1071
1527
  }
1072
1528
  },
1073
1529
  {
1074
- channelName: mistralChannels.embeddingsCreate.channelName,
1530
+ channelName: mistralChannels.embeddingsCreate.channelName,
1531
+ module: {
1532
+ name: "@mistralai/mistralai",
1533
+ versionRange: ">=1.0.0 <2.0.0",
1534
+ filePath: "sdk/embeddings.js"
1535
+ },
1536
+ functionQuery: {
1537
+ className: "Embeddings",
1538
+ methodName: "create",
1539
+ kind: "Async"
1540
+ }
1541
+ },
1542
+ {
1543
+ channelName: mistralChannels.embeddingsCreate.channelName,
1544
+ module: {
1545
+ name: "@mistralai/mistralai",
1546
+ versionRange: ">=2.0.0 <3.0.0",
1547
+ filePath: "esm/sdk/embeddings.js"
1548
+ },
1549
+ functionQuery: {
1550
+ className: "Embeddings",
1551
+ methodName: "create",
1552
+ kind: "Async"
1553
+ }
1554
+ },
1555
+ {
1556
+ channelName: mistralChannels.fimComplete.channelName,
1557
+ module: {
1558
+ name: "@mistralai/mistralai",
1559
+ versionRange: ">=1.0.0 <2.0.0",
1560
+ filePath: "sdk/fim.js"
1561
+ },
1562
+ functionQuery: {
1563
+ className: "Fim",
1564
+ methodName: "complete",
1565
+ kind: "Async"
1566
+ }
1567
+ },
1568
+ {
1569
+ channelName: mistralChannels.fimComplete.channelName,
1570
+ module: {
1571
+ name: "@mistralai/mistralai",
1572
+ versionRange: ">=2.0.0 <3.0.0",
1573
+ filePath: "esm/sdk/fim.js"
1574
+ },
1575
+ functionQuery: {
1576
+ className: "Fim",
1577
+ methodName: "complete",
1578
+ kind: "Async"
1579
+ }
1580
+ },
1581
+ {
1582
+ channelName: mistralChannels.fimStream.channelName,
1583
+ module: {
1584
+ name: "@mistralai/mistralai",
1585
+ versionRange: ">=1.0.0 <2.0.0",
1586
+ filePath: "sdk/fim.js"
1587
+ },
1588
+ functionQuery: {
1589
+ className: "Fim",
1590
+ methodName: "stream",
1591
+ kind: "Async"
1592
+ }
1593
+ },
1594
+ {
1595
+ channelName: mistralChannels.fimStream.channelName,
1596
+ module: {
1597
+ name: "@mistralai/mistralai",
1598
+ versionRange: ">=2.0.0 <3.0.0",
1599
+ filePath: "esm/sdk/fim.js"
1600
+ },
1601
+ functionQuery: {
1602
+ className: "Fim",
1603
+ methodName: "stream",
1604
+ kind: "Async"
1605
+ }
1606
+ },
1607
+ {
1608
+ channelName: mistralChannels.agentsComplete.channelName,
1609
+ module: {
1610
+ name: "@mistralai/mistralai",
1611
+ versionRange: ">=1.0.0 <2.0.0",
1612
+ filePath: "sdk/agents.js"
1613
+ },
1614
+ functionQuery: {
1615
+ className: "Agents",
1616
+ methodName: "complete",
1617
+ kind: "Async"
1618
+ }
1619
+ },
1620
+ {
1621
+ channelName: mistralChannels.agentsComplete.channelName,
1622
+ module: {
1623
+ name: "@mistralai/mistralai",
1624
+ versionRange: ">=2.0.0 <3.0.0",
1625
+ filePath: "esm/sdk/agents.js"
1626
+ },
1627
+ functionQuery: {
1628
+ className: "Agents",
1629
+ methodName: "complete",
1630
+ kind: "Async"
1631
+ }
1632
+ },
1633
+ {
1634
+ channelName: mistralChannels.agentsStream.channelName,
1635
+ module: {
1636
+ name: "@mistralai/mistralai",
1637
+ versionRange: ">=1.0.0 <2.0.0",
1638
+ filePath: "sdk/agents.js"
1639
+ },
1640
+ functionQuery: {
1641
+ className: "Agents",
1642
+ methodName: "stream",
1643
+ kind: "Async"
1644
+ }
1645
+ },
1646
+ {
1647
+ channelName: mistralChannels.agentsStream.channelName,
1648
+ module: {
1649
+ name: "@mistralai/mistralai",
1650
+ versionRange: ">=2.0.0 <3.0.0",
1651
+ filePath: "esm/sdk/agents.js"
1652
+ },
1653
+ functionQuery: {
1654
+ className: "Agents",
1655
+ methodName: "stream",
1656
+ kind: "Async"
1657
+ }
1658
+ }
1659
+ ];
1660
+
1661
+ // src/instrumentation/plugins/google-adk-channels.ts
1662
+ var googleADKChannels = defineChannels("@google/adk", {
1663
+ runnerRunAsync: channel({
1664
+ channelName: "runner.runAsync",
1665
+ kind: "sync-stream"
1666
+ }),
1667
+ agentRunAsync: channel({
1668
+ channelName: "agent.runAsync",
1669
+ kind: "sync-stream"
1670
+ }),
1671
+ toolRunAsync: channel({
1672
+ channelName: "tool.runAsync",
1673
+ kind: "async"
1674
+ })
1675
+ });
1676
+
1677
+ // src/auto-instrumentations/configs/google-adk.ts
1678
+ var googleADKVersionRange = ">=0.1.0";
1679
+ var googleADKBundledIndexV06VersionRange = ">=0.6.1 <0.7.0";
1680
+ var googleADKBundledIndexV1VersionRange = ">=1.0.0 <2.0.0";
1681
+ var googleADKConfigs = [
1682
+ // --- Runner.runAsync --- async generator, kind "Sync" + sync-stream channel
1683
+ // Runner.runAsync — ESM individual module file
1684
+ {
1685
+ channelName: googleADKChannels.runnerRunAsync.channelName,
1686
+ module: {
1687
+ name: "@google/adk",
1688
+ versionRange: googleADKVersionRange,
1689
+ filePath: "dist/esm/runner/runner.js"
1690
+ },
1691
+ functionQuery: {
1692
+ className: "Runner",
1693
+ methodName: "runAsync",
1694
+ kind: "Sync"
1695
+ }
1696
+ },
1697
+ // Runner.runAsync — bundled CJS/ESM indexes
1698
+ // The bundled entrypoints minify class names, so target the 12th sync
1699
+ // `runAsync` method in file order rather than a class name. This mapping is
1700
+ // only validated against the current 0.6.x bundle layout, so keep the range
1701
+ // tight until we verify newer bundled outputs.
1702
+ {
1703
+ channelName: googleADKChannels.runnerRunAsync.channelName,
1704
+ module: {
1705
+ name: "@google/adk",
1706
+ versionRange: googleADKBundledIndexV06VersionRange,
1707
+ filePath: "dist/cjs/index.js"
1708
+ },
1709
+ functionQuery: {
1710
+ methodName: "runAsync",
1711
+ kind: "Sync",
1712
+ index: 11
1713
+ }
1714
+ },
1715
+ {
1716
+ channelName: googleADKChannels.runnerRunAsync.channelName,
1717
+ module: {
1718
+ name: "@google/adk",
1719
+ versionRange: googleADKBundledIndexV06VersionRange,
1720
+ filePath: "dist/esm/index.js"
1721
+ },
1722
+ functionQuery: {
1723
+ methodName: "runAsync",
1724
+ kind: "Sync",
1725
+ index: 11
1726
+ }
1727
+ },
1728
+ // The 1.x bundled entrypoints still inline the runtime into index.js, but
1729
+ // the minified method order changed. These indices are verified against 1.0.0.
1730
+ {
1731
+ channelName: googleADKChannels.runnerRunAsync.channelName,
1732
+ module: {
1733
+ name: "@google/adk",
1734
+ versionRange: googleADKBundledIndexV1VersionRange,
1735
+ filePath: "dist/cjs/index.js"
1736
+ },
1737
+ functionQuery: {
1738
+ methodName: "runAsync",
1739
+ kind: "Sync",
1740
+ index: 12
1741
+ }
1742
+ },
1743
+ {
1744
+ channelName: googleADKChannels.runnerRunAsync.channelName,
1745
+ module: {
1746
+ name: "@google/adk",
1747
+ versionRange: googleADKBundledIndexV1VersionRange,
1748
+ filePath: "dist/esm/index.js"
1749
+ },
1750
+ functionQuery: {
1751
+ methodName: "runAsync",
1752
+ kind: "Sync",
1753
+ index: 12
1754
+ }
1755
+ },
1756
+ // --- BaseAgent.runAsync --- async generator, kind "Sync" + sync-stream channel
1757
+ // BaseAgent.runAsync — ESM individual module file
1758
+ {
1759
+ channelName: googleADKChannels.agentRunAsync.channelName,
1760
+ module: {
1761
+ name: "@google/adk",
1762
+ versionRange: googleADKVersionRange,
1763
+ filePath: "dist/esm/agents/base_agent.js"
1764
+ },
1765
+ functionQuery: {
1766
+ className: "BaseAgent",
1767
+ methodName: "runAsync",
1768
+ kind: "Sync"
1769
+ }
1770
+ },
1771
+ // BaseAgent.runAsync — bundled CJS/ESM indexes
1772
+ // The bundled entrypoints minify class names, so target the first sync
1773
+ // `runAsync` method in file order rather than a class name. This mapping is
1774
+ // only validated against the current 0.6.x bundle layout, so keep the range
1775
+ // tight until we verify newer bundled outputs.
1776
+ {
1777
+ channelName: googleADKChannels.agentRunAsync.channelName,
1778
+ module: {
1779
+ name: "@google/adk",
1780
+ versionRange: googleADKBundledIndexV06VersionRange,
1781
+ filePath: "dist/cjs/index.js"
1782
+ },
1783
+ functionQuery: {
1784
+ methodName: "runAsync",
1785
+ kind: "Sync",
1786
+ index: 0
1787
+ }
1788
+ },
1789
+ {
1790
+ channelName: googleADKChannels.agentRunAsync.channelName,
1791
+ module: {
1792
+ name: "@google/adk",
1793
+ versionRange: googleADKBundledIndexV06VersionRange,
1794
+ filePath: "dist/esm/index.js"
1795
+ },
1796
+ functionQuery: {
1797
+ methodName: "runAsync",
1798
+ kind: "Sync",
1799
+ index: 0
1800
+ }
1801
+ },
1802
+ // The 1.x bundled entrypoints keep BaseAgent.runAsync as the first bundled
1803
+ // async-generator runAsync method in file order.
1804
+ {
1805
+ channelName: googleADKChannels.agentRunAsync.channelName,
1806
+ module: {
1807
+ name: "@google/adk",
1808
+ versionRange: googleADKBundledIndexV1VersionRange,
1809
+ filePath: "dist/cjs/index.js"
1810
+ },
1811
+ functionQuery: {
1812
+ methodName: "runAsync",
1813
+ kind: "Sync",
1814
+ index: 0
1815
+ }
1816
+ },
1817
+ {
1818
+ channelName: googleADKChannels.agentRunAsync.channelName,
1819
+ module: {
1820
+ name: "@google/adk",
1821
+ versionRange: googleADKBundledIndexV1VersionRange,
1822
+ filePath: "dist/esm/index.js"
1823
+ },
1824
+ functionQuery: {
1825
+ methodName: "runAsync",
1826
+ kind: "Sync",
1827
+ index: 0
1828
+ }
1829
+ },
1830
+ // --- FunctionTool.runAsync --- regular async, kind "Async"
1831
+ // FunctionTool.runAsync — ESM individual module file
1832
+ {
1833
+ channelName: googleADKChannels.toolRunAsync.channelName,
1834
+ module: {
1835
+ name: "@google/adk",
1836
+ versionRange: googleADKVersionRange,
1837
+ filePath: "dist/esm/tools/function_tool.js"
1838
+ },
1839
+ functionQuery: {
1840
+ className: "FunctionTool",
1841
+ methodName: "runAsync",
1842
+ kind: "Async"
1843
+ }
1844
+ },
1845
+ // FunctionTool.runAsync — bundled CJS/ESM indexes
1846
+ // The bundled entrypoints minify class names, so target the first async
1847
+ // `runAsync` method in file order rather than a class name. This mapping is
1848
+ // only validated against the current 0.6.x bundle layout, so keep the range
1849
+ // tight until we verify newer bundled outputs.
1850
+ {
1851
+ channelName: googleADKChannels.toolRunAsync.channelName,
1075
1852
  module: {
1076
- name: "@mistralai/mistralai",
1077
- versionRange: ">=1.0.0 <2.0.0",
1078
- filePath: "sdk/embeddings.js"
1853
+ name: "@google/adk",
1854
+ versionRange: googleADKBundledIndexV06VersionRange,
1855
+ filePath: "dist/cjs/index.js"
1079
1856
  },
1080
1857
  functionQuery: {
1081
- className: "Embeddings",
1082
- methodName: "create",
1858
+ methodName: "runAsync",
1859
+ kind: "Async",
1860
+ index: 1
1861
+ }
1862
+ },
1863
+ {
1864
+ channelName: googleADKChannels.toolRunAsync.channelName,
1865
+ module: {
1866
+ name: "@google/adk",
1867
+ versionRange: googleADKBundledIndexV06VersionRange,
1868
+ filePath: "dist/esm/index.js"
1869
+ },
1870
+ functionQuery: {
1871
+ methodName: "runAsync",
1872
+ kind: "Async",
1873
+ index: 1
1874
+ }
1875
+ },
1876
+ // The 1.x bundle moves FunctionTool.runAsync behind one helper method.
1877
+ {
1878
+ channelName: googleADKChannels.toolRunAsync.channelName,
1879
+ module: {
1880
+ name: "@google/adk",
1881
+ versionRange: googleADKBundledIndexV1VersionRange,
1882
+ filePath: "dist/cjs/index.js"
1883
+ },
1884
+ functionQuery: {
1885
+ methodName: "runAsync",
1886
+ kind: "Async",
1887
+ index: 1
1888
+ }
1889
+ },
1890
+ {
1891
+ channelName: googleADKChannels.toolRunAsync.channelName,
1892
+ module: {
1893
+ name: "@google/adk",
1894
+ versionRange: googleADKBundledIndexV1VersionRange,
1895
+ filePath: "dist/esm/index.js"
1896
+ },
1897
+ functionQuery: {
1898
+ methodName: "runAsync",
1899
+ kind: "Async",
1900
+ index: 1
1901
+ }
1902
+ }
1903
+ ];
1904
+
1905
+ // src/instrumentation/plugins/cohere-channels.ts
1906
+ var cohereChannels = defineChannels("cohere-ai", {
1907
+ chat: channel({
1908
+ channelName: "chat",
1909
+ kind: "async"
1910
+ }),
1911
+ chatStream: channel({
1912
+ channelName: "chatStream",
1913
+ kind: "async"
1914
+ }),
1915
+ embed: channel({
1916
+ channelName: "embed",
1917
+ kind: "async"
1918
+ }),
1919
+ rerank: channel({
1920
+ channelName: "rerank",
1921
+ kind: "async"
1922
+ })
1923
+ });
1924
+
1925
+ // src/auto-instrumentations/configs/cohere.ts
1926
+ var cohereConfigs = [
1927
+ {
1928
+ channelName: cohereChannels.chat.channelName,
1929
+ module: {
1930
+ name: "cohere-ai",
1931
+ versionRange: ">=7.0.0 <8.0.0",
1932
+ filePath: "Client.js"
1933
+ },
1934
+ functionQuery: {
1935
+ className: "CohereClient",
1936
+ methodName: "chat",
1083
1937
  kind: "Async"
1084
1938
  }
1085
1939
  },
1086
1940
  {
1087
- channelName: mistralChannels.embeddingsCreate.channelName,
1941
+ channelName: cohereChannels.chat.channelName,
1088
1942
  module: {
1089
- name: "@mistralai/mistralai",
1090
- versionRange: ">=2.0.0 <3.0.0",
1091
- filePath: "esm/sdk/embeddings.js"
1943
+ name: "cohere-ai",
1944
+ versionRange: ">=7.20.0 <8.0.0",
1945
+ filePath: "api/resources/v2/client/Client.js"
1092
1946
  },
1093
1947
  functionQuery: {
1094
- className: "Embeddings",
1095
- methodName: "create",
1948
+ className: "V2Client",
1949
+ methodName: "chat",
1096
1950
  kind: "Async"
1097
1951
  }
1098
1952
  },
1099
1953
  {
1100
- channelName: mistralChannels.fimComplete.channelName,
1954
+ channelName: cohereChannels.chat.channelName,
1101
1955
  module: {
1102
- name: "@mistralai/mistralai",
1103
- versionRange: ">=1.0.0 <2.0.0",
1104
- filePath: "sdk/fim.js"
1956
+ name: "cohere-ai",
1957
+ versionRange: ">=8.0.0 <9.0.0",
1958
+ filePath: "Client.js"
1105
1959
  },
1106
1960
  functionQuery: {
1107
- className: "Fim",
1108
- methodName: "complete",
1961
+ className: "CohereClient",
1962
+ methodName: "chat",
1109
1963
  kind: "Async"
1110
1964
  }
1111
1965
  },
1112
1966
  {
1113
- channelName: mistralChannels.fimComplete.channelName,
1967
+ channelName: cohereChannels.chat.channelName,
1114
1968
  module: {
1115
- name: "@mistralai/mistralai",
1116
- versionRange: ">=2.0.0 <3.0.0",
1117
- filePath: "esm/sdk/fim.js"
1969
+ name: "cohere-ai",
1970
+ versionRange: ">=8.0.0 <9.0.0",
1971
+ filePath: "api/resources/v2/client/Client.js"
1118
1972
  },
1119
1973
  functionQuery: {
1120
- className: "Fim",
1121
- methodName: "complete",
1974
+ className: "V2Client",
1975
+ methodName: "chat",
1122
1976
  kind: "Async"
1123
1977
  }
1124
1978
  },
1125
1979
  {
1126
- channelName: mistralChannels.fimStream.channelName,
1980
+ channelName: cohereChannels.chatStream.channelName,
1127
1981
  module: {
1128
- name: "@mistralai/mistralai",
1129
- versionRange: ">=1.0.0 <2.0.0",
1130
- filePath: "sdk/fim.js"
1982
+ name: "cohere-ai",
1983
+ versionRange: ">=7.0.0 <8.0.0",
1984
+ filePath: "Client.js"
1131
1985
  },
1132
1986
  functionQuery: {
1133
- className: "Fim",
1134
- methodName: "stream",
1987
+ className: "CohereClient",
1988
+ methodName: "chatStream",
1135
1989
  kind: "Async"
1136
1990
  }
1137
1991
  },
1138
1992
  {
1139
- channelName: mistralChannels.fimStream.channelName,
1993
+ channelName: cohereChannels.chatStream.channelName,
1140
1994
  module: {
1141
- name: "@mistralai/mistralai",
1142
- versionRange: ">=2.0.0 <3.0.0",
1143
- filePath: "esm/sdk/fim.js"
1995
+ name: "cohere-ai",
1996
+ versionRange: ">=7.20.0 <8.0.0",
1997
+ filePath: "api/resources/v2/client/Client.js"
1144
1998
  },
1145
1999
  functionQuery: {
1146
- className: "Fim",
1147
- methodName: "stream",
2000
+ className: "V2Client",
2001
+ methodName: "chatStream",
1148
2002
  kind: "Async"
1149
2003
  }
1150
2004
  },
1151
2005
  {
1152
- channelName: mistralChannels.agentsComplete.channelName,
2006
+ channelName: cohereChannels.chatStream.channelName,
1153
2007
  module: {
1154
- name: "@mistralai/mistralai",
1155
- versionRange: ">=1.0.0 <2.0.0",
1156
- filePath: "sdk/agents.js"
2008
+ name: "cohere-ai",
2009
+ versionRange: ">=8.0.0 <9.0.0",
2010
+ filePath: "Client.js"
1157
2011
  },
1158
2012
  functionQuery: {
1159
- className: "Agents",
1160
- methodName: "complete",
2013
+ className: "CohereClient",
2014
+ methodName: "chatStream",
1161
2015
  kind: "Async"
1162
2016
  }
1163
2017
  },
1164
2018
  {
1165
- channelName: mistralChannels.agentsComplete.channelName,
2019
+ channelName: cohereChannels.chatStream.channelName,
1166
2020
  module: {
1167
- name: "@mistralai/mistralai",
1168
- versionRange: ">=2.0.0 <3.0.0",
1169
- filePath: "esm/sdk/agents.js"
2021
+ name: "cohere-ai",
2022
+ versionRange: ">=8.0.0 <9.0.0",
2023
+ filePath: "api/resources/v2/client/Client.js"
1170
2024
  },
1171
2025
  functionQuery: {
1172
- className: "Agents",
1173
- methodName: "complete",
2026
+ className: "V2Client",
2027
+ methodName: "chatStream",
1174
2028
  kind: "Async"
1175
2029
  }
1176
2030
  },
1177
2031
  {
1178
- channelName: mistralChannels.agentsStream.channelName,
2032
+ channelName: cohereChannels.embed.channelName,
1179
2033
  module: {
1180
- name: "@mistralai/mistralai",
1181
- versionRange: ">=1.0.0 <2.0.0",
1182
- filePath: "sdk/agents.js"
2034
+ name: "cohere-ai",
2035
+ versionRange: ">=7.0.0 <8.0.0",
2036
+ filePath: "Client.js"
1183
2037
  },
1184
2038
  functionQuery: {
1185
- className: "Agents",
1186
- methodName: "stream",
2039
+ className: "CohereClient",
2040
+ methodName: "embed",
1187
2041
  kind: "Async"
1188
2042
  }
1189
2043
  },
1190
2044
  {
1191
- channelName: mistralChannels.agentsStream.channelName,
2045
+ channelName: cohereChannels.embed.channelName,
1192
2046
  module: {
1193
- name: "@mistralai/mistralai",
1194
- versionRange: ">=2.0.0 <3.0.0",
1195
- filePath: "esm/sdk/agents.js"
2047
+ name: "cohere-ai",
2048
+ versionRange: ">=7.20.0 <8.0.0",
2049
+ filePath: "api/resources/v2/client/Client.js"
1196
2050
  },
1197
2051
  functionQuery: {
1198
- className: "Agents",
1199
- methodName: "stream",
2052
+ className: "V2Client",
2053
+ methodName: "embed",
2054
+ kind: "Async"
2055
+ }
2056
+ },
2057
+ {
2058
+ channelName: cohereChannels.embed.channelName,
2059
+ module: {
2060
+ name: "cohere-ai",
2061
+ versionRange: ">=8.0.0 <9.0.0",
2062
+ filePath: "Client.js"
2063
+ },
2064
+ functionQuery: {
2065
+ className: "CohereClient",
2066
+ methodName: "embed",
2067
+ kind: "Async"
2068
+ }
2069
+ },
2070
+ {
2071
+ channelName: cohereChannels.embed.channelName,
2072
+ module: {
2073
+ name: "cohere-ai",
2074
+ versionRange: ">=8.0.0 <9.0.0",
2075
+ filePath: "api/resources/v2/client/Client.js"
2076
+ },
2077
+ functionQuery: {
2078
+ className: "V2Client",
2079
+ methodName: "embed",
2080
+ kind: "Async"
2081
+ }
2082
+ },
2083
+ {
2084
+ channelName: cohereChannels.rerank.channelName,
2085
+ module: {
2086
+ name: "cohere-ai",
2087
+ versionRange: ">=7.0.0 <8.0.0",
2088
+ filePath: "Client.js"
2089
+ },
2090
+ functionQuery: {
2091
+ className: "CohereClient",
2092
+ methodName: "rerank",
2093
+ kind: "Async"
2094
+ }
2095
+ },
2096
+ {
2097
+ channelName: cohereChannels.rerank.channelName,
2098
+ module: {
2099
+ name: "cohere-ai",
2100
+ versionRange: ">=7.20.0 <8.0.0",
2101
+ filePath: "api/resources/v2/client/Client.js"
2102
+ },
2103
+ functionQuery: {
2104
+ className: "V2Client",
2105
+ methodName: "rerank",
2106
+ kind: "Async"
2107
+ }
2108
+ },
2109
+ {
2110
+ channelName: cohereChannels.rerank.channelName,
2111
+ module: {
2112
+ name: "cohere-ai",
2113
+ versionRange: ">=8.0.0 <9.0.0",
2114
+ filePath: "Client.js"
2115
+ },
2116
+ functionQuery: {
2117
+ className: "CohereClient",
2118
+ methodName: "rerank",
2119
+ kind: "Async"
2120
+ }
2121
+ },
2122
+ {
2123
+ channelName: cohereChannels.rerank.channelName,
2124
+ module: {
2125
+ name: "cohere-ai",
2126
+ versionRange: ">=8.0.0 <9.0.0",
2127
+ filePath: "api/resources/v2/client/Client.js"
2128
+ },
2129
+ functionQuery: {
2130
+ className: "V2Client",
2131
+ methodName: "rerank",
2132
+ kind: "Async"
2133
+ }
2134
+ }
2135
+ ];
2136
+
2137
+ // src/instrumentation/plugins/groq-channels.ts
2138
+ var groqChannels = defineChannels("groq-sdk", {
2139
+ chatCompletionsCreate: channel({
2140
+ channelName: "chat.completions.create",
2141
+ kind: "async"
2142
+ }),
2143
+ embeddingsCreate: channel(
2144
+ {
2145
+ channelName: "embeddings.create",
2146
+ kind: "async"
2147
+ }
2148
+ )
2149
+ });
2150
+
2151
+ // src/auto-instrumentations/configs/groq.ts
2152
+ var groqConfigs = [
2153
+ {
2154
+ channelName: groqChannels.chatCompletionsCreate.channelName,
2155
+ module: {
2156
+ name: "groq-sdk",
2157
+ versionRange: ">=1.0.0",
2158
+ filePath: "resources/chat/completions.mjs"
2159
+ },
2160
+ functionQuery: {
2161
+ className: "Completions",
2162
+ methodName: "create",
2163
+ kind: "Async"
2164
+ }
2165
+ },
2166
+ {
2167
+ channelName: groqChannels.embeddingsCreate.channelName,
2168
+ module: {
2169
+ name: "groq-sdk",
2170
+ versionRange: ">=1.0.0",
2171
+ filePath: "resources/embeddings.mjs"
2172
+ },
2173
+ functionQuery: {
2174
+ className: "Embeddings",
2175
+ methodName: "create",
1200
2176
  kind: "Async"
1201
2177
  }
1202
2178
  }
@@ -1365,15 +2341,33 @@ var dc = await import(
1365
2341
  dcPath
1366
2342
  );
1367
2343
  patchTracingChannel(dc.tracingChannel);
2344
+ function readDisabledIntegrations() {
2345
+ const raw = process.env.BRAINTRUST_DISABLE_INSTRUMENTATION;
2346
+ if (!raw) {
2347
+ return /* @__PURE__ */ new Set();
2348
+ }
2349
+ return new Set(
2350
+ raw.split(",").map((value) => value.trim().toLowerCase()).filter((value) => value.length > 0)
2351
+ );
2352
+ }
2353
+ function isDisabled(disabled, ...names) {
2354
+ return names.some((name) => disabled.has(name));
2355
+ }
2356
+ var disabledIntegrations = readDisabledIntegrations();
1368
2357
  var allConfigs = [
1369
- ...openaiConfigs,
1370
- ...anthropicConfigs,
1371
- ...aiSDKConfigs,
1372
- ...claudeAgentSDKConfigs,
1373
- ...googleGenAIConfigs,
1374
- ...openRouterConfigs,
1375
- ...openRouterAgentConfigs,
1376
- ...mistralConfigs
2358
+ ...isDisabled(disabledIntegrations, "openai") ? [] : openaiConfigs,
2359
+ ...isDisabled(disabledIntegrations, "anthropic") ? [] : anthropicConfigs,
2360
+ ...isDisabled(disabledIntegrations, "aisdk", "ai-sdk", "vercel-ai") ? [] : aiSDKConfigs,
2361
+ ...isDisabled(disabledIntegrations, "claudeagentsdk", "claude-agent-sdk") ? [] : claudeAgentSDKConfigs,
2362
+ ...isDisabled(disabledIntegrations, "cursor", "cursor-sdk") ? [] : cursorSDKConfigs,
2363
+ ...isDisabled(disabledIntegrations, "google", "google-genai") ? [] : googleGenAIConfigs,
2364
+ ...isDisabled(disabledIntegrations, "huggingface") ? [] : huggingFaceConfigs,
2365
+ ...isDisabled(disabledIntegrations, "openrouter") ? [] : openRouterConfigs,
2366
+ ...isDisabled(disabledIntegrations, "openrouteragent", "openrouter-agent") ? [] : openRouterAgentConfigs,
2367
+ ...isDisabled(disabledIntegrations, "mistral") ? [] : mistralConfigs,
2368
+ ...isDisabled(disabledIntegrations, "googleadk", "google-adk") ? [] : googleADKConfigs,
2369
+ ...isDisabled(disabledIntegrations, "cohere") ? [] : cohereConfigs,
2370
+ ...isDisabled(disabledIntegrations, "groq", "groq-sdk") ? [] : groqConfigs
1377
2371
  ];
1378
2372
  register("./loader/esm-hook.mjs", {
1379
2373
  parentURL: import.meta.url,