fix: 修复bug

browser_utils.py

@@ -11,13 +11,13 @@ class BrowserManager:
     def __init__(self):
         self.browser = None
 
-    def init_browser(self):
+    def init_browser(self, user_agent=None):
         """初始化浏览器"""
-        co = self._get_browser_options()
+        co = self._get_browser_options(user_agent)
         self.browser = Chromium(co)
         return self.browser
 
-    def _get_browser_options(self):
+    def _get_browser_options(self, user_agent=None):
         """获取浏览器配置"""
         co = ChromiumOptions()
         try:
@@ -26,15 +26,19 @@ class BrowserManager:
         except FileNotFoundError as e:
             logging.warning(f"警告: {e}")
 
-
         co.set_pref("credentials_enable_service", False)
         co.set_argument("--hide-crash-restore-bubble")
-        proxy = os.getenv('BROWSER_PROXY')
+        proxy = os.getenv("BROWSER_PROXY")
         if proxy:
             co.set_proxy(proxy)
-        
+
         co.auto_port()
-        co.headless(os.getenv('BROWSER_HEADLESS', 'True').lower() == 'true')  # 生产环境使用无头模式
+        if user_agent:
+            co.set_user_agent(user_agent)
+
+        co.headless(
+            os.getenv("BROWSER_HEADLESS", "True").lower() == "true"
+        )  # 生产环境使用无头模式
 
         # Mac 系统特殊处理
         if sys.platform == "darwin":