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",