formatter_node.py

import re

def formatter_node(llm_output):
    cleaned = llm_output.strip()
    # Remove common prefixes like "Answer:", "Final Answer:"
    cleaned = re.sub(r'^(Answer:|Final Answer:)\s*', '', cleaned, flags=re.IGNORECASE)
    # Remove enclosing quotes if present
    if cleaned.startswith('"') and cleaned.endswith('"'):
        cleaned = cleaned[1:-1]
    return cleaned.strip()