change

agent.py

@@ -1,4 +1,7 @@
 # agent_v6.py
+# Develop an AI agent with LangGraph and LangChain 
+# to answer the questions in the "gaia-benchmark/GAIA" dataset.
+
 from pathlib import Path
 import os, re, base64, mimetypes, tempfile, uuid, subprocess, json
 from urllib.parse import urlparse, unquote
@@ -350,7 +353,7 @@ def solve_multimodal_node(state: AgentState) -> AgentState:
     sys = SystemMessage(content=(
         "You solve GAIA tasks using the provided evidence and attached images.\n"
         "Be precise, quote numbers/strings exactly. If uncertain, say so.\n" 
-        "Your answer to the GAIA tasks should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. 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 of whether the element to be put in the list is a number or a string.\n" + end_instr
+        "Your answer to the GAIA tasks should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. 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 of whether the element to be put in the list is a number or a string. If your answer only include a single word, make the first letter capital.\n" + end_instr
     ))
 
     # Summarized text evidence (ASR/OCR/PDF text)
@@ -399,7 +402,7 @@ def solve_text_only_node(state: "AgentState") -> "AgentState":
     sys = SystemMessage(content=(
         "You solve GAIA tasks. Use careful step-by-step reasoning but keep it concise.\n"
         "You can use the provided textual evidence if there is any. \n" 
-        "Your answer to the GAIA tasks should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. 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 of whether the element to be put in the list is a number or a string.\n" + end_instr
+        "Your answer to the GAIA tasks should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. 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 of whether the element to be put in the list is a number or a string. If your answer only include a single word, make the first letter capital.\n" + end_instr
     ))
 
     user = HumanMessage(content=(
@@ -465,7 +468,7 @@ def has_images(state: AgentState) -> bool:
             return True
     return False
 
-def route_after_preprocess(state: AgentState) -> Literal["visions","text"]:
+def route_after_preprocess(state: AgentState) -> Literal["vision","text"]:
     return "vision" if has_images(state) else "text"
 
 # ---------- Graph ----------