update tests

langchain-ai · Oct 8, 2024 · 97d36f3 · 97d36f3
1 parent 5ace260
commit 97d36f3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/libs/ibm/tests/integration_tests/test_chat_models.py b/libs/ibm/tests/integration_tests/test_chat_models.py
@@ -402,6 +402,7 @@ def test_22a_bind_tools_tool_choice_as_class() -> None:
         model_id=MODEL_ID_TOOL,
         url=URL,  # type: ignore[arg-type]
         project_id=WX_PROJECT_ID,
+        params={"temperature": 0},
     )
 
     class Person(BaseModel):
@@ -428,6 +429,7 @@ def test_22b_bind_tools_tool_choice_as_dict() -> None:
         model_id=MODEL_ID_TOOL,
         url=URL,  # type: ignore[arg-type]
         project_id=WX_PROJECT_ID,
+        params={"temperature": 0},
     )
 
     class Person(BaseModel):
@@ -455,6 +457,7 @@ def test_23a_bind_tools_list_tool_choice_dict() -> None:
         model_id=MODEL_ID_TOOL,
         url=URL,  # type: ignore[arg-type]
         project_id=WX_PROJECT_ID,
+        params={"temperature": 0},
     )
 
     @tool
@@ -495,6 +498,7 @@ def test_23_bind_tools_list_tool_choice_auto() -> None:
         model_id=MODEL_ID_TOOL,
         url=URL,  # type: ignore[arg-type]
         project_id=WX_PROJECT_ID,
+        params={"temperature": 0},
     )
 
     @tool