capo-devops-agent 0.7.0__tar.gz → 0.8.0__tar.gz

{capo_devops_agent-0.7.0/src/capo_devops_agent.egg-info → capo_devops_agent-0.8.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: capo-devops-agent
-Version: 0.7.0
+Version: 0.8.0
 Summary: Python SDK for AWS Devops Agent.
 Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Developers

{capo_devops_agent-0.7.0 → capo_devops_agent-0.8.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "capo-devops-agent"
-version = "0.7.0"
+version = "0.8.0"
 description = "Python SDK for AWS Devops Agent."
 readme = "README.md"
 requires-python = ">=3.10"

{capo_devops_agent-0.7.0 → capo_devops_agent-0.8.0}/src/capo_devops_agent/__init__.py

@@ -16,6 +16,9 @@ from ._auth._providers import (
 from ._auth._providers import (
     ChainedProvider as ChainedProvider,
 )
+from ._auth._providers import (
+    CredentialProcessError as CredentialProcessError,
+)
 from ._auth._providers import (
     CredentialsProvider as CredentialsProvider,
 )
@@ -31,6 +34,9 @@ from ._auth._providers import (
 from ._auth._providers import (
     MissingDependencyError as MissingDependencyError,
 )
+from ._auth._providers import (
+    ProcessCredentialsProvider as ProcessCredentialsProvider,
+)
 from ._auth._providers import (
     ProfileCredentialsProvider as ProfileCredentialsProvider,
 )

{capo_devops_agent-0.7.0 → capo_devops_agent-0.8.0}/src/capo_devops_agent/_auth/_providers.py

@@ -4,6 +4,7 @@ import configparser
 import hashlib
 import json
 import os
+import subprocess
 import time
 from abc import abstractmethod
 from dataclasses import dataclass
@@ -59,6 +60,10 @@ class IdentityNotFound(Exception):
     """Raised when a provider cannot resolve an identity. Chain continues."""
 
 
+class CredentialProcessError(Exception):
+    """Raised when a profile's `credential_process` fails or prints unusable output. Chain stops."""
+
+
 class MissingDependencyError(Exception):
     """Raised when a provider needs the `sso` extra but it is not installed."""
 
@@ -186,13 +191,27 @@ class ProfileCredentialsProvider(CredentialsProvider):
         self, credentials_file: Path | None = None, profile: str | None = None
     ) -> None:
         self._profile = profile or active_profile()
-        self._cred_file = credentials_file or Path(
-            os.environ.get("AWS_SHARED_CREDENTIALS_FILE")
-            or Path.home() / ".aws" / "credentials"
-        )
+        self._cred_file = credentials_file or _shared_credentials_file()
 
     def resolve_identity(self) -> Credentials:
-        section = self._load_section()
+        from_config, _ = _load_profile(self._profile)
+        from_credentials_file = _credentials_file_section(
+            self._profile, self._cred_file
+        )
+        section = {**from_config, **from_credentials_file}
+        if not section:
+            raise IdentityNotFound(
+                f"profile {self._profile!r} not found in credentials/config files"
+            )
+        # botocore's order: keys from the credentials file, then credential_process, then keys from the
+        # config file, so a profile with a process only takes its keys from the credentials file
+        if "credential_process" in section and not (
+            from_credentials_file.get("aws_access_key_id")
+            and from_credentials_file.get("aws_secret_access_key")
+        ):
+            raise IdentityNotFound(
+                f"profile {self._profile!r}: credential_process comes before the keys in the config file"
+            )
         ak = section.get("aws_access_key_id")
         sk = section.get("aws_secret_access_key")
         if not ak or not sk:
@@ -205,19 +224,94 @@ class ProfileCredentialsProvider(CredentialsProvider):
             out["session_token"] = token
         return out
 
-    def _load_section(self) -> dict[str, str]:
-        # config-file profile reuses the loader from _services/_aws_config
-        merged, _ = _load_profile(self._profile)
-        if self._cred_file.is_file():
-            cfg = configparser.ConfigParser(interpolation=None)
-            cfg.read(self._cred_file)
-            if cfg.has_section(self._profile):
-                merged.update(dict(cfg.items(self._profile)))
-        if not merged:
-            raise IdentityNotFound(
-                f"profile {self._profile!r} not found in credentials/config files"
+
+class ProcessCredentialsProvider(CredentialsProvider):
+    """Run the profile's ``credential_process`` command and read the credentials it prints.
+
+    This is how helpers such as ``aws-sso-cli`` hand out short-term credentials. The command
+    runs through the system shell, as in the Rust, Kotlin and Swift SDKs, so a quoted Windows path
+    works as written, and it must print the JSON document the AWS SDKs define (``Version`` 1).
+    The key is read from the credentials file as well as the config file. A profile without
+    ``credential_process`` is skipped; a command that fails or outlives the timeout stops the chain.
+    """
+
+    def __init__(
+        self,
+        credentials_file: Path | None = None,
+        profile: str | None = None,
+        timeout: float = 60.0,
+    ) -> None:
+        self._profile = profile or active_profile()
+        self._cred_file = credentials_file or _shared_credentials_file()
+        self._timeout = timeout
+
+    def resolve_identity(self) -> Credentials:
+        section = _load_merged_profile(self._profile, self._cred_file)
+        command = section.get("credential_process")
+        if not command:
+            raise IdentityNotFound(f"profile {self._profile!r}: no credential_process")
+        try:
+            # the shell is sh -c on POSIX and cmd.exe /c on Windows, which parses the path natively
+            result = subprocess.run(
+                command,
+                shell=True,
+                capture_output=True,
+                text=True,
+                check=False,
+                timeout=self._timeout,
+            )
+        except subprocess.TimeoutExpired:
+            raise CredentialProcessError(
+                f"profile {self._profile!r}: credential_process did not finish within {self._timeout} seconds"
+            ) from None
+        except OSError as e:
+            raise CredentialProcessError(f"profile {self._profile!r}: {e}") from e
+        # stdout carries the secret, so only stderr ever goes into an error message
+        if result.returncode != 0:
+            raise CredentialProcessError(
+                f"profile {self._profile!r}: credential_process exited with {result.returncode}: {result.stderr.strip()}"
+            )
+        try:
+            data = json.loads(result.stdout)
+        except ValueError:
+            raise CredentialProcessError(
+                f"profile {self._profile!r}: credential_process did not print JSON"
+            ) from None
+        if not isinstance(data, dict) or data.get("Version") != 1:
+            raise CredentialProcessError(
+                f"profile {self._profile!r}: credential_process output must be a JSON object with Version 1"
             )
-        return merged
+        try:
+            return _credentials_from_json(data, token_key="SessionToken")
+        except IdentityNotFound as e:
+            raise CredentialProcessError(f"profile {self._profile!r}: {e}") from None
+        except ValueError:
+            raise CredentialProcessError(
+                f"profile {self._profile!r}: credential_process printed an invalid Expiration"
+            ) from None
+
+
+def _shared_credentials_file() -> Path:
+    return Path(
+        os.environ.get("AWS_SHARED_CREDENTIALS_FILE")
+        or Path.home() / ".aws" / "credentials"
+    )
+
+
+def _load_merged_profile(profile: str, credentials_file: Path) -> dict[str, str]:
+    """The profile's keys from the config file, overlaid by the credentials file, as the AWS SDKs merge them."""
+    # config-file profile reuses the loader from _services/_aws_config
+    merged, _ = _load_profile(profile)
+    merged.update(_credentials_file_section(profile, credentials_file))
+    return merged
+
+
+def _credentials_file_section(profile: str, credentials_file: Path) -> dict[str, str]:
+    if not credentials_file.is_file():
+        return {}
+    cfg = configparser.ConfigParser(interpolation=None)
+    cfg.read(credentials_file)
+    return dict(cfg.items(profile)) if cfg.has_section(profile) else {}
 
 
 class EcsContainerCredentialsProvider(CredentialsProvider):
@@ -721,6 +815,7 @@ class AssumeRoleCredentialsProvider(CredentialsProvider):
             ),
             SsoCredentialsProvider(self._client, source_profile),
             ProfileCredentialsProvider(profile=source_profile),
+            ProcessCredentialsProvider(profile=source_profile),
         )
 
 
@@ -853,7 +948,10 @@ def _parse_iso8601(value: str) -> datetime:
     return datetime.fromisoformat(value.replace("Z", "+00:00"))
 
 
-def _credentials_from_json(data: dict[str, object]) -> Credentials:
+def _credentials_from_json(
+    data: dict[str, object], *, token_key: str = "Token"
+) -> Credentials:
+    # ECS and IMDS name the session token Token; credential_process names it SessionToken
     ak = data.get("AccessKeyId")
     sk = data.get("SecretAccessKey")
     if not isinstance(ak, str) or not isinstance(sk, str):
@@ -861,7 +959,7 @@ def _credentials_from_json(data: dict[str, object]) -> Credentials:
             "credentials response missing AccessKeyId/SecretAccessKey"
         )
     out: Credentials = {"access_key": ak, "secret_key": sk}
-    token = data.get("Token")
+    token = data.get(token_key)
     if isinstance(token, str):
         out["session_token"] = token
     exp = data.get("Expiration")
@@ -880,6 +978,7 @@ def default_aws_credentials_chain(
             WebIdentityCredentialsProvider(client),
             SsoCredentialsProvider(client),
             ProfileCredentialsProvider(),
+            ProcessCredentialsProvider(),
             EcsContainerCredentialsProvider(client),
             Ec2InstanceMetadataProvider(client),
         )

{capo_devops_agent-0.7.0 → capo_devops_agent-0.8.0}/src/capo_devops_agent/_operations/dev_ops_agent/get_agent_space.py

@@ -24,6 +24,7 @@ import capo_devops_agent.errors.validation_exception
 import capo_devops_agent.types.agent_space
 import capo_devops_agent.types.get_agent_space_input
 import capo_devops_agent.types.get_agent_space_output
+import capo_devops_agent.types.included_data
 import capo_devops_agent.types.tags
 from capo_devops_agent._protocol.errors import parse_error_metadata_json
 from capo_devops_agent._rule_engine._endpoint_rule_set import EndpointParams, resolve
@@ -138,9 +139,20 @@ def build_request(
             UseFIPS=options.use_fips, Endpoint=options.endpoint, Region=options.region
         )
     )  # noqa: F841
+    import capo_devops_agent.types.included_data
+
     url = endpoint.url.rstrip("/") + "/v1/agentspaces/{agentSpaceId}"
     url = url.replace("{agentSpaceId}", quote(input_["agent_space_id"], safe=""))
     params: list[tuple[str, str]] = []
+    if "included_data" in input_:
+        params.append(
+            (
+                "includedData",
+                capo_devops_agent.types.included_data.serialize_json(
+                    input_["included_data"]
+                ),
+            )
+        )
     headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
     body: bytes | None = b""
     signer = (

{capo_devops_agent-0.7.0 → capo_devops_agent-0.8.0}/src/capo_devops_agent/_operations/dev_ops_agent/list_agent_spaces.py

@@ -21,6 +21,7 @@ import capo_devops_agent.errors.service_quota_exceeded_exception
 import capo_devops_agent.errors.throttling_exception
 import capo_devops_agent.errors.validation_exception
 import capo_devops_agent.types.agent_space_list
+import capo_devops_agent.types.included_data
 import capo_devops_agent.types.list_agent_spaces_input
 import capo_devops_agent.types.list_agent_spaces_output
 from capo_devops_agent._protocol.errors import parse_error_metadata_json
@@ -136,11 +137,22 @@ def build_request(
             UseFIPS=options.use_fips, Endpoint=options.endpoint, Region=options.region
         )
     )  # noqa: F841
