mirror of
https://github.com/xtekky/gpt4free.git
synced 2024-09-17 10:29:29 -04:00
Update helper.py
This commit is contained in:
parent
51a6669210
commit
cbc08d6014
@ -42,7 +42,6 @@ except ImportError:
|
|||||||
|
|
||||||
from ..typing import Dict, Messages, Union, Tuple
|
from ..typing import Dict, Messages, Union, Tuple
|
||||||
from .. import debug
|
from .. import debug
|
||||||
DisplayType = Union[Display, bool]
|
|
||||||
|
|
||||||
# Change event loop policy on windows
|
# Change event loop policy on windows
|
||||||
if sys.platform == 'win32':
|
if sys.platform == 'win32':
|
||||||
@ -140,23 +139,24 @@ def format_prompt(messages: Messages, add_special_tokens=False) -> str:
|
|||||||
|
|
||||||
def get_browser(
|
def get_browser(
|
||||||
user_data_dir: str = None,
|
user_data_dir: str = None,
|
||||||
display: bool = False,
|
hidden_display: bool = False,
|
||||||
proxy: str = None
|
proxy: str = None,
|
||||||
|
options: ChromeOptions = None
|
||||||
) -> Union[Chrome, Tuple[Chrome, Display]] :
|
) -> Union[Chrome, Tuple[Chrome, Display]] :
|
||||||
if user_data_dir == None:
|
if user_data_dir == None:
|
||||||
user_data_dir = user_config_dir("g4f")
|
user_data_dir = user_config_dir("g4f")
|
||||||
|
|
||||||
if display:
|
if hidden_display:
|
||||||
display = Display(visible=0, size=(1920, 1080))
|
display = Display(visible=0, size=(1920, 1080))
|
||||||
display.start()
|
display.start()
|
||||||
|
|
||||||
options = None
|
|
||||||
if proxy:
|
if proxy:
|
||||||
options = ChromeOptions()
|
if not options:
|
||||||
options.add_argument(f'--proxy-server={proxy}')
|
options = ChromeOptions()
|
||||||
|
options.add_argument(f'--proxy-server={proxy}')
|
||||||
|
|
||||||
browser = Chrome(user_data_dir=user_data_dir, options=options)
|
browser = Chrome(user_data_dir=user_data_dir, options=options)
|
||||||
if display:
|
if hidden_display:
|
||||||
return browser, display
|
return browser, display
|
||||||
|
|
||||||
return browser
|
return browser
|
Loading…
Reference in New Issue
Block a user