Update services/report_data_handler.py

services/report_data_handler.py

@@ -60,6 +60,37 @@ def fetch_latest_agentic_analysis(org_urn: str) -> Tuple[Optional[pd.DataFrame],
         logger.exception(f"An unexpected error occurred in fetch_latest_agentic_analysis for org_urn {org_urn}: {e}")
         return None, str(e)
 
+def _get_metrics_for_subject(data_subject: Optional[str], metrics: Dict[str, Any]) -> Optional[Dict[str, Any]]:
+    """
+    Selects the appropriate metrics dictionary from the main metrics object
+    based on the key result's data_subject.
+
+    Args:
+        data_subject: The data subject string from the key result (e.g., 'follower_stats').
+        metrics: The dictionary containing all available metrics for the agents.
+
+    Returns:
+        The relevant metrics dictionary, or None if no match is found.
+    """
+    if not data_subject or not metrics:
+        return None
+
+    # This mapping connects the data_subject string values to the keys in the `metrics` parameter.
+    METRICS_KEY_MAPPING = {
+        "follower_stats": "follower_agent",
+        "posts": "post_agent",
+        "mentions": "mentions_agent",
+    }
+
+    # Find the corresponding key for the metrics dictionary (e.g., 'follower_agent').
+    metrics_key = METRICS_KEY_MAPPING.get(data_subject)
+
+    if not metrics_key:
+        logger.debug(f"No metrics mapping found for data_subject: '{data_subject}'")
+        return None
+
+    # Retrieve and return the actual metrics data using the resolved key.
+    return metrics.get(metrics_key)
 
 def save_report_results(
     org_urn: str,
@@ -163,7 +194,8 @@ def save_objectives(
 
 def save_key_results(
     org_urn: str,
-    objectives_with_ids: List[Tuple[Dict[str, Any], str]]
+    objectives_with_ids: List[Tuple[Dict[str, Any], str]],
+    metrics
 ) -> Optional[List[Tuple[Dict[str, Any], str]]]:
     """
     Saves Key Result records to Bubble, linking them to their parent objectives.
@@ -188,6 +220,10 @@ def save_key_results(
             for kr in objective_data.get("key_results", []):
                 kr_type = kr.get("key_result_type")
                 key_results_to_process.append(kr)
+                data_subject_value = kr.get("data_subject")
+
+                metrics_for_kr = _get_metrics_for_subject(data_subject_value, metrics)
+                
                 key_result_payloads.append({
                     "okr": parent_objective_id,
                     "description": kr.get("key_result_description"),
@@ -196,6 +232,7 @@ def save_key_results(
                     # FIX: Convert Enum to its value before sending.
                     "kr_type": kr_type.value if hasattr(kr_type, 'value') else kr_type,
                     "data_subject": kr.get("data_subject"),
+                    "metriche_data_subject": metrics_for_kr
                 })
 
         if not key_result_payloads:
@@ -299,7 +336,7 @@ def save_tasks(
 
 # --- Orchestrator Function ---
 
-def save_actionable_okrs(org_urn: str, actionable_okrs: Dict[str, Any], report_id: str):
+def save_actionable_okrs(org_urn: str, actionable_okrs: Dict[str, Any], report_id: str, metrics):
     """
     Orchestrates the sequential saving of objectives, key results, and tasks.
     """
@@ -332,7 +369,7 @@ def save_actionable_okrs(org_urn: str, actionable_okrs: Dict[str, Any], report_i
         objectives_with_ids = list(zip(objectives_data, objective_ids))
 
         # Step 2: Save the key results, linking them to the objectives
-        key_results_with_ids = save_key_results(org_urn, objectives_with_ids)
+        key_results_with_ids = save_key_results(org_urn, objectives_with_ids, metrics)
         if key_results_with_ids is None:
             logger.error("OKR save process aborted due to failure in saving key results.")
             return