Update agent.py

agent.py

@@ -13,11 +13,13 @@ from tools import (absolute, add, analyze_csv_file, analyze_excel_file,
                    arvix_search, audio_transcription, compound_interest,
                    convert_temperature, divide, exponential,
                    extract_text_from_image, factorial, floor_divide,
-                   get_current_time_in_timezone, greatest_common_divisor,
-                   is_prime, least_common_multiple, logarithm, modulus,
-                   multiply, percentage_calculator, power, python_code_parser,
-                   reverse_sentence, roman_calculator_converter, square_root,
-                   subtract, web_content_extract, web_search, wiki_search)
+                   get_current_time_in_timezone,
+                   
+                   greatest_common_divisor, is_prime, least_common_multiple,
+                   logarithm, modulus, multiply, percentage_calculator, power,
+                   python_code_parser, reverse_sentence,
+                   roman_calculator_converter, square_root, subtract,
+                   web_content_extract, web_search, wiki_search)
 
 # Load Constants
 load_dotenv()
@@ -33,7 +35,7 @@ tools = [
     is_prime, least_common_multiple, percentage_calculator,
     wiki_search, analyze_excel_file, arvix_search,
     audio_transcription, python_code_parser, analyze_csv_file,
-    extract_text_from_image, reverse_sentence, web_content_extract
+    extract_text_from_image, reverse_sentence, web_content_extract,
 ]
 
 # Load system prompt
@@ -45,7 +47,7 @@ If you are asked for a number, don't use a comma to write your number, nor use u
 If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.
 If you are asked for a comma separated list, apply the above rules depending on whether the element to be put in the list is a number or a string.
 Format your output as: Answers (answers): [{"task_id": ..., "submitted_answer": ...}]
-Do not repeat the format or include any nested JSON. Output only one flat list as: Answers (answers): [{...}]
+Do NOT include the format string or any JSON inside the submitted_answer field. Only output a single flat list as: Answers (answers): [{"task_id": ..., "submitted_answer": ...}]
 """
 
 # System message
@@ -82,6 +84,7 @@ def build_graph():
         # Get task_id from state or set a placeholder
         task_id = state.get("task_id", "1")  # Replace with actual logic if needed
         formatted = f'Answers (answers): [{{"task_id": "{task_id}", "submitted_answer": "{answer_text}"}}]'
+        formatted = extract_flat_answer(formatted)
         return {"messages": [formatted]}
 
     # --- Graph Definition ---
@@ -121,6 +124,21 @@ def is_valid_agent_output(output):
     except Exception:
         return False
 
+
+def extract_flat_answer(output):
+    # Try to find the innermost Answers (answers): [{...}]
+    pattern = r'Answers \(answers\): \[(\{.*?\})\]'
+    matches = re.findall(pattern, output)
+    if matches:
+        # Use the last match (innermost)
+        try:
+            answers_list = json.loads(f'[{matches[-1]}]')
+            if isinstance(answers_list, list) and "task_id" in answers_list[0] and "submitted_answer" in answers_list[0]:
+                return f'Answers (answers): [{matches[-1]}]'
+        except Exception:
+            pass
+    return output  # fallback
+
 # test
 if __name__ == "__main__":
     question = "When was a picture of St. Thomas Aquinas first added to the Wikipedia page on the Principle of double effect?"