From ab2d7821a75b45fc0c6de46c917744a4ab2a07a7 Mon Sep 17 00:00:00 2001 From: ccurme Date: Mon, 15 Jul 2024 09:15:08 -0400 Subject: [PATCH] fireworks[patch]: use firefunction-v2 in standard tests (#24264) --- .../partners/fireworks/tests/integration_tests/test_standard.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/partners/fireworks/tests/integration_tests/test_standard.py b/libs/partners/fireworks/tests/integration_tests/test_standard.py index 6280f619f0..cfefb2445e 100644 --- a/libs/partners/fireworks/tests/integration_tests/test_standard.py +++ b/libs/partners/fireworks/tests/integration_tests/test_standard.py @@ -19,7 +19,7 @@ class TestFireworksStandard(ChatModelIntegrationTests): @property def chat_model_params(self) -> dict: return { - "model": "accounts/fireworks/models/firefunction-v1", + "model": "accounts/fireworks/models/firefunction-v2", "temperature": 0, }