archastro.platform.client

  1# Copyright (c) 2026 ArchAstro Inc. Licensed under the MIT License.
  2# This file is auto-generated by @archastro/sdk-generator. Do not edit.
  3# Content hash: f24da3972099
  4
  5from urllib.parse import urlparse, urlunparse
  6
  7from archastro.phx_channel import Socket
  8
  9from .auth import AsyncAuthClient, AuthClient
 10from .runtime.http_client import HttpClient, SyncHttpClient
 11from .v1 import V1, AsyncV1
 12
 13
 14class AsyncPlatformClient:
 15    def __init__(
 16        self,
 17        *,
 18        base_url: str = "https://platform.archastro.ai",
 19        access_token: str | None = None,
 20        get_access_token=None,
 21        on_refresh_token=None,
 22        path_prefix: str | None = None,
 23        default_headers: dict[str, str] | None = None,
 24    ):
 25        self._base_url = base_url
 26        self._access_token = access_token
 27        self._get_access_token = get_access_token
 28        self._default_headers = default_headers or {}
 29        self._http = HttpClient(
 30            base_url=base_url,
 31            access_token=access_token,
 32            get_access_token=get_access_token,
 33            on_refresh_token=on_refresh_token,
 34            path_prefix=path_prefix,
 35            default_headers=default_headers,
 36        )
 37        self.auth = AsyncAuthClient(self._http)
 38        self.v1 = AsyncV1(self._http)
 39        self.activity_feed = self.v1.activity_feed
 40        self.agent_computers = self.v1.agent_computers
 41        self.agent_env_vars = self.v1.agent_env_vars
 42        self.agent_health_actions = self.v1.agent_health_actions
 43        self.agent_installations = self.v1.agent_installations
 44        self.agent_routine_runs = self.v1.agent_routine_runs
 45        self.agent_routines = self.v1.agent_routines
 46        self.agent_sessions = self.v1.agent_sessions
 47        self.agent_skills = self.v1.agent_skills
 48        self.agent_tools = self.v1.agent_tools
 49        self.agents = self.v1.agents
 50        self.artifacts = self.v1.artifacts
 51        self.automation_runs = self.v1.automation_runs
 52        self.automations = self.v1.automations
 53        self.bug_reports = self.v1.bug_reports
 54        self.config = self.v1.config
 55        self.custom_objects = self.v1.custom_objects
 56        self.event_subscription_deliveries = self.v1.event_subscription_deliveries
 57        self.event_subscriptions = self.v1.event_subscriptions
 58        self.extractions = self.v1.extractions
 59        self.files = self.v1.files
 60        self.installation_sources = self.v1.installation_sources
 61        self.invites = self.v1.invites
 62        self.knowledge_documents = self.v1.knowledge_documents
 63        self.knowledge_ingestions = self.v1.knowledge_ingestions
 64        self.knowledge_sources = self.v1.knowledge_sources
 65        self.kv = self.v1.kv
 66        self.notification_preferences = self.v1.notification_preferences
 67        self.notifications = self.v1.notifications
 68        self.orgs = self.v1.orgs
 69        self.private_service_definitions = self.v1.private_service_definitions
 70        self.private_service_enrollments = self.v1.private_service_enrollments
 71        self.private_services = self.v1.private_services
 72        self.sandboxes = self.v1.sandboxes
 73        self.slack_channel_bindings = self.v1.slack_channel_bindings
 74        self.solution_categories = self.v1.solution_categories
 75        self.solution_instances = self.v1.solution_instances
 76        self.solution_tags = self.v1.solution_tags
 77        self.solutions = self.v1.solutions
 78        self.status = self.v1.status
 79        self.tasks = self.v1.tasks
 80        self.team_memberships = self.v1.team_memberships
 81        self.teams = self.v1.teams
 82        self.thread_messages = self.v1.thread_messages
 83        self.threads = self.v1.threads
 84        self.trajectories = self.v1.trajectories
 85        self.users = self.v1.users
 86        self.work_items = self.v1.work_items
 87        self.ai = self.v1.ai
 88        self.oauth = self.v1.oauth
 89        self._refresh_token: str | None = None
 90        self._extra_http_clients: list[HttpClient] = []
 91        self._sockets: list[Socket] = []
 92
 93    @property
 94    def refresh_token(self) -> str | None:
 95        return self._refresh_token
 96
 97    def set_access_token(self, token: str):
 98        self._access_token = token
 99        self._http.set_access_token(token)
