hyperpocket-llamaindex 0.0.1__tar.gz

hyperpocket_llamaindex-0.0.1/PKG-INFO

@@ -0,0 +1,59 @@
+Metadata-Version: 2.1
+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)
+Description-Content-Type: text/markdown
+
+## Anthropic extensions
+
+### Get Pocket Tools
+
+```python
+import hyperpocket as pk
+from pocket_llamaindex import PocketLlamaindex
+
+# 
+pocket = PocketLlamaindex(tools=[
+    *pk.curated_tools.SLACK,  # SLACK = [slack_get_message, slack_post_message, ..]
+    *pk.curated_tools.LINEAR,
+    "https://github.com/my-org/some-awesome-tool"]
+)
+
+# get tools from pocket
+tools = pocket.get_tools()
+```
+
+
+### Examples
+
+```python
+from llama_index.core.agent import FunctionCallingAgent
+from llama_index.llms.openai import OpenAI
+
+import hyperpocket as pk
+from pocket_llamaindex import PocketLlamaindex
+
+pocket = PocketLlamaindex(tools=[
+    *pk.curated_tools.SLACK,  # SLACK = [slack_get_message, slack_post_message, ..]
+    *pk.curated_tools.LINEAR,
+    "https://github.com/my-org/some-awesome-tool"]
+)
+
+# get tools from pocket
+tools = pocket.get_tools()
+
+llm = OpenAI()
+# pass tools get by pocket to an argument
+agent = FunctionCallingAgent.from_tools(
+    tools=tools, llm=llm, verbose=True
+)
+```

hyperpocket_llamaindex-0.0.1/README.md

@@ -0,0 +1,44 @@
+## Anthropic extensions
+
+### Get Pocket Tools
+
+```python
+import hyperpocket as pk
+from pocket_llamaindex import PocketLlamaindex
+
+# 
+pocket = PocketLlamaindex(tools=[
+    *pk.curated_tools.SLACK,  # SLACK = [slack_get_message, slack_post_message, ..]
+    *pk.curated_tools.LINEAR,
+    "https://github.com/my-org/some-awesome-tool"]
+)
+
+# get tools from pocket
+tools = pocket.get_tools()
+```
+
+
+### Examples
+
+```python
+from llama_index.core.agent import FunctionCallingAgent
+from llama_index.llms.openai import OpenAI
+
+import hyperpocket as pk
+from pocket_llamaindex import PocketLlamaindex
+
+pocket = PocketLlamaindex(tools=[
+    *pk.curated_tools.SLACK,  # SLACK = [slack_get_message, slack_post_message, ..]
+    *pk.curated_tools.LINEAR,
+    "https://github.com/my-org/some-awesome-tool"]
+)
+
+# get tools from pocket
+tools = pocket.get_tools()
+
+llm = OpenAI()
+# pass tools get by pocket to an argument
+agent = FunctionCallingAgent.from_tools(
+    tools=tools, llm=llm, verbose=True
+)
+```

hyperpocket_llamaindex-0.0.1/hyperpocket_llamaindex/__init__.py

@@ -0,0 +1,3 @@
+from hyperpocket_llamaindex.pocket_llamaindex import PocketLlamaindex
+
+__all__ = ["PocketLlamaindex"]

hyperpocket_llamaindex-0.0.1/hyperpocket_llamaindex/pocket_llamaindex.py

@@ -0,0 +1,44 @@
+from typing import List, Any
+
+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) -> List[BaseTool]:
+        tools = [self.get_tool(pk) for pk in self.tools.values()]
+        return tools
+
+    def get_tool(self, pocket_tool: Tool) -> BaseTool:
+        def _invoke(body: Any, thread_id: str = 'default', profile: str = 'default', **kwargs) -> str:
+            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(body: Any, thread_id: str = 'default', profile: str = 'default', **kwargs) -> str:
+            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(
+                description=pocket_tool.description,
+                name=pocket_tool.name,
+                fn_schema=pocket_tool.schema_model(),
+            )
+        )

hyperpocket_llamaindex-0.0.1/pyproject.toml

@@ -0,0 +1,16 @@
+[tool.poetry]
+name = "hyperpocket-llamaindex"
+version = "0.0.1"
+description = ""
+authors = ["moon <moon@vessl.ai>"]
+readme = "README.md"
+
+[tool.poetry.dependencies]
+python = "^3.11"
+llama-index = "^0.12.5"
+hyperpocket = "0.0.3"
+
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"