Upload agent.py

agent.py

@@ -119,6 +119,45 @@ class AgentRouter:
 
         return tool_output
 
+    @staticmethod
+    def _is_unhelpful_web_response(text: str) -> bool:
+        lower = text.lower()
+        bad_patterns = [
+            "i don't have access to real-time",
+            "i do not have access to real-time",
+            "i can't access real-time",
+            "cannot access real-time",
+            "as an ai language model",
+            "you can use any reliable news",
+        ]
+        return any(pattern in lower for pattern in bad_patterns)
+
+    @staticmethod
+    def _summarize_web_tool_output(tool_output: str, message: str) -> str:
+        if tool_output.startswith("Web search unavailable"):
+            return "Web search is currently unavailable. Please try again in a moment."
+
+        if tool_output.startswith("No web results found"):
+            return f"I could not find recent web results for: {message}."
+
+        lines = [line.strip() for line in tool_output.splitlines() if line.strip()]
+        bullets = []
+
+        for line in lines[:5]:
+            # Expected line format:
+            # 1. Title | snippet text | Source: https://...
+            match = re.match(r"^\d+\.\s+(.*?)\s+\|\s+(.*?)\s+\|\s+Source:\s+(.*)$", line)
+            if match:
+                title, snippet, source = match.groups()
+                bullets.append(f"- {title}: {snippet} (Source: {source})")
+            else:
+                bullets.append(f"- {line}")
+
+        if not bullets:
+            return "I found web results, but could not format them cleanly. Please retry."
+
+        return "Here are the latest web results:\n" + "\n".join(bullets)
+
     def respond(self, user_id: str, message: str) -> str:
         memory_context = get_relevant_context(user_id, message)
         route = self._classify(message)
@@ -135,6 +174,24 @@ class AgentRouter:
         else:
             tool_context = f"Tool used: {tool_name}\n{tool_output}"
 
+        if tool_name == "web_search":
+            web_instruction = (
+                "Answer using only the provided web results. "
+                "Do not say you lack real-time access. "
+                "Provide a concise summary with sources."
+            )
+            response = self.model.generate(
+                message=f"{web_instruction}\n\nUser request: {message}",
+                memory_context=memory_context,
+                tool_context=tool_context,
+            )
+
+            if self._is_unhelpful_web_response(response):
+                response = self._summarize_web_tool_output(tool_output, message)
+
+            save_interaction(user_id, message, response)
+            return response
+
         response = self.model.generate(
             message=message,
             memory_context=memory_context,