llama-deploy-appserver 0.3.10__py3-none-any.whl → 0.3.11__py3-none-any.whl

llama_deploy/appserver/workflow_store/agent_data_store.py

@@ -1,6 +1,7 @@
+import asyncio
 import logging
 import os
-from typing import List
+from typing import Any, List
 
 from llama_cloud.client import AsyncLlamaCloud, httpx
 from llama_cloud_services.beta.agent_data import AsyncAgentDataClient
@@ -49,19 +50,7 @@ class AgentDataStore(AbstractWorkflowStore):
 
     @override
     async def query(self, query: HandlerQuery) -> List[PersistentHandler]:
-        filters = {}
-        if query.handler_id_in is not None:
-            filters["handler_id"] = {
-                "includes": query.handler_id_in,
-            }
-        if query.workflow_name_in is not None:
-            filters["workflow_name"] = {
-                "includes": query.workflow_name_in,
-            }
-        if query.status_in is not None:
-            filters["status"] = {
-                "includes": query.status_in,
-            }
+        filters = self._build_filters(query)
         results = await self.client.search(
             filter=filters,
             page_size=1000,
@@ -85,6 +74,15 @@ class AgentDataStore(AbstractWorkflowStore):
                     data=handler,
                 )
 
+    @override
+    async def delete(self, handler: HandlerQuery) -> int:
+        filters = self._build_filters(handler)
+        results = await self.client.search(filter=filters, page_size=1000)
+        await asyncio.gather(
+            *[self.client.delete_item(item_id=x.id) for x in results.items]
+        )
+        return len(results.items)
+
     async def _get_item_id(self, handler: PersistentHandler) -> str | None:
         cached_id = self.cache.get(handler.handler_id, None)
         if cached_id is not None:
@@ -98,3 +96,19 @@ class AgentDataStore(AbstractWorkflowStore):
         id = results.items[0].id
         self.cache.set(handler.handler_id, id)
         return id
+
+    def _build_filters(self, query: HandlerQuery) -> dict[str, Any]:
+        filters = {}
+        if query.handler_id_in is not None:
+            filters["handler_id"] = {
+                "includes": query.handler_id_in,
+            }
+        if query.workflow_name_in is not None:
+            filters["workflow_name"] = {
+                "includes": query.workflow_name_in,
+            }
+        if query.status_in is not None:
+            filters["status"] = {
+                "includes": query.status_in,
+            }
+        return filters

{llama_deploy_appserver-0.3.10.dist-info → llama_deploy_appserver-0.3.11.dist-info}/METADATA

@@ -1,15 +1,15 @@
 Metadata-Version: 2.3
 Name: llama-deploy-appserver
-Version: 0.3.10
+Version: 0.3.11
 Summary: Application server components for LlamaDeploy
 Author: Massimiliano Pippi, Adrian Lyjak
 Author-email: Massimiliano Pippi <mpippi@gmail.com>, Adrian Lyjak <adrianlyjak@gmail.com>
 License: MIT
-Requires-Dist: llama-index-workflows[server]>=2.5.0
+Requires-Dist: llama-index-workflows[server]>=2.6.0
 Requires-Dist: pydantic-settings>=2.10.1
 Requires-Dist: fastapi>=0.100.0
 Requires-Dist: websockets>=12.0
-Requires-Dist: llama-deploy-core>=0.3.10,<0.4.0
+Requires-Dist: llama-deploy-core>=0.3.11,<0.4.0
 Requires-Dist: httpx>=0.24.0,<1.0.0
 Requires-Dist: prometheus-fastapi-instrumentator>=7.1.0
 Requires-Dist: packaging>=25.0

{llama_deploy_appserver-0.3.10.dist-info → llama_deploy_appserver-0.3.11.dist-info}/RECORD

@@ -16,9 +16,9 @@ llama_deploy/appserver/settings.py,sha256=279dad9d80f4b54215cb8073bc46ee2beebfbc
 llama_deploy/appserver/stats.py,sha256=1f3989f6705a6de3e4d61ee8cdd189fbe04a2c53ec5e720b2e5168acc331427f,691
 llama_deploy/appserver/types.py,sha256=4edc991aafb6b8497f068d12387455df292da3ff8440223637641ab1632553ec,2133
 llama_deploy/appserver/workflow_loader.py,sha256=aa06b3cdc9cdbf7e4c58c1b5bbd28025f56869a0065311aac8e98e729e70d38a,14342
-llama_deploy/appserver/workflow_store/agent_data_store.py,sha256=7b8d1b8cb6f741ff631d668fc955ca76a82e8da0bf8a27ee3bc9a8ef71123701,3594
+llama_deploy/appserver/workflow_store/agent_data_store.py,sha256=25f9bd0a78e73e2f096afa40beb9e0b664c37ae31394ae01d29b1cec88ba32a1,4097
 llama_deploy/appserver/workflow_store/keyed_lock.py,sha256=bb1504d9de09d51a8f60721cc77b14d4051ac5a897ace6f9d9cba494f068465e,950
 llama_deploy/appserver/workflow_store/lru_cache.py,sha256=7511231b6aba81ea96044cf644cd9c1f11d78190b7b7f578b1b5a55e2c218f9f,1323
-llama_deploy_appserver-0.3.10.dist-info/WHEEL,sha256=66530aef82d5020ef5af27ae0123c71abb9261377c5bc519376c671346b12918,79
-llama_deploy_appserver-0.3.10.dist-info/METADATA,sha256=66b05451824cc94c35c2a79b6de86a9d394ab47122032d7b06315b4a988f6c97,1009
-llama_deploy_appserver-0.3.10.dist-info/RECORD,,
+llama_deploy_appserver-0.3.11.dist-info/WHEEL,sha256=66530aef82d5020ef5af27ae0123c71abb9261377c5bc519376c671346b12918,79
+llama_deploy_appserver-0.3.11.dist-info/METADATA,sha256=4b72f8c3728bfd119e9949ff0a4fb10162e1e345ac9445a9a4817e62479a6247,1009
+llama_deploy_appserver-0.3.11.dist-info/RECORD,,