braintrust 0.3.14__tar.gz → 0.4.0__tar.gz

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 (126) hide show
  1. {braintrust-0.3.14 → braintrust-0.4.0}/PKG-INFO +3 -2
  2. {braintrust-0.3.14 → braintrust-0.4.0}/setup.py +2 -1
  3. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/__init__.py +4 -0
  4. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/_generated_types.py +1200 -611
  5. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/audit.py +2 -2
  6. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/cli/eval.py +6 -7
  7. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/cli/push.py +11 -11
  8. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/conftest.py +1 -0
  9. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/context.py +12 -17
  10. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/contrib/temporal/__init__.py +16 -27
  11. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/contrib/temporal/test_temporal.py +8 -3
  12. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/devserver/auth.py +8 -8
  13. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/devserver/cache.py +3 -4
  14. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/devserver/cors.py +8 -7
  15. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/devserver/dataset.py +3 -5
  16. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/devserver/eval_hooks.py +7 -6
  17. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/devserver/schemas.py +22 -19
  18. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/devserver/server.py +19 -12
  19. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/devserver/test_cached_login.py +4 -4
  20. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/framework.py +128 -140
  21. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/framework2.py +88 -87
  22. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/functions/invoke.py +93 -53
  23. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/functions/stream.py +3 -2
  24. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/generated_types.py +17 -1
  25. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/git_fields.py +11 -11
  26. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/gitutil.py +2 -3
  27. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/graph_util.py +10 -10
  28. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/id_gen.py +2 -2
  29. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/logger.py +346 -357
  30. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/merge_row_batch.py +10 -9
  31. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/oai.py +107 -24
  32. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/otel/__init__.py +49 -49
  33. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/otel/context.py +16 -30
  34. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/otel/test_distributed_tracing.py +14 -11
  35. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/otel/test_otel_bt_integration.py +32 -31
  36. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/parameters.py +8 -8
  37. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/prompt.py +14 -14
  38. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/prompt_cache/disk_cache.py +5 -4
  39. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/prompt_cache/lru_cache.py +3 -2
  40. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/prompt_cache/prompt_cache.py +13 -14
  41. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/queue.py +4 -4
  42. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/score.py +4 -4
  43. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/serializable_data_class.py +4 -4
  44. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/span_identifier_v1.py +1 -2
  45. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/span_identifier_v2.py +3 -4
  46. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/span_identifier_v3.py +23 -20
  47. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/span_identifier_v4.py +34 -25
  48. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/test_framework.py +16 -6
  49. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/test_helpers.py +5 -5
  50. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/test_id_gen.py +2 -3
  51. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/test_otel.py +61 -53
  52. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/test_queue.py +0 -1
  53. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/test_score.py +1 -3
  54. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/test_span_components.py +29 -44
  55. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/util.py +9 -8
  56. braintrust-0.4.0/src/braintrust/version.py +4 -0
  57. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/_anthropic_utils.py +4 -4
  58. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/agno/__init__.py +3 -4
  59. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/agno/agent.py +1 -2
  60. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/agno/function_call.py +1 -2
  61. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/agno/model.py +1 -2
  62. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/agno/team.py +1 -2
  63. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/agno/utils.py +12 -12
  64. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/anthropic.py +7 -8
  65. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/claude_agent_sdk/__init__.py +3 -4
  66. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/claude_agent_sdk/_wrapper.py +29 -27
  67. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/dspy.py +15 -17
  68. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/google_genai/__init__.py +16 -16
  69. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/langchain.py +22 -24
  70. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/litellm.py +4 -3
  71. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/openai.py +15 -15
  72. braintrust-0.4.0/src/braintrust/wrappers/pydantic_ai.py +1204 -0
  73. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/test_agno.py +0 -1
  74. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/test_dspy.py +0 -1
  75. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/test_google_genai.py +2 -3
  76. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/test_litellm.py +0 -1
  77. braintrust-0.4.0/src/braintrust/wrappers/test_oai_attachments.py +322 -0
  78. braintrust-0.4.0/src/braintrust/wrappers/test_pydantic_ai_integration.py +1788 -0
  79. braintrust-0.3.14/src/braintrust/wrappers/test_pydantic_ai.py → braintrust-0.4.0/src/braintrust/wrappers/test_pydantic_ai_wrap_openai.py +1 -2
  80. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust.egg-info/PKG-INFO +3 -2
  81. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust.egg-info/SOURCES.txt +4 -1
  82. braintrust-0.3.14/src/braintrust/version.py +0 -4
  83. {braintrust-0.3.14 → braintrust-0.4.0}/README.md +0 -0
  84. {braintrust-0.3.14 → braintrust-0.4.0}/setup.cfg +0 -0
  85. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/aws.py +0 -0
  86. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/bt_json.py +0 -0
  87. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/cli/__init__.py +0 -0
  88. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/cli/__main__.py +0 -0
  89. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/cli/install/__init__.py +0 -0
  90. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/cli/install/api.py +0 -0
  91. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/cli/install/bump_versions.py +0 -0
  92. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/cli/install/logs.py +0 -0
  93. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/cli/install/redshift.py +0 -0
  94. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/cli/install/run_migrations.py +0 -0
  95. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/contrib/__init__.py +0 -0
  96. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/db_fields.py +0 -0
  97. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/devserver/__init__.py +0 -0
  98. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/devserver/test_lru_cache.py +0 -0
  99. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/devserver/test_server_integration.py +0 -0
  100. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/functions/__init__.py +0 -0
  101. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/functions/constants.py +0 -0
  102. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/http_headers.py +0 -0
  103. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/object.py +0 -0
  104. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/prompt_cache/__init__.py +0 -0
  105. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/prompt_cache/test_disk_cache.py +0 -0
  106. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/prompt_cache/test_lru_cache.py +0 -0
  107. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/prompt_cache/test_prompt_cache.py +0 -0
  108. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/py.typed +0 -0
  109. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/resource_manager.py +0 -0
  110. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/span_types.py +0 -0
  111. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/test_framework2.py +0 -0
  112. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/test_logger.py +0 -0
  113. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/test_serializable_data_class.py +0 -0
  114. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/test_util.py +0 -0
  115. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/test_version.py +0 -0
  116. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/__init__.py +0 -0
  117. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/claude_agent_sdk/test_wrapper.py +0 -0
  118. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/test_anthropic.py +0 -0
  119. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/test_openai.py +0 -0
  120. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/test_openrouter.py +0 -0
  121. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/wrappers/test_utils.py +0 -0
  122. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust/xact_ids.py +0 -0
  123. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust.egg-info/dependency_links.txt +0 -0
  124. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust.egg-info/entry_points.txt +0 -0
  125. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust.egg-info/requires.txt +0 -0
  126. {braintrust-0.3.14 → braintrust-0.4.0}/src/braintrust.egg-info/top_level.txt +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: braintrust
