From 86a0720310a3ec58d4ff2c4a1112b3971fd21f3c Mon Sep 17 00:00:00 2001 From: ccurme Date: Tue, 21 Jan 2025 11:05:30 -0500 Subject: [PATCH] fireworks[patch]: update model used in integration tests (#29342) No access to firefunction-v1 and -v2. --- .../fireworks/tests/integration_tests/test_chat_models.py | 6 +++--- .../fireworks/tests/integration_tests/test_standard.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/libs/partners/fireworks/tests/integration_tests/test_chat_models.py b/libs/partners/fireworks/tests/integration_tests/test_chat_models.py index 88a1cd46cfcbb..885e3071cff12 100644 --- a/libs/partners/fireworks/tests/integration_tests/test_chat_models.py +++ b/libs/partners/fireworks/tests/integration_tests/test_chat_models.py @@ -15,7 +15,7 @@ def test_chat_fireworks_call() -> None: """Test valid call to fireworks.""" llm = ChatFireworks( # type: ignore[call-arg] - model="accounts/fireworks/models/firefunction-v1", temperature=0 + model="accounts/fireworks/models/llama-v3p1-70b-instruct", temperature=0 ) resp = llm.invoke("Hello!") @@ -27,7 +27,7 @@ def test_chat_fireworks_call() -> None: def test_tool_choice() -> None: """Test that tool choice is respected.""" llm = ChatFireworks( # type: ignore[call-arg] - model="accounts/fireworks/models/firefunction-v1", temperature=0 + model="accounts/fireworks/models/llama-v3p1-70b-instruct", temperature=0 ) class MyTool(BaseModel): @@ -59,7 +59,7 @@ def test_tool_choice_bool() -> None: """Test that tool choice is respected just passing in True.""" llm = ChatFireworks( # type: ignore[call-arg] - model="accounts/fireworks/models/firefunction-v1", temperature=0 + model="accounts/fireworks/models/llama-v3p1-70b-instruct", temperature=0 ) class MyTool(BaseModel): diff --git a/libs/partners/fireworks/tests/integration_tests/test_standard.py b/libs/partners/fireworks/tests/integration_tests/test_standard.py index b40b8476d88cd..024a5f0e7871c 100644 --- a/libs/partners/fireworks/tests/integration_tests/test_standard.py +++ b/libs/partners/fireworks/tests/integration_tests/test_standard.py @@ -20,7 +20,7 @@ def chat_model_class(self) -> Type[BaseChatModel]: @property def chat_model_params(self) -> dict: return { - "model": "accounts/fireworks/models/firefunction-v2", + "model": "accounts/fireworks/models/llama-v3p1-70b-instruct", "temperature": 0, }