Final_Assignment_Template / formatter_node.py
Umer797's picture
Update formatter_node.py
3a66e59 verified
raw
history blame
353 Bytes
import re
def formatter_node(llm_output):
cleaned = llm_output.strip()
# Remove common LLM prefixes
cleaned = re.sub(r'^(Answer:|Final Answer:)\s*', '', cleaned, flags=re.IGNORECASE)
# Remove enclosing quotes if any
if cleaned.startswith('"') and cleaned.endswith('"'):
cleaned = cleaned[1:-1]
return cleaned.strip()