add openai req

app.py

@@ -21,6 +21,27 @@ import yaml
 import os
 from dotenv import load_dotenv
 
+
+import wikipediaapi
+
+@tool
+def wikipedia_search(query: str) -> str:
+    """
+    Searches for a page on Wikipedia and returns a summary of the page.
+
+    Args:
+        query (str): The search query for the Wikipedia page.
+
+    Returns:
+        str: A summary of the Wikipedia page, or an error message if the page is not found.
+    """
+    wiki_wiki = wikipediaapi.Wikipedia('MyProjectName (merci@example.com)', 'en')
+    page = wiki_wiki.page(query)
+    if page.exists():
+        return page.summary
+    else:
+        return "Could not find a Wikipedia page for that query."
+
 # with open("prompts.yaml", 'r') as stream:
 #     prompt_templates = yaml.safe_load(stream)
 # 1) load your .env (skip if you set it in the shell)
@@ -39,10 +60,10 @@ model = OpenAIServerModel(
 )                                            # :contentReference[oaicite:1]{index=1}
 
 # 4) create a CodeAgent powered by that model
-from smolagents import CodeAgent, DuckDuckGoSearchTool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, FinalAnswerTool
 
 OpenAIAgent = CodeAgent(
-    tools=[DuckDuckGoSearchTool()],            # any tools you want
+    tools=[ wikipedia_search],
     model=model,
     max_steps=6,
     verbosity_level=1,
@@ -107,7 +128,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        agent = OpenAIAgent()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None

requirements.txt

@@ -13,4 +13,5 @@ beautifulsoup4
 openai-whisper
 torch
 IMDbPY
-pyyaml
+pyyaml
+wikipedia-api