From 89e096334dfb1049e8cb427a866739d934cd7ad8 Mon Sep 17 00:00:00 2001 From: hlohaus <983577+hlohaus@users.noreply.github.com> Date: Fri, 31 Jan 2025 17:36:48 +0100 Subject: Support reasoning tokens by default Add new default HuggingFace provider Add format_image_prompt and get_last_user_message helper Add stop_browser callable to get_nodriver function Fix content type response in images route --- g4f/Provider/needs_auth/OpenaiChat.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'g4f/Provider/needs_auth/OpenaiChat.py') diff --git a/g4f/Provider/needs_auth/OpenaiChat.py b/g4f/Provider/needs_auth/OpenaiChat.py index adcecb3f..8fde1090 100644 --- a/g4f/Provider/needs_auth/OpenaiChat.py +++ b/g4f/Provider/needs_auth/OpenaiChat.py @@ -98,7 +98,7 @@ class OpenaiChat(AsyncAuthedProvider, ProviderModelMixin): default_model = "auto" default_image_model = "dall-e-3" image_models = [default_image_model] - text_models = [default_model, "gpt-4", "gpt-4o", "gpt-4o-mini", "gpt-4o-canmore", "o1", "o1-preview", "o1-mini"] + text_models = [default_model, "gpt-4", "gpt-4o", "gpt-4o-mini", "o1", "o1-preview", "o1-mini"] vision_models = text_models models = text_models + image_models synthesize_content_type = "audio/mpeg" @@ -598,7 +598,7 @@ class OpenaiChat(AsyncAuthedProvider, ProviderModelMixin): @classmethod async def nodriver_auth(cls, proxy: str = None): - browser = await get_nodriver(proxy=proxy) + browser, stop_browser = await get_nodriver(proxy=proxy) try: page = browser.main_tab def on_request(event: nodriver.cdp.network.RequestWillBeSent): @@ -648,7 +648,7 @@ class OpenaiChat(AsyncAuthedProvider, ProviderModelMixin): cls._create_request_args(RequestConfig.cookies, RequestConfig.headers, user_agent=user_agent) cls._set_api_key(cls._api_key) finally: - browser.stop() + stop_browser() @staticmethod def get_default_headers() -> Dict[str, str]: -- cgit v1.2.3