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