Update main.py

Author: Santhosh-p653

agents/main.py

@@ -1,9 +1,10 @@
 import requests
 from fastapi import FastAPI
+
 from .log_analyser import analyse_logs
-from ml_client import call_ml
-from aggregator import aggregate
-from decider import decide
+from .ml_client import call_ml
+from .aggregator import aggregate
+from .decider import decide
 from shared.signal import Signal
 
 API_URL = "http://api:8000"
@@ -27,37 +28,47 @@ def run(payload: dict):
     record_id = payload.get("record_id")
     file_path = payload.get("file_path")
 
-    # call ml service
-    ml_result = call_ml(file_path, record_id)
+    if not record_id or not file_path:
+        return {"error": "record_id or file_path missing"}
+
+    # 🔹 Step 1 — call ML service
+    try:
+        ml_result = call_ml(file_path, record_id)
+    except Exception as e:
+        return {"error": f"ML call failed: {str(e)}"}
 
-    preprocessing = Signal(**ml_result["preprocessing"])
-    detection = Signal(**ml_result["detection"])
+    try:
+        preprocessing = Signal(**ml_result["preprocessing"])
+        detection = Signal(**ml_result["detection"])
+    except Exception as e:
+        return {"error": f"Invalid ML response: {str(e)}"}
 
-    # analyse logs
+    # 🔹 Step 2 — analyse logs
     log_signal = analyse_logs()
 
-    # aggregate all three signals
+    # 🔹 Step 3 — aggregate signals
     aggregated = aggregate(preprocessing, detection, log_signal)
 
-    # decide verdict
+    # 🔹 Step 4 — decide verdict
     decision = decide(aggregated, record_id)
 
-    # send verdict back to api
+    # 🔹 Step 5 — send verdict back to API
     try:
         requests.post(
             f"{API_URL}/verdict",
             json={
                 "record_id": record_id,
-                "verdict": decision["verdict"],
-                "verdict_score": decision["score"],
+                "verdict": decision.get("verdict"),
+                "verdict_score": decision.get("score"),
             },
             timeout=10,
         )
     except Exception:
+        # don't crash pipeline if API is temporarily unavailable
         pass
 
     return {
         "record_id": record_id,
         "aggregated": aggregated,
         "decision": decision,
-    }
+    }