diff --git a/packages/toolbox-core/src/toolbox_core/client.py b/packages/toolbox-core/src/toolbox_core/client.py index a1c6d1ba..08182743 100644 --- a/packages/toolbox-core/src/toolbox_core/client.py +++ b/packages/toolbox-core/src/toolbox_core/client.py @@ -40,7 +40,7 @@ def __init__( Initializes the ToolboxClient. Args: - url: The base URL for the Toolbox service API (e.g., "http://localhost:8000"). + url: The base URL for the Toolbox service API (e.g., "http://localhost:5000"). session: An optional existing `aiohttp.ClientSession` to use. If None (default), a new session is created internally. Note that if a session is provided, its lifecycle (including closing) diff --git a/packages/toolbox-langchain/tests/test_utils.py b/packages/toolbox-langchain/tests/test_utils.py index 41166b89..488a6aef 100644 --- a/packages/toolbox-langchain/tests/test_utils.py +++ b/packages/toolbox-langchain/tests/test_utils.py @@ -214,7 +214,7 @@ async def test_invoke_tool(self, mock_post): mock_post.return_value.__aenter__.return_value = mock_response result = await _invoke_tool( - "http://localhost:8000", + "http://localhost:5000", aiohttp.ClientSession(), "tool_name", {"input": "data"}, @@ -222,7 +222,7 @@ async def test_invoke_tool(self, mock_post): ) mock_post.assert_called_once_with( - "http://localhost:8000/api/tool/tool_name/invoke", + "http://localhost:5000/api/tool/tool_name/invoke", json={"input": "data"}, headers={}, ) @@ -241,7 +241,7 @@ async def test_invoke_tool_unsecure_with_auth(self, mock_post): match="Sending ID token over HTTP. User data may be exposed. Use HTTPS for secure communication.", ): result = await _invoke_tool( - "http://localhost:8000", + "http://localhost:5000", aiohttp.ClientSession(), "tool_name", {"input": "data"}, @@ -249,7 +249,7 @@ async def test_invoke_tool_unsecure_with_auth(self, mock_post): ) mock_post.assert_called_once_with( - "http://localhost:8000/api/tool/tool_name/invoke", + "http://localhost:5000/api/tool/tool_name/invoke", json={"input": "data"}, headers={"my_test_auth_token": "fake_id_token"}, ) @@ -267,7 +267,7 @@ async def test_invoke_tool_secure_with_auth(self, mock_post): with warnings.catch_warnings(): warnings.simplefilter("error") result = await _invoke_tool( - "https://localhost:8000", + "https://localhost:5000", session, "tool_name", {"input": "data"}, @@ -275,7 +275,7 @@ async def test_invoke_tool_secure_with_auth(self, mock_post): ) mock_post.assert_called_once_with( - "https://localhost:8000/api/tool/tool_name/invoke", + "https://localhost:5000/api/tool/tool_name/invoke", json={"input": "data"}, headers={"my_test_auth_token": "fake_id_token"}, )