hyperpocket-llamaindex 0.0.1__py3-none-any.whl

hyperpocket_llamaindex/__init__.py

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

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.dist-info/METADATA

@@ -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.dist-info/RECORD

@@ -0,0 +1,5 @@
+hyperpocket_llamaindex/__init__.py,sha256=tZQ6ZnSlrc6zkUafpiYTbknXy7cac1fRLGJ6lBsGtvw,102
+hyperpocket_llamaindex/pocket_llamaindex.py,sha256=AGKrJ8lO1wZPig0gEriQiBqlZ-Gk-r0N_cqZBDaWw1g,1774
+hyperpocket_llamaindex-0.0.1.dist-info/METADATA,sha256=ClC4ekflbRsMAk7OGulxqL-7r3tst0sjPxhuK7k_jk0,1457
+hyperpocket_llamaindex-0.0.1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+hyperpocket_llamaindex-0.0.1.dist-info/RECORD,,

hyperpocket_llamaindex-0.0.1.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: poetry-core 1.9.1
+Root-Is-Purelib: true
+Tag: py3-none-any