summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/helper.py
diff options
context:
space:
mode:
Diffstat (limited to 'g4f/Provider/helper.py')
-rw-r--r--g4f/Provider/helper.py16
1 files changed, 8 insertions, 8 deletions
diff --git a/g4f/Provider/helper.py b/g4f/Provider/helper.py
index dee27fc4..cad32f05 100644
--- a/g4f/Provider/helper.py
+++ b/g4f/Provider/helper.py
@@ -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 \ No newline at end of file