codecraftai-restclient-pychestrate 1.2.0__py3-none-any.whl

ccai/pychestrate/rest/client/groups/agents.py

@@ -0,0 +1,171 @@
+# Auto-generated by RestClientGenerator. DO NOT EDIT.
+
+from typing import Any, Dict, List, Optional
+
+from ccai.pychestrate.rest.client.models.agents import Agentrollout
+from ccai.pychestrate.rest.client.models.agents import Agentrolloutcreaterequest
+from ccai.pychestrate.rest.client.models.agents import Clientcapabilities
+from ccai.pychestrate.rest.client.models.agents import Clientinventory
+from ccai.restclient.base.aspectsrest import AspectsRest
+from ccai.restclient.base.restapigroup import RestApiGroup
+
+
+class AgentsApiGroup(RestApiGroup):
+    """
+        Agent rollouts and per-client capability inventory.
+    """
+
+    def agents_create_rollout(self, body: Agentrolloutcreaterequest, *, exp_status: Optional[List[int]] = None, headers: Optional[Dict[str, str]] = None, aspects: Optional[AspectsRest] = None) -> Any:
+        """
+            Create an agent rollout plan.
+        """
+        client = self.client
+        if aspects is None:
+            aspects = client.aspects
+
+        header_params = client.headers
+        if headers is not None:
+            header_params = headers
+
+        call_leaf = "/agent-rollouts"
+
+        query_params = None
+
+
+        if exp_status is None:
+            exp_status = [201]
+
+        resp = client.perform_call(
+            leaf=call_leaf,
+            method="POST",
+            params=query_params if query_params else None,
+            body=body,
+            exp_status=exp_status,
+            headers=header_params if header_params else None,
+            context_msg=None,
+            aspects=aspects,
+        )
+        if resp.status_code in (204, 205) or (resp.status_code == 202 and len(resp.content) == 0):
+            rtnval = None
+        else:
+            rtnval = resp.json()
+        return rtnval
+
+    def agents_get_client_capabilities(self, client_id: str, *, exp_status: Optional[List[int]] = None, headers: Optional[Dict[str, str]] = None, aspects: Optional[AspectsRest] = None) -> Clientcapabilities:
+        """
+            Read advertised capabilities for a client.
+        """
+        client = self.client
+        if aspects is None:
+            aspects = client.aspects
+
+        header_params = client.headers
+        if headers is not None:
+            header_params = headers
+
+        leaf_template = "/clients/{client_id}/capabilities"
+        path_params = {
+            'client_id': client_id,
+        }
+        call_leaf = leaf_template.format(**path_params)
+
+        query_params = None
+
+
+        if exp_status is None:
+            exp_status = [200]
+
+        resp = client.perform_call(
+            leaf=call_leaf,
+            method="GET",
+            params=query_params if query_params else None,
+            body=None,
+            exp_status=exp_status,
+            headers=header_params if header_params else None,
+            context_msg=None,
+            aspects=aspects,
+        )
+        if resp.status_code in (204, 205) or (resp.status_code == 202 and len(resp.content) == 0):
+            rtnobj = None
+        else:
+            raw_data = resp.json()
+            rtnobj = Clientcapabilities.from_dict(raw_data)
+        return rtnobj
+
+    def agents_get_client_inventory(self, *, exp_status: Optional[List[int]] = None, headers: Optional[Dict[str, str]] = None, aspects: Optional[AspectsRest] = None) -> Clientinventory:
+        """
+            Cross-fleet inventory of clients.
+        """
+        client = self.client
+        if aspects is None:
+            aspects = client.aspects
+
+        header_params = client.headers
+        if headers is not None:
+            header_params = headers
+
+        call_leaf = "/clients:inventory"
+
+        query_params = None
+
+
+        if exp_status is None:
+            exp_status = [200]
+
+        resp = client.perform_call(
+            leaf=call_leaf,
+            method="GET",
+            params=query_params if query_params else None,
+            body=None,
+            exp_status=exp_status,
+            headers=header_params if header_params else None,
+            context_msg=None,
+            aspects=aspects,
+        )
+        if resp.status_code in (204, 205) or (resp.status_code == 202 and len(resp.content) == 0):
+            rtnobj = None
+        else:
+            raw_data = resp.json()
+            rtnobj = Clientinventory.from_dict(raw_data)
+        return rtnobj
+
+    def agents_start_rollout(self, rollout_id: str, *, exp_status: Optional[List[int]] = None, headers: Optional[Dict[str, str]] = None, aspects: Optional[AspectsRest] = None) -> Agentrollout:
+        """
+            Start an agent rollout.
+        """
+        client = self.client
+        if aspects is None:
+            aspects = client.aspects
+
+        header_params = client.headers
+        if headers is not None:
+            header_params = headers
+
+        leaf_template = "/agent-rollouts/{rollout_id}:start"
+        path_params = {
+            'rollout_id': rollout_id,
+        }
+        call_leaf = leaf_template.format(**path_params)
+
+        query_params = None
+
+
+        if exp_status is None:
+            exp_status = [200]
+
+        resp = client.perform_call(
+            leaf=call_leaf,
+            method="POST",
+            params=query_params if query_params else None,
+            body=None,
+            exp_status=exp_status,
+            headers=header_params if header_params else None,
+            context_msg=None,
+            aspects=aspects,
+        )
+        if resp.status_code in (204, 205) or (resp.status_code == 202 and len(resp.content) == 0):
+            rtnobj = None
+        else:
+            raw_data = resp.json()
+            rtnobj = Agentrollout.from_dict(raw_data)
+        return rtnobj

