linting

src/gaia_solving_agent/agent.py

@@ -42,12 +42,13 @@ openai_llm = OpenAI(
     max_retries=5,
 )
 mistral_llm = MistralAI(
-        model="mistral-small-latest",
-        api_key=MISTRAL_API_KEY,
-        temperature=.1,
-        max_retries=5,
-        # is_function_calling_model=True,
-    )
+    model="mistral-small-latest",
+    api_key=MISTRAL_API_KEY,
+    temperature=.1,
+    max_retries=5,
+    # is_function_calling_model=True,
+)
+
 
 def get_llm(model_name=cheap_model_name):
     return NebiusLLM(
@@ -60,14 +61,17 @@ def get_llm(model_name=cheap_model_name):
         max_retries=5,
     )
 
+
 class PlanEvent(Event):
     to_do: Literal["Initialize", "Format", "Replan"] = "Initialize"
     plan: str | None = None
     n_retries: int = 0
 
+
 class QueryEvent(Event):
     pass
 
+
 class AnswerEvent(Event):
     plan: str
     answer: str
@@ -177,7 +181,7 @@ The sub-tasks are :
 
 
 gaia_solving_agent = FunctionAgent(
-    tools = [
+    tools=[
         get_text_representation_of_additional_file,
         vllm_ask_image_tool,
         tavily_search_web,
@@ -190,17 +194,17 @@ gaia_solving_agent = FunctionAgent(
     llm=get_llm(reasoning_model_name),
     system_prompt="""
     You are a helpful assistant that uses tools to browse additional information and resources on the web to answer questions.
-    
+
     Tools you have are of three types:
     - External resources getter: get text, images, video, etc. from the internet
     - Resource querier and transformer: query, summarize or transform a resource into a more digestible format.
     - Analyse or compute : specialized tools to provide a specific analysis or computation.
-    
+
     Try to get resources before querying them.
     If it is an additional file, you can access its content through the get_text_representation_of_additional_file tool.
     If you need the original Document, you can use the llamaindex context with ctx.store.get("additional_file").
     If the analysis require a new external resource get it first.(e.g. a set of rules or a process)
-    
+
     You will be provided a question, some known facts summarizing the user provided context and some sub-tasks to complete.
     You should follow the order of the sub-tasks.
     If the tools provides facts that go against your knowledge, you should not use them.

src/gaia_solving_agent/hf_submission_api.py

@@ -154,7 +154,11 @@ def prepare_submission_data(username:str, agent_code: str, answers_payload: list
         search = re.search(pattern, answer)
         extracted_answer = search.group(1).strip() if search else ""
         try:
-            return eval(extracted_answer)
+            evaluated_answer = eval(extracted_answer)
+            if isinstance(evaluated_answer, tuple):
+                return ', '.join(evaluated_answer)
+            else:
+                return evaluated_answer
         except:
             return extracted_answer