From 28896f2a987a835a49248a73d41ef98a65a5e356 Mon Sep 17 00:00:00 2001
From: eminemkun <teruteruwork@gmail.com>
Date: Mon, 18 Dec 2023 21:07:00 +0900
Subject: Fix Bing provider Cookie issue 2

---
 g4f/Provider/Bing.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

(limited to 'g4f/Provider')

diff --git a/g4f/Provider/Bing.py b/g4f/Provider/Bing.py
index af6c5132..56e964c0 100644
--- a/g4f/Provider/Bing.py
+++ b/g4f/Provider/Bing.py
@@ -10,6 +10,7 @@ import base64
 import numpy as np
 import uuid
 import urllib.parse
+import time
 from PIL import Image
 from aiohttp        import ClientSession, ClientTimeout
 from ..typing       import AsyncResult, Messages
@@ -26,7 +27,7 @@ default_cookies = {
     'KievRPSSecAuth': '',
     'SUID'          : '',
     'SRCHUSR'       : '',
-    'SRCHHPGUSR'    : '',
+    'SRCHHPGUSR'    : f'HV={int(time.time())}',
 }
 
 class Bing(AsyncGeneratorProvider):
@@ -52,8 +53,12 @@ class Bing(AsyncGeneratorProvider):
             prompt = messages[-1]["content"]
             context = create_context(messages[:-1])
         
-        if not cookies or "SRCHD" not in cookies:
+        if not cookies:
             cookies = default_cookies
+        else:
+            for key, value in default_cookies.items():
+                if key not in cookies:
+                    cookies[key] = value
         return stream_generate(prompt, tone, image, context, proxy, cookies)
 
 def create_context(messages: Messages):
-- 
cgit v1.2.3