hyperpocket-llamaindex 0.1.8__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]:
+    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.1.8.dist-info → hyperpocket_llamaindex-0.1.9.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: hyperpocket-llamaindex
-Version: 0.1.8
-Author-email: moon <moon@vessl.ai>
+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

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

@@ -1,5 +0,0 @@
-hyperpocket_llamaindex/__init__.py,sha256=tZQ6ZnSlrc6zkUafpiYTbknXy7cac1fRLGJ6lBsGtvw,102
-hyperpocket_llamaindex/pocket_llamaindex.py,sha256=91l-gPUDQUV8MATXZekRor9PL4sSdamWuCQhT7SkXh4,1779
-hyperpocket_llamaindex-0.1.8.dist-info/METADATA,sha256=MT4LEDDolZ8jMtO3XJltEz_RlSiVfR-clC5yNzecuSc,1220
-hyperpocket_llamaindex-0.1.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-hyperpocket_llamaindex-0.1.8.dist-info/RECORD,,