make invoke method return str

Author: twishabansal

packages/toolbox-core/src/toolbox_core/itransport.py

@@ -48,7 +48,7 @@ async def tools_list(
     @abstractmethod
     async def tool_invoke(
         self, tool_name: str, arguments: dict, headers: Mapping[str, str]
-    ) -> dict:
+    ) -> str:
         """Invokes a specific tool on the server."""
         pass
 

packages/toolbox-core/src/toolbox_core/tool.py

@@ -290,7 +290,7 @@ async def __call__(self, *args: Any, **kwargs: Any) -> str:
             payload,
             headers,
         )
-        return body.get("result", body)
+        return body
 
     def add_auth_token_getters(
         self,

packages/toolbox-core/src/toolbox_core/toolbox_transport.py

@@ -88,7 +88,7 @@ async def tool_invoke(
             if not resp.ok:
                 err = body.get("error", f"unexpected status from server: {resp.status}")
                 raise Exception(err)
-        return body
+        return body.get("result")
 
     async def close(self):
         if self.__manage_session and not self.__session.closed:

packages/toolbox-core/tests/test_toolbox_transport.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import AsyncGenerator, Mapping, Optional, Union
+from typing import AsyncGenerator, Mapping, Union, Optional
 from unittest.mock import AsyncMock
 
 import pytest
@@ -135,7 +135,7 @@ async def test_tool_invoke_success(http_session: ClientSession):
         m.post(url, status=200, payload=response_payload)
         result = await transport.tool_invoke(TEST_TOOL_NAME, args, headers)
 
-        assert result == response_payload
+        assert result == "success"
         m.assert_called_once_with(url, method="POST", json=args, headers=headers)