3
- Version: 0.3.14
3
+ Version: 0.4.0
4
4
  Summary: SDK for integrating Braintrust
5
5
  Home-page: https://www.braintrust.dev
6
6
  Author: Braintrust
@@ -8,8 +8,9 @@ Author-email: info@braintrust.dev
8
8
  Project-URL: Source Code, https://github.com/braintrustdata/braintrust-sdk
9
9
  Project-URL: Bug Tracker, https://github.com/braintrustdata/braintrust-sdk/issues
10
10
  Classifier: Programming Language :: Python :: 3
11
+ Classifier: Programming Language :: Python :: 3.10
11
12
  Classifier: Operating System :: OS Independent
12
- Requires-Python: >=3.8.0
13
+ Requires-Python: >=3.10.0
13
14
  Description-Content-Type: text/markdown
14
15
  Requires-Dist: GitPython
15
16
  Requires-Dist: requests
@@ -49,12 +49,13 @@ setuptools.setup(
49
49
  },
50
50
  classifiers=[
51
51
  "Programming Language :: Python :: 3",
52
+ "Programming Language :: Python :: 3.10",
52
53
  "Operating System :: OS Independent",
53
54
  ],
54
55
  package_dir={"": "src"},
55
56
  packages=setuptools.find_packages(where="src"),
56
57
  package_data={"braintrust": ["py.typed"]},
57
- python_requires=">=3.8.0",
58
+ python_requires=">=3.10.0",
58
59
  entry_points={"console_scripts": ["braintrust = braintrust.cli.__main__:main"]},
59
60
  install_requires=install_requires,
60
61
  extras_require=extras_require,
@@ -1,3 +1,4 @@
1
+ # pyright: reportUnusedImport=false
1
2
  """
2
3
  A Python library for interacting with [Braintrust](https://braintrust.dev/). This library
3
4
  contains functionality for running evaluations, logging completions, loading and invoking
@@ -73,3 +74,6 @@ from .wrappers.anthropic import (
73
74
  from .wrappers.litellm import (
74
75
  wrap_litellm, # noqa: F401 # type: ignore[reportUnusedImport]
75
76
  )
77
+ from .wrappers.pydantic_ai import (
78
+ setup_pydantic_ai, # noqa: F401 # type: ignore[reportUnusedImport]
79
+ )