hyperpocket-llamaindex 0.0.1__tar.gz → 0.1.9__tar.gz

hyperpocket_llamaindex-0.1.9/.gitignore

@@ -0,0 +1,157 @@
+.vs/
+.vscode/
+.idea/
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+pip-wheel-metadata/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# Google GitHub Actions credentials files created by:
+# https://github.com/google-github-actions/auth
+#
+# That action recommends adding this gitignore to prevent accidentally committing keys.
+gha-creds-*.json
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+
+# Jupyter Notebook
+.ipynb_checkpoints
+notebooks/
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+.python-version
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.envrc
+.venv*
+venv*
+env/
+ENV/
+env.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.mypy_cache_test/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# macOS display setting files
+.DS_Store
+
+# Wandb directory
+wandb/
+
+# asdf tool versions
+.tool-versions
+/.ruff_cache/
+
+*.pkl
+*.bin
+
+# integration test artifacts
+data_map*
+\[('_type', 'fake'), ('stop', None)]
+
+# Replit files
+*replit*
+
+
+prof
+virtualenv/
+
+# Ignore dynaconf secret files
+**/.secrets.toml
+*.xml
+*.iml
+pocket.lock
+.log/
+.pocket/
+_build/

{hyperpocket_llamaindex-0.0.1 → hyperpocket_llamaindex-0.1.9}/PKG-INFO

@@ -1,16 +1,10 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: hyperpocket-llamaindex
-Version: 0.0.1
-Summary: 
-Author: moon
-Author-email: moon@vessl.ai
-Requires-Python: >=3.11,<4.0
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
-Requires-Dist: hyperpocket (==0.0.3)
-Requires-Dist: llama-index (>=0.12.5,<0.13.0)
+Version: 0.1.9
+Author-email: Hyperpocket Team <hyperpocket@vessl.ai>
+Requires-Python: >=3.10
+Requires-Dist: hyperpocket>=0.0.3
+Requires-Dist: llama-index>=0.12.5
 Description-Content-Type: text/markdown
 
 ## Anthropic extensions
@@ -56,4 +50,4 @@ llm = OpenAI()
 agent = FunctionCallingAgent.from_tools(
     tools=tools, llm=llm, verbose=True
 )
-```
+```

hyperpocket_llamaindex-0.1.9/hyperpocket_llamaindex/pocket_llamaindex.py

@@ -0,0 +1,73 @@
+from typing import List, Optional
+
+from pydantic import BaseModel
+
+try:
+    from llama_index.core.tools import FunctionTool, BaseTool, ToolMetadata
+except ImportError:
+    raise ImportError(
+        "You need to install llama-index to use pocket llamaindex"
+    )
+
+from hyperpocket import Pocket
+from hyperpocket.tool import Tool
+
+
+class PocketLlamaindex(Pocket):
+    def get_tools(self, use_profile: Optional[bool] = None) -> List[BaseTool]:
+        if use_profile is not None:
+            self.use_profile = use_profile
+
+        tools = [self.get_tool(pk) for pk in self.core.tools.values()]
+        return tools
+
+    def get_tool(self, pocket_tool: Tool) -> BaseTool:
+        def _invoke(**kwargs) -> str:
+            if self.use_profile:
+                body = kwargs["body"]
+                thread_id = kwargs.pop("thread_id", "default")
+                profile = kwargs.pop("profile", "default")
+            else:
+                body = kwargs
+                thread_id = "default"
+                profile = "default"
+
+            if isinstance(body, BaseModel):
+                body = body.model_dump()
+
+            result, interrupted = self.invoke_with_state(pocket_tool.name, body=body, thread_id=thread_id,
+                                                         profile=profile, **kwargs)
+            say = result
+            if interrupted:
+                say = f'{say}\n\nThe tool execution interrupted. Please talk to me to resume.'
+            return say
+
+        async def _ainvoke(**kwargs) -> str:
+            if self.use_profile:
+                body = kwargs["body"]
+                thread_id = kwargs.pop("thread_id", "default")
+                profile = kwargs.pop("profile", "default")
+            else:
+                body = kwargs
+                thread_id = "default"
+                profile = "default"
+
+            if isinstance(body, BaseModel):
+                body = body.model_dump()
+
+            result, interrupted = await self.ainvoke_with_state(pocket_tool.name, body=body,
+                                                                thread_id=thread_id, profile=profile, **kwargs)
+            say = result
+            if interrupted:
+                say = f'{say}\n\nThe tool execution interrupted. Please talk to me to resume.'
+            return say
+
+        return FunctionTool.from_defaults(
+            fn=_invoke,
+            async_fn=_ainvoke,
+            tool_metadata=ToolMetadata(
+                name=pocket_tool.name,
+                description=pocket_tool.get_description(use_profile=self.use_profile),
+                fn_schema=pocket_tool.schema_model(use_profile=self.use_profile),
+            )
+        )

hyperpocket_llamaindex-0.1.9/pyproject.toml