+    import capo_devops_agent.types.included_data
+
     url = endpoint.url.rstrip("/") + "/v1/agentspaces/list"
     params: list[tuple[str, str]] = []
     params.append(("maxResults", str(input_.get("max_results", 20))))
     if "next_token" in input_:
         params.append(("nextToken", input_["next_token"]))
+    if "included_data" in input_:
+        params.append(
+            (
+                "includedData",
+                capo_devops_agent.types.included_data.serialize_json(
+                    input_["included_data"]
+                ),
+            )
+        )
     headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
     body: bytes | None = b""
     signer = (

{capo_devops_agent-0.7.0 → capo_devops_agent-0.8.0}/src/capo_devops_agent/_resources/dev_ops_agent/agent_space_resource.py

@@ -35,6 +35,7 @@ if TYPE_CHECKING:
     import capo_devops_agent.types.get_operator_app_output
     import capo_devops_agent.types.idp_client_id
     import capo_devops_agent.types.idp_client_secret
+    import capo_devops_agent.types.included_data
     import capo_devops_agent.types.kms_key_arn
     import capo_devops_agent.types.list_agent_spaces_input
     import capo_devops_agent.types.list_agent_spaces_output
@@ -168,11 +169,15 @@ class AgentSpaceResource:
         agent_space_id: "capo_devops_agent.types.agent_space_id.AgentSpaceId",
         *,
         config_overrides: Optional[DevOpsAgentClientConfig] = None,
+        included_data: Optional[
+            "capo_devops_agent.types.included_data.IncludedData"
+        ] = None,
     ) -> "capo_devops_agent.types.get_agent_space_output.GetAgentSpaceOutput":
         """<p>Retrieves detailed information about a specific AgentSpace.</p>
 
         Args:
             agent_space_id: <p>The unique identifier of the AgentSpace</p>
+            included_data: <p>How much data to include in the response.</p>
 
         Raises:
             capo_devops_agent.errors.access_denied_exception.AccessDeniedException: <p>Access to the requested resource is denied due to insufficient permissions.</p>
@@ -217,6 +222,8 @@ class AgentSpaceResource:
         input_: capo_devops_agent.types.get_agent_space_input.GetAgentSpaceInput = {
             "agent_space_id": agent_space_id
         }
+        if included_data is not None:
+            input_["included_data"] = included_data
 
         response = execute_pipeline(
             OperationRequest(input=input_, options=options_),
@@ -653,12 +660,16 @@ class AgentSpaceResource:
         config_overrides: Optional[DevOpsAgentClientConfig] = None,
         max_results: Optional[int] = None,
         next_token: Optional["capo_devops_agent.types.next_token.NextToken"] = None,
+        included_data: Optional[
+            "capo_devops_agent.types.included_data.IncludedData"
+        ] = None,
     ) -> "capo_devops_agent.types.list_agent_spaces_output.ListAgentSpacesOutput":
         """<p>Lists all AgentSpaces with optional pagination.</p>
 
         Args:
             max_results: <p>Maximum number of results to return in a single call.</p>
             next_token: <p>Token for the next page of results.</p>
