update yaml parsing

main.py

@@ -416,8 +416,11 @@ def parse_followup_response(response):
     if len(clarification_parts) > 1:
         clarification_yaml = clarification_parts[1].split("</clarification>")[0].strip()
         try:
-            clarification = yaml.safe_load(clarification_yaml)
-        except yaml.YAMLError:
+            # Add indentation to make it valid YAML
+            indented_yaml = "\n".join("  " + line for line in clarification_yaml.split("\n"))
+            clarification = yaml.load("questions:\n" + indented_yaml, Loader=SafeLoader)
+        except yaml.YAMLError as e:
+            logger.error(f"YAML parsing error: {e}")
             clarification = None
     else:
         clarification = None
@@ -453,7 +456,7 @@ async def followup_agent(query: FollowupQueryModel, background_tasks: Background
         
         result = {
             "response": response_content,
-            "clarification": clarification
+            "clarification": clarification['questions'] if clarification else None
         }
         
         yield "\n\n" + json.dumps(result)