llama-deploy-core 0.3.22__py3-none-any.whl → 0.3.24__py3-none-any.whl

llama_deploy/core/_compat.py

@@ -0,0 +1,31 @@
+from __future__ import annotations
+
+import logging
+import sys
+from collections.abc import Mapping
+from typing import Any, BinaryIO
+
+if sys.version_info >= (3, 11):
+    # Stdlib TOML parser (Python 3.11+)
+    import tomllib as _toml_backend  # type: ignore[import-not-found]
+else:
+    # Lightweight TOML backport for Python 3.10
+    import tomli as _toml_backend  # type: ignore[import-not-found]
+
+
+def get_logging_level_mapping() -> Mapping[str, int]:
+    """Return a mapping of log level names to their numeric values."""
+    if sys.version_info >= (3, 11):
+        mapping = logging.getLevelNamesMapping()
+        return {k: int(v) for k, v in mapping.items() if isinstance(v, int)}
+
+    return {
+        name: level
+        for name, level in logging._nameToLevel.items()  # type: ignore[attr-defined]
+        if isinstance(level, int)
+    }
+
+
+def load_toml_file(file_obj: BinaryIO) -> dict[str, Any]:
+    """Load TOML data from a binary file object in a version-agnostic way."""
+    return _toml_backend.load(file_obj)

llama_deploy/core/client/manage_client.py

@@ -1,9 +1,10 @@
 from __future__ import annotations
 
 from contextlib import asynccontextmanager
-from typing import AsyncIterator, Callable, List
+from typing import AsyncGenerator, AsyncIterator, Callable, List
 
 import httpx
+from httpx._types import PrimitiveData
 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 (
@@ -250,25 +251,29 @@ class ProjectClient(BaseClient):
         include_init_containers: bool = False,
         since_seconds: int | None = None,
         tail_lines: int | None = None,
-    ) -> AsyncIterator[LogEvent]:
+    ) -> AsyncGenerator[LogEvent, None]:
         """Stream logs as LogEvent items from the control plane using SSE.
 
         Yields `LogEvent` models until the stream ends (e.g., rollout completes).
         """
-        params: dict[str, object] = {
+        params_dict: dict[str, PrimitiveData] = {
             "project_id": self.project_id,
             "include_init_containers": include_init_containers,
         }
         if since_seconds is not None:
-            params["since_seconds"] = since_seconds
+            params_dict["since_seconds"] = since_seconds
         if tail_lines is not None:
-            params["tail_lines"] = tail_lines
+            params_dict["tail_lines"] = tail_lines
 
         url = f"/api/v1beta1/deployments/{deployment_id}/logs"
         headers = {"Accept": "text/event-stream"}
 
         async with self.hookless_client.stream(
-            "GET", url, params=params, headers=headers, timeout=None
+            "GET",
+            url,
+            params=httpx.QueryParams(params_dict),
+            headers=headers,
+            timeout=None,
         ) as response:
             _raise_for_status(response)
 

llama_deploy/core/deployment_config.py

@@ -1,11 +1,11 @@
 from __future__ import annotations
 
 import json
-import tomllib
 from pathlib import Path
 from typing import Any, TypeVar
 
 import yaml
+from llama_deploy.core._compat import load_toml_file
 from llama_deploy.core.git.git_util import get_git_root, is_git_repo
 from llama_deploy.core.path_util import validate_path_traversal
 from pydantic import BaseModel, ConfigDict, Field, ValidationError, model_validator
@@ -54,13 +54,13 @@ def read_deployment_config(source_root: Path, config_path: Path) -> "DeploymentC
     # local TOML format
     if local_toml_path.exists():
         with open(local_toml_path, "rb") as toml_file:
-            toml_data = tomllib.load(toml_file)
+            toml_data = load_toml_file(toml_file)
             if isinstance(toml_data, dict):
                 toml_config = DeploymentConfig.model_validate(toml_data)
     # pyproject.toml format
     elif pyproject_path.exists():
         with open(pyproject_path, "rb") as pyproject_file:
-            pyproject = tomllib.load(pyproject_file)
+            pyproject = load_toml_file(pyproject_file)
             tool = pyproject.get("tool", {})
             project_name: str | None = None
             project_metadata = pyproject.get("project", {})
@@ -179,7 +179,7 @@ class DeploymentConfig(BaseModel):
         description="If true, serving locally expects Llama Cloud access and will inject credentials when possible.",
     )
     app: str | None = Field(
-        None,
+        default=None,
         description="A full bundle of all workflows as an 'app'. \"path.to_import:app_name\"",
     )
     workflows: dict[str, str] = Field(
@@ -202,7 +202,7 @@ class DeploymentConfig(BaseModel):
         ),
     )
     ui: UIConfig | None = Field(
-        None,
+        default=None,
         description="The UI configuration.",
     )
 