+            included_data: <p>How much data to include in the response.</p>
 
         Raises:
             capo_devops_agent.errors.access_denied_exception.AccessDeniedException: <p>Access to the requested resource is denied due to insufficient permissions.</p>
@@ -705,6 +716,8 @@ class AgentSpaceResource:
             input_["max_results"] = max_results
         if next_token is not None:
             input_["next_token"] = next_token
+        if included_data is not None:
+            input_["included_data"] = included_data
 
         response = execute_pipeline(
             OperationRequest(input=input_, options=options_),
@@ -828,11 +841,15 @@ class AsyncAgentSpaceResource:
         agent_space_id: "capo_devops_agent.types.agent_space_id.AgentSpaceId",
         *,
         config_overrides: Optional[AsyncDevOpsAgentClientConfig] = None,
+        included_data: Optional[
+            "capo_devops_agent.types.included_data.IncludedData"
+        ] = None,
     ) -> "capo_devops_agent.types.get_agent_space_output.GetAgentSpaceOutput":
         """<p>Retrieves detailed information about a specific AgentSpace.</p>
 
         Args:
             agent_space_id: <p>The unique identifier of the AgentSpace</p>
+            included_data: <p>How much data to include in the response.</p>
 
         Raises:
             capo_devops_agent.errors.access_denied_exception.AccessDeniedException: <p>Access to the requested resource is denied due to insufficient permissions.</p>
@@ -878,6 +895,8 @@ class AsyncAgentSpaceResource:
         input_: capo_devops_agent.types.get_agent_space_input.GetAgentSpaceInput = {
             "agent_space_id": agent_space_id
         }
+        if included_data is not None:
+            input_["included_data"] = included_data
 
         response = await aexecute_pipeline(
             AsyncOperationRequest(input=input_, options=options_),
@@ -1320,12 +1339,16 @@ class AsyncAgentSpaceResource:
         config_overrides: Optional[AsyncDevOpsAgentClientConfig] = None,
         max_results: Optional[int] = None,
         next_token: Optional["capo_devops_agent.types.next_token.NextToken"] = None,
+        included_data: Optional[
+            "capo_devops_agent.types.included_data.IncludedData"
+        ] = None,
     ) -> "capo_devops_agent.types.list_agent_spaces_output.ListAgentSpacesOutput":
         """<p>Lists all AgentSpaces with optional pagination.</p>
 
         Args:
             max_results: <p>Maximum number of results to return in a single call.</p>
             next_token: <p>Token for the next page of results.</p>
