Update metrics_tracker.py

metrics_tracker.py

@@ -3,12 +3,13 @@ import json
 import os
 
 class MetricsTracker:
+    
     def __init__(self):
         self.total_queries = 0
         self.successful_routings = 0
         self.failed_routings = 0
         self.response_times = []
-        self.routing_logs = []
+        self.logs = []
 
     def record_query(
         self,
@@ -17,10 +18,9 @@ class MetricsTracker:
         use_rag: bool,
         tool_name: str,
         routed_correctly: bool,
-        response_time: float
+        response_time: float,
     ):
         self.total_queries += 1
-
         if routed_correctly:
             self.successful_routings += 1
         else:
@@ -31,20 +31,19 @@ class MetricsTracker:
         log_entry = {
             "query": query,
             "model": model,
-            "rag_enabled": use_rag,
+            "use_rag": use_rag,
             "tool": tool_name,
+            "success": routed_correctly,
             "response_time": round(response_time, 2),
-            "status": "SUCCESS" if routed_correctly else "FAILED"
         }
 
-        self.routing_logs.append(log_entry)
+        self.logs.append(log_entry)
 
-        # 🔍 PRINT ROUTING LOG
-        print("\n=== ROUTING LOG ===")
+        # Print structured log
+        print("\n=== ROUTING EVENT ===")
         for k, v in log_entry.items():
             print(f"{k}: {v}")
-        print("====================\n")
-
+        print("=====================\n")
 
     def avg_response_time(self):
         if self.total_queries == 0: