related check

gemini_agent.py

@@ -214,10 +214,14 @@ 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:
+        # 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):
             prompt = f"Use the following Wikipedia context to answer the question:\n{wiki_context}\n\n{prompt}"
-        elif ddg_context:
+        elif ddg_context and is_relevant_context(ddg_context, question):
             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