Using enums for browser for extendabilty

autogen/agentchat/contrib/web_surfer.py

@@ -9,7 +9,7 @@
 from typing_extensions import Annotated
 
 from ... import Agent, AssistantAgent, ConversableAgent, GroupChat, GroupChatManager, OpenAIWrapper, UserProxyAgent
-from ...browser_utils.browser_creator import TextBrowserCreator
+from ...browser_utils.browser_creator import TextBrowserEnum
 from ...code_utils import content_str
 from ...oai.openai_utils import filter_config
 from ...token_count_utils import count_token, get_max_token_limit
@@ -60,7 +60,7 @@ def __init__(
 
         # Create the browser
         self.browser_name = browser_name
-        self.browser = TextBrowserCreator.create_browser(self.browser_name)
+        self.browser = TextBrowserEnum.get_browser(browser_name)
 
         inner_llm_config = copy.deepcopy(llm_config)
 

autogen/browser_utils/browser_creator.py

@@ -1,19 +1,14 @@
-from bing_browser import BingTextBrowser
-from google_browser import GoogleTextBrowser
+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"""
+class TextBrowserEnum:
+    """Enum class for creating different text browsers. Make sure to add newly registered browsers here"""
 
-    browser_classes = {
-        'BingTextBrowser': BingTextBrowser,
-        'GoogleTextBrowser': GoogleTextBrowser
-    }
+    bing = BingTextBrowser
+    google = 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}")
+    def get_browser(cls, browser_str):
+        return getattr(cls, browser_str)
+