fix some bugs

Author: DVampire

configs/config_example.toml

@@ -29,7 +29,7 @@ time_limit_seconds = 60
 max_follow_ups = 3
 
 [browser_tool]
-model_id = "langchain-gpt-4.1"
+model_id = "gpt-4.1"
 headless = false
 disable_security = true
 extra_chromium_args = []

configs/config_gaia.toml

@@ -29,7 +29,7 @@ time_limit_seconds = 60
 max_follow_ups = 3
 
 [browser_tool]
-model_id = "langchain-gpt-4.1"
+model_id = "gpt-4.1"
 headless = false
 disable_security = true
 extra_chromium_args = []

configs/config_hle.toml

@@ -29,7 +29,7 @@ time_limit_seconds = 60
 max_follow_ups = 3
 
 [browser_tool]
-model_id = "langchain-gpt-4.1"
+model_id = "gpt-4.1"
 headless = false
 disable_security = true
 extra_chromium_args = []

src/tools/auto_browser.py

@@ -67,6 +67,12 @@ async def _browser_task(self, task):
         controller = Controller(http_save_path=self.http_save_path)
 
         model_id = self.browser_tool_config.model_id
+
+        assert model_id in ['gpt-4.1'], f"Model should be in [gpt-4.1, ], but got {model_id}. Please check your config file."
+
+        if "lanchain" not in model_id:
+            model_id = f"lanchain-{model_id}"
+
         model = model_manager.registed_models[model_id]
 
         browser_agent = Agent(