+            included_data: <p>How much data to include in the response.</p>
 
         Raises:
             capo_devops_agent.errors.access_denied_exception.AccessDeniedException: <p>Access to the requested resource is denied due to insufficient permissions.</p>
@@ -1373,6 +1396,8 @@ class AsyncAgentSpaceResource:
             input_["max_results"] = max_results
         if next_token is not None:
             input_["next_token"] = next_token
+        if included_data is not None:
+            input_["included_data"] = included_data
 
         response = await aexecute_pipeline(
             AsyncOperationRequest(input=input_, options=options_),

{capo_devops_agent-0.7.0 → capo_devops_agent-0.8.0}/src/capo_devops_agent/_services/async_dev_ops_agent.py

@@ -139,6 +139,7 @@ if TYPE_CHECKING:
     import capo_devops_agent.types.goal_type
     import capo_devops_agent.types.idp_client_id
     import capo_devops_agent.types.idp_client_secret
+    import capo_devops_agent.types.included_data
     import capo_devops_agent.types.journal_record
     import capo_devops_agent.types.kms_key_arn
     import capo_devops_agent.types.list_agent_spaces_input
@@ -3221,11 +3222,15 @@ class AsyncDevOpsAgentClient:
         agent_space_id: "capo_devops_agent.types.agent_space_id.AgentSpaceId",
         *,
         config_overrides: Optional[AsyncDevOpsAgentClientConfig] = None,
+        included_data: Optional[
+            "capo_devops_agent.types.included_data.IncludedData"
+        ] = None,
     ) -> "capo_devops_agent.types.get_agent_space_output.GetAgentSpaceOutput":
         """<p>Retrieves detailed information about a specific AgentSpace.</p>
 
         Args:
             agent_space_id: <p>The unique identifier of the AgentSpace</p>
+            included_data: <p>How much data to include in the response.</p>
 
         Raises:
             capo_devops_agent.errors.access_denied_exception.AccessDeniedException: <p>Access to the requested resource is denied due to insufficient permissions.</p>
@@ -3271,6 +3276,8 @@ class AsyncDevOpsAgentClient:
         input_: capo_devops_agent.types.get_agent_space_input.GetAgentSpaceInput = {
             "agent_space_id": agent_space_id
         }
+        if included_data is not None:
+            input_["included_data"] = included_data
 
         response = await aexecute_pipeline(
             AsyncOperationRequest(input=input_, options=options_),
@@ -3713,12 +3720,16 @@ class AsyncDevOpsAgentClient:
         config_overrides: Optional[AsyncDevOpsAgentClientConfig] = None,
         max_results: Optional[int] = None,
         next_token: Optional["capo_devops_agent.types.next_token.NextToken"] = None,
+        included_data: Optional[
+            "capo_devops_agent.types.included_data.IncludedData"
+        ] = None,
     ) -> "capo_devops_agent.types.list_agent_spaces_output.ListAgentSpacesOutput":
         """<p>Lists all AgentSpaces with optional pagination.</p>
 
         Args:
             max_results: <p>Maximum number of results to return in a single call.</p>
             next_token: <p>Token for the next page of results.</p>
+            included_data: <p>How much data to include in the response.</p>
 
         Raises:
             capo_devops_agent.errors.access_denied_exception.AccessDeniedException: <p>Access to the requested resource is denied due to insufficient permissions.</p>
@@ -3766,6 +3777,8 @@ class AsyncDevOpsAgentClient:
             input_["max_results"] = max_results
         if next_token is not None:
             input_["next_token"] = next_token
+        if included_data is not None:
+            input_["included_data"] = included_data
 
         response = await aexecute_pipeline(
             AsyncOperationRequest(input=input_, options=options_),
@@ -3781,6 +3794,9 @@ class AsyncDevOpsAgentClient:
         config_overrides: Optional[AsyncDevOpsAgentClientConfig] = None,
         max_results: Optional[int] = None,
         next_token: Optional["capo_devops_agent.types.next_token.NextToken"] = None,
+        included_data: Optional[
+            "capo_devops_agent.types.included_data.IncludedData"
+        ] = None,
     ) -> "AsyncIterator[capo_devops_agent.types.agent_space.AgentSpace]":
         _token = next_token
         while True:
