llama-deploy-core 0.3.0a13__tar.gz → 0.3.0a14__tar.gz

{llama_deploy_core-0.3.0a13 → llama_deploy_core-0.3.0a14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: llama-deploy-core
-Version: 0.3.0a13
+Version: 0.3.0a14
 Summary: Core models and schemas for LlamaDeploy
 License: MIT
 Requires-Dist: fastapi>=0.115.0

{llama_deploy_core-0.3.0a13 → llama_deploy_core-0.3.0a14}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "llama-deploy-core"
-version = "0.3.0a13"
+version = "0.3.0a14"
 description = "Core models and schemas for LlamaDeploy"
 readme = "README.md"
 license = { text = "MIT" }

{llama_deploy_core-0.3.0a13 → llama_deploy_core-0.3.0a14}/src/llama_deploy/core/client/manage_client.py

@@ -1,3 +1,6 @@
+from __future__ import annotations
+
+from contextlib import asynccontextmanager
 from typing import AsyncIterator, Callable, List
 
 import httpx
@@ -70,6 +73,20 @@ class BaseClient:
 class ControlPlaneClient(BaseClient):
     """Unscoped client for non-project endpoints."""
 
+    @classmethod
+    @asynccontextmanager
+    async def ctx(
+        cls, base_url: str, api_key: str | None = None
+    ) -> AsyncIterator[ControlPlaneClient]:
+        client = cls(base_url, api_key)
+        try:
+            yield client
+        finally:
+            try:
+                await client.aclose()
+            except Exception:
+                pass
+
     def __init__(self, base_url: str, api_key: str | None = None) -> None:
         super().__init__(base_url, api_key)
 
@@ -86,6 +103,20 @@ class ControlPlaneClient(BaseClient):
 class ProjectClient(BaseClient):
     """Project-scoped client for deployment operations."""
 
+    @classmethod
+    @asynccontextmanager
+    async def ctx(
+        cls, base_url: str, project_id: str, api_key: str | None = None
+    ) -> AsyncIterator[ProjectClient]:
+        client = cls(base_url, project_id, api_key)
+        try:
+            yield client
+        finally:
+            try:
+                await client.aclose()
+            except Exception:
+                pass
+
     def __init__(
         self,
         base_url: str,

llama_deploy_core-0.3.0a14/src/llama_deploy/core/schema/projects.py

@@ -0,0 +1,28 @@
+from typing import Any
+
+from pydantic import model_validator
+
+from .base import Base
+
+
+class ProjectSummary(Base):
+    """Summary of a project with deployment count"""
+
+    project_id: str
+    project_name: str
+    deployment_count: int
+
+    @model_validator(mode="before")
+    @classmethod
+    def set_default_project_name(cls, data: Any) -> Any:
+        if isinstance(data, dict):
+            if "project_name" not in data or data.get("project_name") is None:
+                if "project_id" in data:
+                    data["project_name"] = data["project_id"]
+        return data
+
+
+class ProjectsListResponse(Base):
+    """Response model for listing projects with deployment counts"""
+
+    projects: list[ProjectSummary]

llama_deploy_core-0.3.0a14/src/llama_deploy/core/schema/public.py

@@ -0,0 +1,7 @@
+from .base import Base
+
+
+class VersionResponse(Base):
+    version: str
+    requires_auth: bool = False
+    min_llamactl_version: str | None = None

llama_deploy_core-0.3.0a13/src/llama_deploy/core/schema/projects.py

@@ -1,15 +0,0 @@
-from .base import Base
-
-
-class ProjectSummary(Base):
-    """Summary of a project with deployment count"""
-
-    project_id: str
-    project_name: str
-    deployment_count: int
-
-
-class ProjectsListResponse(Base):
-    """Response model for listing projects with deployment counts"""
-
-    projects: list[ProjectSummary]

llama_deploy_core-0.3.0a13/src/llama_deploy/core/schema/public.py

@@ -1,5 +0,0 @@
-from .base import Base
-
-
-class VersionResponse(Base):
-    version: str