Add file to the promt

agent.py

@@ -55,26 +55,53 @@ def build_agent_graph():
     # Compile graph
     return builder.compile()
 
+def print_stream(stream):
+    for s in stream:
+        message = s["messages"][-1]
+        if isinstance(message, tuple):
+            print(message)
+        else:
+            message.pretty_print()
 
 
 class MyGAIAAgent:
-    def __init__(self):
+    def __init__(self, verbose: bool = False):
         print("MyAgent initialized.")
         self.graph = build_agent_graph()
-    def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        self.verbose = verbose
+    def __call__(self, task: dict) -> str:
+        
+        '''
         # Wrap the question in a HumanMessage from langchain_core
         
         messages = [HumanMessage(content=question)]
         messages = self.graph.invoke({"messages": messages})
         answer = messages['messages'][-1].content
-        '''
+        
         user_input = {"messages": [("user", question)]}
         answer1 = self.graph.invoke(user_input)["messages"][-1].content
         print (answer)
-        '''
         #print (self._clean_answer(answer))
         return self._clean_answer(answer)
+        '''
+        question = task["question"]
+        task_id = task["task_id"]
+        file_name = task.get("file_name")
+        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        file_ext = None
+        user_prompt = question
+        if file_name:
+            file_ext = os.path.splitext(file_name)[-1].removeprefix(".")
+            user_prompt += f"\nTask ID: {task_id}\nFile extension: {file_ext}"
+
+        user_input = {"messages": [("user", user_prompt)]}
+        if self.verbose:
+            print_stream(self.graph.stream(user_input, stream_mode="values"))
+        else:
+            answer = self.graph.invoke(user_input)["messages"][-1].content
+            return self._clean_answer(answer)
+    
+    
     
     def _clean_answer(self, answer: any) -> str:
         """
@@ -157,5 +184,12 @@ if __name__ == "__main__":
     question4 = "How many at bats did the Yankee with the most walks in the 1977 regular season have that same season?"
     question6 = "What is the first name of the only Malko Competition recipient from the 20th Century (after 1977) whose nationality on record is a country that no longer exists?"
     '''
+    task = {
+        "task_id": "8e867cd7-cff9-4e6c-867a-ff5ddc2550be",
+        "question": "How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? You can use the latest 2022 version of english wikipedia.",
+        "Level": "1",
+        "file_name": "",
+    }
+    
     agent = MyGAIAAgent()
-    print(agent(question18))
+    print(agent(task))

app.py

@@ -73,7 +73,8 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            submitted_answer = agent(question_text)
+            #submitted_answer = agent(question_text)
+            submitted_answer = agent(item)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
         except Exception as e: