SIENTIAPDE-1314

Enhance OPC Metrics Handling and Refactor Write Operations

- Updated the OPC class to return response times for write operations, improving metrics tracking.
- Refactored the Gates activity to incorporate OPC metrics into the metrics writing process.
- Adjusted the manage_output_tags method in OpcRepository to return response times for each tag written.
- Modified tests to validate the new metrics structure and ensure correct behavior of the updated methods.

laborious/activities/gates.py

@@ -604,6 +604,7 @@ class Gates(BaseActivity):
         prediction = DataFrame(input_data['prediction'])
         prediction_confidence = prediction['prediction_confidence'].values[0]
         response_time = prediction['response_time'].values[0]
+        opc_metrics = input_data['opc_metrics']
 
         self.info(f'Writing metrics for model {metadata["model_name"]}', metadata)
 
@@ -625,4 +626,21 @@ class Gates(BaseActivity):
             pipeline_name=metadata['workflow_name'],
         ).observe(response_time)
 
+        for server_id, tags in opc_metrics.items():
+            for tag, response_time in tags.items():
+                metrics.PREDICTION_OPC_WRITING_RESPONSE_TIME_MONITOR.labels(
+                    pod_id=self.pod_id,
+                    model_name=metadata['model_name'],
+                    pipeline_name=metadata['workflow_name'],
+                    opc_server_id=server_id,
+                    tag=tag,
+                ).observe(response_time)
+                metrics.PREDICTION_OPC_WRITING_COUNT.labels(
+                    pod_id=self.pod_id,
+                    model_name=metadata['model_name'],
+                    pipeline_name=metadata['workflow_name'],
+                    opc_server_id=server_id,
+                    tag=tag,
+                ).inc()
+
         self.info(f'Metrics written for model {metadata["model_name"]}', metadata)