removed initial question evaluation

config/prompts.yaml

@@ -71,7 +71,7 @@ prompts:
   final_answer_processor:
     content: |
       You are a general AI assistant. You are given a question and an answer to that question. 
-      Process the answer such that it contains only YOUR FINAL ANSWER and respects the following guidelines. 
+      Process the answer and extract YOUR FINAL ANSWER to be provided to the user. Make sure it respects the following guidelines. 
       YOUR FINAL ANSWER 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. 
@@ -93,16 +93,6 @@ prompts:
     variables: ["summary"]
     version: 1.0
     description: "Prompt for summarization and memory optimization"
-  question_evaluation:
-    content: |
-      You are a general AI assistant. You are given with a question. Evaluate this question and answer with a YES if the question involves the analysis of
-      an attached file or video link, otherwise with a NO.
-      This is the question to be evaluated
-      {{question}}
-    type: question_refinement
-    variables: ["question"]
-    version: 1.0
-    description: "Prompt for evaluating a question"
   audio_evaluation:
     content: |
       You are an audio analysis assistant. Answer questions based on the provided audio. Be precise and factual.

core/agent.py

@@ -45,7 +45,7 @@ class GaiaAgent:
         self.react_graph = builder.compile()
 
     def __call__(self, question: str, attachment: Optional[Attachment] = None) -> str:
-        initial_state = {"messages": [HumanMessage(content=question)]}
+        initial_state = {"messages": [HumanMessage(content=question)], "question": question}
         if attachment:
             initial_state["file_reference"] = attachment.file_path
 
@@ -57,7 +57,7 @@ class GaiaAgent:
         return answer
 
     def __streamed_call__(self, question: str, attachment: Optional[Attachment] = None) -> str:
-        initial_state = {"messages": [HumanMessage(content=question)]}
+        initial_state = {"messages": [HumanMessage(content=question)], "question": question}
         if attachment:
             initial_state["file_reference"] = attachment.file_path
 

core/state.py

@@ -5,5 +5,4 @@ class State(MessagesState):
     summary: str
     question: str
     chunked_last_tool_call: bool
-    attachment: str
     file_reference: str  # Attachment file reference: a path, URL, or unique ID

nodes/nodes.py

@@ -31,19 +31,12 @@ def pre_processor(state: State):
 
     file_reference = state.get("file_reference", "")
     extension = pathlib.Path(file_reference).suffix
-    if extension == "png":
+    if extension == ".png":
         content_bytes = attachmentHandler.fetch_file_from_reference(file_reference)
         mime_type = mimetypes.guess_type(file_reference)[0]
         state["messages"][0].content = [{"type": "text", "text": question},
                                         attachmentHandler.get_representation("image", content_bytes, "png", mime_type)]
 
-    message = prompt_mgmt.render_template("question_evaluation", {"question": question[0]})
-
-    # Add prompt to our history
-    messages = [HumanMessage(content=message)]
-    response = response_processing_model.invoke(messages)
-    if response.content == "YES":
-        return {"question": question, "attachment": "true"}
     return {"question": question}
 
 

utils/prompt_manager.py

@@ -13,7 +13,6 @@ class PromptType(Enum):
     BASE_SYSTEM = "base_system"
     ANSWER_REFINEMENT = "answer_refinement"
     MEMORY_OPTIMIZATION = "memory_optimization"
-    QUESTION_REFINEMENT = "question_refinement"
     TOOL = "tool"