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

hyperpocket_llamaindex-0.1.8/.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.8}/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.8
+Author-email: moon <moon@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.0.1 → hyperpocket_llamaindex-0.1.8}/hyperpocket_llamaindex/pocket_llamaindex.py

@@ -13,7 +13,7 @@ from hyperpocket.tool import Tool
 
 class PocketLlamaindex(Pocket):
     def get_tools(self) -> List[BaseTool]:
-        tools = [self.get_tool(pk) for pk in self.tools.values()]
+        tools = [self.get_tool(pk) for pk in self.core.tools.values()]
         return tools
 
     def get_tool(self, pocket_tool: Tool) -> BaseTool:

hyperpocket_llamaindex-0.1.8/pyproject.toml

@@ -0,0 +1,36 @@
+
+[project]
+name = "hyperpocket-llamaindex"
+version = "0.1.8"
+description = ""
+authors = [{ name = "moon", email = "moon@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.8/tests/test_pocket_llamaindex.py

@@ -0,0 +1,126 @@
+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 TestPocketLlamaindex(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/simple-echo-tool"),
+                self.add,
+                self.sub_pydantic_args
+            ],
+        )
+
+        self.agent = OpenAIAgent.from_tools(
+            tools=self.pocket.get_tools(),
+            llm=OpenAI(model="gpt-4o", api_key=secret["OPENAI_API_KEY"]),
+            verbose=True
+        )
+
+        self.llm = OpenAI(model="gpt-4o", api_key=secret["OPENAI_API_KEY"])
+
+    async def asyncTearDown(self):
+        self.pocket._teardown_server()
+
+    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)
+
+        # then
+        self.assertEqual(tool_name, "simple_echo_text")
+        self.assertEqual(args["body"], {
+            "text": "hello world"
+        })
+        self.assertTrue(result.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