Update main.py

main.py

@@ -43,9 +43,10 @@ def chat():
     role = data.get('role')
     user_input = data.get('message')
     company_code = data.get('companyCode')
+    user_id = data.get('userId')  # ✅
 
-    if not role or not user_input or not company_code:
-        return jsonify({"error": "Missing role, message, or companyCode"}), 400
+    if not role or not user_input or not company_code or not user_id:
+        return jsonify({"error": "Missing role, message, companyCode, or userId"}), 400
 
     print("🧠 Incoming user message:", user_input)