github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-26 08:33:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
No access to firefunction-v1 and -v2.
This commit is contained in:
ccurme 2025-01-21 11:05:30 -05:00 committed by GitHub
parent 32c9c58adf
commit 86a0720310
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libs/partners/fireworks/tests/integration_tests/test_chat_models.py
View File

@ -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):

2
libs/partners/fireworks/tests/integration_tests/test_standard.py
View File

@ -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,
}