🐛 fix(metadata): Enhance YAML parsing error handling

Author: thibaultyou

.github/scripts/generate_metadata.py

@@ -57,30 +57,32 @@ def generate_metadata(prompt_content):
     logger.info(f"Response structure: {type(message)}")
     logger.info(f"Content structure: {type(message.content)}")
 
-    # Extract the YAML content from the AI response
+    # Extract the content from the AI response
     content = message.content[0].text if isinstance(message.content, list) else message.content
     logger.info(f"Extracted content: {content[:100]}...")
 
+    # Check if the content is enclosed in output tags
     output_start = content.find("<output>")
     output_end = content.find("</output>")
+    
     if output_start != -1 and output_end != -1:
         yaml_content = content[output_start + 8:output_end].strip()
-        logger.info(f"Extracted YAML content: {yaml_content[:100]}...")
+    else:
+        yaml_content = content.strip()
+    
+    logger.info(f"Extracted YAML content: {yaml_content[:100]}...")
 
-        class SafeLoaderIgnoreUnknown(yaml.SafeLoader):
-            def ignore_unknown(self, node):
-                return None
-        
-        SafeLoaderIgnoreUnknown.add_constructor(None, SafeLoaderIgnoreUnknown.ignore_unknown)
+    class SafeLoaderIgnoreUnknown(yaml.SafeLoader):
+        def ignore_unknown(self, node):
+            return None
+    
+    SafeLoaderIgnoreUnknown.add_constructor(None, SafeLoaderIgnoreUnknown.ignore_unknown)
 
-        try:
-            metadata = yaml.load(yaml_content, Loader=SafeLoaderIgnoreUnknown)
-            logger.info("YAML content parsed successfully")
-        except yaml.YAMLError as e:
-            logger.error(f"YAML parsing error: {e}")
-            metadata = {}
-    else:
-        logger.error("Could not find metadata output in AI response")
+    try:
+        metadata = yaml.load(yaml_content, Loader=SafeLoaderIgnoreUnknown)
+        logger.info("YAML content parsed successfully")
+    except yaml.YAMLError as e:
+        logger.error(f"YAML parsing error: {e}")
         metadata = {}
 
     return metadata