fireworks[patch]: update model used in integration tests (#29342)

No access to firefunction-v1 and -v2.
langchain-ai · Jan 21, 2025 · 86a0720 · 86a0720
1 parent 32c9c58
commit 86a0720
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 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
@@ -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
@@ -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,
         }