Update app.py

app.py

@@ -11,6 +11,7 @@ from typing import TypedDict, List, Dict, Any, Optional
 from langgraph.graph import StateGraph, END
 from langchain_openai import ChatOpenAI
 from langchain_core.messages import HumanMessage
+import wikipedia
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -99,6 +100,33 @@ def web_search(query: str) -> str:
     result=search_tool(query)
     return f"Search results for '{query}' : {result}."
 
+@tool
+def wikipedia_tool(query: str, llm=openrouter_inference) -> str:
+    """
+    Searches Wikipedia for the given query and uses the LLM to extract a direct answer.
+    Args:
+        query (str): The search term or question.
+        llm (callable, optional): The language model to extract the answer.
+    Returns:
+        str: A direct answer or summary from Wikipedia.
+    """
+    try:
+        summary = wikipedia.summary(query, sentences=5)
+        if llm:
+            # Ask the LLM to extract a direct answer from the summary
+            prompt = (
+                f"Given the following Wikipedia summary, answer the question as directly as possible:\n"
+                f"Question: {query}\n"
+                f"Summary: {summary}\n"
+                f"Answer:"
+            )
+            answer = llm(prompt)
+            return answer.strip()
+        else:
+            return summary
+    except Exception as e:
+        return f"Wikipedia search error: {e}"
+
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 
@@ -114,9 +142,10 @@ class BasicAgent:
         Always return a valid answer after validating the source.
         
         Your final answer should be in the following format:
-        "<your answer here>".
+        "your answer here".
         
         If you are asked question related to numbers,return the answer in numeric format.
+        If you want more details about a topic,person, you can use the wikipedia tool.
         If you want to perform web search to find the answer, use the web_search tool.
         If you are given a task to create an image, you can use the image_generation_tool.