Update gemini_agent.py

gemini_agent.py

@@ -163,20 +163,29 @@ 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 = ""
+        # Only use retrieval for explicit web/Wikipedia questions
+        def is_explicit_retrieval_question(question):
+            q = question.lower()
+            return (
+                "according to wikipedia" in q or
+                "from wikipedia" in q or
+                "search the web" in q or
+                "duckduckgo" in q or
+                "web search" in q
+            )
         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}")
+        if is_explicit_retrieval_question(question):
+            if "wikipedia" in question.lower():
+                try:
+                    wiki_context = self.wiki_tool.run(question)
+                except Exception as e:
+                    print(f"Wikipedia tool failed: {e}")
+            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():
@@ -215,13 +224,11 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
             prompt = f"""Answer this question with only the essential information requested:\n\n{question}\n\nAnswer:"""
         
         # Prepend context to the prompt if available and likely relevant
-        def is_relevant_context(context, question):
-            # Only use context if it contains a main keyword from the question and is not a generic error
-            keywords = [w.lower().strip(',.?') for w in question.split() if len(w) > 3]
-            return context and not any(x in context.lower() for x in ["not found", "no results", "does not contain information"]) and any(kw in context.lower() for kw in keywords)
-        if wiki_context and is_relevant_context(wiki_context, question):
+        def is_good_context(context):
+            return context and not any(x in context.lower() for x in ["not found", "no results", "does not contain information"])
+        if wiki_context and is_good_context(wiki_context):
             prompt = f"Use the following Wikipedia context to answer the question:\n{wiki_context}\n\n{prompt}"
-        elif ddg_context and is_relevant_context(ddg_context, question):
+        elif ddg_context and is_good_context(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