more tools now that we shouldnt hit rate limits

gemini_agent.py

@@ -5,6 +5,9 @@ from excel_parser import ExcelParser
 import re
 import time
 import asyncio
+# Add LangChain tools for Wikipedia and DuckDuckGo
+from langchain.tools import DuckDuckGoSearchRun, WikipediaQueryRun
+from langchain.utilities import WikipediaAPIWrapper
 
 load_dotenv()
 
@@ -22,6 +25,9 @@ class GeminiAgent:
         
         # Initialize parsers
         self.excel_parser = ExcelParser()
+        # Initialize Wikipedia and DuckDuckGo tools
+        self.wiki_tool = WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper())
+        self.ddg_tool = DuckDuckGoSearchRun()
         
     async def __call__(self, question: str) -> str:
         print(f"GeminiAgent received question (first 50 chars): {question}...")
@@ -157,6 +163,20 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
     async def _handle_text_question(self, question: str) -> str:
         """Handle regular text-based questions"""
         prompt = ""
+        wiki_context = ""
+        ddg_context = ""
+        # Wikipedia retrieval
+        if 'wikipedia' in question.lower():
+            try:
+                wiki_context = self.wiki_tool.run(question)
+            except Exception as e:
+                print(f"Wikipedia tool failed: {e}")
+        # DuckDuckGo retrieval
+        if 'duckduckgo' in question.lower() or 'web search' in question.lower():
+            try:
+                ddg_context = self.ddg_tool.run(question)
+            except Exception as e:
+                print(f"DuckDuckGo tool failed: {e}")
         # Handle attached file questions with enhanced prompts
         if 'attached' in question.lower():
             if 'python code' in question.lower():
@@ -194,6 +214,12 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
         else:
             prompt = f"""Answer this question with only the essential information requested:\n\n{question}\n\nAnswer:"""
         
+        # Prepend context to the prompt if available
+        if wiki_context:
+            prompt = f"Use the following Wikipedia context to answer the question:\n{wiki_context}\n\n{prompt}"
+        elif ddg_context:
+            prompt = f"Use the following web search context to answer the question:\n{ddg_context}\n\n{prompt}"
+        
         # Use the constructed prompt for all cases
         await self._rate_limit()
         response = self.model.generate_content(
@@ -242,13 +268,6 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
             if list_match:
                 answer = ', '.join([item.strip() for item in list_match if item.strip()])
 
-        # Wikipedia tool integration (simple version)
-        if 'wikipedia' in question.lower() or 'according to wikipedia' in question.lower():
-            # Add a Wikipedia search instruction to the prompt if not already present
-            if 'wikipedia' not in prompt.lower():
-                prompt += "\nIf you do not know the answer, search the latest English Wikipedia and use only information from there."
-            # Optionally, you could call a real Wikipedia API here for retrieval-augmented generation
-
         return answer
     
     async def _generate_video_answer_from_question(self, question: str, video_id: str) -> str: