pt
/
gpt4free
mirror of https://github.com/xtekky/gpt4free.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update helper.py

This commit is contained in:
H Lohaus 2023-11-16 18:39:34 +01:00 committed by GitHub
parent 51a6669210
commit cbc08d6014
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
g4f/Provider/helper.py
View File

@ -42,7 +42,6 @@ except ImportError:
from ..typing import Dict, Messages, Union, Tuple
from .. import debug
DisplayType = Union[Display, bool]
# Change event loop policy on windows
if sys.platform == 'win32':
@ -140,23 +139,24 @@ def format_prompt(messages: Messages, add_special_tokens=False) -> str:
def get_browser(
user_data_dir: str = None,
display: bool = False,
proxy: str = None
hidden_display: bool = False,
proxy: str = None,
options: ChromeOptions = None
) -> Union[Chrome, Tuple[Chrome, Display]] :
if user_data_dir == None:
user_data_dir = user_config_dir("g4f")
if display:
if hidden_display:
display = Display(visible=0, size=(1920, 1080))
display.start()
options = None
if proxy:
options = ChromeOptions()
options.add_argument(f'--proxy-server={proxy}')
if not options:
options = ChromeOptions()
options.add_argument(f'--proxy-server={proxy}')
browser = Chrome(user_data_dir=user_data_dir, options=options)
if display:
if hidden_display:
return browser, display
return browser