braintrust 0.4.0__py3-none-any.whl → 0.4.2__py3-none-any.whl

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 (23) hide show
  1. braintrust/bt_json.py +178 -19
  2. braintrust/db_fields.py +1 -0
  3. braintrust/framework.py +13 -4
  4. braintrust/logger.py +30 -120
  5. braintrust/otel/__init__.py +24 -15
  6. braintrust/test_bt_json.py +644 -0
  7. braintrust/test_framework.py +81 -0
  8. braintrust/test_logger.py +245 -107
  9. braintrust/test_otel.py +118 -26
  10. braintrust/test_util.py +51 -1
  11. braintrust/util.py +24 -3
  12. braintrust/version.py +2 -2
  13. braintrust/wrappers/google_genai/__init__.py +2 -15
  14. braintrust/wrappers/litellm.py +43 -0
  15. braintrust/wrappers/pydantic_ai.py +209 -95
  16. braintrust/wrappers/test_google_genai.py +62 -1
  17. braintrust/wrappers/test_litellm.py +73 -0
  18. braintrust/wrappers/test_pydantic_ai_integration.py +819 -22
  19. {braintrust-0.4.0.dist-info → braintrust-0.4.2.dist-info}/METADATA +1 -1
  20. {braintrust-0.4.0.dist-info → braintrust-0.4.2.dist-info}/RECORD +23 -22
  21. {braintrust-0.4.0.dist-info → braintrust-0.4.2.dist-info}/WHEEL +0 -0
  22. {braintrust-0.4.0.dist-info → braintrust-0.4.2.dist-info}/entry_points.txt +0 -0
  23. {braintrust-0.4.0.dist-info → braintrust-0.4.2.dist-info}/top_level.txt +0 -0
{braintrust-0.4.0.dist-info → braintrust-0.4.2.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: braintrust
3
- Version: 0.4.0
3
+ Version: 0.4.2
4
4
  Summary: SDK for integrating Braintrust
5
5
  Home-page: https://www.braintrust.dev
6
6
  Author: Braintrust
{braintrust-0.4.0.dist-info → braintrust-0.4.2.dist-info}/RECORD RENAMED
@@ -2,11 +2,11 @@ braintrust/__init__.py,sha256=sw6dLIINeyUVP4IbF49wqXv2cyAWEoO3SGSIvCb_b2o,2238
2
2
  braintrust/_generated_types.py,sha256=k36Qyd3AkTMXA_EJLyj0_zrUJTa7VIJDAQCNPbiIbL8,93868
3
3
  braintrust/audit.py,sha256=3GQKzuTcFquYdrJtABM-k3xMlOIqgVkfG6UyeQ8_028,461
4
4
  braintrust/aws.py,sha256=OBz_SRyopgpCDSNvETLypzGwTXk-bNLn-Eisevnjfwo,377
5
- braintrust/bt_json.py,sha256=f2AkpVxT2LY3PZliJn-jH_1Qd9nujnSJcFronTdlsBE,3650
5
+ braintrust/bt_json.py,sha256=VNunedFUEfbvVEBuACHLRsjDr8lLD3_nFwB54MXtcbY,9385
6
6
  braintrust/conftest.py,sha256=hopKH5M5ts7DXlvRD_P2b94uXTT7vPKNhBWUdoIetL8,1681
7
7
  braintrust/context.py,sha256=bOo1Li29lUsi2DOOllIDar0oRuQNbkLNzJ7cYq5JTbo,4126
8
- braintrust/db_fields.py,sha256=DBGFhfu9B3aLQI6cU6P2WGrdfCIs8AzDFRQEUBt9NCw,439
9
- braintrust/framework.py,sha256=3NLInBWpAc5uBdPP5qrfP_GB3P0dvhY8hqHwUiuF-KE,57202
8
+ braintrust/db_fields.py,sha256=vjVEyDxl2d13lvvTcVwT5paHvaT-1kgI-4Lg-gihRIw,476
9
+ braintrust/framework.py,sha256=U2ubFkX6LS-3iArMT4sRG17oZa0ClM4GVpC0O8xotxQ,57593
10
10
  braintrust/framework2.py,sha256=o0igz4vXbmn0jHJPhDYvx14rFnI3ntV8H6VJfyJYRtM,16542
11
11
  braintrust/generated_types.py,sha256=foqHolHkcqYTNL5tfAR0mRIAtOO8Jl4R022QQqe-M1k,4849
12
12
  braintrust/git_fields.py,sha256=au5ayyuvt7y_ojE9LC98ypTZd3RgFdjhRc8eFxcjnto,1434
@@ -14,7 +14,7 @@ braintrust/gitutil.py,sha256=RsW7cawJMgaAaTw6WeB1sShyfflkPb7yH2x7yuRv10c,5642
14
14
  braintrust/graph_util.py,sha256=Z2Uy8RaOq5iMe5mShhQqRDDIpXVitE-biVxDiFB-0Ds,5545
15
15
  braintrust/http_headers.py,sha256=9ZsDcsAKG04SGowsgchZktD6rG_oSTKWa8QyGUPA4xE,154
16
16
  braintrust/id_gen.py,sha256=4UWLWRhksf76IkYi4cKACSaQ3yNgausrMRlhiurhy74,1590
17
- braintrust/logger.py,sha256=i5WLp8HrpNy5F-bFZvQ9-abIbbx2y42PsTHsRTZyQH4,209519
17
+ braintrust/logger.py,sha256=UuTETw0JeNBtKX2OJlM_GlzUyCKPg7Ar1DdVULhl_E4,205548
18
18
  braintrust/merge_row_batch.py,sha256=tvuz3qdsa7HZBrzzAzoQqAXU7pKsHUebC7sw6RybbmA,9881
19
19
  braintrust/oai.py,sha256=AjHeD0uNYO-ECVUh9RbPBxyKMps06A3dvHAWa41prMc,36259
20
20
  braintrust/object.py,sha256=vYLyYWncsqLD00zffZUJwGTSkcJF9IIXmgIzrx3Np5c,632
@@ -30,20 +30,21 @@ braintrust/span_identifier_v2.py,sha256=2dLc-Vz8iWLISmL_-ebCyWnY-ysA7sMnBsQtKqzM
30
30
  braintrust/span_identifier_v3.py,sha256=l_W8amV0u89X-cHqIhWQdjLZcoeWhwJoAEdYWgCN1nM,9720
31
31
  braintrust/span_identifier_v4.py,sha256=uFT-OdzySo4uCeAaJC3VqH55Oy433xZGBdK2hiEsm2w,10044
32
32
  braintrust/span_types.py,sha256=UvuyjfL_Delao14TJs9IjnpYLcgDSodKJfAYju01aXQ,292
33
- braintrust/test_framework.py,sha256=F5QvwASOEG_c2OuUV3duveitYPZkcJHWW_z-wPImWOw,14666
33
+ braintrust/test_bt_json.py,sha256=iixyk_7QIKiF3VVThD9oiWzAT1b-t2fHPYECmSB8VT0,25304
34
+ braintrust/test_framework.py,sha256=7spRLovwNjZLula8mNW_KspyWuWgpqSfHV6sZqub-ME,17664
34
35
  braintrust/test_framework2.py,sha256=pSEEmBIyszAiYnpEVvDZgJqIe3lQ3T807LmIuBqV98w,7235
35
36
  braintrust/test_helpers.py,sha256=VSelzjkR2IHyy5QD6sYB_79VIXq-wEDslDwyx9H11kI,13528
36
37
  braintrust/test_id_gen.py,sha256=Oqp8Rxd1_9ptpEJSu_xi6STIquZpx9n7Kh-bGUKMQH0,2489
37
- braintrust/test_logger.py,sha256=Z3euqSHmc4kqsyKo838tkYGK8B1rTkIdxl7QZYR1CLk,98516
38
- braintrust/test_otel.py,sha256=D7APSJcFGHo9HWGPFfFng_ID5_KIGLdH4pt1HNVLOxY,27444
38
+ braintrust/test_logger.py,sha256=ARSrWMmIOkuXosI5dzSRMIAKGYWKWOJCeWhTx5cA_ME,103267
39
+ braintrust/test_otel.py,sha256=EyLOBMIZegvGyqqWl4vAlIazZfz1s4tWbgSAM68-2mM,31356
39
40
  braintrust/test_queue.py,sha256=872weDhTfxN1Vu4-cV8pZZ9XNE8UkX4WNcvc2mPtSDI,8431
40
41
  braintrust/test_score.py,sha256=PpfzNeaYhC4shMFqF85EczuMT0FCTr6l1rdEYWLe4g0,5857
41
42
  braintrust/test_serializable_data_class.py,sha256=b04Ym64YtC6GJRGbKIN4J20RG1QN1FlnODwtEQh4sv0,1897
42
43
  braintrust/test_span_components.py,sha256=6w0oDnDLuVbHygNienujk4JDqtSRR5A49AI-S8Pa5hY,16197
43
- braintrust/test_util.py,sha256=gyqe2JspRP7oXlp6ENztZe2fdRTOEMZMKpQi00y1DSc,4538
44
+ braintrust/test_util.py,sha256=xMN84ERVzZYUTAOp0IkFK-ktWOLK4rr6OdhaNeq7Fg0,6382
44
45
  braintrust/test_version.py,sha256=hk5JKjEFbNJ_ONc1VEkqHquflzre34RpFhCEYLTK8iA,1051
45
- braintrust/util.py,sha256=0cEoSJ3GAkAP17Pz4kXSrwqNhiC_rNWxGcfu_hGx8ys,7217
46
- braintrust/version.py,sha256=zbkdMmqjYz16krkw7dIISZIx2wXWFEpvi8SIn4MUAM4,117
46
+ braintrust/util.py,sha256=-q73F17BkaUaDCGtNBTtacqIPwWiXcgWYpzHqFTQdhc,8361
47
+ braintrust/version.py,sha256=8d_QVXySfoJqVPmORR9pWGmW5J1QcpfY-Cp9MlqpGC4,117
47
48
  braintrust/xact_ids.py,sha256=bdyp88HjlyIkglgLSqYlCYscdSH6EWVyE14sR90Xl1s,658
48
49
  braintrust/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
49
50
  braintrust/cli/__main__.py,sha256=wCBKHGVmn3IT_yMXk5qfDwyI2SV2gf1tLr0NTxm9T8k,1519
@@ -73,7 +74,7 @@ braintrust/functions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
73
74
  braintrust/functions/constants.py,sha256=g_EDiSrfCltHr5QaQMQzJ3qy3D29X-11LanDqlicqB0,23
74
75
  braintrust/functions/invoke.py,sha256=BZQv5FXjLvT1nlPRBhEQv4MSKRVZPSunn63pm1vdkRs,10096
75
76
  braintrust/functions/stream.py,sha256=zH3okhkbUBE3-iYD6edLqjDdewc1Zj1MjVx5A3Xwe9Y,6103
76
- braintrust/otel/__init__.py,sha256=R2ZyXu8OTHhAjK1veMg_1CKcSqZLFBu65_7iwgcJ_iw,23764
77
+ braintrust/otel/__init__.py,sha256=sZxGs2khKfL9uFJtB118VJJ7O5QOZJSG3vuKUFB0fiE,24350
77
78
  braintrust/otel/context.py,sha256=EpGk3B1UeAfBeqDe-yYy4LNNCFBKU6aU-0tfVOMrkcs,4611
78
79
  braintrust/otel/test_distributed_tracing.py,sha256=12Cw18_kkon-y5is-b3lyVuzmzmOUpQdSVxk4NzYRUM,5238
79
80
  braintrust/otel/test_otel_bt_integration.py,sha256=t5FLU_wDGOc8h9PnCQvb2eEi8CtJNxXInuFAeJUGVeo,13302
@@ -89,18 +90,18 @@ braintrust/wrappers/_anthropic_utils.py,sha256=tOj_rv5NVRXpCij0F3NPxOZg0XGAAXo8Y
89
90
  braintrust/wrappers/anthropic.py,sha256=sh0UbSWDarU5glic4HIUXNZmMcfn3Odau2K8HzaOIWY,11785
90
91
  braintrust/wrappers/dspy.py,sha256=ckPcMHpbVZP6k4R60ZbgegWz82lsK3lu2u9nvN0EIss,13516
91
92
  braintrust/wrappers/langchain.py,sha256=0aY5LuVA7BPkgWA0N6CwPG9EaPqRmVVfEPaM1kN4XZY,5028
92
- braintrust/wrappers/litellm.py,sha256=peXP58o8waFZqatYPWTGpn1ROgkFxFN0CDvOd9G_wOo,24948
93
+ braintrust/wrappers/litellm.py,sha256=ZE63C8OE5Z0N3mArwOWZusDXBW1MElPOCjODOF-tmB0,26616
93
94
  braintrust/wrappers/openai.py,sha256=SZuT4ouJd8FRNXxy9zM_OGb2HNL9XGsFnwkHCk3LDAM,10563
94
- braintrust/wrappers/pydantic_ai.py,sha256=eRfzs5h2cPQ_kpJMYuRh3HnJ7bJZzJN4chF03I2HFx4,44007
95
+ braintrust/wrappers/pydantic_ai.py,sha256=AgMZS4F7c5Um-P98p13A0uapMFGuYf5YyXuDw1nwPHs,48534
95
96
  braintrust/wrappers/test_agno.py,sha256=e-ss-Lr7HMiGvPzsSSTxMblXgzyrbNebrvM062-lVHY,3668
96
97
  braintrust/wrappers/test_anthropic.py,sha256=tSHAnroaLK1QYhLhhPFAxSDEJUybzffp2o4IptDZRAU,15554
97
98
  braintrust/wrappers/test_dspy.py,sha256=jE-k-joobVbGm8DAyp1eEkbPt_xBleZ10wkWJXLp0nI,1952
98
- braintrust/wrappers/test_google_genai.py,sha256=ZipWAP0vnSbxp23bn4YMf23FXUQcXk7QDeTkxEPkG-o,17595
99
- braintrust/wrappers/test_litellm.py,sha256=dio0iUb9bGQed6mRhSEWxC33F7363DVz4uD4AggXx0w,21427
99
+ braintrust/wrappers/test_google_genai.py,sha256=W4p3TIuJi2_xXqIVFtwQta7ai-WeoSwAd2y_tReC3Z4,19967
100
+ braintrust/wrappers/test_litellm.py,sha256=miOeXEQm6g8ypvx8LpGlPrnXW-U51glOBP49mdwptc0,23777
100
101
  braintrust/wrappers/test_oai_attachments.py,sha256=9UXaXXXzZMLi_Q97QyV2uEtpuTy6BYL6I4olyFfEJ04,11273
101
102
  braintrust/wrappers/test_openai.py,sha256=UqKtoIlZlqiLC6x7Rbv9zROgDdE5zvXZObbFMKpmifw,60237
102
103
  braintrust/wrappers/test_openrouter.py,sha256=9wIYuwObFkADH8h5pzHFB4C9TcL_HTeq1KlODYhSMLs,3987
103
- braintrust/wrappers/test_pydantic_ai_integration.py,sha256=M60jFulGGblFfDnFkobfphvkbTATpnfB3VxfonQ5aR4,71802
104
+ braintrust/wrappers/test_pydantic_ai_integration.py,sha256=zvBqXd7ik-wZEKNMVB6pRTrGZ3XcTSEZ_FG4Ogho9mA,104304
104
105
  braintrust/wrappers/test_pydantic_ai_wrap_openai.py,sha256=cxj6uQaObhQ97-mEjN2OStpL4swlWhaln3n2zpIEA_I,5056
105
106
  braintrust/wrappers/test_utils.py,sha256=Qz7LYG5V0DK2KuTJ_YLGpO_Zr_LJFfJgZX_Ps8tlM_c,505
106
107
  braintrust/wrappers/agno/__init__.py,sha256=0ehV_7L_ILsMfZ1bKHnzcR_CvFM98KMbGFN9UiE94vI,2379
@@ -112,9 +113,9 @@ braintrust/wrappers/agno/utils.py,sha256=_b_s2WzAJ5HIkZ8Qbq5ZKctsdFDh5tX3gFuRiT2
112
113
  braintrust/wrappers/claude_agent_sdk/__init__.py,sha256=dcBP61ijW2B4g902-09ZfIOkAkC2GtPWz9Rg3RWELWQ,4390
113
114
  braintrust/wrappers/claude_agent_sdk/_wrapper.py,sha256=K6iWfnYSgRq0ETGQ7CCh4rF4JJ0CR-G-VkCUfeKEQ-c,15645
114
115
  braintrust/wrappers/claude_agent_sdk/test_wrapper.py,sha256=0NmohdECudFvWtc-5PbANtTXzexkkwIJhGbujydDrT8,6826
115
- braintrust/wrappers/google_genai/__init__.py,sha256=lDQpwOTRHcVOnOCraFhR4ZI1lp7eKJzGpIsmapzd6rY,15994
116
- braintrust-0.4.0.dist-info/METADATA,sha256=bUfDvvvbkgID7g56dTnFMYKbEWE2GdgZ03HqfJV1PvQ,3753
117
- braintrust-0.4.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
118
- braintrust-0.4.0.dist-info/entry_points.txt,sha256=Zpc0_09g5xm8as5jHqqFq7fhwO0xHSNct_TrEMONS7Q,60
119
- braintrust-0.4.0.dist-info/top_level.txt,sha256=hw1-y-UFMf60RzAr8x_eM7SThbIuWfQsQIbVvqSF83A,11
120
- braintrust-0.4.0.dist-info/RECORD,,
116
+ braintrust/wrappers/google_genai/__init__.py,sha256=Yv4hBOUc6TjUpYH3bEW8UaYiZ-sm3TpdVfqUw50NwQU,15826
117
+ braintrust-0.4.2.dist-info/METADATA,sha256=DiX2q-5LiIz53SvCzzwGvoISnjY7YK-peu5Sd6EVM6s,3753
118
+ braintrust-0.4.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
119
+ braintrust-0.4.2.dist-info/entry_points.txt,sha256=Zpc0_09g5xm8as5jHqqFq7fhwO0xHSNct_TrEMONS7Q,60
120
+ braintrust-0.4.2.dist-info/top_level.txt,sha256=hw1-y-UFMf60RzAr8x_eM7SThbIuWfQsQIbVvqSF83A,11
121
+ braintrust-0.4.2.dist-info/RECORD,,