remove delay and print

agent.py

@@ -36,6 +36,7 @@ repl_tool = Tool(
 tavily_search_tool = TavilySearch(
     max_results=5,
     topic="general",
+    search_depth="advanced"
 )
 
 # Initialize Tavily Extract Tool
@@ -92,9 +93,27 @@ def wiki_search(query: str) -> str:
             f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'
             for doc in search_docs
         ])
-    print(formatted_search_docs)
     return {"wiki_results": formatted_search_docs}
 
+@tool
+def solve_logic_or_math_problem(problem: str) -> str:
+    """Solve logic or math problem.
+    
+    Args:
+        problem: The problem statement."""
+    print('solve')
+    llm = ChatGoogleGenerativeAI(
+      model="gemini-2.0-flash",
+      temperature=0.1,
+      max_tokens=None,
+      timeout=None,
+      max_retries=2,
+      # other params...
+    )
+    response = llm.invoke(problem)
+    return response.content
+
+
 # @tool
 # def web_search(query: str) -> str:
 #     """Search Tavily for a query and return maximum 3 results.
@@ -127,19 +146,12 @@ tools = [
     tavily_extract_tool,
     repl_tool,
     read_excel_file,
-    transcribe_audio
+    transcribe_audio,
+    solve_logic_or_math_problem
 ]
 
 
-# llm = ChatGroq(model="qwen-qwq-32b", temperature=0.7)
-llm = ChatGoogleGenerativeAI(
-    model="gemini-2.0-flash",
-    temperature=0.7,
-    max_tokens=None,
-    timeout=None,
-    max_retries=2,
-    # other params...
-)
+llm = ChatGroq(model="qwen-qwq-32b", temperature=0.1)
 llm_with_tools = llm.bind_tools(tools)
 
 def assistant(state: MessagesState):
@@ -147,7 +159,6 @@ def assistant(state: MessagesState):
     return {"messages": [llm_with_tools.invoke(state["messages"])]}
 
 def final_answer(answer):
-  print(f"ANSWER: {answer}")
   return answer.replace("FINAL ANSWER:","")
 
 builder = StateGraph(MessagesState)
@@ -173,6 +184,7 @@ if __name__ == "__main__":
     # question = "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?"
     # question = "Where were the Vietnamese specimens described by Kuznetzov in Nedoshivina's 2010 paper eventually deposited? Just give me the city name without abbreviations."
     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."
+    question ="Given this table defining * on the set S = {a, b, c, d, e}\n\n|*|a|b|c|d|e|\n|---|---|---|---|---|---|\n|a|a|b|c|b|d|\n|b|b|c|a|e|c|\n|c|c|a|b|b|a|\n|d|b|e|b|e|d|\n|e|d|b|a|d|c|\n\nprovide the subset of S involved in any possible counter-examples that prove * is not commutative. Provide your answer as a comma separated list of the elements in the set in alphabetical order."
     # getmessages = [HumanMessage(content=question)]
     # messages = graph.invoke({"messages": messages})
     # for m in messages["messages"]:

app.py

@@ -77,7 +77,6 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     answers_payload = []
     print(f"Running agent on {len(questions_data)} questions...")
     for item in questions_data:
-        time.sleep(10)
         task_id = item.get("task_id")
         question_text = item.get("question")
         if not task_id or question_text is None: