@@ -47,16 +47,21 @@ 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  =  (
53+         nipyapi .system .get_system_diagnostics ().system_diagnostics .aggregate_snapshot .version_info .ni_fi_version 
54+     )
5255    task  =  nipyapi .nifi .models .reporting_task_entity .ReportingTaskEntity (
5356        revision = nipyapi .nifi .models .revision_dto .RevisionDTO (version = 0 ),
5457        disconnected_node_acknowledged = False ,
5558        component = nipyapi .nifi .models .reporting_task_dto .ReportingTaskDTO (
5659            name = name ,
5760            type = "org.apache.nifi.reporting.prometheus.PrometheusReportingTask" ,
5861            bundle = nipyapi .nifi .models .bundle_dto .BundleDTO (
59-                 group = "org.apache.nifi" , artifact = "nifi-prometheus-nar" , version = version 
62+                 group = "org.apache.nifi" ,
63+                 artifact = "nifi-prometheus-nar" ,
64+                 version = nifi_version ,
6065            ),
6166            properties = {
6267                "prometheus-reporting-task-metrics-endpoint-port" : port ,
@@ -159,7 +164,7 @@ def main():
159164
160165    if  reporting_task  is  None :
161166        reporting_task  =  create_reporting_task (
162-             name = task_name , port = port ,  version = args [ "nifi_version" ] 
167+             name = task_name , port = port 
163168        )
164169        print (
165170            get_reporting_task_name (task = reporting_task )
0 commit comments