Google Search API

autogen/agentchat/contrib/web_surfer.py

@@ -9,16 +9,13 @@
 from typing_extensions import Annotated
 
 from ... import Agent, AssistantAgent, ConversableAgent, GroupChat, GroupChatManager, OpenAIWrapper, UserProxyAgent
-from ...browser_utils.bing_browser import BingTextBrowser
-from ...browser_utils.google_broswer import GoogleTextBrowser
+from ...browser_utils.browser_creator import TextBrowserCreator
 from ...code_utils import content_str
 from ...oai.openai_utils import filter_config
 from ...token_count_utils import count_token, get_max_token_limit
 
 logger = logging.getLogger(__name__)
 
-BROWSERS = {"google": GoogleTextBrowser, "bing": BingTextBrowser}
-
 
 class WebSurferAgent(ConversableAgent):
     """(In preview) An agent that acts as a basic web surfer that can search the web and visit web pages."""
@@ -43,7 +40,7 @@ def __init__(
         llm_config: Optional[Union[Dict, Literal[False]]] = None,
         summarizer_llm_config: Optional[Union[Dict, Literal[False]]] = None,
         default_auto_reply: Optional[Union[str, Dict, None]] = "",
-        browser_name: str = "bing",
+        browser_name: Literal['bing', 'google'] = "bing",
         browser_config: Optional[Union[Dict, None]] = None,
     ):
         super().__init__(
@@ -63,8 +60,7 @@ def __init__(
 
         # Create the browser
         self.browser_name = browser_name
-        chosen_browser = BROWSERS[self.browser_name]
-        self.browser = chosen_browser(**(browser_config if browser_config else {}))
+        self.browser = TextBrowserCreator.create_browser(self.browser_name)
 
         inner_llm_config = copy.deepcopy(llm_config)
 

autogen/browser_utils.py

@@ -28,7 +28,7 @@
     pass
 
 
-class SimpleTextBrowser:
+class TextBrowserBase:
     """(In preview) An extremely simple text-based web browser comparable to Lynx. Suitable for Agentic use."""
 
     def __init__(

autogen/browser_utils/base_browser.py

@@ -27,7 +27,7 @@
     pass
 
 
-class SimpleTextBrowser:
+class TextBrowserBase:
     """(In preview) An extremely simple text-based web browser comparable to Lynx. Suitable for Agentic use."""
 
     def __init__(

autogen/browser_utils/bing_browser.py

@@ -3,10 +3,10 @@
 
 import requests
 
-from .base_browser import SimpleTextBrowser
+from .base_browser import TextBrowserBase
 
 
-class BingTextBrowser(SimpleTextBrowser):
+class BingTextBrowser(TextBrowserBase):
     """(In preview) An extremely simple text-based web browser comparable to Lynx. Suitable for Agentic use."""
 
     def __init__(
@@ -19,6 +19,8 @@ def __init__(
         request_kwargs: Optional[Union[Dict[str, Any], None]] = None,
     ):
         super().__init__(start_page, viewport_size, downloads_folder, base_url, api_key, request_kwargs)
+        self.name = 'bing'
+
 
     def set_address(self, uri_or_path: str) -> None:
         self.history.append(uri_or_path)

autogen/browser_utils/browser_creator.py

@@ -0,0 +1,19 @@
+from bing_browser import BingTextBrowser
+from google_browser import GoogleTextBrowser
+
+
+class TextBrowserCreator:
+    """Creator class for creating different text browsers at runtime. Make sure to add newly registered browsers here"""
+
+    browser_classes = {
+        'BingTextBrowser': BingTextBrowser,
+        'GoogleTextBrowser': GoogleTextBrowser
+    }
+
+    @classmethod
+    def create_browser(cls, name: str):
+        """Factory method to create a text browser instance based on the name."""
+        if name in cls.browser_classes:
+            return cls.browser_classes[name]()
+        else:
+            raise ValueError(f"Unknown browser name: {name}")

autogen/browser_utils/google_broswer.py → autogen/browser_utils/google_browser.py

@@ -3,10 +3,10 @@
 
 import requests
 
-from .base_browser import SimpleTextBrowser
+from .base_browser import TextBrowserBase
 
 
-class GoogleTextBrowser(SimpleTextBrowser):
+class GoogleTextBrowser(TextBrowserBase):
     """(In preview) An extremely simple text-based web browser comparable to Lynx. Suitable for Agentic use."""
 
     def __init__(
@@ -22,6 +22,7 @@ def __init__(
     ):
         super().__init__(start_page, viewport_size, downloads_folder, base_url, api_key, request_kwargs)
         self.cx = cx
+        self.name = 'google'
 
     def set_address(self, uri_or_path: str) -> None:
         self.history.append(uri_or_path)

test/test_browser_utils_google.py

@@ -10,7 +10,7 @@
 import requests
 from agentchat.test_assistant_agent import KEY_LOC  # noqa: E402
 
-from autogen.browser_utils.google_broswer import GoogleTextBrowser
+from autogen.browser_utils.google_browser import GoogleTextBrowser
 
 BLOG_POST_URL = "https://microsoft.github.io/autogen/blog/2023/04/21/LLM-tuning-math"
 BLOG_POST_TITLE = "Does Model and Inference Parameter Matter in LLM Applications? - A Case Study for MATH | AutoGen"
@@ -36,7 +36,7 @@
 
 
 try:
-    from autogen.browser_utils.google_broswer import GoogleTextBrowser
+    from autogen.browser_utils.google_browser import GoogleTextBrowser
 except ImportError:
     skip_all = True
 else: