summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/helper.py
diff options
context:
space:
mode:
authorTekky <98614666+xtekky@users.noreply.github.com>2023-11-17 09:28:55 +0100
committerGitHub <noreply@github.com>2023-11-17 09:28:55 +0100
commit053cfa6630a64caac04dbf0bc2aab177224ef807 (patch)
tree095bf4dae94e1b96e2bc2ac638d2608e1f3a4e37 /g4f/Provider/helper.py
parentMerge pull request #1261 from hlohaus/any (diff)
parentUpdate Readme (diff)
downloadgpt4free-053cfa6630a64caac04dbf0bc2aab177224ef807.tar
gpt4free-053cfa6630a64caac04dbf0bc2aab177224ef807.tar.gz
gpt4free-053cfa6630a64caac04dbf0bc2aab177224ef807.tar.bz2
gpt4free-053cfa6630a64caac04dbf0bc2aab177224ef807.tar.lz
gpt4free-053cfa6630a64caac04dbf0bc2aab177224ef807.tar.xz
gpt4free-053cfa6630a64caac04dbf0bc2aab177224ef807.tar.zst
gpt4free-053cfa6630a64caac04dbf0bc2aab177224ef807.zip
Diffstat (limited to '')
-rw-r--r--g4f/Provider/helper.py22
1 files changed, 3 insertions, 19 deletions
diff --git a/g4f/Provider/helper.py b/g4f/Provider/helper.py
index b10c912a..f879816a 100644
--- a/g4f/Provider/helper.py
+++ b/g4f/Provider/helper.py
@@ -31,14 +31,6 @@ except ImportError:
class ChromeOptions():
def add_argument():
pass
-try:
- from pyvirtualdisplay import Display
-except ImportError:
- class Display():
- def start():
- pass
- def stop():
- pass
from ..typing import Dict, Messages, Union, Tuple
from .. import debug
@@ -139,24 +131,16 @@ def format_prompt(messages: Messages, add_special_tokens=False) -> str:
def get_browser(
user_data_dir: str = None,
- hidden_display: bool = False,
+ headless: bool = False,
proxy: str = None,
options: ChromeOptions = None
-) -> Union[Chrome, Tuple[Chrome, Display]] :
+) -> Chrome:
if user_data_dir == None:
user_data_dir = user_config_dir("g4f")
- if hidden_display:
- display = Display(visible=0, size=(1920, 1080))
- display.start()
-
if proxy:
if not options:
options = ChromeOptions()
options.add_argument(f'--proxy-server={proxy}')
- browser = Chrome(user_data_dir=user_data_dir, options=options)
- if hidden_display:
- return browser, display
-
- return browser \ No newline at end of file
+ return Chrome(user_data_dir=user_data_dir, options=options, headless=headless) \ No newline at end of file