Revert "Update generate_legal_chat_responses"

This reverts commit e932545e890f4a06702734e49cff051130cb6bb8.

api/main.py

@@ -260,7 +260,8 @@ async def generate_legal_chat_responses(message: str, session_id: Optional[str]
                 try:
                     if isinstance(state, MultiCountryLegalState):
                         final_state = state
-                        state_dict = state.model_dump(exclude_unset=True) if hasattr(state, "model_dump") else state.dict(exclude_unset=True)
+                        # Use model_dump_json for Pydantic v2 compatibility
+                        state_dict = json.loads(state.model_dump_json(exclude_unset=True)) if hasattr(state, "model_dump_json") else state.dict(exclude_unset=True)
                         if "messages" in state_dict and isinstance(state_dict["messages"], list):
                             state_dict["messages"] = [
                                 msg if isinstance(msg, dict) else {"role": "unknown", "content": str(msg), "meta": {}}
@@ -278,7 +279,8 @@ async def generate_legal_chat_responses(message: str, session_id: Optional[str]
     # Yield final state if captured
     if final_state and isinstance(final_state, MultiCountryLegalState):
         try:
-            final_state_dict = final_state.model_dump(exclude_unset=True) if hasattr(final_state, "model_dump") else final_state.dict(exclude_unset=True)
+            # Use model_dump_json for Pydantic v2 compatibility
+            final_state_dict = json.loads(final_state.model_dump_json(exclude_unset=True)) if hasattr(final_state, "model_dump_json") else final_state.dict(exclude_unset=True)
             if "messages" in final_state_dict and isinstance(final_state_dict["messages"], list):
                 final_state_dict["messages"] = [
                     msg if isinstance(msg, dict) else {"role": "unknown", "content": str(msg), "meta": {}}
@@ -290,7 +292,7 @@ async def generate_legal_chat_responses(message: str, session_id: Optional[str]
             yield f"data: {json.dumps({'type': 'error', 'message': 'An internal error occurred'})}\n\n"
 
     yield f"data: {json.dumps({'type': 'end'})}\n\n"
-    
+
 @app.get("/chat")
 async def chat_stream(
     message: str = Query(..., description="User message"),