From a2b5446b2e449f3107d9980d8eaac0a3ae07730f Mon Sep 17 00:00:00 2001
From: Heiner Lohaus <hlohaus@users.noreply.github.com>
Date: Wed, 17 Apr 2024 10:33:23 +0200
Subject: Fix DuckDuckGo Provider issues Fix PerplexityLabs, FlowGpt Provider
 Update Bing, Gemini Provider

---
 g4f/Provider/needs_auth/Gemini.py | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'g4f/Provider/needs_auth/Gemini.py')

diff --git a/g4f/Provider/needs_auth/Gemini.py b/g4f/Provider/needs_auth/Gemini.py
index fc9d9575..ebf5f413 100644
--- a/g4f/Provider/needs_auth/Gemini.py
+++ b/g4f/Provider/needs_auth/Gemini.py
@@ -60,6 +60,7 @@ class Gemini(AsyncGeneratorProvider):
         model: str,
         messages: Messages,
         proxy: str = None,
+        api_key: str = None,
         cookies: Cookies = None,
         connector: BaseConnector = None,
         image: ImageType = None,
@@ -67,6 +68,10 @@ class Gemini(AsyncGeneratorProvider):
         **kwargs
     ) -> AsyncResult:
         prompt = format_prompt(messages)
+        if api_key is not None:
+            if cookies is None:
+                cookies = {}
+            cookies["__Secure-1PSID"] = api_key
         cookies = cookies if cookies else get_cookies(".google.com", False, True)
         base_connector = get_connector(connector, proxy)
         async with ClientSession(
-- 
cgit v1.2.3