llama-deploy-core 0.3.11__tar.gz → 0.3.13__tar.gz

{llama_deploy_core-0.3.11 → llama_deploy_core-0.3.13}/PKG-INFO

@@ -1,12 +1,13 @@
 Metadata-Version: 2.3
 Name: llama-deploy-core
-Version: 0.3.11
+Version: 0.3.13
 Summary: Core models and schemas for LlamaDeploy
 License: MIT
 Requires-Dist: fastapi>=0.115.0
 Requires-Dist: overrides>=7.7.0
 Requires-Dist: pydantic>=2.0.0
 Requires-Dist: pyyaml>=6.0.2
+Requires-Dist: truststore>=0.10.4
 Requires-Dist: types-pyyaml>=6.0.12.20250822
 Requires-Dist: httpx>=0.24.0,<1.0.0 ; extra == 'client'
 Requires-Dist: fastapi>=0.115.0 ; extra == 'server'

{llama_deploy_core-0.3.11 → llama_deploy_core-0.3.13}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "llama-deploy-core"
-version = "0.3.11"
+version = "0.3.13"
 description = "Core models and schemas for LlamaDeploy"
 readme = "README.md"
 license = { text = "MIT" }
@@ -10,6 +10,7 @@ dependencies = [
     "overrides>=7.7.0",
     "pydantic>=2.0.0",
     "pyyaml>=6.0.2",
+    "truststore>=0.10.4",
     "types-pyyaml>=6.0.12.20250822",
 ]
 

{llama_deploy_core-0.3.11 → llama_deploy_core-0.3.13}/src/llama_deploy/core/client/manage_client.py

@@ -4,6 +4,7 @@ from contextlib import asynccontextmanager
 from typing import AsyncIterator, Callable, List
 
 import httpx
+from llama_deploy.core.client.ssl_util import get_httpx_verify_param
 from llama_deploy.core.schema import LogEvent
 from llama_deploy.core.schema.deployments import (
     DeploymentCreate,
@@ -30,13 +31,15 @@ class BaseClient:
         if api_key:
             headers["Authorization"] = f"Bearer {api_key}"
 
+        verify = get_httpx_verify_param()
         self.client = httpx.AsyncClient(
             base_url=self.base_url,
             headers=headers,
             auth=auth,
+            verify=verify,
         )
         self.hookless_client = httpx.AsyncClient(
-            base_url=self.base_url, headers=headers, auth=auth
+            base_url=self.base_url, headers=headers, auth=auth, verify=verify
         )
 
     async def aclose(self) -> None:

llama_deploy_core-0.3.13/src/llama_deploy/core/client/ssl_util.py

@@ -0,0 +1,32 @@
+"""Utility functions for SSL/TLS configuration with optional truststore support."""
+
+from __future__ import annotations
+
+import os
+import ssl
+from typing import Any
+
+import truststore
+
+
+def get_ssl_context() -> ssl.SSLContext | bool:
+    """Get SSL context for httpx clients.
+
+    Returns an SSL context using truststore if LLAMA_DEPLOY_USE_TRUSTSTORE is set,
+    otherwise returns True (default SSL verification).
+
+    Truststore allows Python to use the system certificate store, which is useful
+    for corporate environments with MITM proxies.
+    """
+    if os.getenv("LLAMA_DEPLOY_USE_TRUSTSTORE", "").lower() in ("1", "true", "yes"):
+        return truststore.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
+    return True
+
+
+def get_httpx_verify_param() -> Any:
+    """Get the verify parameter for httpx clients.
+
+    Returns an SSL context using truststore if configured, otherwise returns True.
+    This can be passed directly to httpx.Client/AsyncClient's verify parameter.
+    """
+    return get_ssl_context()