From 783440dcd3954500be98e948a97ac32ee406a287 Mon Sep 17 00:00:00 2001
From: Kenny Wong <wong@coda.io>
Date: Tue, 2 Jun 2026 08:22:31 -0400
Subject: [PATCH] [Deps] Pin braintrust <0.13 to unbreak Python CI

braintrust 0.13.0 removed the wrapper classes (ChatCompletionV0Wrapper,
OpenAIV1Wrapper, ...) and changed wrap_openai to no longer return a
NamedWrapper. That breaks test_oai.py's imports (aborting pytest collection)
and LLMClient.is_wrapped detection. Cap at the last compatible release
(0.12.1) until autoevals is adapted to the new braintrust API.
---
 setup.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index d8b70803..ab9a0c08 100644
--- a/setup.py
+++ b/setup.py
@@ -16,7 +16,10 @@
 extras_require = {
     "dev": [
         "black==22.6.0",
-        "braintrust",  # used for testing
+        # 0.13.0 removed the wrapper classes (ChatCompletionV0Wrapper, OpenAIV1Wrapper, ...)
+        # and changed wrap_openai so it no longer returns a NamedWrapper, which breaks
+        # test_oai.py imports and LLMClient.is_wrapped detection. Cap until autoevals is updated.
+        "braintrust<0.13",  # used for testing
         "build",
         "flake8",
         "flake8-isort",