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

hyperpocket/cli/__main__.py

@@ -1,10 +1,7 @@
 import click
 
-from hyperpocket.cli.eject import eject
-from hyperpocket.cli.pull import pull
-from hyperpocket.cli.eject import eject
-from hyperpocket.cli.auth_token import create_token_auth_template
 from hyperpocket.cli.auth_oauth2 import create_oauth2_auth_template
+from hyperpocket.cli.auth_token import create_token_auth_template
 from hyperpocket.cli.tool_create import create_tool_template, build_tool
 from hyperpocket.cli.tool_export import export_tool
 
@@ -26,7 +23,4 @@ devtool.add_command(create_tool_template)
 devtool.add_command(build_tool)
 devtool.add_command(export_tool)
 
-cli.add_command(pull)
-cli.add_command(eject)
-
 cli()

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,58 @@ 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(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)}")
+    def load_tools(self, tools: Union[List[ToolLike], ToolLike]) -> List[Tool]:
+        """
+        Load a list of tools into the pocket.
+
+        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.
+        """
+        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) 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) -> Tool:
+        if isinstance(tool_like, str) or isinstance(tool_like, tuple):
+            dock = self._default_dock()
+            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/util/function_to_model.py

@@ -38,18 +38,15 @@ def function_to_model(func: callable) -> Type[BaseModel]:
             raise Exception(
                 f"Should all arguments be annotated but {param_name} is not annotated"
             )
+        
+        if param.annotation.__module__ == "typing":
+            for arg in param.annotation.__args__:
+                if arg.__module__ != "builtins" and not issubclass(arg, BaseModel):
+                    raise Exception(
+                        f"currently only support builtin types and pydantic BaseModel but {param_name} is not builtin type"
+                    )
 
-        if (
-            param.annotation.__module__ == "typing"
-            and param.annotation.__name__ == "Optional"
-        ):
-            fields[param_name] = (
-                param.annotation.__args__[0],
-                FieldInfo(
-                    default=param.default,
-                    description=param_desc_map.get(param_name, ""),
-                ),
-            )
+        # if annotation type isn't typing, check this type directly.
         elif param.annotation.__module__ != "builtins" and not issubclass(
             param.annotation, BaseModel
         ):

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyperpocket
-Version: 0.5.1
+Version: 0.5.3
 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
@@ -14,10 +14,9 @@ Requires-Dist: httpx==0.27
 Requires-Dist: jinja2>=3.1.4
 Requires-Dist: multiprocess>=0.70.17
 Requires-Dist: nest-asyncio>=1.6.0
-Requires-Dist: playwright>=1.49.0
 Requires-Dist: pydantic>=2.10.2
 Requires-Dist: pygithub>=2.5.0
-Requires-Dist: python-multipart>=0.0.19
+Requires-Dist: python-multipart>=0.0.18
 Requires-Dist: redis>=5.2.1
 Requires-Dist: requests>=2.32.3
 Requires-Dist: toml>=0.10.2

