llamactl 0.3.0a10__py3-none-any.whl → 0.3.0a12__py3-none-any.whl

llama_deploy/cli/client.py

@@ -1,267 +1,29 @@
-import contextlib
-from typing import Iterator, List
-
-import httpx
-from llama_deploy.core.schema.base import LogEvent
-from llama_deploy.core.schema.deployments import (
-    DeploymentCreate,
-    DeploymentResponse,
-    DeploymentsListResponse,
-    DeploymentUpdate,
-)
-from llama_deploy.core.schema.git_validation import (
-    RepositoryValidationRequest,
-    RepositoryValidationResponse,
-)
-from llama_deploy.core.schema.projects import ProjectsListResponse, ProjectSummary
-from rich.console import Console
-
-from .config import config_manager
-
-
-class ClientError(Exception):
-    """Base class for client errors."""
-
-    def __init__(self, message: str) -> None:
-        super().__init__(message)
-
-
-class BaseClient:
-    def __init__(self, base_url: str, console: Console) -> None:
-        self.base_url = base_url.rstrip("/")
-        self.console = console
-        self.client = httpx.Client(
-            base_url=self.base_url,
-            event_hooks={"response": [self._handle_response]},
-        )
-        self.hookless_client = httpx.Client(base_url=self.base_url)
-
-    def _handle_response(self, response: httpx.Response) -> None:
-        try:
-            response.raise_for_status()
-        except httpx.HTTPStatusError as e:
-            try:
-                response.read()
-                error_data = e.response.json()
-                if isinstance(error_data, dict) and "detail" in error_data:
-                    error_message = error_data["detail"]
-                else:
-                    error_message = str(error_data)
-            except (ValueError, KeyError):
-                error_message = e.response.text
-            raise ClientError(f"HTTP {e.response.status_code}: {error_message}") from e
-        except httpx.RequestError as e:
-            raise ClientError(f"Request failed: {e}") from e
-
-
-class ControlPlaneClient(BaseClient):
-    """Unscoped client for non-project endpoints."""
-
-    def health_check(self) -> dict:
-        response = self.client.get("/health")
-        return response.json()
-
-    def server_version(self) -> dict:
-        response = self.client.get("/version")
-        return response.json()
-
-    def list_projects(self) -> List[ProjectSummary]:
-        response = self.client.get("/api/v1beta1/deployments/list-projects")
-        projects_response = ProjectsListResponse.model_validate(response.json())
-        return [project for project in projects_response.projects]
-
-
-class ProjectClient(BaseClient):
-    """Project-scoped client for deployment operations."""
-
-    def __init__(
-        self,
-        base_url: str | None = None,
-        project_id: str | None = None,
-        console: Console | None = None,
-    ) -> None:
-        # Allow default construction using active profile (for tests and convenience)
-        if base_url is None or project_id is None:
-            profile = config_manager.get_current_profile()
-            if not profile:
-                # Match previous behavior for missing profiles
-                (console or Console()).print(
-                    "\n[bold red]No profile configured![/bold red]"
-                )
-                (console or Console()).print("\nTo get started, create a profile with:")
-                (console or Console()).print("[cyan]llamactl profile create[/cyan]")
-                raise SystemExit(1)
-            base_url = base_url or profile.api_url or ""
-            project_id = project_id or profile.active_project_id
-        if not base_url:
-            raise ValueError("API URL is required")
-        if not project_id:
-            raise ValueError("Project ID is required")
-        resolved_console = console or Console()
-        super().__init__(base_url, resolved_console)
-        self.project_id = project_id
-
-    def list_deployments(self) -> List[DeploymentResponse]:
-        response = self.client.get(
-            "/api/v1beta1/deployments",
-            params={"project_id": self.project_id},
-        )
-        deployments_response = DeploymentsListResponse.model_validate(response.json())
-        return [deployment for deployment in deployments_response.deployments]
-
-    def get_deployment(
-        self, deployment_id: str, include_events: bool = False
-    ) -> DeploymentResponse:
-        response = self.client.get(
-            f"/api/v1beta1/deployments/{deployment_id}",
-            params={"project_id": self.project_id, "include_events": include_events},
-        )
-        return DeploymentResponse.model_validate(response.json())
-
-    def create_deployment(
-        self, deployment_data: DeploymentCreate
-    ) -> DeploymentResponse:
-        response = self.client.post(
-            "/api/v1beta1/deployments",
-            params={"project_id": self.project_id},
-            json=deployment_data.model_dump(exclude_none=True),
-        )
-        return DeploymentResponse.model_validate(response.json())
-
-    def delete_deployment(self, deployment_id: str) -> None:
-        self.client.delete(
-            f"/api/v1beta1/deployments/{deployment_id}",
-            params={"project_id": self.project_id},
-        )
-
-    def update_deployment(
-        self,
-        deployment_id: str,
-        update_data: DeploymentUpdate,
-    ) -> DeploymentResponse:
-        response = self.client.patch(
-            f"/api/v1beta1/deployments/{deployment_id}",
-            params={"project_id": self.project_id},
-            json=update_data.model_dump(),
-        )
-        return DeploymentResponse.model_validate(response.json())
-
-    def validate_repository(
-        self,
-        repo_url: str,
-        deployment_id: str | None = None,
-        pat: str | None = None,
-    ) -> RepositoryValidationResponse:
-        response = self.client.post(
-            "/api/v1beta1/deployments/validate-repository",
-            params={"project_id": self.project_id},
-            json=RepositoryValidationRequest(
-                repository_url=repo_url,
-                deployment_id=deployment_id,
-                pat=pat,
-            ).model_dump(),
-        )
-        return RepositoryValidationResponse.model_validate(response.json())
-
-    def stream_deployment_logs(
-        self,
-        deployment_id: str,
-        *,
-        include_init_containers: bool = False,
-        since_seconds: int | None = None,
-        tail_lines: int | None = None,
-    ) -> tuple["Closer", Iterator[LogEvent]]:
-        """Stream logs as LogEvent items from the control plane using SSE.
-
-        This yields `LogEvent` models until the stream ends (e.g. rollout).
-        """
-        # Use a separate client without response hooks so we don't consume the stream
-
-        params = {
-            "project_id": self.project_id,
-            "include_init_containers": include_init_containers,
-        }
-        if since_seconds is not None:
-            params["since_seconds"] = since_seconds
-        if tail_lines is not None:
-            params["tail_lines"] = tail_lines
-
-        url = f"/api/v1beta1/deployments/{deployment_id}/logs"
-        headers = {"Accept": "text/event-stream"}
-
-        stack = contextlib.ExitStack()
-        response = stack.enter_context(
-            self.hookless_client.stream(
-                "GET", url, params=params, headers=headers, timeout=None
-            )
-        )
-        try:
-            response.raise_for_status()
-        except Exception:
-            stack.close()
-            raise
-
-        return stack.close, _iterate_log_stream(response, stack.close)
-
-
-def _iterate_log_stream(
-    response: httpx.Response, closer: "Closer"
-) -> Iterator[LogEvent]:
-    event_name: str | None = None
-    data_lines: list[str] = []
-
-    try:
-        for line in response.iter_lines():
-            if line is None:
-                continue
-            line = line.decode() if isinstance(line, (bytes, bytearray)) else line
-            print("got line", line)
-            if line.startswith("event:"):
-                event_name = line[len("event:") :].strip()
-            elif line.startswith("data:"):
-                data_lines.append(line[len("data:") :].lstrip())
-            elif line.strip() == "":
-                if event_name == "log" and data_lines:
-                    data_str = "\n".join(data_lines)
-                    try:
-                        yield LogEvent.model_validate_json(data_str)
-                        print("yielded log event", data_str)
-                    except Exception:
-                        # If parsing fails, skip malformed event
-                        pass
-                # reset for next event
-                event_name = None
-                data_lines = []
-    finally:
-        try:
-            closer()
-        except Exception:
-            pass
+from llama_deploy.cli.config import config_manager
+from llama_deploy.core.client.manage_client import ControlPlaneClient, ProjectClient
+from rich import print as rprint
 
 
 def get_control_plane_client(base_url: str | None = None) -> ControlPlaneClient:
