@@ -47,16 +47,19 @@ def find_reporting_task(name: str, port: str):
4747    return  None 
4848
4949
50- def  create_reporting_task (name : str , port : str ,  version :  str ):
50+ def  create_reporting_task (name : str , port : str ):
5151    """Create a ReportingTask""" 
52+     nifi_version  =  nipyapi .system .get_system_diagnostics ().system_diagnostics .aggregate_snapshot .version_info .ni_fi_version 
5253    task  =  nipyapi .nifi .models .reporting_task_entity .ReportingTaskEntity (
5354        revision = nipyapi .nifi .models .revision_dto .RevisionDTO (version = 0 ),
5455        disconnected_node_acknowledged = False ,
5556        component = nipyapi .nifi .models .reporting_task_dto .ReportingTaskDTO (
5657            name = name ,
5758            type = "org.apache.nifi.reporting.prometheus.PrometheusReportingTask" ,
5859            bundle = nipyapi .nifi .models .bundle_dto .BundleDTO (
59-                 group = "org.apache.nifi" , artifact = "nifi-prometheus-nar" , version = version 
60+                 group = "org.apache.nifi" ,
61+                 artifact = "nifi-prometheus-nar" ,
62+                 version = nifi_version ,
6063            ),
6164            properties = {
6265                "prometheus-reporting-task-metrics-endpoint-port" : port ,
@@ -158,9 +161,7 @@ def main():
158161    reporting_task  =  find_reporting_task (name = task_name , port = port )
159162
160163    if  reporting_task  is  None :
161-         reporting_task  =  create_reporting_task (
162-             name = task_name , port = port , version = args ["nifi_version" ]
163-         )
164+         reporting_task  =  create_reporting_task (name = task_name , port = port )
164165        print (
165166            get_reporting_task_name (task = reporting_task )
166167            +  " [%s] -> CREATED"  %  reporting_task .id 
0 commit comments