hyperpocket 0.5.1__py3-none-any.whl → 0.5.2__py3-none-any.whl

hyperpocket/pocket_main.py

@@ -32,18 +32,22 @@ class Pocket(object):
 
     def __init__(
         self,
-        tools: list[ToolLike],
+        tools: list[ToolLike] = None,
         auth: PocketAuth = None,
         use_profile: bool = False,
     ):
         try:
             if auth is None:
                 auth = PocketAuth()
+            if tools is None:
+                tools = []
+
             self.auth = auth
             self.use_profile = use_profile
             self.server = PocketServer.get_instance()
+            self.tools = {}
 
-            self._load_tools(tools)
+            self.load_tools(tools)
             pocket_logger.info(
                 f"All Registered Tools Loaded successfully. total registered tools : {len(self.tools)}"
             )
@@ -468,25 +472,61 @@ class Pocket(object):
                 tool_by_provider[auth_provider_name] = [tool]
         return tool_by_provider
 
-    def _load_tools(self, tools):
-        self.tools = dict()
-        dock = self._default_dock()
+    def load_tools(self, tools: Union[List[ToolLike], ToolLike]) -> List[Tool]:
+        """
+        Load a list of tools into the pocket.
 
-        def _load(tool_like):
-            if isinstance(tool_like, str) or isinstance(tool_like, tuple):
-                return dock(tool_like)
-            elif isinstance(tool_like, Tool):
-                return tool_like
-            elif isinstance(tool_like, Callable):
-                return from_func(tool_like)
-            else:
-                raise ValueError(f"Invalid tool type: {type(tool_like)}")
+        This method takes a list of tool identifiers(tool like) and loads them into the
+        pocket for use.
+
+        Args:
+            tools (Union[List[ToolLike], ToolLike]): A list of tool identifiers to be loaded.
+        """
+        dock = self._default_dock()
+        if not isinstance(tools, list):
+            tools = [tools]
 
+        loaded_tools = []
         with concurrent.futures.ThreadPoolExecutor(max_workers=10, thread_name_prefix="tool-loader") as executor:
-            futures = [executor.submit(_load, tool_like) for tool_like in tools]
-            for future in concurrent.futures.as_completed(futures):
-                tool = future.result()
-                self.tools[tool.name] = tool
+            futures = [executor.submit(self._load_tool, tool_like, dock) for tool_like in tools]
+            loaded_tools = [future.result() for future in concurrent.futures.as_completed(futures)]
+
+        for tool in loaded_tools:
+            if tool.name in self.tools:
+                raise RuntimeError(f"{tool.name} already exists. duplicated tool name.")
+            self.tools[tool.name] = tool
+
+        return loaded_tools
+
+    def _load_tool(self, tool_like, dock: Dock = None) -> Tool:
+        if dock is None:
+            dock = self._default_dock()
+
+        if isinstance(tool_like, str) or isinstance(tool_like, tuple):
+            return dock(tool_like)
+        elif isinstance(tool_like, Tool):
+            return tool_like
+        elif isinstance(tool_like, Callable):
+            return from_func(tool_like)
+        else:
+            raise ValueError(f"Invalid tool type: {type(tool_like)}")
+
+    def remove_tool(self, tool_name: str) -> bool:
+        """
+        Remove a tool from the pocket.
+
+        This method removes a tool identified by the given tool_name.
+
+        Args:
+            tool_name (str): The tool name to be removed.
+
+        Returns:
+            bool: True if the tool is removed successfully, False otherwise.
+        """
+        if not tool_name in self.tools:
+            return False
+        del self.tools[tool_name]
+        return True
 
     def _tool_instance(self, tool_name: str) -> Tool:
         return self.tools[tool_name]

{hyperpocket-0.5.1.dist-info → hyperpocket-0.5.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyperpocket
-Version: 0.5.1
+Version: 0.5.2
 Summary: Building AI agent with hyperpocket tool in a flash
 Project-URL: Homepage, https://vessl-ai.github.io/hyperpocket
 Project-URL: Repository, https://github.com/vessl-ai/hyperpocket

{hyperpocket-0.5.1.dist-info → hyperpocket-0.5.2.dist-info}/RECORD

@@ -2,7 +2,7 @@ hyperpocket/__init__.py,sha256=VVLbApRTiULqEVQp6lCNOcuXKx9V62O_7C9VNKBQ0G0,137
 hyperpocket/builtin.py,sha256=w7OLxf5RCKVpLma9HieSdw6Uky5701ae6g31VPvFoZk,2439
 hyperpocket/constants.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hyperpocket/pocket_auth.py,sha256=eRRTzJh3et65EDJnQHB_Kpmco4dgCR2KL8f-FnbuGX0,17819
-hyperpocket/pocket_main.py,sha256=pCbJjT7r-kzYIxzh24jKcGwh-SOnbB9ID8kwDpB1HG4,16467
+hyperpocket/pocket_main.py,sha256=1kNesVy5Z-WRET6-GfnRFpPtUdyxQaPBsjhXRflzh18,17702
 hyperpocket/prompts.py,sha256=N1bCzCLZvGUVhH1Vn_cgeBPsdY3MdIU7ZGqVgexoj5E,472
 hyperpocket/tool_like.py,sha256=Foa-iWTnVb54JEq20Becadbz-TSbYkZk6TxexSzaaRM,116
 hyperpocket/auth/README.md,sha256=zn4QqnFZCA_4X3x8Wb6lE3OP5otYxpByZaCiUkBvaNs,11562
@@ -597,7 +597,7 @@ hyperpocket/util/get_objects_from_subpackage.py,sha256=4mR_S8eaJSdU68YfCkiXeIcXx
 hyperpocket/util/git_parser.py,sha256=y96nhgZXtRgA_u_0GTPo95PGkpG-n_oMIrkbckdxiR8,2496
 hyperpocket/util/json_schema_to_model.py,sha256=nc5AmnqkrdeFLELu-7_O9sEAaUaD8_KGlvIMDRobt-4,3751
 hyperpocket/util/short_hashing_str.py,sha256=ahLUT8iQr-MJVbDJXrSt0cXnnSEeJ8EU3A0PDn6e0gs,119
-hyperpocket-0.5.1.dist-info/METADATA,sha256=pg-5bLmexfszjmU_OB4c96vHKtR6sU2_eo2oycXLA50,13078
-hyperpocket-0.5.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-hyperpocket-0.5.1.dist-info/entry_points.txt,sha256=KpBleaYr0SaENXOa-dFvJ_cvFCHYFEQ4LMl11ShAcBI,61
-hyperpocket-0.5.1.dist-info/RECORD,,
+hyperpocket-0.5.2.dist-info/METADATA,sha256=OBQOkr9W9KmvlePHr68rsnhFkf_OV_BxiV12-H3LQbM,13078
+hyperpocket-0.5.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+hyperpocket-0.5.2.dist-info/entry_points.txt,sha256=KpBleaYr0SaENXOa-dFvJ_cvFCHYFEQ4LMl11ShAcBI,61
+hyperpocket-0.5.2.dist-info/RECORD,,