From 97d36f3e80d01cef9568d55ff56c78802e6e08d9 Mon Sep 17 00:00:00 2001
From: Mateusz Szewczyk <Mateusz.Szewczyk@ibm.com>
Date: Tue, 8 Oct 2024 11:07:06 +0200
Subject: [PATCH] update tests

---
 libs/ibm/tests/integration_tests/test_chat_models.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/libs/ibm/tests/integration_tests/test_chat_models.py b/libs/ibm/tests/integration_tests/test_chat_models.py
index 4372fbe..f37c4af 100644
--- 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