@@ -238,12 +238,12 @@ class DeploymentConfig(BaseModel):
     def is_valid(self) -> bool:
         """Check if the config is valid."""
         try:
-            self.validate()
+            self.validate_config()
             return True
         except ValueError:
             return False
 
-    def validate(self) -> None:
+    def validate_config(self) -> None:
         """Validate the config."""
         if self.has_no_workflows():
             raise ValueError("Config must have at least one workflow.")
@@ -276,23 +276,23 @@ class UIConfig(BaseModel):
         description="The directory containing the UI, relative to the pyproject.toml directory",
     )
     build_output_dir: str | None = Field(
-        None,
+        default=None,
         description="The directory containing the built UI, relative to the pyproject.toml directory. Defaults to 'dist' relative to the ui_directory, if defined",
     )
     package_manager: str = Field(
-        DEFAULT_UI_PACKAGE_MANAGER,
+        default=DEFAULT_UI_PACKAGE_MANAGER,
         description=f"The package manager to use to build the UI. Defaults to '{DEFAULT_UI_PACKAGE_MANAGER}'",
     )
     build_command: str = Field(
-        DEFAULT_UI_BUILD_COMMAND,
+        default=DEFAULT_UI_BUILD_COMMAND,
         description=f"The npm script command to build the UI. Defaults to '{DEFAULT_UI_BUILD_COMMAND}' if not specified",
     )
     serve_command: str = Field(
-        DEFAULT_UI_SERVE_COMMAND,
+        default=DEFAULT_UI_SERVE_COMMAND,
         description=f"The command to serve the UI. Defaults to '{DEFAULT_UI_SERVE_COMMAND}' if not specified",
     )
     proxy_port: int = Field(
-        DEFAULT_UI_PROXY_PORT,
+        default=DEFAULT_UI_PROXY_PORT,
         description=f"The port to proxy the UI to. Defaults to '{DEFAULT_UI_PROXY_PORT}' if not specified",
     )
 
@@ -334,11 +334,11 @@ class ServiceSourceV0(BaseModel):
 class DerecatedService(BaseModel):
     """Configuration for a single service."""
 
-    source: ServiceSourceV0 | None = Field(None)
-    import_path: str | None = Field(None)
-    env: dict[str, str] | None = Field(None)
-    env_files: list[str] | None = Field(None)
-    python_dependencies: list[str] | None = Field(None)
+    source: ServiceSourceV0 | None = Field(default=None)
+    import_path: str | None = Field(default=None)
+    env: dict[str, str] | None = Field(default=None)
+    env_files: list[str] | None = Field(default=None)
+    python_dependencies: list[str] | None = Field(default=None)
 
     @model_validator(mode="before")
     @classmethod
@@ -372,7 +372,7 @@ class DeprecatedDeploymentConfig(BaseModel):
     model_config = ConfigDict(populate_by_name=True, extra="ignore")
 
     name: str
-    default_service: str | None = Field(None)
+    default_service: str | None = Field(default=None)
     services: dict[str, DerecatedService]
     ui: DerecatedService | None = None
 

llama_deploy/core/iter_utils.py

