Add additional_middlewares argument into run_workflow of triage agent

Author: TomasKorbar

agents/tests/e2e/test_triage.py

@@ -3,6 +3,7 @@
 import os
 
 from agents.triage_agent import run_workflow, TriageState
+from agents.metrics_middleware import MetricsMiddleware
 from agents.observability import setup_observability
 from common.models import TriageOutputSchema, Resolution, BackportData
 
@@ -14,8 +15,10 @@ def __init__(self, input, expected_output):
         self.metrics: dict = None
 
     async def run(self) -> TriageState:
-        return await run_workflow(self.input, False)
-
+        metrics_middleware = MetricsMiddleware()
+        finished_state = await run_workflow(self.input, False, [metrics_middleware])
+        self.metrics = metrics_middleware.get_metrics()
+        return finished_state
 
 test_cases=[
     TriageAgentTestCase(input="RHEL-15216",
@@ -88,5 +91,4 @@ def verify_result(real_output: TriageOutputSchema, expected_output: TriageOutput
         assert real_output.data.fix_version == expected_output.data.fix_version
 
     finished_state = await test_case.run()
-    test_case.metrics = finished_state.metrics
     verify_result(finished_state.triage_result, test_case.expected_output)