mobileops-managed-agent 0.3.1__tar.gz → 0.3.2__tar.gz

{mobileops_managed_agent-0.3.1/src/mobileops_managed_agent.egg-info → mobileops_managed_agent-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mobileops-managed-agent
-Version: 0.3.1
+Version: 0.3.2
 Summary: Server-side SDK for the MobileOPS Managed Agent API.
 Project-URL: Homepage, https://github.com/InnoMindX/deepagent/tree/main/managed-agent-sdk/python
 Project-URL: Repository, https://github.com/InnoMindX/deepagent

{mobileops_managed_agent-0.3.1 → mobileops_managed_agent-0.3.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "mobileops-managed-agent"
-version = "0.3.1"
+version = "0.3.2"
 description = "Server-side SDK for the MobileOPS Managed Agent API."
 readme = "README.md"
 requires-python = ">=3.10"

{mobileops_managed_agent-0.3.1 → mobileops_managed_agent-0.3.2}/src/mobileops_managed_agent/__init__.py

@@ -65,6 +65,14 @@ from .vaults import (
 )
 
 __all__ = [
+    "ModelRetryEvent",
+    "ModelRetryTracker",
+    "MobileOpsArtifactsClient",
+    "MobileOpsStorageClient",
+    "Artifact",
+    "ArtifactFile",
+    "ArtifactJob",
+    "StorageUsage",
     "MobileOpsManagedAgentClient",
     "MobileOpsAgentClient",
     "MobileOpsFilesClient",
@@ -116,3 +124,14 @@ __all__ = [
 ]
 
 from .tool_permissions import validate_tool_permissions, TOOL_PERMISSIONS_CONFIG_KEY
+
+from .artifacts import (
+    MobileOpsArtifactsClient,
+    MobileOpsStorageClient,
+    Artifact,
+    ArtifactFile,
+    ArtifactJob,
+    StorageUsage,
+)
+
+from .model_retry import ModelRetryEvent, ModelRetryTracker

mobileops_managed_agent-0.3.2/src/mobileops_managed_agent/artifacts.py

@@ -0,0 +1,134 @@
+"""Persistent deliverables, independent of the sandbox and thread lifetime."""
+
+from __future__ import annotations
+from collections.abc import AsyncIterator
+from typing import Any, TypedDict, Literal
+from urllib.parse import quote, urlencode
+from .resources import MobileOpsResourcesClient, MobileOpsResourcesError
+
+
+class ArtifactFile(TypedDict):
+    path: str
+    size: int
+    sha256: str
+
+
+class Artifact(TypedDict):
+    id: str
+    thread_id: str
+    run_id: str
+    fingerprint: str
+    size_bytes: int
+    files: list[ArtifactFile]
+    created_at: str
+    expires_at: str
+
+
+class ArtifactJob(TypedDict):
+    id: str
+    thread_id: str
+    run_id: str
+    status: Literal["pending", "downloading", "saved", "unchanged", "failed", "cancelled"]
+    artifact_id: str | None
+    error_code: str | None
+    error_message: str | None
+    error_status: int | None
+    created_at: str
+    expires_at: str
+    finished_at: str | None
+
+
+class MobileOpsArtifactJobsClient:
+    def __init__(self, transport: MobileOpsResourcesClient, base: str):
+        self._transport = transport
+        self._base = base
+
+    async def list(self, *, thread_id: str | None = None, run_id: str | None = None, limit: int = 50, offset: int = 0) -> list[ArtifactJob]:
+        query = {"limit": str(limit), "offset": str(offset)}
+        if thread_id is not None:
+            query["thread_id"] = thread_id
+        if run_id is not None:
+            query["run_id"] = run_id
+        return (await self._transport._request_json("GET", self._base + "?" + urlencode(query)))["jobs"]
+
+    async def retrieve(self, job_id: str) -> ArtifactJob:
+        return (await self._transport._request_json("GET", self._base + "/" + quote(job_id, safe="")))["job"]
+
+
+class StorageUsage(TypedDict):
+    used_bytes: int
+    reserved_bytes: int
+    limit_bytes: int | None
+    default_limit_bytes: int
+    effective_limit_bytes: int
+
+
+class MobileOpsArtifactsClient:
+    def __init__(self, **options: Any):
+        self._transport = MobileOpsResourcesClient(**options)
+        self._base = options["api_url"].rstrip("/") + "/context-hub/v1/artifacts"
+        self.jobs = MobileOpsArtifactJobsClient(self._transport, options["api_url"].rstrip("/") + "/context-hub/v1/artifact-jobs")
+
+    async def list(
+        self,
+        *,
+        thread_id: str | None = None,
+        run_id: str | None = None,
+        limit: int = 50,
+        offset: int = 0,
+    ) -> list[Artifact]:
+        query = {"limit": str(limit), "offset": str(offset)}
+        if thread_id is not None:
+            query["thread_id"] = thread_id
+        if run_id is not None:
+            query["run_id"] = run_id
+        return (
+            await self._transport._request_json(
+                "GET", self._base + "?" + urlencode(query)
+            )
+        )["artifacts"]
+
+    async def retrieve(self, artifact_id: str) -> Artifact:
+        return (
+            await self._transport._request_json(
+                "GET", self._base + "/" + quote(artifact_id, safe="")
+            )
+        )["artifact"]
+
+    async def download(self, artifact_id: str) -> AsyncIterator[bytes]:
+        """Stream the ZIP without buffering the whole archive in the SDK."""
+        t = self._transport
+        async with t._client.stream(
+            "GET",
+            self._base + "/" + quote(artifact_id, safe="") + "/content",
+            headers=t._headers,
+        ) as response:
+            if response.is_error:
+                await response.aread()
+                try:
+                    body = response.json()
+                except ValueError:
+                    body = response.text
+                raise MobileOpsResourcesError(response, body)
+            async for chunk in response.aiter_bytes():
+                yield chunk
+
+    async def delete(self, artifact_id: str) -> None:
+        await self._transport._request(
+            "DELETE", self._base + "/" + quote(artifact_id, safe="")
+        )
+
+    async def aclose(self) -> None:
+        await self._transport.aclose()
+
+
+class MobileOpsStorageClient:
+    def __init__(self, **options: Any):
+        self._transport = MobileOpsResourcesClient(**options)
+        self._url = options["api_url"].rstrip("/") + "/context-hub/v1/storage/usage"
+
+    async def usage(self) -> StorageUsage:
+        return await self._transport._request_json("GET", self._url)
+
+    async def aclose(self) -> None:
+        await self._transport.aclose()

{mobileops_managed_agent-0.3.1 → mobileops_managed_agent-0.3.2}/src/mobileops_managed_agent/assistant_model.py

@@ -80,7 +80,7 @@ def _map_hub(
 ) -> dict[str, Any]:
     entrypoints = item.get("entrypoints")
     if kind == "memory":
-        if not entrypoints:
+        if "entrypoints" in item and not entrypoints:
             raise TypeError("memory attachments require a non-empty entrypoints list")
     elif entrypoints is not None:
         raise TypeError(

{mobileops_managed_agent-0.3.1 → mobileops_managed_agent-0.3.2}/src/mobileops_managed_agent/client.py

@@ -11,6 +11,7 @@ from .assistants import MobileOpsAssistantsClient
 from .resources import MobileOpsResourcesClient
 from .sandbox_profiles import MobileOpsSandboxProfilesClient
 from .vaults import MobileOpsVaultsClient
+from .artifacts import MobileOpsArtifactsClient, MobileOpsStorageClient
 
 
 class MobileOpsAgentClient:
@@ -52,6 +53,9 @@ class MobileOpsManagedAgentClient:
                 headers=merged_headers,
             )
         )
+        artifact_options = dict(api_url=api_url, api_key=api_key, headers=headers, http_client=http_client)
+        self.artifacts = MobileOpsArtifactsClient(**artifact_options)
+        self.storage = MobileOpsStorageClient(**artifact_options)
         self.files = MobileOpsFilesClient(
             api_url=api_url,
             api_key=api_key,
@@ -92,7 +96,13 @@ class MobileOpsManagedAgentClient:
                     try:
                         await self.vaults.aclose()
                     finally:
-                        await self.sandbox_profiles.aclose()
+                        try:
+                            await self.sandbox_profiles.aclose()
+                        finally:
+                            try:
+                                await self.artifacts.aclose()
+                            finally:
+                                await self.storage.aclose()
 
     async def __aenter__(self) -> "MobileOpsManagedAgentClient":
         return self

mobileops_managed_agent-0.3.2/src/mobileops_managed_agent/model_retry.py

@@ -0,0 +1,64 @@
+"""Track custom model retry events while consuming the native Run stream."""
+
+from __future__ import annotations
+from collections.abc import Mapping, Sequence
+from typing import Any, Literal, TypedDict
+
+
+class ModelRetryEvent(TypedDict):
+    type: Literal["model_retry"]
+    call_id: str
+    namespace: str
+    phase: Literal["retrying", "resumed", "succeeded", "exhausted"]
+    attempt: int
+    max_retries: int
+    failed_message_ids: list[str]
+    delay_ms: int
+    reason: str | None
+
+
+class ModelRetryTracker:
+    def __init__(self) -> None:
+        self._discarded: set[str] = set()
+        self._active: dict[str, ModelRetryEvent] = {}
+
+    def update(self, value: Any) -> ModelRetryEvent | None:
+        if not isinstance(value, dict) or value.get("type") != "model_retry":
+            return None
+        if (
+            not isinstance(value.get("call_id"), str)
+            or not isinstance(value.get("namespace"), str)
+            or value.get("phase")
+            not in ("retrying", "resumed", "succeeded", "exhausted")
+            or any(
+                type(value.get(k)) is not int or value[k] < 0
+                for k in ("attempt", "max_retries", "delay_ms")
+            )
+            or not isinstance(value.get("failed_message_ids"), list)
+            or not all(isinstance(i, str) for i in value["failed_message_ids"])
+            or "reason" not in value
+            or not (value.get("reason") is None or isinstance(value.get("reason"), str))
+        ):
+            return None
+        self._discarded.update(value["failed_message_ids"])
+        if value["phase"] in ("retrying", "resumed"):
+            self._active[value["call_id"]] = value
+        else:
+            self._active.pop(value["call_id"], None)
+        return value
+
+    @property
+    def pending(self) -> list[ModelRetryEvent]:
+        return list(self._active.values())
+
+    def filter(self, messages: Sequence[Mapping[str, Any]]) -> list[Mapping[str, Any]]:
+        return [m for m in messages if m.get("id") not in self._discarded]
+
+    def reset(self) -> None:
+        """Start tracking a different conversation."""
+        self._discarded.clear()
+        self._active.clear()
+
+    def finish(self) -> None:
+        """Clear status at Run end, keeping failed messages hidden."""
+        self._active.clear()

{mobileops_managed_agent-0.3.1 → mobileops_managed_agent-0.3.2}/src/mobileops_managed_agent/resources.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import json
 from collections.abc import Mapping
 from typing import Any, NotRequired, TypedDict
 from urllib.parse import quote, urlencode
@@ -16,6 +17,8 @@ class ResourceVersion(TypedDict):
     ref: str
     file_count: NotRequired[int]
     created_at: NotRequired[str]
+    size_bytes: NotRequired[int]
+    files: NotRequired[list[ResourceFile]]
 
 
 class ResourceSummary(TypedDict):
@@ -65,10 +68,21 @@ class MobileOpsResourceVersionsClient:
         return payload["resource"]
 
     async def list(
-        self, kind: CapabilityResourceKind, slug: str
+        self,
+        kind: CapabilityResourceKind,
+        slug: str,
+        *,
+        versions: list[str] | None = None,
+        include_files: bool = False,
     ) -> list[ResourceVersion]:
-        resource = await self._resources.retrieve(kind, slug)
-        return resource.get("versions", [])
+        params = {"include_files": str(include_files).lower()}
+        if versions is not None:
+            params["versions"] = json.dumps(versions)
+        payload = await self._resources._request_json(
+            "GET",
+            f"{self._resources._resource_url(kind, slug)}/versions?{urlencode(params)}",
+        )
+        return payload["versions"]
 
 
 class MobileOpsResourceFilesClient:

{mobileops_managed_agent-0.3.1 → mobileops_managed_agent-0.3.2/src/mobileops_managed_agent.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mobileops-managed-agent
-Version: 0.3.1
+Version: 0.3.2
 Summary: Server-side SDK for the MobileOPS Managed Agent API.
 Project-URL: Homepage, https://github.com/InnoMindX/deepagent/tree/main/managed-agent-sdk/python
 Project-URL: Repository, https://github.com/InnoMindX/deepagent

{mobileops_managed_agent-0.3.1 → mobileops_managed_agent-0.3.2}/src/mobileops_managed_agent.egg-info/SOURCES.txt

@@ -2,6 +2,7 @@ MANIFEST.in
 README.md
 pyproject.toml
 src/mobileops_managed_agent/__init__.py
+src/mobileops_managed_agent/artifacts.py
 src/mobileops_managed_agent/assistant_model.py
 src/mobileops_managed_agent/assistants.py
 src/mobileops_managed_agent/auth.py
@@ -9,6 +10,7 @@ src/mobileops_managed_agent/capabilities.py
 src/mobileops_managed_agent/client.py
 src/mobileops_managed_agent/credential_vault.py
 src/mobileops_managed_agent/files.py
+src/mobileops_managed_agent/model_retry.py
 src/mobileops_managed_agent/resources.py
 src/mobileops_managed_agent/response_schema.py
 src/mobileops_managed_agent/sandbox_profile.py