diff --git a/playbooks/robusta_playbooks/krr.py b/playbooks/robusta_playbooks/krr.py index be589dac6..13bc38928 100644 --- a/playbooks/robusta_playbooks/krr.py +++ b/playbooks/robusta_playbooks/krr.py @@ -10,7 +10,6 @@ from hikaru.model.rel_1_26 import Container, EnvVar, EnvVarSource, PodSpec, ResourceRequirements, SecretKeySelector from prometrix import AWSPrometheusConfig, CoralogixPrometheusConfig, PrometheusAuthorization, PrometheusConfig from pydantic import BaseModel, ValidationError, validator - from robusta.api import ( IMAGE_REGISTRY, RUNNER_SERVICE_ACCOUNT, @@ -32,6 +31,7 @@ from robusta.core.reporting.consts import ScanState from robusta.integrations.openshift import IS_OPENSHIFT from robusta.integrations.prometheus.utils import generate_prometheus_config +from robusta.utils.parsing import format_event_templated_string IMAGE: str = os.getenv("KRR_IMAGE_OVERRIDE", f"{IMAGE_REGISTRY}/krr:v1.23.0") KRR_MEMORY_LIMIT: str = os.getenv("KRR_MEMORY_LIMIT", "2Gi") @@ -279,8 +279,13 @@ def krr_scan(event: ExecutionBaseEvent, params: KRRParams): scan_id = str(uuid.uuid4()) prom_config = generate_prometheus_config(params) additional_flags = get_krr_additional_flags(params) + args_sanitized = params.args_sanitized + + if params.args_sanitized and hasattr(event, "obj") and event.obj is not None: + subject = event.get_subject() + args_sanitized = format_event_templated_string(subject, params.args_sanitized) - python_command = f"python krr.py {params.strategy} {params.args_sanitized} {additional_flags} " + python_command = f"python krr.py {params.strategy} {args_sanitized} {additional_flags} " verbose_str = "-v" if params.krr_verbose else "" python_command += f"--max-workers {params.max_workers} {verbose_str} -f json --width 2048"