Update app.py

app.py

@@ -23,7 +23,7 @@ RESULT_COLUMNS = ["Task ID", "Question", "Submitted Answer"]
 HTTP_TIMEOUT = 45
 MAX_EXTRACTED_CHARS = 35_000
 DEFAULT_HF_MODEL = "huggingface/openai/gpt-oss-120b"
-DEFAULT_GEMINI_MODEL = "gemini/gemini-3.5-flash"
+DEFAULT_GROQ_REVIEW_MODEL = "groq/qwen/qwen3.6-27b"
 TASK_FILE_CACHE = {}
 
 
@@ -458,90 +458,27 @@ class AnalyzeGaiaImageTool(Tool):
             return f"Could not analyze the GAIA image: {exc}"
 
 
-class ConsultGeminiTool(Tool):
-    name = "consult_gemini"
-    description = (
-        "Asks Gemini for an independent second opinion on a difficult question "
-        "or on evidence already collected. Use it to verify reasoning, resolve "
-        "conflicting sources, or check the exact requested answer format. Do "
-        "not use it as a substitute for inspecting official attachments."
-    )
-    inputs = {
-        "request": {
-            "type": "string",
-            "description": (
-                "The complete question plus any relevant evidence and the "
-                "specific point Gemini should verify."
-            ),
-        }
-    }
-    output_type = "string"
-
-    def forward(self, request: str) -> str:
-        api_key = os.getenv("GEMINI_API_KEY")
-        if not api_key:
-            return "Gemini consultation unavailable: GEMINI_API_KEY is missing."
-
-        model = os.getenv(
-            "GAIA_GEMINI_TOOL_MODEL", DEFAULT_GEMINI_MODEL
-        )
-        try:
-            response = completion(
-                model=model,
-                api_key=api_key,
-                messages=[
-                    {
-                        "role": "system",
-                        "content": (
-                            "You are a verification specialist assisting another "
-                            "GAIA agent. Analyze the supplied question and evidence "
-                            "critically. Identify uncertainty or contradictions, "
-                            "then provide your recommended exact answer. Be concise "
-                            "and never claim to have opened sources that were not "
-                            "included in the request."
-                        ),
-                    },
-                    {"role": "user", "content": str(request)},
-                ],
-                temperature=0,
-                max_tokens=900,
-            )
-            return str(response.choices[0].message.content).strip()
-        except Exception as exc:
-            return f"Gemini consultation failed: {exc}"
-
-
 class BasicAgent:
     def __init__(self):
         print("Inicializando o agente GAIA...")
 
         hf_token = os.getenv("HF_TOKEN")
-        gemini_api_key = os.getenv("GEMINI_API_KEY")
         configured_model = os.getenv("GAIA_MODEL_ID")
 
         model_id = configured_model or DEFAULT_HF_MODEL
-
-        if model_id.startswith("gemini/"):
-            model_api_key = gemini_api_key
-            required_secret = "GEMINI_API_KEY"
-        else:
-            model_api_key = hf_token
-            required_secret = "HF_TOKEN"
-
-        if not model_api_key:
+        if not hf_token:
             raise RuntimeError(
-                f"O secret {required_secret} não está configurado. "
+                "O secret HF_TOKEN não está configurado. "
                 "Adicione a chave em Settings > Variables and secrets > Secrets."
             )
 
         self.model = LiteLLMModel(
             model_id=model_id,
-            api_key=model_api_key,
+            api_key=hf_token,
             temperature=0,
             max_tokens=2_000,
         )
         self.hf_token = hf_token
-        self.gemini_api_key = gemini_api_key
         self.model_id = model_id
         print(f"Modelo principal selecionado: {model_id}")
 
@@ -573,8 +510,6 @@ class BasicAgent:
             YouTubeTranscriptTool(),
             AnalyzeGaiaImageTool(),
         ]