{hyperpocket-0.5.1.dist-info → hyperpocket-0.5.3.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=PGMx5MovFSD9Xy9gJbCnF4yLPL4LimfTqDta0SGSroU,17615
 hyperpocket/prompts.py,sha256=N1bCzCLZvGUVhH1Vn_cgeBPsdY3MdIU7ZGqVgexoj5E,472
 hyperpocket/tool_like.py,sha256=Foa-iWTnVb54JEq20Becadbz-TSbYkZk6TxexSzaaRM,116
 hyperpocket/auth/README.md,sha256=zn4QqnFZCA_4X3x8Wb6lE3OP5otYxpByZaCiUkBvaNs,11562
@@ -468,11 +468,9 @@ hyperpocket/auth/zoom/oauth2_context.py,sha256=f0vj1zeRzTnT-4cwe8SeEn2xz03NBy9W_
 hyperpocket/auth/zoom/oauth2_handler.py,sha256=dMoZjuSRSBaoYLzYkjxc0KZxm9CALLm9aaEVydLBW-0,5111
 hyperpocket/auth/zoom/oauth2_schema.py,sha256=OlwlEdigiqFX3-XdVp0mhrK7ZKWxVw0v1qNtfx3wN9c,469
 hyperpocket/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-hyperpocket/cli/__main__.py,sha256=sVGU32c27NCaqysM4TsOf6D-Tj7a3mDhUmm7CRajg_g,753
+hyperpocket/cli/__main__.py,sha256=9myHxl6CdY5A1w41q7WkD4u1PyZ7C_PlpRFpIYKmXCM,589
 hyperpocket/cli/auth_oauth2.py,sha256=FTsaIrJY9U1uDEtZL456zOWQKfcqzOPg33ikFhFFjyA,4899
 hyperpocket/cli/auth_token.py,sha256=YV1AK70Q_hNsm_xwHP5LTaUWI-omssnX8214JyD3m_w,4875
-hyperpocket/cli/eject.py,sha256=utzeL4t02ML-EIgpPUiFjCbycjRpvkP1905bLeZAT2Y,321
-hyperpocket/cli/pull.py,sha256=JnoIsmQwJODWOuilmKR5KKFsdMZ0Ht7pCWIbENJuYfw,275
 hyperpocket/cli/tool_create.py,sha256=1tsVw_sy28JmeCSXT1j-7dqOsDXzvZPTNqcmHchtcpI,3996
 hyperpocket/cli/tool_export.py,sha256=NRfssftdb1HEj9DtdM7rh6_rFs1hydYyjpDe7V8kFzU,8060
 hyperpocket/cli/codegen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -592,12 +590,12 @@ hyperpocket/util/extract_func_param_desc_from_docstring.py,sha256=eGDrbigmVg1zau
 hyperpocket/util/find_all_leaf_class_in_package.py,sha256=06n8R47BDPovxhCOnzu9GuEfeQzEbv-HdG-zfMu1gBw,533
 hyperpocket/util/find_all_subclass_in_package.py,sha256=TtEb41-nzCNhC9pgelTS6MMxLT_JNZkFPJe5z2H9yik,978
 hyperpocket/util/flatten_json_schema.py,sha256=iuNBEmMSKFtPi-uqo6fb3RWN0koHOAihWAAofWbd1U8,1671
-hyperpocket/util/function_to_model.py,sha256=TXUs-qPbzL8C9-qqpz4Ad4D9MOPP61n_p0iPU6SoBeM,2318
+hyperpocket/util/function_to_model.py,sha256=HubCCZOdTWcGy4OpDSEws5pEkkuiqT_aYqM663tdU6U,2381
 hyperpocket/util/get_objects_from_subpackage.py,sha256=4mR_S8eaJSdU68YfCkiXeIcXxb6q7LjFGsY_IHeNIZw,929
 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.3.dist-info/METADATA,sha256=D1F09X8C9Ewl20lLIN2i6pEzdZp8_KtkTrA36rUVxpY,13044
+hyperpocket-0.5.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+hyperpocket-0.5.3.dist-info/entry_points.txt,sha256=KpBleaYr0SaENXOa-dFvJ_cvFCHYFEQ4LMl11ShAcBI,61
+hyperpocket-0.5.3.dist-info/RECORD,,

hyperpocket/cli/eject.py

@@ -1,14 +0,0 @@
-import pathlib
-from typing import Optional
-
-import click
-
-import hyperpocket.repository as repository
-
-
-@click.command()
-@click.argument("url", type=str)
-@click.argument("ref", type=str)
-@click.argument("remote_path", type=str)
-def eject(url: str, ref: str, remote_path: str):
-    repository.eject(url, ref, remote_path)

hyperpocket/cli/pull.py

@@ -1,13 +0,0 @@
-import pathlib
-from typing import Optional
-
-import click
-
-import hyperpocket.repository as repository
-
-
-@click.command()
-@click.argument("url", type=str)
-@click.option("--git-ref", type=str, default="HEAD")
-def pull(url: str, git_ref: str):
-    repository.pull(url, git_ref)