Update main.py

main.py

@@ -471,7 +471,7 @@ async def followup_agent(query: FollowupQueryModel, background_tasks: Background
         
         result = {
             "response": response_content,
-            "clarification": clarification['questions'] if clarification else None
+            "clarification": clarification
         }
         
         yield "\n\n" + json.dumps(result)