braintrust 3.9.0 → 3.11.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 (46) hide show
  1. package/dev/dist/index.d.mts +157 -3
  2. package/dev/dist/index.d.ts +157 -3
  3. package/dev/dist/index.js +10819 -7204
  4. package/dev/dist/index.mjs +7458 -3843
  5. package/dist/auto-instrumentations/bundler/esbuild.cjs +686 -1
  6. package/dist/auto-instrumentations/bundler/esbuild.mjs +2 -2
  7. package/dist/auto-instrumentations/bundler/rollup.cjs +686 -1
  8. package/dist/auto-instrumentations/bundler/rollup.mjs +2 -2
  9. package/dist/auto-instrumentations/bundler/vite.cjs +686 -1
  10. package/dist/auto-instrumentations/bundler/vite.mjs +2 -2
  11. package/dist/auto-instrumentations/bundler/webpack-loader.cjs +634 -0
  12. package/dist/auto-instrumentations/bundler/webpack.cjs +686 -1
  13. package/dist/auto-instrumentations/bundler/webpack.mjs +2 -2
  14. package/dist/auto-instrumentations/{chunk-KIMMUFAK.mjs → chunk-DIV5TO4S.mjs} +679 -1
  15. package/dist/auto-instrumentations/{chunk-G7F6HZGE.mjs → chunk-G6ZWXGZB.mjs} +19 -2
  16. package/dist/auto-instrumentations/{chunk-ITP7RAUY.mjs → chunk-MWZXZQUO.mjs} +23 -1
  17. package/dist/auto-instrumentations/hook.mjs +793 -11
  18. package/dist/auto-instrumentations/index.cjs +774 -7
  19. package/dist/auto-instrumentations/index.d.mts +18 -1
  20. package/dist/auto-instrumentations/index.d.ts +18 -1
  21. package/dist/auto-instrumentations/index.mjs +102 -8
  22. package/dist/auto-instrumentations/loader/cjs-patch.cjs +26 -1
  23. package/dist/auto-instrumentations/loader/cjs-patch.mjs +28 -1
  24. package/dist/auto-instrumentations/loader/esm-hook.mjs +16 -1
  25. package/dist/browser.d.mts +600 -265
  26. package/dist/browser.d.ts +600 -265
  27. package/dist/browser.js +6171 -1851
  28. package/dist/browser.mjs +6171 -1851
  29. package/dist/cli.js +6417 -2794
  30. package/dist/edge-light.d.mts +1 -1
  31. package/dist/edge-light.d.ts +1 -1
  32. package/dist/edge-light.js +6171 -1851
  33. package/dist/edge-light.mjs +6171 -1851
  34. package/dist/index.d.mts +613 -278
  35. package/dist/index.d.ts +613 -278
  36. package/dist/index.js +6507 -2187
  37. package/dist/index.mjs +6171 -1851
  38. package/dist/instrumentation/index.d.mts +19 -0
  39. package/dist/instrumentation/index.d.ts +19 -0
  40. package/dist/instrumentation/index.js +10265 -6841
  41. package/dist/instrumentation/index.mjs +10265 -6841
  42. package/dist/workerd.d.mts +1 -1
  43. package/dist/workerd.d.ts +1 -1
  44. package/dist/workerd.js +6171 -1851
  45. package/dist/workerd.mjs +6171 -1851
  46. package/package.json +16 -22
@@ -1,7 +1,8 @@
1
1
  import {
2
+ OPENAI_API_PROMISE_PATCH,
2
3
  getPackageName,
3
4
  getPackageVersion
4
- } from "./chunk-ITP7RAUY.mjs";
5
+ } from "./chunk-MWZXZQUO.mjs";
5
6
 
6
7
  // src/auto-instrumentations/hook.mts
7
8
  import { register } from "module";
@@ -368,6 +369,49 @@ var openaiConfigs = [
368
369
  }
369
370
  ];
370
371
 
372
+ // src/instrumentation/plugins/openai-codex-channels.ts
373
+ var openAICodexChannels = defineChannels("@openai/codex-sdk", {
374
+ run: channel({
375
+ channelName: "Thread.run",
376
+ kind: "async"
377
+ }),
378
+ runStreamed: channel({
379
+ channelName: "Thread.runStreamed",
380
+ kind: "async"
381
+ })
382
+ });
383
+
384
+ // src/auto-instrumentations/configs/openai-codex.ts
385
+ var openAICodexVersionRange = ">=0.128.0 <1.0.0";
386
+ var openAICodexConfigs = [
387
+ {
388
+ channelName: openAICodexChannels.run.channelName,
389
+ module: {
390
+ name: "@openai/codex-sdk",
391
+ versionRange: openAICodexVersionRange,
392
+ filePath: "dist/index.js"
393
+ },
394
+ functionQuery: {
395
+ className: "Thread",
396
+ methodName: "run",
397
+ kind: "Async"
398
+ }
399
+ },
400
+ {
401
+ channelName: openAICodexChannels.runStreamed.channelName,
402
+ module: {
403
+ name: "@openai/codex-sdk",
404
+ versionRange: openAICodexVersionRange,
405
+ filePath: "dist/index.js"
406
+ },
407
+ functionQuery: {
408
+ className: "Thread",
409
+ methodName: "runStreamed",
410
+ kind: "Async"
411
+ }
412
+ }
413
+ ];
414
+
371
415
  // src/instrumentation/plugins/anthropic-channels.ts