@@ -37,15 +37,15 @@ async def debounced_sorted_prefix(
     async for item in merged:
         if item == "__COMPLETE__":
             buffer.sort(key=key)
-            for item in buffer:
-                yield item
+            for buffered_item in buffer:
+                yield buffered_item
             buffer = []
         else:
             if debouncer.is_complete:
-                yield item
+                yield item  # type: ignore[misc]  # item is T after checking != "__COMPLETE__"
             else:
                 debouncer.extend_window()
-                buffer.append(item)
+                buffer.append(item)  # type: ignore[arg-type]  # item is T after checking != "__COMPLETE__"
 
 
 COMPLETE = Literal["__COMPLETE__"]
@@ -67,7 +67,7 @@ async def merge_generators(
         return
 
     active_generators: dict[int, AsyncGenerator[T, None]] = {
-        index: gen for index, gen in enumerate(generators)
+        index: generator for index, generator in enumerate(generators)
     }
 
     next_item_tasks: dict[int, asyncio.Task[T]] = {}
@@ -75,8 +75,8 @@ async def merge_generators(
     stopped_on_first_completion = False
 
     # Prime one pending task per generator to maintain fairness
-    for index, gen in active_generators.items():
-        next_item_tasks[index] = asyncio.create_task(anext(gen))
+    for index, generator in active_generators.items():
+        next_item_tasks[index] = asyncio.create_task(anext(generator))
 
     try:
         while next_item_tasks and exception_to_raise is None:
@@ -119,9 +119,13 @@ async def merge_generators(
                     next_item_tasks.pop(task_index, None)
                     yield value
                     # Schedule the next item fetch for this generator
-                    gen = active_generators.get(task_index)
-                    if gen is not None:
-                        next_item_tasks[task_index] = asyncio.create_task(anext(gen))
+                    active_gen: AsyncGenerator[T, None] | None = active_generators.get(
+                        task_index
+                    )
+                    if active_gen is not None:
+                        next_item_tasks[task_index] = asyncio.create_task(
+                            anext(active_gen)
+                        )
             # If we are configured to stop on first completion and observed one,
             # exit the outer loop to perform cleanup in the finally block.
             if stopped_on_first_completion:
@@ -168,7 +172,7 @@ class Debouncer:
         self.max_complete_time = self.start_time + self.max_window_seconds
         asyncio.create_task(self._loop())
 
-    async def _loop(self):
+    async def _loop(self) -> None:
         while not self.complete_signal.is_set():
             now = self.get_time()
             remaining = min(self.complete_time, self.max_complete_time) - now

llama_deploy/core/server/manage_api/_create_deployments_router.py

@@ -1,4 +1,5 @@
 import logging
+from collections.abc import AsyncGenerator
 from typing import Awaitable, Callable
 
 from fastapi import APIRouter, Depends, HTTPException, Request, Response, params
@@ -166,7 +167,7 @@ def create_v1beta1_deployments_router(
         include_init_containers: Annotated[bool, Query()] = False,
         since_seconds: Annotated[int | None, Query()] = None,
         tail_lines: Annotated[int | None, Query()] = None,
-    ):
+    ) -> StreamingResponse:
         """Stream logs for the latest ReplicaSet of a deployment.
 
         The stream ends when the latest ReplicaSet changes (e.g., a new rollout occurs).
@@ -181,7 +182,7 @@ def create_v1beta1_deployments_router(
                 tail_lines=tail_lines,
             )
 
-            async def sse_lines():
+            async def sse_lines() -> AsyncGenerator[str, None]:
                 async for data in inner:
                     yield "event: log\n"
                     yield f"data: {data.model_dump_json()}\n\n"

{llama_deploy_core-0.3.22.dist-info → llama_deploy_core-0.3.24.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: llama-deploy-core
-Version: 0.3.22
+Version: 0.3.24
 Summary: Core models and schemas for LlamaDeploy
 License: MIT
 Requires-Dist: fastapi>=0.115.0
@@ -9,9 +9,10 @@ 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: tomli>=2.0.1 ; python_full_version < '3.11'
 Requires-Dist: httpx>=0.24.0,<1.0.0 ; extra == 'client'
 Requires-Dist: fastapi>=0.115.0 ; extra == 'server'
-Requires-Python: >=3.11, <4
+Requires-Python: >=3.10, <4
 Provides-Extra: client
 Provides-Extra: server
 Description-Content-Type: text/markdown

{llama_deploy_core-0.3.22.dist-info → llama_deploy_core-0.3.24.dist-info}/RECORD

@@ -1,10 +1,11 @@
 llama_deploy/core/__init__.py,sha256=112612bf2e928c2e0310d6556bb13fc28c00db70297b90a8527486cd2562e408,43
-llama_deploy/core/client/manage_client.py,sha256=d33613cc3355896dc15d20ead713b7766fd900f43c5f4d547e8e8df9e29a660f,10366
+llama_deploy/core/_compat.py,sha256=c4b88be801b052a4b357673c0127115e54a9d37ea108c436b8fa3f31be4a4d73,1023
+llama_deploy/core/client/manage_client.py,sha256=81a44ad8f528b6628f447a369667e70d9f479829e71c69cedf21b81a4879e143,10523
 llama_deploy/core/client/ssl_util.py,sha256=b9743dc828fa27c18ba0867b1348662cdf0d855965c5a33db63505f23eef5d7b,1010
 llama_deploy/core/config.py,sha256=69bb0ea8ac169eaa4e808cd60a098b616bddd3145d26c6c35e56db38496b0e6a,35
-llama_deploy/core/deployment_config.py,sha256=69f8556fce84eddc033a7c881cb9d4d3e33661466074afbfb6b3913535679e92,16382
+llama_deploy/core/deployment_config.py,sha256=724fd2bc210ba11385c965b2f8882fe7e735894e75799778c1c06c8580dc0b9d,16542
 llama_deploy/core/git/git_util.py,sha256=40436142a955d39c95a5b526df57c32a6f31bbb6a421c090710c9c1c9d741373,11883
-llama_deploy/core/iter_utils.py,sha256=68ac9dbf09f58315ffcfec6bd01ed50b88a954b00458da9bc28a0535b79ba29e,7042
+llama_deploy/core/iter_utils.py,sha256=c73bb1a38d68793cdde895674aa2caef3e91373b9853006ba993fb0d490ba4f4,7385
 llama_deploy/core/path_util.py,sha256=14d50c0c337c8450ed46cafc88436027056b365a48370a69cdb76c88d7c26fd1,798
 llama_deploy/core/py.typed,sha256=e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855,0
 llama_deploy/core/schema/__init__.py,sha256=d466fb068415a890de01be1696237cbd62e5fb0d83a6b2976e80c4d29381cd29,910
@@ -15,9 +16,9 @@ llama_deploy/core/schema/projects.py,sha256=726f91e90ff8699c90861d9740819c44c3f0
 llama_deploy/core/schema/public.py,sha256=022129c8fc09192f5e503b0500ccf54d106f5712b9cf8ce84b3b1c37e186f930,147
 llama_deploy/core/server/manage_api/__init__.py,sha256=e477ccab59cfd084edbad46f209972a282e623eb314d0847a754a46a16361db5,457
 llama_deploy/core/server/manage_api/_abstract_deployments_service.py,sha256=4b55c9628a9374a68bb78b7f6ae362b7cf4d1c5fc13d35e5cac191130fc13731,5181
-llama_deploy/core/server/manage_api/_create_deployments_router.py,sha256=2db7d59a43e8690cf6d3d925940212228498f0a7c5728cfcf4253830d6fa1b52,7554
+llama_deploy/core/server/manage_api/_create_deployments_router.py,sha256=784e8e4285a5e209eddb8ea92e75404e22f96fdc8fae9c6d792b585f16465f77,7647
 llama_deploy/core/server/manage_api/_exceptions.py,sha256=ee71cd9c2354a665e6905cd9cc752d2d65f71f0b936d33fec3c1c5229c38accf,246
 llama_deploy/core/ui_build.py,sha256=290dafa951918e5593b9035570fa4c66791d7e5ea785bd372ad11e99e8283857,1514
-llama_deploy_core-0.3.22.dist-info/WHEEL,sha256=66530aef82d5020ef5af27ae0123c71abb9261377c5bc519376c671346b12918,79
-llama_deploy_core-0.3.22.dist-info/METADATA,sha256=f6c8a95f64b556d7efbf4a6da0ba4079f9eb513275653e613dd8270843cdc480,760
-llama_deploy_core-0.3.22.dist-info/RECORD,,
+llama_deploy_core-0.3.24.dist-info/WHEEL,sha256=66530aef82d5020ef5af27ae0123c71abb9261377c5bc519376c671346b12918,79
+llama_deploy_core-0.3.24.dist-info/METADATA,sha256=343c128d288d6710f57b5c006fdfbda346e583582c5c727da9e14d2140cd1732,819
+llama_deploy_core-0.3.24.dist-info/RECORD,,