add Wikipedia QA tool

gemini_agent.py

@@ -22,6 +22,9 @@ from PIL import Image
 import google.generativeai as genai
 from pydantic import Field
 
+from transformers import pipeline
+import wikipedia
+
 from smolagents import WikipediaSearchTool
 
 class SmolagentToolWrapper(BaseTool):
@@ -278,6 +281,18 @@ def analyze_excel_file(file_path: str, query: str) -> str:
         return "Error: pandas and openpyxl are not installed. Please install them with 'pip install pandas openpyxl'."
     except Exception as e:
         return f"Error analyzing Excel file: {str(e)}"
+    
+qa_pipeline = pipeline(
+    "question-answering",
+    model="distilbert-base-cased-distilled-squad",
+    tokenizer="distilbert-base-cased-distilled-squad"
+)
+
+def wikipedia_qa(question: str) -> str:
+    # 先搜条目——可以传“Mercedes Sosa”或直接用 query
+    page = wikipedia.page("Mercedes Sosa").content
+    out = qa_pipeline(question=question, context=page)
+    return out["answer"]    
 
 class GeminiAgent:
     def __init__(self, api_key: str, model_name: str = "gemini-2.0-flash"):
@@ -324,7 +339,12 @@ class GeminiAgent:
                 name="web_search",
                 func=self._web_search,
                 description="Search the web for information"
-            )
+            ),
+            Tool(
+            name="wikipedia_qa",
+            func=wikipedia_qa,
+            description="给定问题 + 维基文章，直接抽取精确答案"
+)
         ]
         
         # Setup memory
@@ -591,7 +611,7 @@ New question: {input}
             agent=agent,
             tools=self.tools,
             memory=self.memory,
-            max_iterations=20,
+            max_iterations=10,
             verbose=True,
             handle_parsing_errors=True,
             return_only_outputs=True  # This ensures we only get the final output