Skip to content

batch_get_threat_models async

batch_get_threat_models(
    threat_model_ids: ThreatModelIdList,
    agent_space_id: str,
    *,
    config_overrides: Optional[AsyncSecurityAgentClientConfig] = None,
) -> BatchGetThreatModelsOutput

Retrieves information about one or more threat models in an agent space.

Parameters:

  • threat_model_ids (ThreatModelIdList) –

    The list of threat model identifiers to retrieve.

  • agent_space_id (str) –

    The unique identifier of the agent space that contains the threat models.

Raises:

Source code in docs_src/capo_securityagent/_services/async_security_agent.py
async def batch_get_threat_models(
    self,
    threat_model_ids: "capo_securityagent.types.threat_model_id_list.ThreatModelIdList",
    agent_space_id: str,
    *,
    config_overrides: Optional[AsyncSecurityAgentClientConfig] = None,
) -> "capo_securityagent.types.batch_get_threat_models_output.BatchGetThreatModelsOutput":
    """<p>Retrieves information about one or more threat models in an agent space.</p>

    Args:
        threat_model_ids: <p>The list of threat model identifiers to retrieve.</p>
        agent_space_id: <p>The unique identifier of the agent space that contains the threat models.</p>

    Raises:
        capo_securityagent.errors.UnknownServiceError: The service returned an error code this client does not model.
    """

    async def _handler(
        req: "AsyncOperationRequest[capo_securityagent.types.batch_get_threat_models_input.BatchGetThreatModelsInput]",
    ) -> AsyncOperationResponse[
        "capo_securityagent.types.batch_get_threat_models_output.BatchGetThreatModelsOutput"
    ]:
        import capo_securityagent._operations.security_agent.batch_get_threat_models

        (
            output,
            http_response,
        ) = await capo_securityagent._operations.security_agent.batch_get_threat_models.async_batch_get_threat_models(
            req.options, req.input
        )
        return AsyncOperationResponse(output=output, response=http_response)

    def _redact_output(
        output: "capo_securityagent.types.batch_get_threat_models_output.BatchGetThreatModelsOutput",
    ) -> dict[str, Any]:
        import capo_securityagent.types.batch_get_threat_models_output

        return capo_securityagent.types.batch_get_threat_models_output.redact_sensitive(
            output
        )

    interceptors_, options_ = self.operation_options(
        config_overrides,
        OperationLogging("BatchGetThreatModels", redact_output=_redact_output),
    )
    input_: capo_securityagent.types.batch_get_threat_models_input.BatchGetThreatModelsInput = {
        "threat_model_ids": threat_model_ids,
        "agent_space_id": agent_space_id,
    }

    response = await aexecute_pipeline(
        AsyncOperationRequest(input=input_, options=options_),
        handler=_handler,
        interceptors=list(interceptors_),
    )
    await response.response.aclose()
    return response.output