llama-deploy-core 0.3.6__tar.gz → 0.3.8__tar.gz

{llama_deploy_core-0.3.6 → llama_deploy_core-0.3.8}/PKG-INFO

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

{llama_deploy_core-0.3.6 → llama_deploy_core-0.3.8}/pyproject.toml

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

{llama_deploy_core-0.3.6 → llama_deploy_core-0.3.8}/src/llama_deploy/core/client/manage_client.py

@@ -68,16 +68,56 @@ class ControlPlaneClient(BaseClient):
 
     async def server_version(self) -> VersionResponse:
         response = await self.client.get("/api/v1beta1/deployments-public/version")
-        response.raise_for_status()
+        _raise_for_status(response)
         return VersionResponse.model_validate(response.json())
 
     async def list_projects(self) -> List[ProjectSummary]:
         response = await self.client.get("/api/v1beta1/deployments/list-projects")
-        response.raise_for_status()
+        _raise_for_status(response)
         projects_response = ProjectsListResponse.model_validate(response.json())
         return [project for project in projects_response.projects]
 
 
+def _raise_for_status(response: httpx.Response) -> None:
+    """
+    Custom raise for status that adds response body information to the error message, but still uses the httpx
+    error classes
+    """
+    try:
+        response.raise_for_status()
+    except httpx.HTTPStatusError as e:
+        body = _response_body_snippet(response, limit=250)
+        request_id = response.headers.get("x-request-id") or response.headers.get(
+            "x-correlation-id"
+        )
+        rid = f" [request id: {request_id}]" if request_id else ""
+        body_part = f" - {body}" if body else ""
+        raise httpx.HTTPStatusError(
+            f"HTTP {response.status_code} for url {response.url}{body_part}{rid}",
+            request=e.request or response.request,
+            response=e.response or response,
+        )
+
+
+def _response_body_snippet(response: httpx.Response, limit: int = 500) -> str:
+    try:
+        text = response.text
+        if not text:
+            # fallback attempt if body not read
+            try:
+                data = response.json()
+            except Exception:
+                data = None
+            if data is not None:
+                text = str(data)
+        text = (text or "").strip()
+        if len(text) > limit:
+            return text[: limit - 3] + "..."
+        return text
+    except Exception:
+        return ""
+
+
 class ProjectClient(BaseClient):
     """Project-scoped client for deployment operations."""
 
@@ -114,7 +154,7 @@ class ProjectClient(BaseClient):
             "/api/v1beta1/deployments",
             params={"project_id": self.project_id},
         )
-        response.raise_for_status()
+        _raise_for_status(response)
         deployments_response = DeploymentsListResponse.model_validate(response.json())
         return [deployment for deployment in deployments_response.deployments]
 
@@ -125,7 +165,7 @@ class ProjectClient(BaseClient):
             f"/api/v1beta1/deployments/{deployment_id}",
             params={"project_id": self.project_id, "include_events": include_events},
         )
-        response.raise_for_status()
+        _raise_for_status(response)
         return DeploymentResponse.model_validate(response.json())
 
     async def create_deployment(
@@ -136,7 +176,7 @@ class ProjectClient(BaseClient):
             params={"project_id": self.project_id},
             json=deployment_data.model_dump(exclude_none=True),
         )
-        response.raise_for_status()
+        _raise_for_status(response)
         return DeploymentResponse.model_validate(response.json())
 
     async def delete_deployment(self, deployment_id: str) -> None:
@@ -144,7 +184,7 @@ class ProjectClient(BaseClient):
             f"/api/v1beta1/deployments/{deployment_id}",
             params={"project_id": self.project_id},
         )
-        response.raise_for_status()
+        _raise_for_status(response)
 
     async def update_deployment(
         self,
@@ -156,7 +196,7 @@ class ProjectClient(BaseClient):
             params={"project_id": self.project_id},
             json=update_data.model_dump(),
         )
-        response.raise_for_status()
+        _raise_for_status(response)
         return DeploymentResponse.model_validate(response.json())
 
     async def validate_repository(
@@ -174,7 +214,7 @@ class ProjectClient(BaseClient):
                 pat=pat,
             ).model_dump(),
         )
-        response.raise_for_status()
+        _raise_for_status(response)
         return RepositoryValidationResponse.model_validate(response.json())
 
     async def stream_deployment_logs(
@@ -204,7 +244,7 @@ class ProjectClient(BaseClient):
         async with self.hookless_client.stream(
             "GET", url, params=params, headers=headers, timeout=None
         ) as response:
-            response.raise_for_status()
+            _raise_for_status(response)
 
             event_name: str | None = None
             data_lines: list[str] = []