Updated Query Reformulator

insucompass/core/agent_orchestrator.py

@@ -76,11 +76,11 @@ def reformulate_query_node(state: AgentState) -> Dict[str, Any]:
     history = state["conversation_history"]
     user_profile = state["user_profile"]
     
-    profile_summary = f"User profile context: State={user_profile.get('state')}, Age={user_profile.get('age')}, History={user_profile.get('medical_history')}"
+    profile_str = str(user_profile)
     prompt = load_prompt("query_reformulator")
     history_str = "\n".join(history)
     
-    full_prompt = f"{prompt}\n\n### User Profile Summary\n{profile_summary}\n\n### Conversation History:\n{history_str}\n\n### Follow-up Question:\n{question}"
+    full_prompt = f"{prompt}\n\n### User Profile\n{profile_str}\n\n### Conversation History:\n{history_str}\n\n### Follow-up Question:\n{question}"
     
     response = llm.invoke(full_prompt)
     standalone_question = response.content.strip()

insucompass/prompts/query_reformulator.txt

@@ -143,30 +143,4 @@ You are an expert query assistant. Your task is to take a conversation history,
 - **Your Correct Output:** "What documents are needed to enroll a spouse under 65 in Medicare in Miami-Dade County, Florida, zip code 33101, for a 64-year-old female, retired, with an income of $60,000 and no reported medical history?"
 
 ### YOUR TASK
-Reformulate the `Follow-up Question` into a self-contained question using the `Conversation History` and `User Profile`. Ensure the question captures all relevant details from the history and profile (e.g., location, timeframe, specific entities, or constraints) to make it contextual, precise, and suitable for accurate document retrieval or response generation. Output only the reformulated question.
-
-### YOUR INPUT
-**Conversation History:**
-- User: "What are the best health insurance plans for a single 30-year-old in New York with no pre-existing conditions?"
-- Agent: "For a single 30-year-old in New York with no pre-existing conditions, top health insurance plans include Blue Cross Blue Shield’s HMO plans and Oscar’s EPO plans, available through the ACA marketplace, with premiums ranging from $300 to $600 per month."
-- User: "Which of these have the best coverage for preventive care?"
-**User Profile:**
-```json
-{
-  "zip_code": "30303",
-  "county": "Fulton",
-  "state": "Georgia",
-  "age": 34,
-  "gender": "Female",
-  "household_size": 1,
-  "income": 65000,
-  "employment_status": "employed_without_coverage",
-  "citizenship": "US Citizen",
-  "medical_history": "None reported.",
-  "medications": null,
-  "special_cases": "Reports smoking tobacco."
-}
-```
-**Follow-up Question:** "What about dental coverage?"
-
-**Your Correct Output:** "Which health insurance plans for a single 34-year-old female in Fulton County, Georgia, zip code 30303, employed without coverage, with an income of $65,000, no reported medical history, and who reports smoking tobacco, such as Blue Cross Blue Shield’s HMO plans or Oscar’s EPO plans, offer the best dental coverage through the ACA marketplace?"
+Reformulate the `Follow-up Question` into a self-contained question using the `Conversation History` and `User Profile`. Ensure the question captures all relevant details from the history and profile (e.g., location, timeframe, specific entities, or constraints) to make it contextual, precise, and suitable for accurate document retrieval or response generation. Output only the reformulated question.