@@ -0,0 +1,36 @@
+
+[project]
+name = "hyperpocket-llamaindex"
+version = "0.1.9"
+description = ""
+authors = [{ name = "Hyperpocket Team", email = "hyperpocket@vessl.ai" }]
+requires-python = ">=3.10"
+readme = "README.md"
+dependencies = ["llama-index>=0.12.5", "hyperpocket>=0.0.3"]
+
+[tool.uv.sources]
+hyperpocket = { path = "../../hyperpocket", editable = true }
+
+[dependency-groups]
+dev = ["pytest>=8.3.4", "ruff>=0.8.6"]
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.ruff.lint]
+select = [
+    "E", # pycodestyle errors,
+    "F", # pyflakes errors,
+    "I", # isort errors,
+]
+ignore = [
+    "E501", # line too long, handled by formatting
+]
+
+[tool.ruff]
+line-length = 88
+target-version = "py310"
+
+[tool.ruff.lint.per-file-ignores]
+"__init__.py" = ["F401"]

hyperpocket_llamaindex-0.1.9/tests/test_pocket_llamaindex_no_profile.py

@@ -0,0 +1,137 @@
+import ast
+import json
+from unittest.async_case import IsolatedAsyncioTestCase
+
+from llama_index.agent.openai import OpenAIAgent
+from llama_index.llms.openai import OpenAI
+from pydantic import BaseModel
+
+from hyperpocket.config import config, secret
+from hyperpocket.tool import from_git
+from hyperpocket_llamaindex import PocketLlamaindex
+
+
+class TestPocketLlamaindexNoProfile(IsolatedAsyncioTestCase):
+
+    async def asyncSetUp(self):
+        config.public_server_port = "https"
+        config.public_hostname = "localhost"
+        config.public_server_port = 8001
+        config.internal_server_port = 8000
+        config.enable_local_callback_proxy = True
+
+        self.pocket = PocketLlamaindex(
+            tools=[
+                from_git("https://github.com/vessl-ai/hyperawesometools", "main",
+                         "managed-tools/none/simple-echo-tool"),
+                self.add,
+                self.sub_pydantic_args
+            ],
+            use_profile=False
+        )
+
+        self.llm = OpenAI(model="gpt-4o", api_key=secret["OPENAI_API_KEY"])
+
+    async def asyncTearDown(self):
+        self.pocket._teardown_server()
+
+    async def test_agent_no_profile(self):
+        # given
+        agent = OpenAIAgent.from_tools(
+            tools=self.pocket.get_tools(),
+            llm=OpenAI(model="gpt-4o", api_key=secret["OPENAI_API_KEY"]),
+            verbose=True
+        )
+
+        # when
+        agent.query("add 1, 2")
+        agent.query("sub 1, 2")
+        agent.query("echo 'hello world'")
+
+    async def test_function_tool_no_profile(self):
+        # when
+        response = self.llm.chat_with_tools(user_msg="add 1, 2", tools=self.pocket.get_tools(), verbose=True)
+        message = response.message
+        tool_calls = message.additional_kwargs["tool_calls"]
+
+        tool_name = tool_calls[0].function.name
+        args = tool_calls[0].function.arguments
+        args = json.loads(args)
+
+        result = await self.pocket.ainvoke(tool_name=tool_name, body=args)
+
+        # then
+        self.assertEqual(tool_name, "add")
+        self.assertEqual(args, {
+            "a": 1,
+            "b": 2,
+        })
+        self.assertEqual(result, '3')
+
+    async def test_pydantic_function_tool_no_profile(self):
+        # when
+        response = self.llm.chat_with_tools(user_msg="sub 1, 2", tools=self.pocket.get_tools(), verbose=True)
+        tool_calls = response.message.additional_kwargs["tool_calls"]
+
+        tool_name = tool_calls[0].function.name
+        args = tool_calls[0].function.arguments
+        args = json.loads(args)
+
+        result = await self.pocket.ainvoke(tool_name=tool_name, body=args)
+
+        # then
+        self.assertEqual(tool_name, "sub_pydantic_args")
+        self.assertEqual(args, {
+            "a": {"first": 1},
+            "b": {"second": 2},
+        })
+        self.assertEqual(result, '-1')
+
+    async def test_wasm_tool_no_profile(self):
+        # when
+        response = self.llm.chat_with_tools(user_msg="echo 'hello world'", tools=self.pocket.get_tools(), verbose=True)
+        tool_calls = response.message.additional_kwargs["tool_calls"]
+
+        tool_name = tool_calls[0].function.name
+        args = tool_calls[0].function.arguments
+        args = json.loads(args)
+
+        result = await self.pocket.ainvoke(tool_name=tool_name, body=args)
+        output = ast.literal_eval(result)
+
+        # then
+        self.assertEqual(tool_name, "simple_echo_text")
+        self.assertEqual(args, {
+            "text": "hello world"
+        })
+        self.assertTrue(output["stdout"].startswith("echo message : hello world"))
+
+    @staticmethod
+    def add(a: int, b: int) -> int:
+        """
+        Add two numbers
+
+        Args:
+            a(int): first number
+            b(int): second number
+
+        """
+
+        return a + b
+
+    class FirstNumber(BaseModel):
+        first: int
+
+    class SecondNumber(BaseModel):
+        second: int
+
+    @staticmethod
+    def sub_pydantic_args(a: FirstNumber, b: SecondNumber):
+        """
+        sub two numbers
+
+        Args:
+            a(FirstNumber): first number
+            b(SecondNumber): second number
+        """
+        return a.first - b.second

hyperpocket_llamaindex-0.1.9/tests/test_pocket_llamaindex_use_profile.py

@@ -0,0 +1,137 @@
+import ast
+import json
+from unittest.async_case import IsolatedAsyncioTestCase
+
+from llama_index.agent.openai import OpenAIAgent
+from llama_index.llms.openai import OpenAI
+from pydantic import BaseModel
+
+from hyperpocket.config import config, secret
+from hyperpocket.tool import from_git
+from hyperpocket_llamaindex import PocketLlamaindex
+
+
+class TestPocketLlamaindexUseProfile(IsolatedAsyncioTestCase):
+
+    async def asyncSetUp(self):
+        config.public_server_port = "https"
+        config.public_hostname = "localhost"
+        config.public_server_port = 8001
+        config.internal_server_port = 8000
+        config.enable_local_callback_proxy = True
+
+        self.pocket = PocketLlamaindex(
+            tools=[
+                from_git("https://github.com/vessl-ai/hyperawesometools", "main",
+                         "managed-tools/none/simple-echo-tool"),
+                self.add,
+                self.sub_pydantic_args
+            ],
+            use_profile=True
+        )
+
+        self.llm = OpenAI(model="gpt-4o", api_key=secret["OPENAI_API_KEY"])
+
+    async def asyncTearDown(self):
+        self.pocket._teardown_server()
+
+    async def test_agent_use_profile(self):
+        # given
+        agent = OpenAIAgent.from_tools(
+            tools=self.pocket.get_tools(),
+            llm=OpenAI(model="gpt-4o", api_key=secret["OPENAI_API_KEY"]),
+            verbose=True
+        )
+
+        # when
+        agent.query("add 1, 2")
+        agent.query("sub 1, 2")
+        agent.query("echo 'hello world'")
+
+    async def test_function_tool(self):
+        # when
+        response = self.llm.chat_with_tools(user_msg="add 1, 2", tools=self.pocket.get_tools(), verbose=True)
+        message = response.message
+        tool_calls = message.additional_kwargs["tool_calls"]
+
+        tool_name = tool_calls[0].function.name
+        args = tool_calls[0].function.arguments
+        args = json.loads(args)
+
+        result = await self.pocket.ainvoke(tool_name=tool_name, **args)
+
+        # then
+        self.assertEqual(tool_name, "add")
+        self.assertEqual(args["body"], {
+            "a": 1,
+            "b": 2,
+        })
+        self.assertEqual(result, '3')
+
+    async def test_pydantic_function_tool(self):
+        # when
+        response = self.llm.chat_with_tools(user_msg="sub 1, 2", tools=self.pocket.get_tools(), verbose=True)
+        tool_calls = response.message.additional_kwargs["tool_calls"]
+
+        tool_name = tool_calls[0].function.name
+        args = tool_calls[0].function.arguments
+        args = json.loads(args)
+
+        result = await self.pocket.ainvoke(tool_name=tool_name, **args)
+
+        # then
+        self.assertEqual(tool_name, "sub_pydantic_args")
+        self.assertEqual(args["body"], {
+            "a": {"first": 1},
+            "b": {"second": 2},
+        })
+        self.assertEqual(result, '-1')
+
+    async def test_wasm_tool(self):
+        # when
+        response = self.llm.chat_with_tools(user_msg="echo 'hello world'", tools=self.pocket.get_tools(), verbose=True)
+        tool_calls = response.message.additional_kwargs["tool_calls"]
+
+        tool_name = tool_calls[0].function.name
+        args = tool_calls[0].function.arguments
+        args = json.loads(args)
+
+        result = await self.pocket.ainvoke(tool_name=tool_name, **args)
+        output = ast.literal_eval(result)
+
+        # then
+        self.assertEqual(tool_name, "simple_echo_text")
+        self.assertEqual(args["body"], {
+            "text": "hello world"
+        })
+        self.assertTrue(output["stdout"].startswith("echo message : hello world"))
+
+    @staticmethod
+    def add(a: int, b: int) -> int:
+        """
+        Add two numbers
+
+        Args:
+            a(int): first number
+            b(int): second number
+
+        """
+
+        return a + b
+
+    class FirstNumber(BaseModel):
+        first: int
+
+    class SecondNumber(BaseModel):
+        second: int
+
+    @staticmethod
+    def sub_pydantic_args(a: FirstNumber, b: SecondNumber):
+        """
+        sub two numbers
+
+        Args:
+            a(FirstNumber): first number
+            b(SecondNumber): second number
+        """
+        return a.first - b.second