fix : stuck for and back same webpage link

sources/agents/browser_agent.py

@@ -336,7 +336,7 @@ class BrowserAgent(Agent):
                 complete = True
                 break
 
-            if link == None or Action.GO_BACK.value in answer:
+            if link == None or Action.GO_BACK.value in answer or link in self.search_history:
                 pretty_print(f"Going back to results. Still {len(unvisited)}", color="status")
                 unvisited = self.select_unvisited(search_result)
                 prompt = self.make_newsearch_prompt(user_prompt, unvisited)

sources/router.py

@@ -128,6 +128,7 @@ class AgentRouter:
             ("Check if a file named ‘project_proposal.pdf’ exists in my Documents", "LOW"),
             ("Search the web for tips on improving coding skills", "LOW"),
             ("Write a Python script to count words in a text file", "LOW"),
+            ("Search the web for restaurant", "LOW"),
             ("Find a public API for sports scores and build a web app to show live updates", "HIGH"),
             ("Create a simple HTML page with CSS styling", "LOW"),
             ("hi", "LOW"),

tests/test_browser_agent_parsing.py

@@ -17,12 +17,13 @@ class TestBrowserAgentParsing(unittest.TestCase):
         # Test various link formats
         test_text = """
         Check this out: https://thriveonai.com/15-ai-startups-in-japan-to-take-note-of, and www.google.com!
-        Also try https://test.org/about?page=1.
+        Also try https://test.org/about?page=1, hey this one as well bro https://weatherstack.com/documentation.
         """
         expected = [
             "https://thriveonai.com/15-ai-startups-in-japan-to-take-note-of",
             "www.google.com",
-            "https://test.org/about?page=1"
+            "https://test.org/about?page=1",
+            "https://weatherstack.com/documentation"
         ]
         result = self.agent.extract_links(test_text)
         self.assertEqual(result, expected)