Update evaluator.py

evaluator.py

@@ -199,7 +199,15 @@ def generate_executive_summary(reports: list) -> dict:
         'errors_detected': sum(len(r['failures']) for r in reports)
     }
 
-def evaluate_trajectories_batch(input_data: Union[str, List[Dict[str, Any]]], profile: str = None, run_stress_test: bool = False) -> dict:
+MAX_FREE_STEPS = 20
+
+def evaluate_trajectories_batch(
+    input_data: Union[str, List[Dict[str, Any]]], 
+    profile: str = None, 
+    run_stress_test: bool = False,
+    plan: str = "free"  
+) -> dict:
+    
     if isinstance(input_data, str):
         with open(input_data, 'r', encoding='utf-8') as f:
             sessions_data = pd.DataFrame(json.load(f))
@@ -208,6 +216,15 @@ def evaluate_trajectories_batch(input_data: Union[str, List[Dict[str, Any]]], pr
     else:
         raise ValueError("Invalid input format.")
 
+    if plan == "free":
+        for _, session in sessions_data.iterrows():
+            node_count = len(session.get("nodes", []))
+            if node_count > MAX_FREE_STEPS:
+                return {
+                    "error": f"Free Tier Limit Exceeded: Session [{session.get('session_id', 'unknown')}] has {node_count} steps (Max allowed on Free: {MAX_FREE_STEPS}). Upgrade to Pro for unlimited trajectory depth.",
+                    "status_code": 429
+                }
+
     active_profile = profile or DEFAULT_STRICTNESS_PROFILE
     profile_cfg = PROFILES_CONFIG.get(active_profile.lower(), PROFILES_CONFIG["standard"])
     max_tool_latency = profile_cfg["max_latency_ms"]