372
416
  var anthropicChannels = defineChannels("@anthropic-ai/sdk", {
373
417
  messagesCreate: channel({
@@ -893,6 +937,71 @@ var claudeAgentSDKConfigs = [
893
937
  }
894
938
  ];
895
939
 
940
+ // src/instrumentation/plugins/cursor-sdk-channels.ts
941
+ var cursorSDKChannels = defineChannels("@cursor/sdk", {
942
+ create: channel({
943
+ channelName: "Agent.create",
944
+ kind: "async"
945
+ }),
946
+ resume: channel({
947
+ channelName: "Agent.resume",
948
+ kind: "async"
949
+ }),
950
+ prompt: channel({
951
+ channelName: "Agent.prompt",
952
+ kind: "async"
953
+ }),
954
+ send: channel({
955
+ channelName: "agent.send",
956
+ kind: "async"
957
+ })
958
+ });
959
+
960
+ // src/auto-instrumentations/configs/cursor-sdk.ts
961
+ var cursorSDKVersionRange = ">=1.0.7 <2.0.0";
962
+ var cursorSDKEntrypoints = ["dist/esm/index.js", "dist/cjs/index.js"];
963
+ var cursorSDKConfigs = cursorSDKEntrypoints.flatMap((filePath) => [
964
+ {
965
+ channelName: cursorSDKChannels.create.channelName,
966
+ module: {
967
+ name: "@cursor/sdk",
968
+ versionRange: cursorSDKVersionRange,
969
+ filePath
970
+ },
971
+ functionQuery: {
972
+ className: "Agent",
973
+ methodName: "create",
974
+ kind: "Async"
975
+ }
976
+ },
977
+ {
978
+ channelName: cursorSDKChannels.resume.channelName,
979
+ module: {
980
+ name: "@cursor/sdk",
981
+ versionRange: cursorSDKVersionRange,
982
+ filePath
983
+ },
984
+ functionQuery: {
985
+ className: "Agent",
986
+ methodName: "resume",
987
+ kind: "Async"
988
+ }
989
+ },
990
+ {
991
+ channelName: cursorSDKChannels.prompt.channelName,
992
+ module: {
993
+ name: "@cursor/sdk",
994
+ versionRange: cursorSDKVersionRange,
995
+ filePath
996
+ },
997
+ functionQuery: {
998
+ className: "Agent",
999
+ methodName: "prompt",
1000
+ kind: "Async"
1001
+ }
1002
+ }
1003
+ ]);
1004
+
896
1005
  // src/instrumentation/plugins/google-genai-channels.ts
897
1006
  var googleGenAIChannels = defineChannels("@google/genai", {
898
1007
  generateContent: channel({
@@ -1389,6 +1498,22 @@ var mistralChannels = defineChannels("@mistralai/mistralai", {
1389
1498
  channelName: "embeddings.create",
1390
1499
  kind: "async"
1391
1500
  }),
1501
+ classifiersModerate: channel({
1502
+ channelName: "classifiers.moderate",
1503
+ kind: "async"
1504
+ }),
1505
+ classifiersModerateChat: channel({
1506
+ channelName: "classifiers.moderateChat",
1507
+ kind: "async"
1508
+ }),
1509
+ classifiersClassify: channel({
1510
+ channelName: "classifiers.classify",
1511
+ kind: "async"
1512
+ }),
1513
+ classifiersClassifyChat: channel({
1514
+ channelName: "classifiers.classifyChat",
1515
+ kind: "async"
1516
+ }),
1392
1517
  fimComplete: channel({
1393
1518
  channelName: "fim.complete",
1394
1519
  kind: "async"
@@ -1487,6 +1612,110 @@ var mistralConfigs = [
1487
1612
  kind: "Async"
1488
1613
  }
1489
1614
  },
1615
+ {
1616
+ channelName: mistralChannels.classifiersModerate.channelName,
1617
+ module: {
1618
+ name: "@mistralai/mistralai",
1619
+ versionRange: ">=1.0.0 <2.0.0",
1620
+ filePath: "sdk/classifiers.js"
1621
+ },
1622
+ functionQuery: {
1623
+ className: "Classifiers",
1624
+ methodName: "moderate",
1625
+ kind: "Async"
1626
+ }
1627
+ },
1628
+ {
1629
+ channelName: mistralChannels.classifiersModerate.channelName,
1630
+ module: {
1631
+ name: "@mistralai/mistralai",
1632
+ versionRange: ">=2.0.0 <3.0.0",
1633
+ filePath: "esm/sdk/classifiers.js"
1634
+ },
1635
+ functionQuery: {
1636
+ className: "Classifiers",
1637
+ methodName: "moderate",
1638
+ kind: "Async"
1639
+ }
1640
+ },
1641
+ {
1642
+ channelName: mistralChannels.classifiersModerateChat.channelName,
1643
+ module: {
1644
+ name: "@mistralai/mistralai",
1645
+ versionRange: ">=1.0.0 <2.0.0",
1646
+ filePath: "sdk/classifiers.js"
1647
+ },
1648
+ functionQuery: {
1649
+ className: "Classifiers",
1650
+ methodName: "moderateChat",
1651
+ kind: "Async"
1652
+ }
1653
+ },
1654
+ {
1655
+ channelName: mistralChannels.classifiersModerateChat.channelName,
1656
+ module: {
1657
+ name: "@mistralai/mistralai",
1658
+ versionRange: ">=2.0.0 <3.0.0",
1659
+ filePath: "esm/sdk/classifiers.js"
1660
+ },
1661
+ functionQuery: {
1662
+ className: "Classifiers",
1663
+ methodName: "moderateChat",
1664
+ kind: "Async"
1665
+ }
1666
+ },
1667
+ {
1668
+ channelName: mistralChannels.classifiersClassify.channelName,
1669
+ module: {
1670
+ name: "@mistralai/mistralai",
1671
+ versionRange: ">=1.10.0 <2.0.0",
1672
+ filePath: "sdk/classifiers.js"
1673
+ },
1674
+ functionQuery: {
1675
+ className: "Classifiers",
1676
+ methodName: "classify",
1677
+ kind: "Async"
1678
+ }
1679
+ },
1680
+ {
1681
+ channelName: mistralChannels.classifiersClassify.channelName,
1682
+ module: {
1683
+ name: "@mistralai/mistralai",
1684
+ versionRange: ">=2.0.0 <3.0.0",
1685
+ filePath: "esm/sdk/classifiers.js"
1686
+ },
1687
+ functionQuery: {
1688
+ className: "Classifiers",
1689
+ methodName: "classify",
1690
+ kind: "Async"
1691
+ }
1692
+ },
1693
+ {
1694
+ channelName: mistralChannels.classifiersClassifyChat.channelName,
1695
+ module: {
1696
+ name: "@mistralai/mistralai",
1697
+ versionRange: ">=1.10.0 <2.0.0",
1698
+ filePath: "sdk/classifiers.js"
1699
+ },
1700
+ functionQuery: {
1701
+ className: "Classifiers",
1702
+ methodName: "classifyChat",
1703
+ kind: "Async"
1704
+ }
1705
+ },
1706
+ {
1707
+ channelName: mistralChannels.classifiersClassifyChat.channelName,
1708
+ module: {
1709
+ name: "@mistralai/mistralai",
1710
+ versionRange: ">=2.0.0 <3.0.0",
1711
+ filePath: "esm/sdk/classifiers.js"
1712
+ },
1713
+ functionQuery: {
1714
+ className: "Classifiers",
1715
+ methodName: "classifyChat",
1716
+ kind: "Async"
1717
+ }
1718
+ },
1490
1719
  {
1491
1720
  channelName: mistralChannels.fimComplete.channelName,
1492
1721
  module: {
@@ -1611,7 +1840,8 @@ var googleADKChannels = defineChannels("@google/adk", {
1611
1840
 
1612
1841
  // src/auto-instrumentations/configs/google-adk.ts
1613
1842
  var googleADKVersionRange = ">=0.1.0";
1614
- var googleADKBundledIndexVersionRange = ">=0.6.1 <0.7.0";
1843
+ var googleADKBundledIndexV06VersionRange = ">=0.6.1 <0.7.0";
1844
+ var googleADKBundledIndexV1VersionRange = ">=1.0.0 <2.0.0";
1615
1845
  var googleADKConfigs = [
1616
1846
  // --- Runner.runAsync --- async generator, kind "Sync" + sync-stream channel
1617
1847
  // Runner.runAsync — ESM individual module file
@@ -1637,7 +1867,7 @@ var googleADKConfigs = [
1637
1867
  channelName: googleADKChannels.runnerRunAsync.channelName,
1638
1868
  module: {
1639
1869
  name: "@google/adk",
1640
- versionRange: googleADKBundledIndexVersionRange,
1870
+ versionRange: googleADKBundledIndexV06VersionRange,
1641
1871
  filePath: "dist/cjs/index.js"
1642
1872
  },
1643
1873
  functionQuery: {
@@ -1650,7 +1880,7 @@ var googleADKConfigs = [
1650
1880
  channelName: googleADKChannels.runnerRunAsync.channelName,
1651
1881
  module: {
1652
1882
  name: "@google/adk",
1653
- versionRange: googleADKBundledIndexVersionRange,
1883
+ versionRange: googleADKBundledIndexV06VersionRange,
1654
1884
  filePath: "dist/esm/index.js"
1655
1885
  },
1656
1886
  functionQuery: {
@@ -1659,6 +1889,34 @@ var googleADKConfigs = [
1659
1889
  index: 11
1660
1890
  }
1661
1891
  },
1892
+ // The 1.x bundled entrypoints still inline the runtime into index.js, but
1893
+ // the minified method order changed. These indices are verified against 1.0.0.
1894
+ {
1895
+ channelName: googleADKChannels.runnerRunAsync.channelName,
1896
+ module: {
1897
+ name: "@google/adk",
1898
+ versionRange: googleADKBundledIndexV1VersionRange,
1899
+ filePath: "dist/cjs/index.js"
1900
+ },
1901
+ functionQuery: {
1902
+ methodName: "runAsync",
1903
+ kind: "Sync",
1904
+ index: 12
1905
+ }
1906
+ },
1907
+ {
1908
+ channelName: googleADKChannels.runnerRunAsync.channelName,
1909
+ module: {
1910
+ name: "@google/adk",
1911
+ versionRange: googleADKBundledIndexV1VersionRange,
1912
+ filePath: "dist/esm/index.js"
1913
+ },
1914
+ functionQuery: {
1915
+ methodName: "runAsync",
1916
+ kind: "Sync",
1917
+ index: 12
1918
+ }
1919
+ },
1662
1920
  // --- BaseAgent.runAsync --- async generator, kind "Sync" + sync-stream channel
1663
1921
  // BaseAgent.runAsync — ESM individual module file
1664
1922
  {
@@ -1683,7 +1941,7 @@ var googleADKConfigs = [
1683
1941
  channelName: googleADKChannels.agentRunAsync.channelName,
1684
1942
  module: {
1685
1943
  name: "@google/adk",
1686
- versionRange: googleADKBundledIndexVersionRange,
1944
+ versionRange: googleADKBundledIndexV06VersionRange,
1687
1945
  filePath: "dist/cjs/index.js"
1688
1946
  },
1689
1947
  functionQuery: {
@@ -1696,7 +1954,35 @@ var googleADKConfigs = [
1696
1954
  channelName: googleADKChannels.agentRunAsync.channelName,
1697
1955
  module: {
1698
1956
  name: "@google/adk",
1699
- versionRange: googleADKBundledIndexVersionRange,
1957
+ versionRange: googleADKBundledIndexV06VersionRange,
1958
+ filePath: "dist/esm/index.js"
1959
+ },
1960
+ functionQuery: {
1961
+ methodName: "runAsync",
1962
+ kind: "Sync",
1963
+ index: 0
1964
+ }
1965
+ },
1966
+ // The 1.x bundled entrypoints keep BaseAgent.runAsync as the first bundled
1967
+ // async-generator runAsync method in file order.
1968
+ {
1969
+ channelName: googleADKChannels.agentRunAsync.channelName,
1970
+ module: {
1971
+ name: "@google/adk",
1972
+ versionRange: googleADKBundledIndexV1VersionRange,
1973
+ filePath: "dist/cjs/index.js"
1974
+ },
1975
+ functionQuery: {
1976
+ methodName: "runAsync",
1977
+ kind: "Sync",
1978
+ index: 0
1979
+ }
1980
+ },
1981
+ {
1982
+ channelName: googleADKChannels.agentRunAsync.channelName,
1983
+ module: {
1984
+ name: "@google/adk",
1985
+ versionRange: googleADKBundledIndexV1VersionRange,
1700
1986
  filePath: "dist/esm/index.js"
1701
1987
  },
1702
1988
  functionQuery: {
@@ -1729,7 +2015,34 @@ var googleADKConfigs = [
1729
2015
  channelName: googleADKChannels.toolRunAsync.channelName,
1730
2016
  module: {
1731
2017
  name: "@google/adk",
1732
- versionRange: googleADKBundledIndexVersionRange,
2018
+ versionRange: googleADKBundledIndexV06VersionRange,
2019
+ filePath: "dist/cjs/index.js"
2020
+ },
2021
+ functionQuery: {
2022
+ methodName: "runAsync",
2023
+ kind: "Async",
2024
+ index: 1
2025
+ }
2026
+ },
2027
+ {
2028
+ channelName: googleADKChannels.toolRunAsync.channelName,
2029
+ module: {
2030
+ name: "@google/adk",
2031
+ versionRange: googleADKBundledIndexV06VersionRange,
2032
+ filePath: "dist/esm/index.js"
2033
+ },
2034
+ functionQuery: {
2035
+ methodName: "runAsync",
2036
+ kind: "Async",
2037
+ index: 1
2038
+ }
2039
+ },
2040
+ // The 1.x bundle moves FunctionTool.runAsync behind one helper method.
2041
+ {
2042
+ channelName: googleADKChannels.toolRunAsync.channelName,
2043
+ module: {
2044
+ name: "@google/adk",
2045
+ versionRange: googleADKBundledIndexV1VersionRange,
1733
2046
  filePath: "dist/cjs/index.js"
1734
2047
  },
1735
2048
  functionQuery: {
@@ -1742,7 +2055,7 @@ var googleADKConfigs = [
1742
2055
  channelName: googleADKChannels.toolRunAsync.channelName,
1743
2056
  module: {
1744
2057
  name: "@google/adk",
1745
- versionRange: googleADKBundledIndexVersionRange,
2058
+ versionRange: googleADKBundledIndexV1VersionRange,
1746
2059
  filePath: "dist/esm/index.js"
1747
2060
  },
1748
2061
  functionQuery: {
@@ -1788,6 +2101,32 @@ var cohereConfigs = [
1788
2101
  kind: "Async"
1789
2102
  }
1790
2103
  },
2104
+ {
2105
+ channelName: cohereChannels.chat.channelName,
2106
+ module: {
2107
+ name: "cohere-ai",
2108
+ versionRange: ">=7.0.0 <7.21.0",
2109
+ filePath: "api/resources/v2/client/Client.js"
2110
+ },
2111
+ functionQuery: {
2112
+ className: "V2",
2113
+ methodName: "chat",
2114
+ kind: "Async"
2115
+ }
2116
+ },
2117
+ {
2118
+ channelName: cohereChannels.chat.channelName,
2119
+ module: {
2120
+ name: "cohere-ai",
2121
+ versionRange: ">=7.21.0 <8.0.0",
2122
+ filePath: "api/resources/v2/client/Client.js"
2123
+ },
2124
+ functionQuery: {
2125
+ className: "V2Client",
2126
+ methodName: "chat",
2127
+ kind: "Async"
2128
+ }
2129
+ },
1791
2130
  {
1792
2131
  channelName: cohereChannels.chat.channelName,
1793
2132
  module: {
@@ -1827,6 +2166,32 @@ var cohereConfigs = [
1827
2166
  kind: "Async"
1828
2167
  }
1829
2168
  },
2169
+ {
2170
+ channelName: cohereChannels.chatStream.channelName,
2171
+ module: {
2172
+ name: "cohere-ai",
2173
+ versionRange: ">=7.0.0 <7.21.0",
2174
+ filePath: "api/resources/v2/client/Client.js"
2175
+ },
2176
+ functionQuery: {
2177
+ className: "V2",
2178
+ methodName: "chatStream",
2179
+ kind: "Async"
2180
+ }
2181
+ },
2182
+ {
2183
+ channelName: cohereChannels.chatStream.channelName,
2184
+ module: {
2185
+ name: "cohere-ai",
2186
+ versionRange: ">=7.21.0 <8.0.0",
2187
+ filePath: "api/resources/v2/client/Client.js"
2188
+ },
2189
+ functionQuery: {
2190
+ className: "V2Client",
2191
+ methodName: "chatStream",
2192
+ kind: "Async"
2193
+ }
2194
+ },
1830
2195
  {
1831
2196
  channelName: cohereChannels.chatStream.channelName,
1832
2197
  module: {
@@ -1866,6 +2231,32 @@ var cohereConfigs = [
1866
2231
  kind: "Async"
1867
2232
  }
1868
2233
  },
2234
+ {
2235
+ channelName: cohereChannels.embed.channelName,
2236
+ module: {
2237
+ name: "cohere-ai",
2238
+ versionRange: ">=7.0.0 <7.21.0",
2239
+ filePath: "api/resources/v2/client/Client.js"
2240
+ },
2241
+ functionQuery: {
2242
+ className: "V2",
2243
+ methodName: "embed",
2244
+ kind: "Async"
2245
+ }
2246
+ },
2247
+ {
2248
+ channelName: cohereChannels.embed.channelName,
2249
+ module: {
2250
+ name: "cohere-ai",
2251
+ versionRange: ">=7.21.0 <8.0.0",
2252
+ filePath: "api/resources/v2/client/Client.js"
2253
+ },
2254
+ functionQuery: {
2255
+ className: "V2Client",
2256
+ methodName: "embed",
2257
+ kind: "Async"
2258
+ }
2259
+ },
1869
2260
  {
1870
2261
  channelName: cohereChannels.embed.channelName,
1871
2262
  module: {
@@ -1905,6 +2296,32 @@ var cohereConfigs = [
1905
2296
  kind: "Async"
1906
2297
  }
1907
2298
  },
2299
+ {
2300
+ channelName: cohereChannels.rerank.channelName,
2301
+ module: {
2302
+ name: "cohere-ai",
2303
+ versionRange: ">=7.0.0 <7.21.0",
2304
+ filePath: "api/resources/v2/client/Client.js"
2305
+ },
2306
+ functionQuery: {
2307
+ className: "V2",
2308
+ methodName: "rerank",
2309
+ kind: "Async"
2310
+ }
2311
+ },
2312
+ {
2313
+ channelName: cohereChannels.rerank.channelName,
2314
+ module: {
2315
+ name: "cohere-ai",
2316
+ versionRange: ">=7.21.0 <8.0.0",
2317
+ filePath: "api/resources/v2/client/Client.js"
2318
+ },
2319
+ functionQuery: {
2320
+ className: "V2Client",
2321
+ methodName: "rerank",
2322
+ kind: "Async"
2323
+ }
2324
+ },
1908
2325
  {
1909
2326
  channelName: cohereChannels.rerank.channelName,
1910
2327
  module: {
@@ -1933,6 +2350,347 @@ var cohereConfigs = [
1933
2350
  }
1934
2351
  ];
1935
2352
 
2353
+ // src/instrumentation/plugins/groq-channels.ts
2354
+ var groqChannels = defineChannels("groq-sdk", {
2355
+ chatCompletionsCreate: channel({
2356
+ channelName: "chat.completions.create",
2357
+ kind: "async"
2358
+ }),
2359
+ embeddingsCreate: channel(
2360
+ {
2361
+ channelName: "embeddings.create",
2362
+ kind: "async"
2363
+ }
2364
+ )
2365
+ });
2366
+
2367
+ // src/auto-instrumentations/configs/groq.ts
2368
+ var groqConfigs = [
2369
+ {
2370
+ channelName: groqChannels.chatCompletionsCreate.channelName,
2371
+ module: {
2372
+ name: "groq-sdk",
2373
+ versionRange: ">=1.0.0",
2374
+ filePath: "resources/chat/completions.mjs"
2375
+ },
2376
+ functionQuery: {
2377
+ className: "Completions",
2378
+ methodName: "create",
2379
+ kind: "Async"
2380
+ }
2381
+ },
2382
+ {
2383
+ channelName: groqChannels.embeddingsCreate.channelName,
2384
+ module: {
2385
+ name: "groq-sdk",
2386
+ versionRange: ">=1.0.0",
2387
+ filePath: "resources/embeddings.mjs"
2388
+ },
2389
+ functionQuery: {
2390
+ className: "Embeddings",
2391
+ methodName: "create",
2392
+ kind: "Async"
2393
+ }
2394
+ }
2395
+ ];
2396
+
2397
+ // src/instrumentation/plugins/genkit-channels.ts
2398
+ var genkitChannels = defineChannels("@genkit-ai/ai", {
2399
+ generate: channel({
2400
+ channelName: "generate",
2401
+ kind: "async"
2402
+ }),
2403
+ generateStream: channel({
2404
+ channelName: "generateStream",
2405
+ kind: "sync-stream"
2406
+ }),
2407
+ embed: channel({
2408
+ channelName: "embed",
2409
+ kind: "async"
2410
+ }),
2411
+ embedMany: channel({
2412
+ channelName: "embedMany",
2413
+ kind: "async"
2414
+ }),
2415
+ actionRun: channel({
2416
+ channelName: "action.run",
2417
+ kind: "async"
2418
+ }),
2419
+ actionStream: channel({
2420
+ channelName: "action.stream",
2421
+ kind: "sync-stream"
2422
+ })
2423
+ });
2424
+ var genkitCoreChannels = defineChannels("@genkit-ai/core", {
2425
+ actionSpan: channel({
2426
+ channelName: "action.span",
2427
+ kind: "async"
2428
+ })
2429
+ });
2430
+
2431
+ // src/auto-instrumentations/configs/genkit.ts
2432
+ var genkitVersionRange = ">=1.0.0 <2.0.0";
2433
+ var genkitConfigs = [
2434
+ {
2435
+ channelName: genkitChannels.generate.channelName,
2436
+ module: {
2437
+ name: "@genkit-ai/ai",
2438
+ versionRange: genkitVersionRange,
2439
+ filePath: "lib/genkit-ai.mjs"
2440
+ },
2441
+ functionQuery: {
2442
+ className: "GenkitAI",
2443
+ methodName: "generate",
2444
+ kind: "Async"
2445
+ }
2446
+ },
2447
+ {
2448
+ channelName: genkitChannels.generate.channelName,
2449
+ module: {
2450
+ name: "@genkit-ai/ai",
2451
+ versionRange: genkitVersionRange,
2452
+ filePath: "lib/genkit-ai.js"
2453
+ },
2454
+ functionQuery: {
2455
+ className: "GenkitAI",
2456
+ methodName: "generate",
2457
+ kind: "Async"
2458
+ }
2459
+ },
2460
+ {
2461
+ channelName: genkitChannels.generateStream.channelName,
2462
+ module: {
2463
+ name: "@genkit-ai/ai",
2464
+ versionRange: genkitVersionRange,
2465
+ filePath: "lib/genkit-ai.mjs"
2466
+ },
2467
+ functionQuery: {
2468
+ className: "GenkitAI",
2469
+ methodName: "generateStream",
2470
+ kind: "Sync"
2471
+ }
2472
+ },
2473
+ {
2474
+ channelName: genkitChannels.generateStream.channelName,
2475
+ module: {
2476
+ name: "@genkit-ai/ai",
2477
+ versionRange: genkitVersionRange,
2478
+ filePath: "lib/genkit-ai.js"
2479
+ },
2480
+ functionQuery: {
2481
+ className: "GenkitAI",
2482
+ methodName: "generateStream",
2483
+ kind: "Sync"
2484
+ }
2485
+ },
2486
+ {
2487
+ channelName: genkitChannels.embed.channelName,
2488
+ module: {
2489
+ name: "@genkit-ai/ai",
2490
+ versionRange: genkitVersionRange,
2491
+ filePath: "lib/genkit-ai.mjs"
2492
+ },
2493
+ functionQuery: {
2494
+ className: "GenkitAI",
2495
+ methodName: "embed",
2496
+ kind: "Async"
2497
+ }
2498
+ },
2499
+ {
2500
+ channelName: genkitChannels.embed.channelName,
2501
+ module: {
2502
+ name: "@genkit-ai/ai",
2503
+ versionRange: genkitVersionRange,
2504
+ filePath: "lib/genkit-ai.js"
2505
+ },
2506
+ functionQuery: {
2507
+ className: "GenkitAI",
2508
+ methodName: "embed",
2509
+ kind: "Async"
2510
+ }
2511
+ },
2512
+ {
2513
+ channelName: genkitChannels.embedMany.channelName,
2514
+ module: {
2515
+ name: "@genkit-ai/ai",
2516
+ versionRange: genkitVersionRange,
2517
+ filePath: "lib/genkit-ai.mjs"
2518
+ },
2519
+ functionQuery: {
2520
+ className: "GenkitAI",
2521
+ methodName: "embedMany",
2522
+ kind: "Async"
2523
+ }
2524
+ },
2525
+ {
2526
+ channelName: genkitChannels.embedMany.channelName,
2527
+ module: {
2528
+ name: "@genkit-ai/ai",
2529
+ versionRange: genkitVersionRange,
2530
+ filePath: "lib/genkit-ai.js"
2531
+ },
2532
+ functionQuery: {
2533
+ className: "GenkitAI",
2534
+ methodName: "embedMany",
2535
+ kind: "Async"
2536
+ }
2537
+ },
2538
+ {
2539
+ channelName: genkitChannels.actionRun.channelName,
2540
+ module: {
2541
+ name: "@genkit-ai/ai",
2542
+ versionRange: genkitVersionRange,
2543
+ filePath: "lib/genkit-ai.mjs"
2544
+ },
2545
+ functionQuery: {
2546
+ className: "GenkitAI",
2547
+ methodName: "run",
2548
+ kind: "Async"
2549
+ }
2550
+ },
2551
+ {
2552
+ channelName: genkitChannels.actionRun.channelName,
2553
+ module: {
2554
+ name: "@genkit-ai/ai",
2555
+ versionRange: genkitVersionRange,
2556
+ filePath: "lib/genkit-ai.js"
2557
+ },
2558
+ functionQuery: {
2559
+ className: "GenkitAI",
2560
+ methodName: "run",
2561
+ kind: "Async"
2562
+ }
2563
+ },
2564
+ {
2565
+ channelName: genkitCoreChannels.actionSpan.channelName,
2566
+ module: {
2567
+ name: "@genkit-ai/core",
2568
+ versionRange: genkitVersionRange,
2569
+ filePath: "lib/tracing/instrumentation.mjs"
2570
+ },
2571
+ functionQuery: {
2572
+ functionName: "runInNewSpan",
2573
+ kind: "Async"
2574
+ }
2575
+ },
2576
+ {
2577
+ channelName: genkitCoreChannels.actionSpan.channelName,
2578
+ module: {
2579
+ name: "@genkit-ai/core",
2580
+ versionRange: genkitVersionRange,
2581
+ filePath: "lib/tracing/instrumentation.js"
2582
+ },
2583
+ functionQuery: {
2584
+ functionName: "runInNewSpan",
2585
+ kind: "Async"
2586
+ }
2587
+ }
2588
+ ];
2589
+
2590
+ // src/instrumentation/plugins/github-copilot-channels.ts
2591
+ var gitHubCopilotChannels = defineChannels("@github/copilot-sdk", {
2592
+ createSession: channel({
2593
+ channelName: "client.createSession",
2594
+ kind: "async"
2595
+ }),
2596
+ resumeSession: channel({
2597
+ channelName: "client.resumeSession",
2598
+ kind: "async"
2599
+ }),
2600
+ sendAndWait: channel({
2601
+ channelName: "session.sendAndWait",
2602
+ kind: "async"
2603
+ })
2604
+ });
2605
+
2606
+ // src/auto-instrumentations/configs/github-copilot.ts
2607
+ var gitHubCopilotConfigs = [
2608
+ // ESM: CopilotClient.createSession
2609
+ {
2610
+ channelName: gitHubCopilotChannels.createSession.channelName,
2611
+ module: {
2612
+ name: "@github/copilot-sdk",
2613
+ versionRange: ">=0.3.0",
2614
+ filePath: "dist/client.js"
2615
+ },
2616
+ functionQuery: {
2617
+ className: "CopilotClient",
2618
+ methodName: "createSession",
2619
+ kind: "Async"
2620
+ }
2621
+ },
2622
+ // CJS: CopilotClient.createSession
2623
+ {
2624
+ channelName: gitHubCopilotChannels.createSession.channelName,
2625
+ module: {
2626
+ name: "@github/copilot-sdk",
2627
+ versionRange: ">=0.3.0",
2628
+ filePath: "dist/cjs/client.js"
2629
+ },
2630
+ functionQuery: {
2631
+ className: "CopilotClient",
2632
+ methodName: "createSession",
2633
+ kind: "Async"
2634
+ }
2635
+ },
2636
+ // ESM: CopilotClient.resumeSession
2637
+ {
2638
+ channelName: gitHubCopilotChannels.resumeSession.channelName,
2639
+ module: {
2640
+ name: "@github/copilot-sdk",
2641
+ versionRange: ">=0.3.0",
2642
+ filePath: "dist/client.js"
2643
+ },
2644
+ functionQuery: {
2645
+ className: "CopilotClient",
2646
+ methodName: "resumeSession",
2647
+ kind: "Async"
2648
+ }
2649
+ },
2650
+ // CJS: CopilotClient.resumeSession
2651
+ {
2652
+ channelName: gitHubCopilotChannels.resumeSession.channelName,
2653
+ module: {
2654
+ name: "@github/copilot-sdk",
2655
+ versionRange: ">=0.3.0",
2656
+ filePath: "dist/cjs/client.js"
2657
+ },
2658
+ functionQuery: {
2659
+ className: "CopilotClient",
2660
+ methodName: "resumeSession",
2661
+ kind: "Async"
2662
+ }
2663
+ },
2664
+ // ESM: CopilotSession.sendAndWait
2665
+ {
2666
+ channelName: gitHubCopilotChannels.sendAndWait.channelName,
2667
+ module: {
2668
+ name: "@github/copilot-sdk",
2669
+ versionRange: ">=0.3.0",
2670
+ filePath: "dist/session.js"
2671
+ },
2672
+ functionQuery: {
2673
+ className: "CopilotSession",
2674
+ methodName: "sendAndWait",
2675
+ kind: "Async"
2676
+ }
2677
+ },
2678
+ // CJS: CopilotSession.sendAndWait
2679
+ {
2680
+ channelName: gitHubCopilotChannels.sendAndWait.channelName,
2681
+ module: {
2682
+ name: "@github/copilot-sdk",
2683
+ versionRange: ">=0.3.0",
2684
+ filePath: "dist/cjs/session.js"
2685
+ },
2686
+ functionQuery: {
2687
+ className: "CopilotSession",
2688
+ methodName: "sendAndWait",
2689
+ kind: "Async"
2690
+ }
2691
+ }
2692
+ ];
2693
+
1936
2694
  // src/auto-instrumentations/loader/cjs-patch.ts
1937
2695
  import {
1938
2696
  create
@@ -1970,8 +2728,12 @@ var ModulePatch = class {
1970
2728
  if (!self.packages.has(packageName)) {
1971
2729
  return self.originalCompile.apply(this, args);
1972
2730
  }
1973
- const version = getPackageVersion(resolvedModule.basedir);
1974
2731
  const normalizedModulePath = resolvedModule.path.replace(/\\/g, "/");
2732
+ if (packageName === "openai" && normalizedModulePath.includes("api-promise")) {
2733
+ args[0] = content + OPENAI_API_PROMISE_PATCH;
2734
+ return self.originalCompile.apply(this, args);
2735
+ }
2736
+ const version = getPackageVersion(resolvedModule.basedir);
1975
2737
  const transformer = self.instrumentator.getTransformer(
1976
2738
  packageName,
1977
2739
  version,
@@ -2027,7 +2789,11 @@ function patchTracingChannel(tracingChannelFn) {
2027
2789
  }
2028
2790
  if (TracingChannel.prototype.tracePromise) {
2029
2791
  TracingChannel.prototype.tracePromise = function(fn, context = {}, thisArg, ...args) {
2030
- const { start, end, asyncStart, asyncEnd, error } = this;
2792
+ const start = this.start;
2793
+ const end = this.end;
2794
+ const asyncStart = this.asyncStart;
2795
+ const asyncEnd = this.asyncEnd;
2796
+ const error = this.error;
2031
2797
  function publishRejected(err) {
2032
2798
  context.error = err;
2033
2799
  error?.publish(context);
@@ -2111,16 +2877,32 @@ function isDisabled(disabled, ...names) {
2111
2877
  var disabledIntegrations = readDisabledIntegrations();
2112
2878
  var allConfigs = [
2113
2879
  ...isDisabled(disabledIntegrations, "openai") ? [] : openaiConfigs,
2880
+ ...isDisabled(
2881
+ disabledIntegrations,
2882
+ "openai-codex",
2883
+ "openai-codex-sdk",
2884
+ "codex",
2885
+ "codex-sdk"
2886
+ ) ? [] : openAICodexConfigs,
2114
2887
  ...isDisabled(disabledIntegrations, "anthropic") ? [] : anthropicConfigs,
2115
2888
  ...isDisabled(disabledIntegrations, "aisdk", "ai-sdk", "vercel-ai") ? [] : aiSDKConfigs,
2116
2889
  ...isDisabled(disabledIntegrations, "claudeagentsdk", "claude-agent-sdk") ? [] : claudeAgentSDKConfigs,
2890
+ ...isDisabled(disabledIntegrations, "cursor", "cursor-sdk") ? [] : cursorSDKConfigs,
2117
2891
  ...isDisabled(disabledIntegrations, "google", "google-genai") ? [] : googleGenAIConfigs,
2118
2892
  ...isDisabled(disabledIntegrations, "huggingface") ? [] : huggingFaceConfigs,
2119
2893
  ...isDisabled(disabledIntegrations, "openrouter") ? [] : openRouterConfigs,
2120
2894
  ...isDisabled(disabledIntegrations, "openrouteragent", "openrouter-agent") ? [] : openRouterAgentConfigs,
2121
2895
  ...isDisabled(disabledIntegrations, "mistral") ? [] : mistralConfigs,
2122
2896
  ...isDisabled(disabledIntegrations, "googleadk", "google-adk") ? [] : googleADKConfigs,
2123
- ...isDisabled(disabledIntegrations, "cohere") ? [] : cohereConfigs
2897
+ ...isDisabled(disabledIntegrations, "cohere") ? [] : cohereConfigs,
2898
+ ...isDisabled(disabledIntegrations, "groq", "groq-sdk") ? [] : groqConfigs,
2899
+ ...isDisabled(disabledIntegrations, "genkit", "firebase-genkit") ? [] : genkitConfigs,
2900
+ ...isDisabled(
2901
+ disabledIntegrations,
2902
+ "githubcopilot",
2903
+ "github-copilot",
2904
+ "copilot-sdk"
2905
+ ) ? [] : gitHubCopilotConfigs
2124
2906
  ];
2125
2907
  register("./loader/esm-hook.mjs", {
2126
2908
  parentURL: import.meta.url,