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

hyperpocket_llamaindex/pocket_llamaindex.py

@@ -3,11 +3,9 @@ from typing import List, Optional
 from pydantic import BaseModel
 
 try:
-    from llama_index.core.tools import FunctionTool, BaseTool, ToolMetadata
+    from llama_index.core.tools import BaseTool, FunctionTool, ToolMetadata
 except ImportError:
-    raise ImportError(
-        "You need to install llama-index to use pocket llamaindex"
-    )
+    raise ImportError("You need to install llama-index to use pocket llamaindex")
 
 from hyperpocket import Pocket
 from hyperpocket.tool import Tool
@@ -35,11 +33,16 @@ class PocketLlamaindex(Pocket):
             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)
+            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.'
+                say = f"{say}\n\nThe tool execution interrupted. Please talk to me to resume."
             return say
 
         async def _ainvoke(**kwargs) -> str:
@@ -55,11 +58,16 @@ class PocketLlamaindex(Pocket):
             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)
+            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.'
+                say = f"{say}\n\nThe tool execution interrupted. Please talk to me to resume."
             return say
 
         return FunctionTool.from_defaults(
@@ -69,5 +77,5 @@ class PocketLlamaindex(Pocket):
                 name=pocket_tool.name,
                 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.9.dist-info → hyperpocket_llamaindex-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyperpocket-llamaindex
-Version: 0.1.9
+Version: 0.2.0
 Author-email: Hyperpocket Team <hyperpocket@vessl.ai>
 Requires-Python: >=3.10
 Requires-Dist: hyperpocket>=0.0.3

hyperpocket_llamaindex-0.2.0.dist-info/RECORD

@@ -0,0 +1,5 @@
+hyperpocket_llamaindex/__init__.py,sha256=tZQ6ZnSlrc6zkUafpiYTbknXy7cac1fRLGJ6lBsGtvw,102
+hyperpocket_llamaindex/pocket_llamaindex.py,sha256=Q5pw_0ZVQWh3U2bkJeEr73v5goWEx20mhtx7blkPq98,2724
+hyperpocket_llamaindex-0.2.0.dist-info/METADATA,sha256=A2u5VjTdcIzLtXoGxWeQf4UIyYLYqs8NQ2rQsEYd9bI,1239
+hyperpocket_llamaindex-0.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+hyperpocket_llamaindex-0.2.0.dist-info/RECORD,,

hyperpocket_llamaindex-0.1.9.dist-info/RECORD

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