fix tests

Author: twishabansal

packages/toolbox-llamaindex/tests/test_async_tools.py

@@ -255,7 +255,7 @@ async def test_toolbox_tool_call(self, toolbox_tool):
         )
 
         core_tool = toolbox_tool._AsyncToolboxTool__core_tool
-        session = core_tool._ToolboxTool__transport._ToolboxHttpTransport__session
+        session = core_tool._ToolboxTool__transport._ToolboxTransport__session
         session.post.assert_called_once_with(
             "http://test_url/api/tool/test_tool/invoke",
             json={"param1": "test-value", "param2": 123},
@@ -281,7 +281,7 @@ async def test_toolbox_tool_call_with_bound_params(
             raw_output="test-result",
         )
         core_tool = tool._AsyncToolboxTool__core_tool
-        session = core_tool._ToolboxTool__transport._ToolboxHttpTransport__session
+        session = core_tool._ToolboxTool__transport._ToolboxTransport__session
         session.post.assert_called_once_with(
             "http://test_url/api/tool/test_tool/invoke",
             json={"param1": expected_value, "param2": 123},
@@ -301,7 +301,7 @@ async def test_toolbox_tool_call_with_auth_tokens(self, auth_toolbox_tool):
         )
 
         core_tool = tool._AsyncToolboxTool__core_tool
-        session = core_tool._ToolboxTool__transport._ToolboxHttpTransport__session
+        session = core_tool._ToolboxTool__transport._ToolboxTransport__session
         session.post.assert_called_once_with(
             "https://test-url/api/tool/test_tool/invoke",
             json={"param2": 123},
@@ -342,7 +342,7 @@ async def test_toolbox_tool_call_with_auth_tokens_insecure(
 
         modified_core_tool = tool_with_getter._AsyncToolboxTool__core_tool
         transport = modified_core_tool._ToolboxTool__transport
-        session = transport._ToolboxHttpTransport__session
+        session = transport._ToolboxTransport__session
 
         assert transport.base_url == "http://test-url"