diff --git a/libs/ibm/tests/integration_tests/test_chat_models.py b/libs/ibm/tests/integration_tests/test_chat_models.py index f3ce95a..2a27aac 100644 --- a/libs/ibm/tests/integration_tests/test_chat_models.py +++ b/libs/ibm/tests/integration_tests/test_chat_models.py @@ -429,7 +429,7 @@ class Person(BaseModel): with_tool = chat.bind_tools([Person], tool_choice=tool_choice) - result = with_tool.invoke("Erick, 27 years old") + result = with_tool.invoke("Erick, 27 years old. Make sure to use correct name") assert isinstance(result, AIMessage) assert result.content == "" # should just be tool call tool_call = result.tool_calls[0] @@ -726,6 +726,7 @@ def test_structured_output() -> None: model_id=MODEL_ID_TOOL, url=URL, # type: ignore[arg-type] project_id=WX_PROJECT_ID, + temperature=0, ) schema = { "title": "AnswerWithJustification", @@ -950,7 +951,7 @@ def test_invoke_with_params_5() -> None: def test_init_and_invoke_with_params_1() -> None: - params_1 = None + params_1 = {"max_tokens": 11} chat = ChatWatsonx( model_id=MODEL_ID_TOOL, url=URL, # type: ignore[arg-type] @@ -961,7 +962,7 @@ def test_init_and_invoke_with_params_1() -> None: completion_tokens = resp.response_metadata.get("token_usage", {}).get( "completion_tokens" ) - assert chat.params == {} + assert chat.params == params_1 assert 7 < completion_tokens < 11