fireworks[patch]: use firefunction-v2 in standard tests (#24264)

2 months ago · ab2d7821a7
parent 6fc7610b1c
commit ab2d7821a7
1 changed files with 1 additions and 1 deletions
--- 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,
        }