@@ -3788,6 +3804,7 @@ class AsyncDevOpsAgentClient:
                 config_overrides=config_overrides,
                 max_results=max_results,
                 next_token=_token,
+                included_data=included_data,
             )
             _page = _resolve_path(_response, ("agent_spaces",))
             for _item in _page or []:

{capo_devops_agent-0.7.0 → capo_devops_agent-0.8.0}/src/capo_devops_agent/_services/dev_ops_agent.py

@@ -135,6 +135,7 @@ if TYPE_CHECKING:
     import capo_devops_agent.types.goal_type
     import capo_devops_agent.types.idp_client_id
     import capo_devops_agent.types.idp_client_secret
+    import capo_devops_agent.types.included_data
     import capo_devops_agent.types.journal_record
     import capo_devops_agent.types.kms_key_arn
     import capo_devops_agent.types.list_agent_spaces_input
@@ -3180,11 +3181,15 @@ class DevOpsAgentClient:
         agent_space_id: "capo_devops_agent.types.agent_space_id.AgentSpaceId",
         *,
         config_overrides: Optional[DevOpsAgentClientConfig] = None,
+        included_data: Optional[
+            "capo_devops_agent.types.included_data.IncludedData"
+        ] = None,
     ) -> "capo_devops_agent.types.get_agent_space_output.GetAgentSpaceOutput":
         """<p>Retrieves detailed information about a specific AgentSpace.</p>
 
         Args:
             agent_space_id: <p>The unique identifier of the AgentSpace</p>
+            included_data: <p>How much data to include in the response.</p>
 
         Raises:
             capo_devops_agent.errors.access_denied_exception.AccessDeniedException: <p>Access to the requested resource is denied due to insufficient permissions.</p>
@@ -3229,6 +3234,8 @@ class DevOpsAgentClient:
         input_: capo_devops_agent.types.get_agent_space_input.GetAgentSpaceInput = {
             "agent_space_id": agent_space_id
         }
+        if included_data is not None:
+            input_["included_data"] = included_data
 
         response = execute_pipeline(
             OperationRequest(input=input_, options=options_),
@@ -3665,12 +3672,16 @@ class DevOpsAgentClient:
         config_overrides: Optional[DevOpsAgentClientConfig] = None,
         max_results: Optional[int] = None,
         next_token: Optional["capo_devops_agent.types.next_token.NextToken"] = None,
+        included_data: Optional[
+            "capo_devops_agent.types.included_data.IncludedData"
+        ] = None,
     ) -> "capo_devops_agent.types.list_agent_spaces_output.ListAgentSpacesOutput":
         """<p>Lists all AgentSpaces with optional pagination.</p>
 
         Args:
             max_results: <p>Maximum number of results to return in a single call.</p>
             next_token: <p>Token for the next page of results.</p>
+            included_data: <p>How much data to include in the response.</p>
 
         Raises:
             capo_devops_agent.errors.access_denied_exception.AccessDeniedException: <p>Access to the requested resource is denied due to insufficient permissions.</p>
@@ -3717,6 +3728,8 @@ class DevOpsAgentClient:
             input_["max_results"] = max_results
         if next_token is not None:
             input_["next_token"] = next_token
+        if included_data is not None:
+            input_["included_data"] = included_data
 
         response = execute_pipeline(
             OperationRequest(input=input_, options=options_),
@@ -3732,6 +3745,9 @@ class DevOpsAgentClient:
         config_overrides: Optional[DevOpsAgentClientConfig] = None,
         max_results: Optional[int] = None,
         next_token: Optional["capo_devops_agent.types.next_token.NextToken"] = None,
+        included_data: Optional[
+            "capo_devops_agent.types.included_data.IncludedData"
+        ] = None,
     ) -> "Iterator[capo_devops_agent.types.agent_space.AgentSpace]":
         _token = next_token
         while True:
@@ -3739,6 +3755,7 @@ class DevOpsAgentClient:
                 config_overrides=config_overrides,
                 max_results=max_results,
                 next_token=_token,
+                included_data=included_data,
             )
             _page = _resolve_path(_response, ("agent_spaces",))
             for _item in _page or []:

