braintrust 0.3.13__py3-none-any.whl → 0.3.15__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.
- braintrust/__init__.py +4 -0
- braintrust/_generated_types.py +596 -72
- braintrust/conftest.py +1 -0
- braintrust/functions/invoke.py +35 -2
- braintrust/generated_types.py +15 -1
- braintrust/gitutil.py +4 -0
- braintrust/logger.py +1 -1
- braintrust/oai.py +88 -6
- braintrust/score.py +1 -0
- braintrust/test_score.py +157 -0
- braintrust/version.py +2 -2
- braintrust/wrappers/pydantic_ai.py +1203 -0
- braintrust/wrappers/test_oai_attachments.py +322 -0
- braintrust/wrappers/test_pydantic_ai_integration.py +1788 -0
- braintrust/wrappers/{test_pydantic_ai.py → test_pydantic_ai_wrap_openai.py} +1 -2
- {braintrust-0.3.13.dist-info → braintrust-0.3.15.dist-info}/METADATA +1 -1
- {braintrust-0.3.13.dist-info → braintrust-0.3.15.dist-info}/RECORD +20 -16
- {braintrust-0.3.13.dist-info → braintrust-0.3.15.dist-info}/WHEEL +0 -0
- {braintrust-0.3.13.dist-info → braintrust-0.3.15.dist-info}/entry_points.txt +0 -0
- {braintrust-0.3.13.dist-info → braintrust-0.3.15.dist-info}/top_level.txt +0 -0
|
@@ -15,11 +15,10 @@ except ImportError:
|
|
|
15
15
|
from pydantic_ai.models.openai import OpenAIModel # pylint: disable=import-error
|
|
16
16
|
|
|
17
17
|
OpenAIModelClass = OpenAIModel
|
|
18
|
-
from pydantic_ai.providers.openai import OpenAIProvider # pylint: disable=import-error
|
|
19
|
-
|
|
20
18
|
from braintrust import logger, wrap_openai
|
|
21
19
|
from braintrust.span_types import SpanTypeAttribute
|
|
22
20
|
from braintrust.test_helpers import init_test_logger
|
|
21
|
+
from pydantic_ai.providers.openai import OpenAIProvider # pylint: disable=import-error
|
|
23
22
|
|
|
24
23
|
PROJECT_NAME = "test-pydantic-ai"
|
|
25
24
|
MODEL = "gpt-3.5-turbo" # Use a cheaper model for testing
|
|
@@ -1,29 +1,29 @@
|
|
|
1
|
-
braintrust/__init__.py,sha256
|
|
2
|
-
braintrust/_generated_types.py,sha256=
|
|
1
|
+
braintrust/__init__.py,sha256=sw6dLIINeyUVP4IbF49wqXv2cyAWEoO3SGSIvCb_b2o,2238
|
|
2
|
+
braintrust/_generated_types.py,sha256=KjVWSAPwPKbpVMobs24u7O3GRxJhRBpZiVAm3vrcsK8,93555
|
|
3
3
|
braintrust/audit.py,sha256=Na3LJhpHj8Nd1az41HMQLLbHeWQkDZIOYHLLmVZdAdQ,467
|
|
4
4
|
braintrust/aws.py,sha256=OBz_SRyopgpCDSNvETLypzGwTXk-bNLn-Eisevnjfwo,377
|
|
5
5
|
braintrust/bt_json.py,sha256=f2AkpVxT2LY3PZliJn-jH_1Qd9nujnSJcFronTdlsBE,3650
|
|
6
|
-
braintrust/conftest.py,sha256=
|
|
6
|
+
braintrust/conftest.py,sha256=hopKH5M5ts7DXlvRD_P2b94uXTT7vPKNhBWUdoIetL8,1681
|
|
7
7
|
braintrust/context.py,sha256=ZXIOc3zXIXOzAopkuPIM9tq2prfsTjH0-e_FEEFTTJI,4235
|
|
8
8
|
braintrust/db_fields.py,sha256=DBGFhfu9B3aLQI6cU6P2WGrdfCIs8AzDFRQEUBt9NCw,439
|
|
9
9
|
braintrust/framework.py,sha256=Uf2DogzQG4suCcmABeQzc-5SWCO5tB2ic1OeQFZo_o4,57618
|
|
10
10
|
braintrust/framework2.py,sha256=DTnCql0_Be9kYZ7QkgJHEUAW4_tU-vuqni3vxdCcnyk,16784
|
|
11
|
-
braintrust/generated_types.py,sha256=
|
|
11
|
+
braintrust/generated_types.py,sha256=Gdjl7-o26YNqTyjlnzzLTRspHC8ZIaOkZwF2Y8jXPsY,4787
|
|
12
12
|
braintrust/git_fields.py,sha256=oT1h1nVkidGrIIamtIQfOpbuEzd1n-nup9bx54J_u0A,1480
|
|
13
|
-
braintrust/gitutil.py,sha256=
|
|
13
|
+
braintrust/gitutil.py,sha256=01hCZG291992Knj9q1gOjb42XvfzruMm5yeA68FufAw,5676
|
|
14
14
|
braintrust/graph_util.py,sha256=lABIOMzxHf6E5LfDYfqa4OUR4uaW7xgUYNq5WGewD4w,5594
|
|
15
15
|
braintrust/http_headers.py,sha256=9ZsDcsAKG04SGowsgchZktD6rG_oSTKWa8QyGUPA4xE,154
|
|
16
16
|
braintrust/id_gen.py,sha256=PVkz-pS-9AzgmnAgpV-jgOFFo4hfl6e3IP9dVt6FouQ,1595
|
|
17
|
-
braintrust/logger.py,sha256=
|
|
17
|
+
braintrust/logger.py,sha256=Pesk94By7gPiVLoqA7NJjlLq4XslP7EmioRrusjVd6c,210576
|
|
18
18
|
braintrust/merge_row_batch.py,sha256=NX4jRE9uuFB3Z7btrarQp_di84_NGTjvzpJhksn82W8,9882
|
|
19
|
-
braintrust/oai.py,sha256=
|
|
19
|
+
braintrust/oai.py,sha256=dr7c1vm54ricuTrZfbF1bXfHciJnjUDW-K_EyqlZKc8,36303
|
|
20
20
|
braintrust/object.py,sha256=vYLyYWncsqLD00zffZUJwGTSkcJF9IIXmgIzrx3Np5c,632
|
|
21
21
|
braintrust/parameters.py,sha256=E1NJGJZ4Qjhp2h4Rghrh4qdE0FiaXLGPapS_iAk8es8,5946
|
|
22
22
|
braintrust/prompt.py,sha256=c7UR-UeJ8Hf-DGyFRyUxSgXEOkZFhVd9-IhvyPhy1xI,1938
|
|
23
23
|
braintrust/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
24
24
|
braintrust/queue.py,sha256=5Mjb5FvqC9hq9f4qykCQIxrZ8MFrQlCvf7XIlKyYRGA,3738
|
|
25
25
|
braintrust/resource_manager.py,sha256=Rs4yEd_ShV5YV6WC0YqxkQKXEfBtbTUZ4S4G78oF5Mk,723
|
|
26
|
-
braintrust/score.py,sha256=
|
|
26
|
+
braintrust/score.py,sha256=44MyCUiMb9Bn1kyYqa3Kej33Cx2-5MnQ-_Rcl5pJfQw,3663
|
|
27
27
|
braintrust/serializable_data_class.py,sha256=TjlLpUmVAJ_VW7-aiEecBVQ5d3k4bgQsVeJo_5yIAU0,2471
|
|
28
28
|
braintrust/span_identifier_v1.py,sha256=_i33QR-LIalyBkV28cWWh0OUPtA7cjv-VCc894-XdD0,5209
|
|
29
29
|
braintrust/span_identifier_v2.py,sha256=qLi2iXs-KmIfrQ3BjW6_yLpMYqKh-7_LrotiR4cplwc,9024
|
|
@@ -37,12 +37,13 @@ braintrust/test_id_gen.py,sha256=mgArTyEBV-Xv21ARHPSHEPBsJshZrvIiPjBLNOKsAko,249
|
|
|
37
37
|
braintrust/test_logger.py,sha256=Z3euqSHmc4kqsyKo838tkYGK8B1rTkIdxl7QZYR1CLk,98516
|
|
38
38
|
braintrust/test_otel.py,sha256=janmEtu6dyFQL8N68WjRTS-65Kr5vSNSSwIaBWqXlyw,27243
|
|
39
39
|
braintrust/test_queue.py,sha256=MdH6R9uSk_4akY4Db514Cpukwiy2RJ76Lqts1nQwZJY,8432
|
|
40
|
+
braintrust/test_score.py,sha256=3hgwO_g_yq9VxX6no2YXfEMpBegVeUYLPh0NX4T5rHs,5879
|
|
40
41
|
braintrust/test_serializable_data_class.py,sha256=b04Ym64YtC6GJRGbKIN4J20RG1QN1FlnODwtEQh4sv0,1897
|
|
41
42
|
braintrust/test_span_components.py,sha256=UnF6ZL4k41XZ-CnfbjuqLeK4MZLtHTMdID3CMh3g1os,16344
|
|
42
43
|
braintrust/test_util.py,sha256=gyqe2JspRP7oXlp6ENztZe2fdRTOEMZMKpQi00y1DSc,4538
|
|
43
44
|
braintrust/test_version.py,sha256=hk5JKjEFbNJ_ONc1VEkqHquflzre34RpFhCEYLTK8iA,1051
|
|
44
45
|
braintrust/util.py,sha256=Ec6sRkQw5BckGrFjdA4YTyu_2BaKmHh4tWDwAi_ysOw,7227
|
|
45
|
-
braintrust/version.py,sha256=
|
|
46
|
+
braintrust/version.py,sha256=W7VpEGqwXbRPRYCdJG7RW1HBGYQPQaUQMgv7kRRYS6E,118
|
|
46
47
|
braintrust/xact_ids.py,sha256=bdyp88HjlyIkglgLSqYlCYscdSH6EWVyE14sR90Xl1s,658
|
|
47
48
|
braintrust/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
48
49
|
braintrust/cli/__main__.py,sha256=wCBKHGVmn3IT_yMXk5qfDwyI2SV2gf1tLr0NTxm9T8k,1519
|
|
@@ -70,7 +71,7 @@ braintrust/devserver/test_lru_cache.py,sha256=5YYJ5uFj7k4Z4PQQ-UOV7bLP5zBYVo-5jV
|
|
|
70
71
|
braintrust/devserver/test_server_integration.py,sha256=ygIYDsTCSqwa_rzQDtdX0mjkXQVy7qyJjntdoCgtjKM,6711
|
|
71
72
|
braintrust/functions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
72
73
|
braintrust/functions/constants.py,sha256=g_EDiSrfCltHr5QaQMQzJ3qy3D29X-11LanDqlicqB0,23
|
|
73
|
-
braintrust/functions/invoke.py,sha256=
|
|
74
|
+
braintrust/functions/invoke.py,sha256=fKho8cHpV6KECp5jUzrjdj4sPBBOr9s78BuoxqX5BTA,9911
|
|
74
75
|
braintrust/functions/stream.py,sha256=-c2q1dP0Kk2W_RKECjZJ3-t9ivTjZhC8jSBxLNX23Ik,6088
|
|
75
76
|
braintrust/otel/__init__.py,sha256=h3vGw9vaCHDJpmvXyrL923lN4gnhReTCfWavfCLWt_g,23743
|
|
76
77
|
braintrust/otel/context.py,sha256=FZQJ4IEWEexIoZYrNzoLpE_txvETnsf0fDd3XkiWtDo,4823
|
|
@@ -90,14 +91,17 @@ braintrust/wrappers/dspy.py,sha256=5NPVNRXLPUi4gXxgVyrhsZm6dB9yG4T4kErLuOs3p1A,1
|
|
|
90
91
|
braintrust/wrappers/langchain.py,sha256=-yStNscKz1RecKZoTHT0XGtwgD3KkYZcVPeSuddR0pc,5106
|
|
91
92
|
braintrust/wrappers/litellm.py,sha256=qqOxNmBgq1KrTJeWQoojvfdvFgEDiZjf5kLZoJ0m63g,24947
|
|
92
93
|
braintrust/wrappers/openai.py,sha256=JRyokeMLyoAXSeKJRNLI6baWuo6uNqf5rzVnjlHe9X0,10609
|
|
94
|
+
braintrust/wrappers/pydantic_ai.py,sha256=SZjD6B37Pu_FRuCA-W7KKGU_CdESFIWcPEEKvJHjel4,44057
|
|
93
95
|
braintrust/wrappers/test_agno.py,sha256=A4dd45nIkcy-7xZOdDAEbMejGgwSyMc6mnrHIRTcOcA,3669
|
|
94
96
|
braintrust/wrappers/test_anthropic.py,sha256=tSHAnroaLK1QYhLhhPFAxSDEJUybzffp2o4IptDZRAU,15554
|
|
95
97
|
braintrust/wrappers/test_dspy.py,sha256=x5vXu-NoFFjWGIcvFvXICVsDasQEQzPuSlxgzbHJcwo,1953
|
|
96
98
|
braintrust/wrappers/test_google_genai.py,sha256=20e4DqLUZmcpVGircj7vN0IxsS5FVjvveqOu0JNZ5jo,17596
|
|
97
99
|
braintrust/wrappers/test_litellm.py,sha256=sNId6qnbQibT66aB_lsxX_HlaPF6GoMFaawgnF1A2a4,21428
|
|
100
|
+
braintrust/wrappers/test_oai_attachments.py,sha256=9UXaXXXzZMLi_Q97QyV2uEtpuTy6BYL6I4olyFfEJ04,11273
|
|
98
101
|
braintrust/wrappers/test_openai.py,sha256=UqKtoIlZlqiLC6x7Rbv9zROgDdE5zvXZObbFMKpmifw,60237
|
|
99
102
|
braintrust/wrappers/test_openrouter.py,sha256=9wIYuwObFkADH8h5pzHFB4C9TcL_HTeq1KlODYhSMLs,3987
|
|
100
|
-
braintrust/wrappers/
|
|
103
|
+
braintrust/wrappers/test_pydantic_ai_integration.py,sha256=M60jFulGGblFfDnFkobfphvkbTATpnfB3VxfonQ5aR4,71802
|
|
104
|
+
braintrust/wrappers/test_pydantic_ai_wrap_openai.py,sha256=cxj6uQaObhQ97-mEjN2OStpL4swlWhaln3n2zpIEA_I,5056
|
|
101
105
|
braintrust/wrappers/test_utils.py,sha256=Qz7LYG5V0DK2KuTJ_YLGpO_Zr_LJFfJgZX_Ps8tlM_c,505
|
|
102
106
|
braintrust/wrappers/agno/__init__.py,sha256=V0dnEhu2U0FabyEmwDbWv5x6rP1rdI8z9JLAB_8XuXk,2416
|
|
103
107
|
braintrust/wrappers/agno/agent.py,sha256=WjqtiWt8mRAH5JiZNjWT_EG25A04FC3rRj20D0Xg_kc,6474
|
|
@@ -109,8 +113,8 @@ braintrust/wrappers/claude_agent_sdk/__init__.py,sha256=CSXJWy-z2fHF7h4VJjLSnXJv
|
|
|
109
113
|
braintrust/wrappers/claude_agent_sdk/_wrapper.py,sha256=uzElIOwwPmF_Y5fbWcKWEPC8HnSzW7byzpiuVKK0TXE,15613
|
|
110
114
|
braintrust/wrappers/claude_agent_sdk/test_wrapper.py,sha256=0NmohdECudFvWtc-5PbANtTXzexkkwIJhGbujydDrT8,6826
|
|
111
115
|
braintrust/wrappers/google_genai/__init__.py,sha256=PGFMuR3c4Gc3SUt24eP7z5AzdS2Dc1uF1d3QPCnLnuo,16018
|
|
112
|
-
braintrust-0.3.
|
|
113
|
-
braintrust-0.3.
|
|
114
|
-
braintrust-0.3.
|
|
115
|
-
braintrust-0.3.
|
|
116
|
-
braintrust-0.3.
|
|
116
|
+
braintrust-0.3.15.dist-info/METADATA,sha256=t8wUMpwwpSacNL0H7pP6e5YyPMSs85-mNAZak33NnQo,3702
|
|
117
|
+
braintrust-0.3.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
|
118
|
+
braintrust-0.3.15.dist-info/entry_points.txt,sha256=Zpc0_09g5xm8as5jHqqFq7fhwO0xHSNct_TrEMONS7Q,60
|
|
119
|
+
braintrust-0.3.15.dist-info/top_level.txt,sha256=hw1-y-UFMf60RzAr8x_eM7SThbIuWfQsQIbVvqSF83A,11
|
|
120
|
+
braintrust-0.3.15.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|