diff options
| author | H Lohaus <hlohaus@users.noreply.github.com> | 2024-02-09 18:51:27 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-02-09 18:51:27 +0100 |
| commit | 6c7c293fd16b970d857c888b5108334f00b27a19 (patch) | |
| tree | 3deecb18f982e133cb3b0337ffb0bdcbb607a212 /g4f/Provider/needs_auth | |
| parent | Update OpenaiChat.py (diff) | |
| parent | Update Gemini.py (diff) | |
| download | gpt4free-6c7c293fd16b970d857c888b5108334f00b27a19.tar gpt4free-6c7c293fd16b970d857c888b5108334f00b27a19.tar.gz gpt4free-6c7c293fd16b970d857c888b5108334f00b27a19.tar.bz2 gpt4free-6c7c293fd16b970d857c888b5108334f00b27a19.tar.lz gpt4free-6c7c293fd16b970d857c888b5108334f00b27a19.tar.xz gpt4free-6c7c293fd16b970d857c888b5108334f00b27a19.tar.zst gpt4free-6c7c293fd16b970d857c888b5108334f00b27a19.zip | |
Diffstat (limited to 'g4f/Provider/needs_auth')
| -rw-r--r-- | g4f/Provider/needs_auth/Gemini.py | 4 | ||||
| -rw-r--r-- | g4f/Provider/needs_auth/OpenaiChat.py | 3 |
2 files changed, 4 insertions, 3 deletions
diff --git a/g4f/Provider/needs_auth/Gemini.py b/g4f/Provider/needs_auth/Gemini.py index 402fc02f..32510505 100644 --- a/g4f/Provider/needs_auth/Gemini.py +++ b/g4f/Provider/needs_auth/Gemini.py @@ -66,6 +66,8 @@ class Gemini(AsyncGeneratorProvider): prompt = format_prompt(messages) if not cookies: + cookies = get_cookies(".google.com", False, True) + if "__Secure-1PSID" not in cookies or "__Secure-1PSIDCC" not in cookies: driver = None try: driver = get_browser(proxy=proxy) @@ -88,8 +90,6 @@ class Gemini(AsyncGeneratorProvider): if driver: driver.close() - if not cookies: - cookies = get_cookies(".google.com", False) if "__Secure-1PSID" not in cookies: raise MissingAuthError('Missing "__Secure-1PSID" cookie') diff --git a/g4f/Provider/needs_auth/OpenaiChat.py b/g4f/Provider/needs_auth/OpenaiChat.py index 7c707b5e..b1b267fd 100644 --- a/g4f/Provider/needs_auth/OpenaiChat.py +++ b/g4f/Provider/needs_auth/OpenaiChat.py @@ -38,6 +38,7 @@ class OpenaiChat(AsyncGeneratorProvider, ProviderModelMixin): supports_gpt_4 = True default_model = None models = ["gpt-3.5-turbo", "gpt-4", "gpt-4-gizmo"] + model_aliases = {"text-davinci-002-render-sha": "gpt-3.5-turbo"} _cookies: dict = {} @classmethod @@ -403,7 +404,7 @@ class OpenaiChat(AsyncGeneratorProvider, ProviderModelMixin): if "message_type" not in line["message"]["metadata"]: continue try: - image_response = await cls.get_generated_image(session, headers, line) + image_response = await cls.get_generated_image(session, auth_headers, line) if image_response: yield image_response except Exception as e: |