llama-deploy-appserver 0.3.10__tar.gz → 0.3.11__tar.gz

{llama_deploy_appserver-0.3.10 → llama_deploy_appserver-0.3.11}/PKG-INFO

@@ -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 → llama_deploy_appserver-0.3.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "llama-deploy-appserver"
-version = "0.3.10"
+version = "0.3.11"
 description = "Application server components for LlamaDeploy"
 readme = "README.md"
 license = { text = "MIT" }
@@ -10,11 +10,11 @@ authors = [
 ]
 requires-python = ">=3.11, <4"
 dependencies = [
-    "llama-index-workflows[server]>=2.5.0",
+    "llama-index-workflows[server]>=2.6.0",
     "pydantic-settings>=2.10.1",
     "fastapi>=0.100.0",
     "websockets>=12.0",
-    "llama-deploy-core>=0.3.10,<0.4.0",
+    "llama-deploy-core>=0.3.11,<0.4.0",
     "httpx>=0.24.0,<1.0.0",
     "prometheus-fastapi-instrumentator>=7.1.0",
     "packaging>=25.0",

{llama_deploy_appserver-0.3.10 → llama_deploy_appserver-0.3.11}/src/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