-        if gemini_api_key:
-            agent_tools.append(ConsultGeminiTool())
 
         self.agent = CodeAgent(
             tools=agent_tools,
@@ -613,9 +548,6 @@ TOOL ROUTING POLICY:
    chess positions, or questions that visually depend on an attached image.
 6. youtube_transcript retrieves spoken subtitles. Use it when asked what a
    person said. It cannot answer purely visual video questions.
-7. consult_gemini gives a second opinion. Use it after gathering evidence when
-   sources conflict or the candidate answer is uncertain. Include the complete
-   question, evidence, and candidate answer.
 
 Research carefully, prefer primary or official sources, and cross-check
 uncertain facts. A search snippet alone is insufficient when the source page
@@ -681,12 +613,12 @@ include reasoning, explanations, labels, Markdown, citations, or the words
             ):
                 raise RuntimeError(
                     f"Falha de autenticação no modelo {self.model_id}. "
-                    "Verifique a chave secreta correspondente ao provedor "
-                    "(GEMINI_API_KEY para Gemini ou HF_TOKEN para Hugging Face)."
+                    "Verifique se HF_TOKEN possui permissão para usar "
+                    "Inference Providers."
                 ) from exc
             raise
         candidate = self.format_exact_answer(question, str(result))
-        return self.review_answer_with_gemini(
+        return self.review_answer_with_groq(
             question=question,
             candidate=candidate,
             task_id=task_id,
@@ -754,18 +686,19 @@ include reasoning, explanations, labels, Markdown, citations, or the words
 
         return self.deterministic_answer_cleanup(cleaned)
 
-    def review_answer_with_gemini(
+    def review_answer_with_groq(
         self, question: str, candidate: str, task_id: str | None = None
     ) -> str:
-        """Revisa obrigatoriamente conteúdo e formato antes de salvar a resposta."""
-        if not self.gemini_api_key:
+        """Solicita uma segunda opinião gratuita no Groq, com fallback local."""
+        groq_api_key = os.getenv("GROQ_API_KEY")
+        if not groq_api_key:
             fallback = self.deterministic_answer_cleanup(candidate)
-            print("Gemini review status: SKIPPED — GEMINI_API_KEY is missing")
+            print("Groq review status: SKIPPED — GROQ_API_KEY is missing")
             print(f"Primary answer preserved: {fallback}")
             return fallback
 
         reviewer_model = os.getenv(
-            "GAIA_GEMINI_REVIEW_MODEL", DEFAULT_GEMINI_MODEL
+            "GAIA_GROQ_REVIEW_MODEL", DEFAULT_GROQ_REVIEW_MODEL
         )
         review_prompt = f"""
 You are the mandatory final reviewer for a GAIA exact-match answer.
@@ -811,7 +744,7 @@ Do not return JSON. Do not wrap the fields in Markdown or a code block.
                 )
                 response = completion(
                     model=reviewer_model,
-                    api_key=self.gemini_api_key,
+                    api_key=groq_api_key,
                     messages=[
                         {
                             "role": "user",
@@ -837,13 +770,13 @@ Do not return JSON. Do not wrap the fields in Markdown or a code block.
                 )
                 if not answer_match:
                     raise ValueError(
-                        "Gemini did not return the <final_answer> field."
+                        "Groq did not return the <final_answer> field."
                     )
                 final_answer = self.deterministic_answer_cleanup(
                     answer_match.group(1)
                 )
                 if not final_answer:
-                    raise ValueError("Gemini returned an empty final_answer.")
+                    raise ValueError("Groq returned an empty final_answer.")
 
                 changed = final_answer != candidate
                 note = (
@@ -852,9 +785,9 @@ Do not return JSON. Do not wrap the fields in Markdown or a code block.
                     else "Review completed without a note."
                 )
                 print(f"Candidate answer: {candidate}")
-                print(f"Gemini reviewed answer: {final_answer}")
-                print(f"Gemini changed answer: {changed}")
-                print(f"Gemini review note: {note}")
+                print(f"Groq reviewed answer: {final_answer}")
+                print(f"Groq changed answer: {changed}")
+                print(f"Groq review note: {note}")
                 return final_answer
             except Exception as exc:
                 last_error = exc
@@ -862,13 +795,13 @@ Do not return JSON. Do not wrap the fields in Markdown or a code block.
         fallback = self.deterministic_answer_cleanup(candidate)
         if not fallback:
             raise RuntimeError(
-                "A revisão do Gemini falhou e a resposta primária estava vazia. "
+                "A revisão do Groq falhou e a resposta primária estava vazia. "
                 f"Detalhe: {last_error}"
             )
 
         print(f"Candidate answer: {fallback}")
-        print("Gemini review status: FAILED — primary answer preserved")
-        print(f"Gemini review error: {last_error}")
+        print("Groq review status: FAILED — primary answer preserved")
+        print(f"Groq review error: {last_error}")
         return fallback