-    console = Console()
     profile = config_manager.get_current_profile()
     if not profile and not base_url:
-        console.print("\n[bold red]No profile configured![/bold red]")
-        console.print("\nTo get started, create a profile with:")
-        console.print("[cyan]llamactl profile create[/cyan]")
+        rprint("\n[bold red]No profile configured![/bold red]")
+        rprint("\nTo get started, create a profile with:")
+        rprint("[cyan]llamactl profile create[/cyan]")
         raise SystemExit(1)
     resolved_base_url = (base_url or (profile.api_url if profile else "")).rstrip("/")
     if not resolved_base_url:
         raise ValueError("API URL is required")
-    return ControlPlaneClient(resolved_base_url, console)
+    return ControlPlaneClient(resolved_base_url)
 
 
 def get_project_client(
     base_url: str | None = None, project_id: str | None = None
 ) -> ProjectClient:
-    console = Console()
     profile = config_manager.get_current_profile()
     if not profile:
-        console.print("\n[bold red]No profile configured![/bold red]")
-        console.print("\nTo get started, create a profile with:")
-        console.print("[cyan]llamactl profile create[/cyan]")
+        rprint("\n[bold red]No profile configured![/bold red]")
+        rprint("\nTo get started, create a profile with:")
+        rprint("[cyan]llamactl profile create[/cyan]")
         raise SystemExit(1)
     resolved_base_url = (base_url or profile.api_url or "").rstrip("/")
     if not resolved_base_url:
@@ -269,7 +31,4 @@ def get_project_client(
     resolved_project_id = project_id or profile.active_project_id
     if not resolved_project_id:
         raise ValueError("Project ID is required")
-    return ProjectClient(resolved_base_url, resolved_project_id, console)
-
-
-type Closer = callable[tuple[()], None]
+    return ProjectClient(resolved_base_url, resolved_project_id)

llama_deploy/cli/textual/deployment_monitor.py

@@ -9,8 +9,8 @@ import time
 from pathlib import Path
 from typing import Iterator
 
-from llama_deploy.cli.client import Closer
 from llama_deploy.cli.client import get_project_client as get_client
+from llama_deploy.core.client.manage_client import Closer
 from llama_deploy.core.schema.base import LogEvent
 from llama_deploy.core.schema.deployments import DeploymentResponse
 from rich.text import Text

{llamactl-0.3.0a10.dist-info → llamactl-0.3.0a12.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.3
 Name: llamactl
-Version: 0.3.0a10
+Version: 0.3.0a12
 Summary: A command-line interface for managing LlamaDeploy projects and deployments
 Author: Adrian Lyjak
 Author-email: Adrian Lyjak <adrianlyjak@gmail.com>
 License: MIT
-Requires-Dist: llama-deploy-core>=0.3.0a10,<0.4.0
-Requires-Dist: llama-deploy-appserver>=0.3.0a10,<0.4.0
+Requires-Dist: llama-deploy-core[client]>=0.3.0a12,<0.4.0
+Requires-Dist: llama-deploy-appserver>=0.3.0a12,<0.4.0
 Requires-Dist: httpx>=0.24.0
 Requires-Dist: rich>=13.0.0
 Requires-Dist: questionary>=2.0.0
@@ -16,7 +16,7 @@ Requires-Dist: tenacity>=9.1.2
 Requires-Dist: textual>=5.3.0
 Requires-Dist: aiohttp>=3.12.14
 Requires-Dist: copier>=9.9.0
-Requires-Python: >=3.12, <4
+Requires-Python: >=3.11, <4
 Description-Content-Type: text/markdown
 
 # llamactl

{llamactl-0.3.0a10.dist-info → llamactl-0.3.0a12.dist-info}/RECORD

@@ -1,6 +1,6 @@
 llama_deploy/cli/__init__.py,sha256=274c45e48048bf60668ab564ae8e7c5e6daf1d7779005f87d07ce9fa7d04936c,422
 llama_deploy/cli/app.py,sha256=5200b4ac01b0ad0c405ce841fc01a12ed32f7b6474472f00a7d6c75fe274ea45,2324
-llama_deploy/cli/client.py,sha256=a648bfcd07b1d4e9ad1e7d9e3ef591066ea5f45bb6fe4b8ad2c495347563794a,10110
+llama_deploy/cli/client.py,sha256=f88cc30cf6df39fa68fb0aefe668634e4fd7216895d524b9af6f7c5727ba9da4,1510
 llama_deploy/cli/commands/aliased_group.py,sha256=bc41007c97b7b93981217dbd4d4591df2b6c9412a2d9ed045b0ec5655ed285f2,1066
 llama_deploy/cli/commands/deployment.py,sha256=7874f4a499ce1bfd6ae14833410cc75c4c954463d96064cfd045421358479d4c,8810
 llama_deploy/cli/commands/init.py,sha256=51b2de1e35ff34bc15c9dfec72fbad08aaf528c334df168896d36458a4e9401c,6307
@@ -14,14 +14,14 @@ llama_deploy/cli/options.py,sha256=e71d1a306e9e302b92ab55ace75f3a9273be267ae92a7
 llama_deploy/cli/py.typed,sha256=e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855,0
 llama_deploy/cli/textual/deployment_form.py,sha256=33d4f3f0741aeaf0072f5f91368b3c0b0456c72370b72964363240d790b3fa06,20929
 llama_deploy/cli/textual/deployment_help.py,sha256=d43e9ff29db71a842cf8b491545763d581ede3132b8af518c73af85a40950046,2464
-llama_deploy/cli/textual/deployment_monitor.py,sha256=4a36103310b22d68c11e2fe1e5dc0c2af847772d42c8f15b32a61261aeaa4104,16767
+llama_deploy/cli/textual/deployment_monitor.py,sha256=a2c4e48c5494f63a2b8dcc9ab1b11b4ce50d78ef94c0f54095448ce9d01fa59b,16782
 llama_deploy/cli/textual/git_validation.py,sha256=fcbe5477c99e8e669b31c563572d4894f61475ef7e968a59d9f172642d390cf7,13329
 llama_deploy/cli/textual/github_callback_server.py,sha256=dc74c510f8a98ef6ffaab0f6d11c7ea86ee77ca5adbc7725a2a29112bae24191,7556
 llama_deploy/cli/textual/llama_loader.py,sha256=33cb32a46dd40bcf889c553e44f2672c410e26bd1d4b17aa6cca6d0a5d59c2c4,1468
 llama_deploy/cli/textual/profile_form.py,sha256=747644895774e7416620d2071f6f054b06ec8e398ac0e7649386caa2a83fe2aa,5995
 llama_deploy/cli/textual/secrets_form.py,sha256=a43fbd81aad034d0d60906bfd917c107f9ace414648b0f63ac0b29eeba4050db,7061
 llama_deploy/cli/textual/styles.tcss,sha256=536cec7627d2a16dd03bf25bb9b6e4d53f1e0d18272b07ec0dc3bf76b0a7c2e0,3056
-llamactl-0.3.0a10.dist-info/WHEEL,sha256=66530aef82d5020ef5af27ae0123c71abb9261377c5bc519376c671346b12918,79
-llamactl-0.3.0a10.dist-info/entry_points.txt,sha256=b67e1eb64305058751a651a80f2d2268b5f7046732268421e796f64d4697f83c,52
-llamactl-0.3.0a10.dist-info/METADATA,sha256=64f72086ae8fbbce81d6af9beff116a4a40cb5dce008dc548a656d54ec436c15,3169
-llamactl-0.3.0a10.dist-info/RECORD,,
+llamactl-0.3.0a12.dist-info/WHEEL,sha256=66530aef82d5020ef5af27ae0123c71abb9261377c5bc519376c671346b12918,79
+llamactl-0.3.0a12.dist-info/entry_points.txt,sha256=b67e1eb64305058751a651a80f2d2268b5f7046732268421e796f64d4697f83c,52
+llamactl-0.3.0a12.dist-info/METADATA,sha256=9211df91592467022d69271234445d3cfe20593a0dc2961f7a47fb8fdbc19185,3177
+llamactl-0.3.0a12.dist-info/RECORD,,