Update services/report_data_handler.py

services/report_data_handler.py

@@ -35,34 +35,26 @@ def fetch_latest_agentic_analysis(org_urn: str):
     return report_data_df, None  # Return full dataframe and no error
 
 
-def save_agentic_analysis_results(
+def save_report_results(
     org_urn: str,
     report_markdown: str,
-    orchestration_raw_results: dict, # This is the full output from agentic pipeline
-    # key_results_for_selection: list, # Pass this too if you want to save it
-    generation_timestamp: pd.Timestamp
+    quarter,
+    year,
 ):
     """Saves the agentic pipeline results to Bubble."""
     if not org_urn:
         logger.error("Cannot save agentic results: org_urn missing.")
         return False
 
-    # Extract what you need from orchestration_raw_results
-    actionable_okrs_dict = orchestration_raw_results.get("actionable_okrs_and_tasks", {})
-    # Assuming key_results_for_selection is also part of orchestration_raw_results or passed separately
-    key_results_selection_list = orchestration_raw_results.get("key_results_for_selection", [])
-
-
     payload = {
-        "org_urn": org_urn,
-        BUBBLE_AGENTIC_TIMESTAMP_COLUMN: generation_timestamp.isoformat(), # Store as ISO string
+        "organization_urn": org_urn,
         "report_markdown": report_markdown if report_markdown else "N/A",
-        "okrs_json": json.dumps(actionable_okrs_dict) if actionable_okrs_dict else json.dumps({}),
-        "key_results_selection_json": json.dumps(key_results_selection_list) if key_results_selection_list else json.dumps([]),
+        "quarter": quarter,
+        "year": year,
         # Add other fields as needed
     }
     logger.info(f"Attempting to save agentic analysis to Bubble for org_urn: {org_urn}")
-    success_ids = bulk_upload_to_bubble([payload], BUBBLE_AGENTIC_RESULTS_TABLE_NAME)
+    success_ids = bulk_upload_to_bubble([payload], BUBBLE_REPORT_TABLE_NAME)
 
     if success_ids and success_ids[0]: # bulk_upload_to_bubble returns list of IDs or False
         logger.info(f"Successfully saved agentic analysis to Bubble. Record ID: {success_ids[0]}")