100
101    def set_refresh_token(self, token: str):
102        self._refresh_token = token
103
104    async def __aenter__(self):
105        return self
106
107    async def __aexit__(self, exc_type, exc, tb):
108        await self.close()
109
110    async def close(self):
111        try:
112            try:
113                for socket in self._sockets:
114                    await socket.disconnect()
115            finally:
116                self._sockets.clear()
117            await self._http.close()
118        finally:
119            for http in self._extra_http_clients:
120                await http.close()
121            self._extra_http_clients.clear()
122
123    def _current_access_token(self) -> str | None:
124        if self._get_access_token:
125            return self._get_access_token()
126        return self._access_token
127
128    def _api_key(self) -> str | None:
129        for key, value in self._default_headers.items():
130            if key.lower() == "x-archastro-api-key".lower():
131                return value
132        return None
133
134    def _socket_params(self, params: dict[str, str] | None = None) -> dict[str, str]:
135        socket_params: dict[str, str] = {}
136        api_key = self._api_key()
137        if api_key:
138            socket_params["api_key"] = api_key
139        token = self._current_access_token()
140        if token:
141            socket_params["token"] = token
142        if params:
143            socket_params.update(params)
144        return socket_params
145
146    def _default_websocket_url(self) -> str:
147        parsed = urlparse(self._base_url)
148        scheme = (
149            "wss"
150            if parsed.scheme == "https"
151            else "ws"
152            if parsed.scheme == "http"
153            else parsed.scheme
154        )
155        path = parsed.path.rstrip("/") + "/socket/api/websocket"
156        return urlunparse(
157            parsed._replace(scheme=scheme, path=path, params="", query="", fragment="")
158        )
159
160    async def open_socket(
161        self,
162        *,
163        url: str | None = None,
164        params: dict[str, str] | None = None,
165        connect: bool = True,
166        heartbeat_interval: float = 30,
167        timeout: float = 10,
168        reconnect_backoff_ms: list[int] | None = None,
169        auto_reconnect: bool = True,
170    ) -> Socket:
171        socket_params = self._socket_params(params)
172        if connect and not socket_params.get("token"):
173            raise ValueError("AsyncPlatformClient.open_socket requires an access token")
174        socket = Socket(
175            url or self._default_websocket_url(),
176            params=socket_params,
177            heartbeat_interval=heartbeat_interval,
178            timeout=timeout,
179            reconnect_backoff_ms=reconnect_backoff_ms,
180            auto_reconnect=auto_reconnect,
181        )
182        if connect:
183            await socket.connect()
184            if not socket.is_connected:
185                raise ConnectionError("WebSocket connection failed")
186        self._sockets.append(socket)
187        return socket
188
189    # ─── Factory constructors (generated from auth schemes) ───
190
191    @classmethod
192    def with_secret_key(cls, key: str, base_url: str | None = None) -> "AsyncPlatformClient":
193        """Secret API key — full admin access, no user JWT required"""
194        kwargs = {}
195        if base_url:
196            kwargs["base_url"] = base_url
197        return cls(default_headers={"x-archastro-api-key": key}, **kwargs)
198
199    @classmethod
200    def with_token(
201        cls, api_key: str, access_token: str, base_url: str | None = None
202    ) -> "AsyncPlatformClient":
203        """Create a client with a publishable key and pre-existing access token."""
204        kwargs = {}
205        if base_url:
206            kwargs["base_url"] = base_url
207        return cls(
208            access_token=access_token, default_headers={"x-archastro-api-key": api_key}, **kwargs
209        )
210
211    @classmethod
212    async def with_credentials(
213        cls, api_key: str, email: str, password: str, base_url: str | None = None
214    ) -> "AsyncPlatformClient":
215        """Create a client by logging in with email/password"""
216        kwargs = {}
217        if base_url:
218            kwargs["base_url"] = base_url
219        client = cls(default_headers={"x-archastro-api-key": api_key}, **kwargs)
220        tokens = await client.auth.login(email, password)
221        if not tokens.access_token:
222            raise ValueError("Login did not return an access token")
223        client.set_access_token(tokens.access_token)
224        if tokens.refresh_token:
225            client.set_refresh_token(tokens.refresh_token)
226        refresh_http = HttpClient(
227            base_url=base_url or "https://platform.archastro.ai",
228            default_headers={"x-archastro-api-key": api_key},
229            refresh_only=True,
230        )
231        client._extra_http_clients.append(refresh_http)
232        refresh_auth = AsyncAuthClient(refresh_http)
233
234        async def _refresh() -> str:
235            rt = client.refresh_token
236            if not rt:
237                raise ValueError("No refresh token available")
238            refreshed = await refresh_auth.refresh(rt)
239            if not refreshed.access_token:
240                raise ValueError("Refresh did not return an access token")
241            client.set_access_token(refreshed.access_token)
242            if refreshed.refresh_token:
243                client.set_refresh_token(refreshed.refresh_token)
244            return refreshed.access_token
245
246        client._http.set_refresh_handler(_refresh)
247        return client
248
249
250class PlatformClient:
251    def __init__(
252        self,
253        *,
254        base_url: str = "https://platform.archastro.ai",
255        access_token: str | None = None,
256        get_access_token=None,
257        on_refresh_token=None,
258        path_prefix: str | None = None,
259        default_headers: dict[str, str] | None = None,
260    ):
261        self._closed = False
262        self._http = SyncHttpClient(
263            base_url=base_url,
264            access_token=access_token,
265            get_access_token=get_access_token,
266            on_refresh_token=on_refresh_token,
267            path_prefix=path_prefix,
268            default_headers=default_headers,
269        )
270        self.auth = AuthClient(self._http)
271        self.v1 = V1(self._http)
272        self.activity_feed = self.v1.activity_feed
273        self.agent_computers = self.v1.agent_computers
274        self.agent_env_vars = self.v1.agent_env_vars
275        self.agent_health_actions = self.v1.agent_health_actions
276        self.agent_installations = self.v1.agent_installations
277        self.agent_routine_runs = self.v1.agent_routine_runs
278        self.agent_routines = self.v1.agent_routines
279        self.agent_sessions = self.v1.agent_sessions
280        self.agent_skills = self.v1.agent_skills
281        self.agent_tools = self.v1.agent_tools
282        self.agents = self.v1.agents
283        self.artifacts = self.v1.artifacts
284        self.automation_runs = self.v1.automation_runs
285        self.automations = self.v1.automations
286        self.bug_reports = self.v1.bug_reports
287        self.config = self.v1.config
288        self.custom_objects = self.v1.custom_objects
289        self.event_subscription_deliveries = self.v1.event_subscription_deliveries
290        self.event_subscriptions = self.v1.event_subscriptions
291        self.extractions = self.v1.extractions
292        self.files = self.v1.files
293        self.installation_sources = self.v1.installation_sources
294        self.invites = self.v1.invites
295        self.knowledge_documents = self.v1.knowledge_documents
296        self.knowledge_ingestions = self.v1.knowledge_ingestions
297        self.knowledge_sources = self.v1.knowledge_sources
298        self.kv = self.v1.kv
299        self.notification_preferences = self.v1.notification_preferences
300        self.notifications = self.v1.notifications
301        self.orgs = self.v1.orgs
302        self.private_service_definitions = self.v1.private_service_definitions
303        self.private_service_enrollments = self.v1.private_service_enrollments
304        self.private_services = self.v1.private_services
305        self.sandboxes = self.v1.sandboxes
306        self.slack_channel_bindings = self.v1.slack_channel_bindings
307        self.solution_categories = self.v1.solution_categories
308        self.solution_instances = self.v1.solution_instances
309        self.solution_tags = self.v1.solution_tags
310        self.solutions = self.v1.solutions
311        self.status = self.v1.status
312        self.tasks = self.v1.tasks
313        self.team_memberships = self.v1.team_memberships
314        self.teams = self.v1.teams
315        self.thread_messages = self.v1.thread_messages
316        self.threads = self.v1.threads
317        self.trajectories = self.v1.trajectories
318        self.users = self.v1.users
319        self.work_items = self.v1.work_items
320        self.ai = self.v1.ai
321        self.oauth = self.v1.oauth
322        self._refresh_token: str | None = None
323        self._extra_http_clients: list[SyncHttpClient] = []
324
325    @property
326    def refresh_token(self) -> str | None:
327        return self._refresh_token
328
329    def set_access_token(self, token: str):
330        self._http.set_access_token(token)
331
332    def set_refresh_token(self, token: str):
333        self._refresh_token = token
334
335    def __enter__(self):
336        return self
337
338    def __exit__(self, exc_type, exc, tb):
339        self.close()
340
341    def close(self):
342        if self._closed:
343            return
344        self._closed = True
345        try:
346            self._http.close()
347        finally:
348            for http in self._extra_http_clients:
349                http.close()
350            self._extra_http_clients.clear()
351
352    # ─── Factory constructors (generated from auth schemes) ───
353
354    @classmethod
355    def with_secret_key(cls, key: str, base_url: str | None = None) -> "PlatformClient":
356        """Secret API key — full admin access, no user JWT required"""
357        kwargs = {}
358        if base_url:
359            kwargs["base_url"] = base_url
360        return cls(default_headers={"x-archastro-api-key": key}, **kwargs)
361
362    @classmethod
363    def with_token(
364        cls, api_key: str, access_token: str, base_url: str | None = None
365    ) -> "PlatformClient":
366        """Create a client with a publishable key and pre-existing access token."""
367        kwargs = {}
368        if base_url:
369            kwargs["base_url"] = base_url
370        return cls(
371            access_token=access_token, default_headers={"x-archastro-api-key": api_key}, **kwargs
372        )
373
374    @classmethod
375    def with_credentials(
376        cls, api_key: str, email: str, password: str, base_url: str | None = None
377    ) -> "PlatformClient":
378        kwargs = {}
379        if base_url:
380            kwargs["base_url"] = base_url
381        client = cls(default_headers={"x-archastro-api-key": api_key}, **kwargs)
382        tokens = client.auth.login(email, password)
383        if not tokens.access_token:
384            raise ValueError("Login did not return an access token")
385        client.set_access_token(tokens.access_token)
386        if tokens.refresh_token:
387            client.set_refresh_token(tokens.refresh_token)
388        refresh_http = SyncHttpClient(
389            base_url=base_url or "https://platform.archastro.ai",
390            default_headers={"x-archastro-api-key": api_key},
391            refresh_only=True,
392        )
393        client._extra_http_clients.append(refresh_http)
394        refresh_auth = AuthClient(refresh_http)
395
396        def _refresh() -> str:
397            rt = client.refresh_token
398            if not rt:
399                raise ValueError("No refresh token available")
400            refreshed = refresh_auth.refresh(rt)
401            if not refreshed.access_token:
402                raise ValueError("Refresh did not return an access token")
403            client.set_access_token(refreshed.access_token)
404            if refreshed.refresh_token:
405                client.set_refresh_token(refreshed.refresh_token)
406            return refreshed.access_token
407
408        client._http.set_refresh_handler(_refresh)
409        return client