From 10a2ce2a26fc99b4c45d5e07c3681f174f81cb35 Mon Sep 17 00:00:00 2001 From: ccurme Date: Wed, 21 Aug 2024 10:26:25 -0400 Subject: [PATCH] together[patch]: use mixtral in standard integration tests (#25619) Mistral 7B occasionally fails tool-calling tests. Updating to Mixtral appears to improve this. --- .../tests/integration_tests/test_chat_models_standard.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/partners/together/tests/integration_tests/test_chat_models_standard.py b/libs/partners/together/tests/integration_tests/test_chat_models_standard.py index 18c167f8a91dc..f4c8cc7002de6 100644 --- a/libs/partners/together/tests/integration_tests/test_chat_models_standard.py +++ b/libs/partners/together/tests/integration_tests/test_chat_models_standard.py @@ -24,7 +24,7 @@ def chat_model_class(self) -> Type[BaseChatModel]: @property def chat_model_params(self) -> dict: return { - "model": "mistralai/Mistral-7B-Instruct-v0.1", + "model": "mistralai/Mixtral-8x7B-Instruct-v0.1", "rate_limiter": rate_limiter, }