diff --git a/libs/partners/groq/tests/integration_tests/test_standard.py b/libs/partners/groq/tests/integration_tests/test_standard.py index 04e68fa567f..90b9f16a5e5 100644 --- a/libs/partners/groq/tests/integration_tests/test_standard.py +++ b/libs/partners/groq/tests/integration_tests/test_standard.py @@ -29,14 +29,10 @@ class BaseTestGroq(ChatModelIntegrationTests): return True -class TestGroqLlama(BaseTestGroq): +class TestGroqGemma(BaseTestGroq): @property def chat_model_params(self) -> dict: - return { - "model": "llama-3.1-8b-instant", - "temperature": 0, - "rate_limiter": rate_limiter, - } + return {"model": "gemma2-9b-it", "rate_limiter": rate_limiter} @property def supports_json_mode(self) -> bool: