Enhance check_format function to disallow nested lists (let LLM self-correct) and clarify return types in documentation.

app.py

@@ -47,28 +47,21 @@ def load_file(path: str) -> list | dict:
     else:
         return {"raw document text": text, "file path": path}
     
-def check_format(answer: str | list, *args, **kwargs ) -> list:
-    """ Check if the answer is a list. """
-    
+def check_format(answer: str | list, *args, **kwargs) -> list:
+    """Check if the answer is a list and not a nested list."""
     print("Checking format of the answer:", answer)
-    def flatten(lst):
-        for item in lst:
+    if isinstance(answer, list):
+        for item in answer:
             if isinstance(item, list):
-                yield from flatten(item)
-            else:
-                yield item
-    try:
-        if isinstance(answer, list):
-            flat = list(flatten(answer))
-            return flat
-    except Exception as e:
-        if isinstance(answer, dict):
-            raise TypeError(f"Final answer must be a list, not a dict. Please check the answer format. Error: {e}")
+                print("list detected")
+                raise TypeError("Nested lists are not allowed in the final answer.")
+    elif isinstance(answer, str):
+        return [answer]
+    elif isinstance(answer, dict):
+        raise TypeError(f"Final answer must be a list, not a dict. Please check the answer format. Error: {e}")
         
         
 
-    
-
 ## tools definition
 @tool
 def download_images(image_urls: str) -> list:
@@ -151,6 +144,14 @@ def generate_image(prompt: str, neg_prompt: str) -> Image.Image:
 
     return gr.Image(value=image, label="Generated Image")
 
+"""@tool
+def generate_audio(prompt: str) -> object:
+    space = smolagents.load_tool(
+        
+    )"""
+
+
+
 
 ## agent definition
 class Agent:

system_prompt.txt

@@ -320,10 +320,11 @@ It is MANDATORY to use these rules for the 'final_answer' tool:
 	3.	Any text or explanation must come after the component, as string elements in the same list.
 	4.	If there is no component to return, return a list whose only element is the text.
 	5.	Examples of valid returns:
-• [gr.Image(value=“output.png”), “Here is the chart.”]
-• [gr.File(value=“report.pdf”), “Download the report.”]
+        • [gr.Image(value=“output.png”), “Here is the chart.”]
+        • [gr.File(value=“report.pdf”), “Download the report.”]
 	7.  Any deviation (returning a dict, tuple, raw PIL image, etc.) is invalid.
-    8.  Always put generated images in the list, do NOT create external link.
+    8.  Nested list are forbidden.
+    9.  Always put generated images in the list, do NOT create external link.
 
 Now Begin!