ccai/pychestrate/rest/client/groups/artifacts.py

@@ -0,0 +1,103 @@
+# Auto-generated by RestClientGenerator. DO NOT EDIT.
+
+from typing import Dict, List, Optional
+
+from ccai.pychestrate.rest.client.models.artifacts import Artifactentry
+from ccai.pychestrate.rest.client.models.artifacts import Artifactentrypage
+from ccai.restclient.base.aspectsrest import AspectsRest
+from ccai.restclient.base.restapigroup import RestApiGroup
+
+
+class ArtifactsApiGroup(RestApiGroup):
+    """
+        Artifact listings and browsing for a run.
+    """
+
+    def artifacts_get_entry(self, run_id: str, artifact_path: str, *, exp_status: Optional[List[int]] = None, headers: Optional[Dict[str, str]] = None, aspects: Optional[AspectsRest] = None) -> Artifactentry:
+        """
+            Read metadata for one artifact entry.
+        """
+        client = self.client
+        if aspects is None:
+            aspects = client.aspects
+
+        header_params = client.headers
+        if headers is not None:
+            header_params = headers
+
+        leaf_template = "/runs/{run_id}/artifacts/{artifact_path}"
+        path_params = {
+            'run_id': run_id,
+            'artifact_path': artifact_path,
+        }
+        call_leaf = leaf_template.format(**path_params)
+
+        query_params = None
+
+
+        if exp_status is None:
+            exp_status = [200]
+
+        resp = client.perform_call(
+            leaf=call_leaf,
+            method="GET",
+            params=query_params if query_params else None,
+            body=None,
+            exp_status=exp_status,
+            headers=header_params if header_params else None,
+            context_msg=None,
+            aspects=aspects,
+        )
+        if resp.status_code in (204, 205) or (resp.status_code == 202 and len(resp.content) == 0):
+            rtnobj = None
+        else:
+            raw_data = resp.json()
+            rtnobj = Artifactentry.from_dict(raw_data)
+        return rtnobj
+
+    def artifacts_list_for_run(self, run_id: str, *, prefix: Optional[str] = None, page: Optional[int] = 1, per_page: Optional[int] = 50, exp_status: Optional[List[int]] = None, headers: Optional[Dict[str, str]] = None, aspects: Optional[AspectsRest] = None) -> Artifactentrypage:
+        """
+            List artifacts for a run.
+        """
+        client = self.client
+        if aspects is None:
+            aspects = client.aspects
+
+        header_params = client.headers
+        if headers is not None:
+            header_params = headers
+
+        leaf_template = "/runs/{run_id}/artifacts"
+        path_params = {
+            'run_id': run_id,
+        }
+        call_leaf = leaf_template.format(**path_params)
+
+        query_params = {}
+        if prefix is not None:
+            query_params['prefix'] = prefix
+        if page is not None:
+            query_params['page'] = page
+        if per_page is not None:
+            query_params['per_page'] = per_page
+
+
+        if exp_status is None:
+            exp_status = [200]
+
+        resp = client.perform_call(
+            leaf=call_leaf,
+            method="GET",
+            params=query_params if query_params else None,
+            body=None,
+            exp_status=exp_status,
+            headers=header_params if header_params else None,
+            context_msg=None,
+            aspects=aspects,
+        )
+        if resp.status_code in (204, 205) or (resp.status_code == 202 and len(resp.content) == 0):
+            rtnobj = None
+        else:
+            raw_data = resp.json()
+            rtnobj = Artifactentrypage.from_dict(raw_data)
+        return rtnobj

ccai/pychestrate/rest/client/groups/audit.py

