Update app.py

app.py

@@ -4,11 +4,32 @@ import requests
 import pandas as pd
 import datetime
 from smolagents import CodeAgent, OpenAIServerModel, DuckDuckGoSearchTool, VisitWebpageTool, tool
+from langchain_community.document_loaders import WikipediaLoader, ArxivLoader
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 
+@tool
+def wikipedia_search(query: str) -> str:
+    """Search Wikipedia and return a concise summary."""
+    try:
+        docs = WikipediaLoader(query=query, load_max_docs=2).load()
+        return "\n\n".join(doc.page_content[:1500] for doc in docs)
+    except Exception as e:
+        return f"Wikipedia error: {e}"
+
+@tool
+def arxiv_search(query: str) -> str:
+    """Search arXiv and return paper abstracts."""
+    try:
+        docs = ArxivLoader(query=query, load_max_docs=2).load()
+        return "\n\n".join(
+            f"Title: {doc.metadata.get('Title', '')}\n{doc.page_content[:1500]}"
+            for doc in docs
+        )
+    except Exception as e:
+        return f"arXiv error: {e}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -33,13 +54,13 @@ class BasicAgent:
         self.visit_tool = VisitWebpageTool()
 
         self.model = OpenAIServerModel(
-            model_id="openai/gpt-oss-120b",
+            model_id="qwen/qwen3-32b",
             api_base="https://api.groq.com/openai/v1",
             api_key=os.getenv("GROQ_API_KEY")
         )
 
         self.agent = CodeAgent(
-            tools=[self.search_tool, self.visit_tool, get_current_time_in_timezone],
+            tools=[self.search_tool, self.visit_tool, wikipedia_search, arxiv_search, get_current_time_in_timezone],
             model=self.model,
             add_base_tools=True,
             max_steps=20,