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 88a1cd46cfc..885e3071cff 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 @@ from langchain_fireworks import ChatFireworks 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 b40b8476d88..024a5f0e787 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 @@ class TestFireworksStandard(ChatModelIntegrationTests): @property def chat_model_params(self) -> dict: return { - "model": "accounts/fireworks/models/firefunction-v2", + "model": "accounts/fireworks/models/llama-v3p1-70b-instruct", "temperature": 0, }