capo_devops_agent-0.8.0/src/capo_devops_agent/types/custom_header_names.py

@@ -0,0 +1,19 @@
+"""Generated from Smithy shape ``com.amazonaws.devopsagent#CustomHeaderNames``."""
+
+from typing import TYPE_CHECKING, TypeAlias
+
+if TYPE_CHECKING:
+    import capo_devops_agent.types.custom_header_name
+
+CustomHeaderNames: TypeAlias = list[
+    "capo_devops_agent.types.custom_header_name.CustomHeaderName"
+]
+
+
+# --- restJson1 ser/de ---
+def serialize_json(value: CustomHeaderNames) -> list:
+    return list(value)
+
+
+def deserialize_json(data: list) -> CustomHeaderNames:
+    return [item for item in data if item is not None]

{capo_devops_agent-0.7.0 → capo_devops_agent-0.8.0}/src/capo_devops_agent/types/get_account_usage_output.py

@@ -28,6 +28,9 @@ class GetAccountUsageOutput(TypedDict, closed=True):
         "capo_devops_agent.types.usage_metric.UsageMetric"
     ]
     """<p>Monthly on-demand hours usage and limit for an account</p>"""
+    monthly_account_grader_hours: NotRequired[
+        "capo_devops_agent.types.usage_metric.UsageMetric"
+    ]
     usage_period_start_time: "capo_devops_agent.types.date_time.DateTime"
     """<p>The start time of the usage tracking period</p>"""
     usage_period_end_time: "capo_devops_agent.types.date_time.DateTime"
@@ -69,6 +72,14 @@ def serialize_json(value: GetAccountUsageOutput) -> dict:
                 value["monthly_account_on_demand_hours"]
             )
         )
+    if "monthly_account_grader_hours" in value:
+        import capo_devops_agent.types.usage_metric
+
+        out["monthlyAccountGraderHours"] = (
+            capo_devops_agent.types.usage_metric.serialize_json(
+                value["monthly_account_grader_hours"]
+            )
+        )
     import capo_devops_agent.types.date_time
 
     out["usagePeriodStartTime"] = capo_devops_agent.types.date_time.serialize_json(
@@ -116,6 +127,14 @@ def deserialize_json(data: dict) -> GetAccountUsageOutput:
                 data["monthlyAccountOnDemandHours"]
             )
         )
+    if data.get("monthlyAccountGraderHours") is not None:
+        import capo_devops_agent.types.usage_metric
+
+        out["monthly_account_grader_hours"] = (
+            capo_devops_agent.types.usage_metric.deserialize_json(
+                data["monthlyAccountGraderHours"]
+            )
+        )
     if data.get("usagePeriodStartTime") is not None:
         import capo_devops_agent.types.date_time
 

{capo_devops_agent-0.7.0 → capo_devops_agent-0.8.0}/src/capo_devops_agent/types/get_agent_space_input.py

@@ -2,15 +2,18 @@
 
 from typing import TYPE_CHECKING
 
-from typing_extensions import TypedDict
+from typing_extensions import NotRequired, TypedDict
 
 if TYPE_CHECKING:
     import capo_devops_agent.types.agent_space_id
+    import capo_devops_agent.types.included_data
 
 
 class GetAgentSpaceInput(TypedDict, closed=True):
     agent_space_id: "capo_devops_agent.types.agent_space_id.AgentSpaceId"
     """<p>The unique identifier of the AgentSpace</p>"""
+    included_data: NotRequired["capo_devops_agent.types.included_data.IncludedData"]
+    """<p>How much data to include in the response.</p>"""
 
 
 # --- restJson1 ser/de ---

capo_devops_agent-0.8.0/src/capo_devops_agent/types/included_data.py

@@ -0,0 +1,18 @@
+"""Generated from Smithy shape ``com.amazonaws.devopsagent#IncludedData``."""
+
+from typing import Literal, TypeAlias, cast
+
+"""<p>How much data to include in the response. Defaults to ALL_DATA.</p>"""
+IncludedData: TypeAlias = Literal[
+    "METADATA_ONLY",
+    "ALL_DATA",
+]
+
+
+# --- restJson1 ser/de ---
+def serialize_json(value: IncludedData) -> str:
+    return value
+
+
+def deserialize_json(data: str) -> IncludedData:
+    return cast(IncludedData, data)