@@ -0,0 +1,99 @@
+# Auto-generated by RestClientGenerator. DO NOT EDIT.
+
+from typing import Any, Dict, List, Optional
+
+from ccai.pychestrate.rest.client.models.audit import Auditrecord
+from ccai.restclient.base.aspectsrest import AspectsRest
+from ccai.restclient.base.restapigroup import RestApiGroup
+
+
+class AuditApiGroup(RestApiGroup):
+    """
+        Append-only audit log access.
+    """
+
+    def audit_get_by_sequence(self, sequence_number: int, *, exp_status: Optional[List[int]] = None, headers: Optional[Dict[str, str]] = None, aspects: Optional[AspectsRest] = None) -> Auditrecord:
+        """
+            Read a single audit record by sequence number.
+        """
+        client = self.client
+        if aspects is None:
+            aspects = client.aspects
+
+        header_params = client.headers
+        if headers is not None:
+            header_params = headers
+
+        leaf_template = "/audit/{sequence_number}"
+        path_params = {
+            'sequence_number': sequence_number,
+        }
+        call_leaf = leaf_template.format(**path_params)
+
+        query_params = None
+
+
+        if exp_status is None:
+            exp_status = [200]
+
+        resp = client.perform_call(
+            leaf=call_leaf,
+            method="GET",
+            params=query_params if query_params else None,
+            body=None,
+            exp_status=exp_status,
+            headers=header_params if header_params else None,
+            context_msg=None,
+            aspects=aspects,
+        )
+        if resp.status_code in (204, 205) or (resp.status_code == 202 and len(resp.content) == 0):
+            rtnobj = None
+        else:
+            raw_data = resp.json()
+            rtnobj = Auditrecord.from_dict(raw_data)
+        return rtnobj
+
+    def audit_list(self, *, limit: Optional[int] = 100, after_sequence: Optional[int] = None, actor_id: Optional[str] = None, action: Optional[str] = None, exp_status: Optional[List[int]] = None, headers: Optional[Dict[str, str]] = None, aspects: Optional[AspectsRest] = None) -> Dict[str, Any]:
+        """
+            List audit-log records.
+        """
+        client = self.client
+        if aspects is None:
+            aspects = client.aspects
+
+        header_params = client.headers
+        if headers is not None:
+            header_params = headers
+
+        call_leaf = "/audit"
+
+        query_params = {}
+        if limit is not None:
+            query_params['limit'] = limit
+        if after_sequence is not None:
+            query_params['after_sequence'] = after_sequence
+        if actor_id is not None:
+            query_params['actor_id'] = actor_id
+        if action is not None:
+            query_params['action'] = action
+
+
+        if exp_status is None:
+            exp_status = [200]
+
+        resp = client.perform_call(
+            leaf=call_leaf,
+            method="GET",
+            params=query_params if query_params else None,
+            body=None,
+            exp_status=exp_status,
+            headers=header_params if header_params else None,
+            context_msg=None,
+            aspects=aspects,
+        )
+        if resp.status_code in (204, 205) or (resp.status_code == 202 and len(resp.content) == 0):
+            rtnval = None
+        else:
+            raw_data = resp.json()
+            rtnval = self._parse_item(raw_data, Dict[str, Any])
+        return rtnval

ccai/pychestrate/rest/client/groups/auth.py

@@ -0,0 +1,55 @@
+# Auto-generated by RestClientGenerator. DO NOT EDIT.
+
+from typing import Dict, List, Optional
+
+from ccai.pychestrate.rest.client.models.auth import Logoutrequest
+from ccai.pychestrate.rest.client.models.auth import Logoutresponse
+from ccai.restclient.base.aspectsrest import AspectsRest
+from ccai.restclient.base.restapigroup import RestApiGroup
+
+
+class AuthApiGroup(RestApiGroup):
+    """
+        Local-credential authentication and session lifecycle.
+    """
+
+    def auth_logout(self, body: Logoutrequest, *, exp_status: Optional[List[int]] = None, headers: Optional[Dict[str, str]] = None, aspects: Optional[AspectsRest] = None) -> Logoutresponse:
+        """
+            Close the current session.
+            
+            Close the session named in the request body. Returns
+`{session_id, status: "closed"}` on success.
+
+        """
+        client = self.client
+        if aspects is None:
+            aspects = client.aspects
+
+        header_params = client.headers
+        if headers is not None:
+            header_params = headers
+
+        call_leaf = "/auth/logout"
+
+        query_params = None
+
+
+        if exp_status is None:
+            exp_status = [200]
+
+        resp = client.perform_call(
+            leaf=call_leaf,
+            method="POST",
+            params=query_params if query_params else None,
+            body=body,
+            exp_status=exp_status,
+            headers=header_params if header_params else None,
+            context_msg=None,
+            aspects=aspects,
+        )
+        if resp.status_code in (204, 205) or (resp.status_code == 202 and len(resp.content) == 0):
+            rtnobj = None
+        else:
+            raw_data = resp.json()
+            rtnobj = Logoutresponse.from_dict(raw_data)
+        return rtnobj