socaity-cli 0.1.8__tar.gz → 0.1.9__tar.gz

{socaity_cli-0.1.8 → socaity_cli-0.1.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: socaity-cli
-Version: 0.1.8
+Version: 0.1.9
 Summary: Socaity platform CLI: login, catalog browsing, SDK installs and APIPod deployment commands.
 Author: SocAIty
 License: MIT

{socaity_cli-0.1.8 → socaity_cli-0.1.9}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "socaity-cli"
-version = "0.1.8"
+version = "0.1.9"
 description = "Socaity platform CLI: login, catalog browsing, SDK installs and APIPod deployment commands."
 requires-python = ">=3.8"
 authors = [

{socaity_cli-0.1.8 → socaity_cli-0.1.9}/socaity_cli/backend_client.py

@@ -6,6 +6,7 @@ from socaity_cli.clients.deployment import DeploymentClient
 from socaity_cli.clients.files import FilesClient
 from socaity_cli.clients.interrupts import InterruptsClient
 from socaity_cli.clients.jobs import JobsClient
+from socaity_cli.clients.profile import ProfileClient
 from socaity_cli.clients.projects import ProjectsClient
 from socaity_cli.clients.sdk_install import SdkInstallClient
 from socaity_cli.clients.workflows import WorkflowsClient
@@ -24,6 +25,7 @@ class SocaityBackendClient(
     ConversationsClient,
     InterruptsClient,
     WorkflowsClient,
+    ProfileClient,
 ):
     """Facade over domain mixins. Prefer importing a domain client when you only need one surface."""
     pass

{socaity_cli-0.1.8 → socaity_cli-0.1.9}/socaity_cli/cli.py

@@ -16,7 +16,7 @@ from socaity_cli.sdk_commands import run_install, run_update, run_workflow_comma
 def _run_list(entity: str, category: str | None, family: str | None, limit: int) -> None:
     client = SocaityBackendClient()
     if entity == "services":
-        services = client.list_services(category=category, limit=limit)
+        services = client.query_services(category=category, limit=limit)
         if not services:
             print("No services found.")
             return
@@ -26,7 +26,7 @@ def _run_list(entity: str, category: str | None, family: str | None, limit: int)
             desc = f" - {service.short_desc}" if service.short_desc else ""
             print(f"{name}{official}{desc}")
     else:
-        models = client.list_models(family=family, limit=limit)
+        models = client.query_models(family=family, limit=limit)
         if not models:
             print("No models found.")
             return
@@ -137,7 +137,7 @@ def _run_workflows(args) -> None:
     command = args.workflows_command
 
     if command == "list":
-        workflows = client.search_workflows(query=args.q, limit=args.limit)
+        workflows = client.query_workflows(q=args.q, limit=args.limit)
         if not workflows:
             print("No workflows found.")
             return
@@ -224,7 +224,12 @@ def _run_search(query: str, collections: str | None, limit: int) -> None:
     targets = [part.strip() for part in (collections or "services,models").split(",") if part.strip()]
     found = False
     for collection in targets:
-        hits = client.search(query, collection=collection, limit=limit)
+        if collection == "models":
+            hits = client.query_models(q=query, limit=limit)
+        elif collection == "jobs":
+            hits = client.query_jobs(q=query, limit=limit)
+        else:
+            hits = client.query_services(q=query, limit=limit)
         if not hits:
             continue
         found = True

{socaity_cli-0.1.8 → socaity_cli-0.1.9}/socaity_cli/clients/__init__.py

@@ -3,16 +3,24 @@ from socaity_cli.clients.analytics import AnalyticsClient
 from socaity_cli.clients.catalog import CatalogClient
 from socaity_cli.clients.deployment import DeploymentClient
 from socaity_cli.clients.files import FilesClient
+from socaity_cli.clients.conversations import ConversationsClient
+from socaity_cli.clients.interrupts import InterruptsClient
 from socaity_cli.clients.jobs import JobsClient
+from socaity_cli.clients.profile import ProfileClient
 from socaity_cli.clients.projects import ProjectsClient
 from socaity_cli.clients.sdk_install import SdkInstallClient
+from socaity_cli.clients.workflows import WorkflowsClient
 
 __all__ = [
     "AnalyticsClient",
     "CatalogClient",
+    "ConversationsClient",
     "DeploymentClient",
     "FilesClient",
+    "InterruptsClient",
     "JobsClient",
+    "ProfileClient",
     "ProjectsClient",
     "SdkInstallClient",
+    "WorkflowsClient",
 ]

{socaity_cli-0.1.8 → socaity_cli-0.1.9}/socaity_cli/clients/catalog.py

@@ -1,8 +1,8 @@
 """AI catalog reads (`/v1/catalog`)."""
 import re
-from typing import Dict, List, Optional, Union
+from typing import List, Optional
 
-from socaity_schemas.platform import AIModel, AIService, Job, ServiceCategory
+from socaity_schemas.platform import AIModel, AIService, ServiceCategory
 
 SERVICE_LIST_FIELDS = ["id", "name", "display_name", "short_desc", "categories", "is_official", "n_usages"]
 SERVICE_DETAIL_EXPAND = ["deployments.contract", "endpoints", "models"]
@@ -10,38 +10,42 @@ MODEL_LIST_FIELDS = ["id", "name", "display_name", "family", "release_date", "hu
 
 
 class CatalogClient:
-    """Mixin: catalog list/get/search. Requires ``BackendHttp`` methods on ``self``."""
+    """Mixin: catalog query/get. Requires ``BackendHttp`` methods on ``self``."""
 
     SERVICE_DETAIL_EXPAND = SERVICE_DETAIL_EXPAND
 
-    def search_services(
+    def query_services(
         self,
-        query: Optional[str] = None,
+        q: Optional[str] = None,
         category: Optional[str] = None,
         filters: Optional[List[str]] = None,
         expand: Optional[List[str]] = None,
         fields: Optional[List[str]] = None,
         sort: Optional[str] = None,
-        limit: int = 20,
+        limit: int = 100,
         offset: int = 0,
+        known_ids: Optional[List[str]] = None,
+        known_fieldset: Optional[str] = None,
         mine: bool = False,
     ) -> List[AIService]:
-        """Find AI services in the socaity catalog.
+        """Query AI services in the socaity catalog.
 
-        Start here for any generation task. Search by capability ("text to image",
-        "voice cloning"). Results are slim catalog rows; call ``get_service`` with
-        a hit ``id`` for endpoints and the call signature under
-        ``deployments.contract``.
+        Start here for any generation task. Query by capability ("text to image",
+        "voice cloning") or browse with filters. Results are slim catalog rows;
+        call ``get_service`` with a hit ``id`` for endpoints and the call
+        signature under ``deployments.contract``.
 
         Args:
-            query: Free-text search. Omit to browse.
-            category: Category id (not display name).
+            q: Free-text query. Omit to browse.
+            category: Category id (not display name). Appended as a filter.
             filters: Repeatable ``field:operator:value`` strings.
             expand: Relations to embed (``deployments``, ``models``, ``pricing``).
             fields: Sparse fieldset, e.g. ``["id", "name"]``.
             sort: ``field:asc`` or ``field:desc``.
             limit: Page size, 1 to 100.
             offset: Skip this many rows; increase by ``limit`` for the next page.
+            known_ids: Client cache ids for delta fetch.
+            known_fieldset: Client cache fieldset for delta fetch.
             mine: Only services owned by the caller.
 
         Returns:
@@ -50,51 +54,9 @@ class CatalogClient:
         service_filters = list(filters or [])
         if category:
             service_filters.append(f"categories:contains:{category}")
-        rows = self.query_services(
-            q=query,
-            filters=service_filters or None,
-            expand=expand,
-            fields=fields,
-            sort=sort,
-            limit=limit,
-            offset=offset,
-            mine=mine,
-        )
-        # Stale Typesense (watermark after empty first sync, or a later
-        # write-through of one service) can return [] or the wrong hit while
-        # GET /services still lists the row. Browse + local token match fills
-        # the gap without waiting for an index rebuild.
-        if query and not mine:
-            matched = [row for row in rows if _service_matches_query(row, query)]
-            if matched:
-                return matched[:limit]
-            browsed = self.query_services(
-                filters=service_filters or None,
-                expand=expand,
-                fields=fields,
-                sort=sort,
-                limit=max(limit, 100),
-                offset=0,
-            )
-            return [row for row in browsed if _service_matches_query(row, query)][:limit]
-        return rows
-
-    def query_services(
-        self,
-        q: Optional[str] = None,
-        filters: Optional[List[str]] = None,
-        expand: Optional[List[str]] = None,
-        fields: Optional[List[str]] = None,
-        sort: Optional[str] = None,
-        limit: int = 100,
-        offset: int = 0,
-        known_ids: Optional[List[str]] = None,
-        known_fieldset: Optional[str] = None,
-        mine: bool = False,
-    ) -> List[AIService]:
         rows = self._unwrap_list(self._get("v1/catalog/services", params={
             "q": q,
-            "filter": filters or [],
+            "filter": service_filters,
             "expand": expand or [],
             "fields": fields if fields is not None else SERVICE_LIST_FIELDS,
             "sort": sort,
@@ -104,24 +66,27 @@ class CatalogClient:
             "known_fieldset": known_fieldset,
             "mine": mine,
         }))
-        return [AIService(**row) for row in rows]
-
-    def list_services(
-        self,
-        category: Optional[str] = None,
-        fields: Optional[List[str]] = None,
-        expand: Optional[List[str]] = None,
-        limit: int = 100,
-        offset: int = 0,
-    ) -> List[AIService]:
-        filters = [f"categories:contains:{category}"] if category else None
-        return self.query_services(
-            filters=filters,
-            expand=expand,
-            fields=fields,
-            limit=limit,
-            offset=offset,
-        )
+        services = [AIService(**row) for row in rows]
+        # Stale Typesense (watermark after empty first sync, or a later
+        # write-through of one service) can return [] or the wrong hit while
+        # GET /services still lists the row. Browse + local token match fills
+        # the gap without waiting for an index rebuild.
+        if q and not mine:
+            matched = [row for row in services if _service_matches_query(row, q)]
+            if matched:
+                return matched[:limit]
+            browsed = self._unwrap_list(self._get("v1/catalog/services", params={
+                "filter": service_filters,
+                "expand": expand or [],
+                "fields": fields if fields is not None else SERVICE_LIST_FIELDS,
+                "sort": sort,
+                "limit": max(limit, 100),
+                "offset": 0,
+                "mine": mine,
+            }))
+            browsed_services = [AIService(**row) for row in browsed]
+            return [row for row in browsed_services if _service_matches_query(row, q)][:limit]
+        return services
 
     def get_service(
         self,
@@ -137,7 +102,7 @@ class CatalogClient:
         ``deployments[].contract.endpoints[].parameters``.
 
         Args:
-            identifier: Service id or slug from ``search_services``.
+            identifier: Service id or slug from ``query_services``.
             expand: Override the default expand list.
             fields: Sparse fieldset.
             filters: Post-fetch filters.
@@ -155,6 +120,8 @@ class CatalogClient:
     def query_models(
         self,
         q: Optional[str] = None,
+        family: Optional[str] = None,
+        task: Optional[str] = None,
         filters: Optional[List[str]] = None,
         expand: Optional[List[str]] = None,
         fields: Optional[List[str]] = None,
@@ -164,9 +131,32 @@ class CatalogClient:
         known_ids: Optional[List[str]] = None,
         known_fieldset: Optional[str] = None,
     ) -> List[AIModel]:
+        """Query AI models in the catalog.
+
+        Args:
+            q: Free-text query. Omit to browse.
+            family: Model family, e.g. ``flux``. Appended as a filter.
+            task: Task slug, e.g. ``text2image``. Appended as a filter.
+            filters: Repeatable ``field:operator:value`` strings.
+            expand: Relations to embed (``scores``, ``capabilities``).
+            fields: Sparse fieldset.
+            sort: ``field:asc`` or ``field:desc``.
+            limit: Page size.
+            offset: Skip this many rows.
+            known_ids: Client cache ids for delta fetch.
+            known_fieldset: Client cache fieldset for delta fetch.
+
+        Returns:
+            Model rows, or an empty list when nothing matches.
+        """
+        model_filters = list(filters or [])
+        if family:
+            model_filters.append(f"family:eq:{family}")
+        if task:
+            model_filters.append(f"tasks:contains:{task}")
         rows = self._unwrap_list(self._get("v1/catalog/models", params={
             "q": q,
-            "filter": filters or [],
+            "filter": model_filters,
             "expand": expand or [],
             "fields": fields if fields is not None else MODEL_LIST_FIELDS,
             "sort": sort,
@@ -177,22 +167,6 @@ class CatalogClient:
         }))
         return [AIModel(**row) for row in rows]
 
-    def list_models(
-        self,
-        family: Optional[str] = None,
-        task: Optional[str] = None,
-        fields: Optional[List[str]] = None,
-        expand: Optional[List[str]] = None,
-        limit: int = 100,
-        offset: int = 0,
-    ) -> List[AIModel]:
-        filters: List[str] = []
-        if family:
-            filters.append(f"family:eq:{family}")
-        if task:
-            filters.append(f"tasks:contains:{task}")
-        return self.query_models(filters=filters or None, expand=expand, fields=fields, limit=limit, offset=offset)
-
     def get_model(
         self,
         id_or_name: str,
@@ -205,33 +179,19 @@ class CatalogClient:
         })
         return AIModel(**row) if row else None
 
-    def list_categories(self) -> List[ServiceCategory]:
+    def query_categories(self) -> List[ServiceCategory]:
+        """Query catalog task categories."""
         rows = self._get("v1/catalog/categories") or []
         if isinstance(rows, dict):
             rows = rows.get("entities") or rows.get("data") or []
         return [ServiceCategory(**row) for row in rows]
 
-    def list_pricing_rules(self, active_only: bool = True) -> List[Dict]:
+    def list_pricing_rules(self, active_only: bool = True) -> List[dict]:
         rows = self._get("v1/catalog/pricing-rules", params={"active_only": active_only}) or []
         if isinstance(rows, dict):
             rows = rows.get("entities") or rows.get("data") or []
         return rows if isinstance(rows, list) else []
 
-    def search(
-        self,
-        query: str,
-        collection: str = "services",
-        filters: Optional[List[str]] = None,
-        limit: int = 20,
-        visibility: str = "user_visible",
-    ) -> List[Union[AIService, AIModel, Job]]:
-        """Full-text search via catalog or jobs ``q`` param (Typesense)."""
-        if collection == "models":
-            return self.query_models(q=query, filters=filters, limit=limit)
-        if collection == "jobs":
-            return self.query_jobs(q=query, filters=filters, visibility=visibility, limit=limit)
-        return self.search_services(query=query, filters=filters, limit=limit)
-
 
 def _service_matches_query(service: AIService, text: str) -> bool:
     tokens = _query_tokens(text)

{socaity_cli-0.1.8 → socaity_cli-0.1.9}/socaity_cli/clients/conversations.py

@@ -6,9 +6,9 @@ from socaity_schemas.platform import Chat, ChatItem
 
 
 class ConversationsClient:
-    """Mixin: conversation metadata and items. Requires ``BackendHttp`` methods on ``self``."""
+    """Mixin: conversation query/get/items. Requires ``BackendHttp`` methods on ``self``."""
 
-    def list_conversations(
+    def query_conversations(
         self,
         q: Optional[str] = None,
         filters: Optional[List[str]] = None,
@@ -43,7 +43,7 @@ class ConversationsClient:
         })
         return Chat(**row) if row else None
 
-    def list_conversation_items(
+    def query_conversation_items(
         self,
         conversation_id: str,
         branch: str = "active",
@@ -67,15 +67,21 @@ class ConversationsClient:
         conversation_id: str,
         title: Optional[str] = None,
         status: Optional[str] = None,
+        active_item_id: Optional[str] = None,
     ) -> Optional[Chat]:
-        """Patch title and/or status on a conversation the caller owns."""
+        """Patch title, status, and/or active_item_id (1/N switch) on a conversation the caller owns."""
         row = self._request(
             "PATCH",
             f"v1/conversations/{conversation_id}",
-            params={"title": title, "status": status},
+            params={"title": title, "status": status, "active_item_id": active_item_id},
         )
         return Chat(**row) if row else None
 
+    def fork_conversation(self, conversation_id: str) -> Optional[Chat]:
+        """Copy the conversation tree into a new chats.id."""
+        row = self._request("POST", f"v1/conversations/{conversation_id}/fork")
+        return Chat(**row) if row else None
+
     def delete_conversation(self, conversation_id: str) -> bool:
         """Soft-delete a conversation and clear its linked job data."""
         return bool(self._request("DELETE", f"v1/conversations/{conversation_id}"))

{socaity_cli-0.1.8 → socaity_cli-0.1.9}/socaity_cli/clients/files.py

@@ -4,11 +4,11 @@ from typing import Dict, List, Optional, Union
 
 import httpx
 
-FILE_LIST_PURPOSES = ["USER_UPLOAD", "JOB_RESULT"]
+FILE_LIST_PURPOSES = ["USER_UPLOAD", "JOB_OUTPUT"]
 
 
 class FilesClient:
-    """Mixin: storage usage, list, get, upload, update, delete. Requires ``BackendHttp`` on ``self``."""
+    """Mixin: storage usage, query, get, upload, update, delete. Requires ``BackendHttp`` on ``self``."""
 
     def get_storage_usage(self) -> Optional[Dict]:
         """Return the caller's storage quota and current usage.
@@ -18,7 +18,7 @@ class FilesClient:
         """
         return self._get("v1/files/usage")
 
-    def list_files(
+    def query_files(
         self,
         purpose: Union[str, List[str]] = FILE_LIST_PURPOSES,
         include_expired: bool = False,
@@ -28,7 +28,7 @@ class FilesClient:
         limit: int = 25,
         offset: int = 0,
     ) -> List[Dict]:
-        """List stored files visible to the caller.
+        """Query stored files visible to the caller.
 
         Args:
             purpose: Filter by reference purpose. Default: uploads and job results.
@@ -83,7 +83,7 @@ class FilesClient:
         """Patch file metadata the caller owns. Requires approval in agent mode.
 
         Args:
-            file_id: File id from ``list_files`` or ``get_file``.
+            file_id: File id from ``query_files`` or ``get_file``.
             is_public: Publish or unpublish the file.
             expires_at: ISO timestamp when the file should expire.
             clear_expires_at: Clear expiry and keep the file permanently.
@@ -101,7 +101,7 @@ class FilesClient:
         """Hard-delete a file the caller owns. Requires approval in agent mode.
 
         Args:
-            file_id: File id from ``list_files`` or ``get_file``.
+            file_id: File id from ``query_files`` or ``get_file``.
 
         Returns:
             ``True`` when the delete succeeded, ``False`` otherwise.

{socaity_cli-0.1.8 → socaity_cli-0.1.9}/socaity_cli/clients/jobs.py

@@ -24,6 +24,26 @@ class JobsClient:
         known_ids: Optional[List[str]] = None,
         known_fieldset: Optional[str] = None,
     ) -> List[Job]:
+        """Search and list jobs.
+
+        Last N jobs: ``limit=N`` (catalog default sort is newest first).
+        Pass ``q`` to search display names and prompts.
+
+        Args:
+            q: Optional search text.
+            filters: Optional catalog filter strings.
+            expand: Extra relations. Empty keeps the list slim.
+            fields: Optional field projection.
+            visibility: ``user_visible`` (default) or ``public``.
+            sort: Sort spec. Empty keeps the catalog default (newest first).
+            limit: Maximum rows (default 100).
+            offset: Skip this many rows.
+            known_ids: Client cache ids for delta fetch.
+            known_fieldset: Client cache fieldset for delta fetch.
+
+        Returns:
+            Job records the caller is allowed to see.
+        """
         rows = self._unwrap_list(self._get("v1/jobs", params={
             "q": q,
             "visibility": visibility,
@@ -54,7 +74,7 @@ class JobsClient:
         earlier result.
 
         Args:
-            job_id: Platform job id from ``run_service`` or ``list_jobs``.
+            job_id: Platform job id from ``run_service`` or ``query_jobs``.
 
         Returns:
             The job record, or ``None`` when not found or not visible.

socaity_cli-0.1.9/socaity_cli/clients/profile.py

@@ -0,0 +1,29 @@
+"""Identity and token exchange (`/v1/profile`, `/v1/cli-auth`)."""
+from typing import Any, Dict, Optional
+
+
+class ProfileClient:
+    """Mixin: caller identity. Requires ``BackendHttp`` methods on ``self``."""
+
+    def whoami(self) -> Optional[Dict[str, Any]]:
+        """Return the authenticated caller's identity.
+
+        Used by hosts that need to verify an opaque ``sk_``/``tk_`` key.
+        The backend owns the answer; there is nothing to check locally.
+
+        Returns:
+            Identity dict with ``user_id`` when the Bearer key is valid,
+            or ``None`` when the key is unknown.
+        """
+        return self._get("v1/profile/whoami")
+
+    def exchange_cli_auth(self) -> Optional[Dict[str, Any]]:
+        """Exchange the current Bearer token (JWT) for a platform ``tk_`` key.
+
+        The Authorization header is the token to exchange, not a platform key.
+        MCP OAuth uses this after verifying a Supabase JWT.
+
+        Returns:
+            Payload containing ``api_key``, or ``None`` when exchange fails.
+        """
+        return self._post("v1/cli-auth/exchange")

{socaity_cli-0.1.8 → socaity_cli-0.1.9}/socaity_cli/clients/workflows.py

@@ -11,7 +11,7 @@ from socaity_schemas.platform.workflow import (
     WorkflowUpsertResult,
 )
 
-WORKFLOW_DETAIL_EXPAND = ["document"]
+WORKFLOW_DETAIL_EXPAND = ["document", "layout"]
 
 
 def _dump_model(value: Union[Dict[str, Any], Any, None]) -> Optional[Dict[str, Any]]:
@@ -26,9 +26,9 @@ class WorkflowsClient:
 
     WORKFLOW_DETAIL_EXPAND = WORKFLOW_DETAIL_EXPAND
 
-    def search_workflows(
+    def query_workflows(
         self,
-        query: Optional[str] = None,
+        q: Optional[str] = None,
         filters: Optional[List[str]] = None,
         expand: Optional[List[str]] = None,
         fields: Optional[List[str]] = None,
@@ -38,7 +38,7 @@ class WorkflowsClient:
     ) -> List[Workflow]:
         """Find workflows: your own plus public ones.
 
-        ``query`` searches title, goal, slug and node text. Results are slim
+        ``q`` searches title, goal, slug and node text. Results are slim
         catalog rows; call ``get_workflow`` with a hit ``id`` (or slug) for the
         full document.
 
@@ -46,7 +46,7 @@ class WorkflowsClient:
         ``slug:eq:<slug>`` and ``visibility:eq:mine|public|blended``.
 
         Args:
-            query: Free-text search. Omit to browse.
+            q: Free-text query. Omit to browse.
             filters: Repeatable ``field:operator:value`` strings.
             expand: Relations to embed (``document``, ``revisions``).
             fields: Sparse fieldset, e.g. ``["id", "slug", "title"]``.
@@ -58,7 +58,7 @@ class WorkflowsClient:
             Workflow rows, or an empty list when nothing matches.
         """
         rows = self._unwrap_list(self._get("v1/workflows", params={
-            "q": query,
+            "q": q,
             "filter": filters or [],
             "expand": expand or [],
             "fields": fields or [],
@@ -82,8 +82,8 @@ class WorkflowsClient:
         expects. Pin an older state via ``revision_id`` or ``version``.
 
         Args:
-            workflow_id: ``wf_`` id or slug from ``search_workflows``.
-            expand: Override the default expand (``document``, ``revisions``).
+            workflow_id: ``wf_`` id or slug from ``query_workflows``.
+            expand: Override the default expand (``document``, ``layout``, ``revisions``).
             revision_id: ``rv_`` id to pin; wins over ``version``.
             version: Valid version int; default is the latest.
 
@@ -157,7 +157,7 @@ class WorkflowsClient:
         )
         return WorkflowUpsertResult(**row) if row else None
 
-    def list_workflow_revisions(
+    def query_workflow_revisions(
         self,
         workflow_id: str,
         expand: Optional[List[str]] = None,

{socaity_cli-0.1.8 → socaity_cli-0.1.9}/socaity_cli/errors.py

@@ -40,3 +40,7 @@ class RegistryQuotaExceededError(BackendApiError):
 
     def __init__(self, message: str, *, detail: Any = None):
         super().__init__(413, message, code="REGISTRY_QUOTA_EXCEEDED", detail=detail)
+
+
+class BackendTransportError(Exception):
+    """Network failure talking to the Socaity backend."""

{socaity_cli-0.1.8 → socaity_cli-0.1.9}/socaity_cli/http.py

@@ -6,7 +6,12 @@ from typing import Any, Dict, Optional
 import httpx
 
 from socaity_cli.credentials import DEFAULT_BACKEND_URL, get_api_key, load_credentials
-from socaity_cli.errors import BackendApiError, PrivateSlotLimitError, RegistryQuotaExceededError
+from socaity_cli.errors import (
+    BackendApiError,
+    BackendTransportError,
+    PrivateSlotLimitError,
+    RegistryQuotaExceededError,
+)
 
 
 def resolve_backend_url(backend_url: Optional[str] = None) -> str:
@@ -82,16 +87,18 @@ class BackendHttp:
                     headers=self._auth_headers,
                     timeout=400,
                 )
-            if response.status_code in (200, 202):
-                return response.json() if response.content else {}
-            self._raise_for_structured_error(response.status_code, response.text)
-            print(f"Request to {path} failed with status {response.status_code}: {response.text[:300]}")
-            return None
-        except (PrivateSlotLimitError, RegistryQuotaExceededError, BackendApiError):
-            raise
-        except Exception as e:
-            print(f"Request to {path} failed: {e}")
+        except httpx.HTTPError as exc:
+            raise BackendTransportError(f"Request to {path} failed: {exc}") from exc
+
+        if response.status_code in (200, 201, 202):
+            return response.json() if response.content else {}
+        self._raise_for_structured_error(response.status_code, response.text)
+        if response.status_code == 404:
             return None
+        raise BackendApiError(
+            response.status_code,
+            (response.text or f"HTTP {response.status_code}")[:300],
+        )
 
     @staticmethod
     def _catalog_params(params: Dict) -> Dict:

{socaity_cli-0.1.8 → socaity_cli-0.1.9}/socaity_cli/sdk_commands.py

@@ -33,22 +33,25 @@ def run_workflow_command(workflow: str, inputs: dict, timeout_s: float | None =
     """Run a workflow through the inference gateway and print the terminal result."""
     import json
 
-    from socaity.tools.workflows import execute_workflow
+    from socaity.client import SocaityClient
+    from socaity.core.serialize import serialize_job
 
-    def print_progress(progress: float, message: str) -> None:
-        print(f"[{progress:>4.0%}] {message}")
+    client = SocaityClient()
 
-    def print_job(job_id: str, _envelope) -> None:
-        print(f"Job {job_id} submitted. Waiting...")
+    def on_progress(event) -> None:
+        if event.kind == "started":
+            print(f"Job {event.job_id} submitted. Waiting...")
+        elif event.kind == "progress" and event.message:
+            print(f"[{(event.progress or 0):>4.0%}] {event.message}")
 
     try:
-        result = execute_workflow(
-            workflow,
-            inputs=inputs,
-            timeout_s=timeout_s,
-            on_progress=print_progress,
-            on_job_start=print_job,
-        )
+        job = client.run_workflow(workflow, inputs=inputs)
+        unsubscribe = job.subscribe(on_started=on_progress, on_progress=on_progress, replay=True)
+        try:
+            job.get_result(timeout_s=timeout_s)
+        finally:
+            unsubscribe()
+        result = serialize_job(job)
     except (ValueError, RuntimeError) as error:
         print(str(error))
         raise SystemExit(1)

{socaity_cli-0.1.8 → socaity_cli-0.1.9}/socaity_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: socaity-cli
-Version: 0.1.8
+Version: 0.1.9
 Summary: Socaity platform CLI: login, catalog browsing, SDK installs and APIPod deployment commands.
 Author: SocAIty
 License: MIT

{socaity_cli-0.1.8 → socaity_cli-0.1.9}/socaity_cli.egg-info/SOURCES.txt

@@ -29,6 +29,7 @@ socaity_cli/clients/deployment.py
 socaity_cli/clients/files.py
 socaity_cli/clients/interrupts.py
 socaity_cli/clients/jobs.py
+socaity_cli/clients/profile.py
 socaity_cli/clients/projects.py
 socaity_cli/clients/sdk_install.py
 socaity_cli/clients/workflows.py