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

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

@@ -1,15 +1,15 @@
 Metadata-Version: 2.3
 Name: llama-deploy-appserver
-Version: 0.3.10
+Version: 0.3.12
 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.12,<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.12}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "llama-deploy-appserver"
-version = "0.3.10"
+version = "0.3.12"
 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.12,<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.12}/src/llama_deploy/appserver/app.py

@@ -236,6 +236,7 @@ def start_server_in_target_venv(
     persistence: Literal["memory", "local", "cloud"] | None = None,
     local_persistence_path: str | None = None,
     cloud_persistence_name: str | None = None,
+    host: str | None = None,
 ) -> None:
     # Ensure settings reflect the intended working directory before computing paths
 
@@ -247,6 +248,7 @@ def start_server_in_target_venv(
         persistence=persistence,
         local_persistence_path=local_persistence_path,
         cloud_persistence_name=cloud_persistence_name,
+        host=host,
     )
     base_dir = cwd or Path.cwd()
     path = settings.resolved_config_parent.relative_to(base_dir)

{llama_deploy_appserver-0.3.10 → llama_deploy_appserver-0.3.12}/src/llama_deploy/appserver/deployment.py

@@ -121,7 +121,6 @@ class Deployment:
         for route in server.app.routes:
             # add routes directly rather than mounting, so that we can share a root (only one ASGI app can be mounted at a path)
             if isinstance(route, Route):
-                logger.info(f"Adding route {route.path} to app")
                 app.add_api_route(
                     f"/deployments/{config.name}{route.path}",
                     route.endpoint,

{llama_deploy_appserver-0.3.10 → llama_deploy_appserver-0.3.12}/src/llama_deploy/appserver/routers/ui_proxy.py

@@ -21,6 +21,7 @@ from llama_deploy.appserver.interrupts import (
     wait_or_abort,
 )
 from llama_deploy.appserver.settings import ApiserverSettings
+from llama_deploy.core.client.ssl_util import get_httpx_verify_param
 from llama_deploy.core.deployment_config import DeploymentConfig
 
 logger = logging.getLogger(__name__)
@@ -193,7 +194,7 @@ def create_ui_proxy_router(name: str, port: int) -> APIRouter:
         }
 
         try:
-            client = httpx.AsyncClient(timeout=None)
+            client = httpx.AsyncClient(timeout=None, verify=get_httpx_verify_param())
 
             req = client.build_request(
                 request.method,

{llama_deploy_appserver-0.3.10 → llama_deploy_appserver-0.3.12}/src/llama_deploy/appserver/settings.py

@@ -103,6 +103,7 @@ def configure_settings(
     persistence: Literal["memory", "local", "cloud"] | None = None,
     local_persistence_path: str | None = None,
     cloud_persistence_name: str | None = None,
+    host: str | None = None,
 ) -> None:
     if proxy_ui is not None:
         settings.proxy_ui = proxy_ui
@@ -131,3 +132,6 @@ def configure_settings(
         os.environ["LLAMA_DEPLOY_APISERVER_CLOUD_PERSISTENCE_NAME"] = (
             cloud_persistence_name
         )
+    if host is not None:
+        settings.host = host
+        os.environ["LLAMA_DEPLOY_APISERVER_HOST"] = host

{llama_deploy_appserver-0.3.10 → llama_deploy_appserver-0.3.12}/src/llama_deploy/appserver/workflow_store/agent_data_store.py

@@ -1,10 +1,12 @@
+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
 from llama_deploy.appserver.settings import ApiserverSettings
+from llama_deploy.core.client.ssl_util import get_httpx_verify_param
 from llama_deploy.core.deployment_config import DeploymentConfig
 from typing_extensions import override
 from workflows.server import AbstractWorkflowStore, HandlerQuery, PersistentHandler
@@ -40,6 +42,7 @@ class AgentDataStore(AbstractWorkflowStore):
                 token=os.getenv("LLAMA_CLOUD_API_KEY"),
                 httpx_client=httpx.AsyncClient(
                     headers={"Project-Id": project_id} if project_id else None,
+                    verify=get_httpx_verify_param(),
                 ),
             ),
         )
@@ -49,19 +52,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 +76,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 +98,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