.

Signed-off-by: Bryan Qiu <bryan.qiu@databricks.com>

Author: bbqiu

integrations/langchain/src/databricks_langchain/chat_models.py

@@ -1055,7 +1055,6 @@ def _convert_message_to_dict(message: BaseMessage) -> dict:
         return {"role": "user", **message_dict}
     elif isinstance(message, AIMessage):
         if tool_calls := _get_tool_calls_from_ai_message(message):
-            print(tool_calls)
             message_dict["tool_calls"] = tool_calls  # type: ignore[assignment]
             # If tool calls present, content null value should be None not empty string.
             message_dict["content"] = message_dict["content"] or None  # type: ignore[assignment]

integrations/langchain/tests/integration_tests/agent.py

@@ -906,5 +906,3 @@ def test_chat_databricks_with_gemini():
     )
     assert result is not None
     assert result.output is not None
-    print(result.model_dump())
-    assert False