not saving bug attempt 2

ai_chat.py

@@ -190,6 +190,10 @@ async def process_message(user_message: str, user_info: dict = None, session_id:
     if session_id:
         session_messages = await get_session_messages(session_id, limit=30)
     
+    # Check if we should ask persona questions
+    conversation_context = " ".join([msg["content"] for msg in session_messages[-5:]]) + " " + user_message
+    should_ask, field_to_ask = await should_ask_persona_question(persona, conversation_context)
+    
     # Check if this is a response to a persona question
     is_persona_question = False
     current_field = None
@@ -206,12 +210,6 @@ async def process_message(user_message: str, user_info: dict = None, session_id:
                 parsed_response = await parse_user_response(user_message, field)
                 break
     
-    # Only check for new persona questions if this isn't a response to one
-    should_ask, field_to_ask = (False, None)
-    if not is_persona_question:
-        conversation_context = " ".join([msg["content"] for msg in session_messages[-5:]]) + " " + user_message
-        should_ask, field_to_ask = await should_ask_persona_question(persona, conversation_context)
-    
     # Process with AI
     result = await chat_graph.ainvoke({
         "user_message": user_message,
@@ -227,8 +225,12 @@ async def process_message(user_message: str, user_info: dict = None, session_id:
         "persona_check": (should_ask, field_to_ask)
     })
     
-    # Handle persona updates
-    if result.get("update_persona") and wa_id:
+    print("LangGraph result:", result)  # Debugging
+    # Save persona field if present
+    if result.get("persona_field") and result.get("persona_value") is not None and wa_id:
+        print(f"Saving persona field {result['persona_field']} with value {result['persona_value']} for user {wa_id}")
+        await update_persona_field(wa_id, result["persona_field"], result["persona_value"])
+    elif result.get("update_persona") and wa_id:
         await update_persona_field(wa_id, result["persona_field"], result["persona_value"])
     
     # Save messages to database

whatsapp.py

@@ -15,7 +15,8 @@ async def send_whatsapp_message(wa_id: str, message: str):
         "text": {"body": message}
     }
     try:
-        resp = await httpx.post(url, headers=headers, json=payload)
+        async with httpx.AsyncClient() as client:
+            resp = await client.post(url, headers=headers, json=payload)
         print(f"Sent message → {resp.status_code}: {resp.text}")
         return resp.status_code == 200
     except Exception as e: