hyperpocket-llamaindex 0.0.1__py3-none-any.whl → 0.1.9__py3-none-any.whl

hyperpocket_llamaindex/pocket_llamaindex.py

@@ -1,4 +1,6 @@
-from typing import List, Any
+from typing import List, Optional
+
+from pydantic import BaseModel
 
 try:
     from llama_index.core.tools import FunctionTool, BaseTool, ToolMetadata
@@ -12,12 +14,27 @@ 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()]
+    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(body: Any, thread_id: str = 'default', profile: str = 'default', **kwargs) -> str:
+        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
@@ -25,7 +42,19 @@ class PocketLlamaindex(Pocket):
                 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:
+        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
@@ -37,8 +66,8 @@ class PocketLlamaindex(Pocket):
             fn=_invoke,
             async_fn=_ainvoke,
             tool_metadata=ToolMetadata(
-                description=pocket_tool.description,
                 name=pocket_tool.name,
-                fn_schema=pocket_tool.schema_model(),
+                description=pocket_tool.get_description(use_profile=self.use_profile),
+                fn_schema=pocket_tool.schema_model(use_profile=self.use_profile),
             )
         )

{hyperpocket_llamaindex-0.0.1.dist-info → hyperpocket_llamaindex-0.1.9.dist-info}/METADATA

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

@@ -0,0 +1,5 @@
+hyperpocket_llamaindex/__init__.py,sha256=tZQ6ZnSlrc6zkUafpiYTbknXy7cac1fRLGJ6lBsGtvw,102
+hyperpocket_llamaindex/pocket_llamaindex.py,sha256=36O1S64e_4XgRb-WDx-lCTaaDqMvy4M_XCj4UdIUdtQ,2668
+hyperpocket_llamaindex-0.1.9.dist-info/METADATA,sha256=qechw2kDiE_UDKVfdXsoNdlmQNkhz66kUBr8GyLe9-E,1239
+hyperpocket_llamaindex-0.1.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+hyperpocket_llamaindex-0.1.9.dist-info/RECORD,,

{hyperpocket_llamaindex-0.0.1.dist-info → hyperpocket_llamaindex-0.1.9.dist-info}/WHEEL

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

hyperpocket_llamaindex-0.0.1.dist-info/RECORD

@@ -1,5 +0,0 @@
-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,,