返回提交历史
Modified
g4f/Provider/helper.py
+8
-8
XFEstudio/gpt4free
Update helper.py
cbc08d60
代码差异
1 个文件
+8
-8
@@ -42,7 +42,6 @@ except ImportError:
42
42
43
43
from ..typing import Dict, Messages, Union, Tuple
44
44
from .. import debug
45
DisplayType = Union[Display, bool]
46
45
47
46
# Change event loop policy on windows
48
47
if sys.platform == 'win32':
@@ -140,23 +139,24 @@ def format_prompt(messages: Messages, add_special_tokens=False) -> str:
140
139
141
140
def get_browser(
142
141
user_data_dir: str = None,
143
display: bool = False,
144
proxy: str = None
142
hidden_display: bool = False,
143
proxy: str = None,
144
options: ChromeOptions = None
145
145
) -> Union[Chrome, Tuple[Chrome, Display]] :
146
146
if user_data_dir == None:
147
147
user_data_dir = user_config_dir("g4f")
148
148
149
if display:
149
if hidden_display:
150
150
display = Display(visible=0, size=(1920, 1080))
151
151
display.start()
152
152
153
options = None
154
153
if proxy:
155
options = ChromeOptions()
156
options.add_argument(f'--proxy-server={proxy}')
154
if not options:
155
options = ChromeOptions()
156
options.add_argument(f'--proxy-server={proxy}')
157
157
158
158
browser = Chrome(user_data_dir=user_data_dir, options=options)
159
if display:
159
if hidden_display:
160
160
return browser, display
161
161
162
162
return browser