Update data_planner.py

data_planner.py

@@ -2,7 +2,7 @@ import json
 from openai_client import ask_gpt
 from schema_map import schema  # 👈 Import the full schema
 
-def determine_data_requirements(user_prompt: str, company_code: str):
+def determine_data_requirements(user_prompt: str, company_code: str, user_id: str):
     system_msg = {
         "role": "system",
         "content": (
@@ -20,8 +20,11 @@ def determine_data_requirements(user_prompt: str, company_code: str):
 
     user_msg = {
         "role": "user",
-        "content": f"User question: {user_prompt}\n\nCompanyCode: {company_code}"
-    }
+        "content": (
+            f"User question: {user_prompt}\n"
+            f"CompanyCode: {company_code}\n"
+            f"UserID: {user_id}"
+        )
 
     try:
         result = ask_gpt([system_msg, user_msg])