Fix key error in set_cookies

xtekky · Feb 1, 2024 · 2a591b8 · 2a591b8
1 parent d6b3d0f
commit 2a591b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/g4f/Provider/helper.py b/g4f/Provider/helper.py
@@ -51,7 +51,7 @@ def get_cookies(domain_name: str = '', raise_requirements_error: bool = True) ->
 def set_cookies(domain_name: str, cookies: Cookies = None) -> None:
     if cookies:
         _cookies[domain_name] = cookies
-    else:
+    elif domain_name in _cookies:
         _cookies.pop(domain_name)
 
 def load_cookies_from_browsers(domain_name: str, raise_requirements_error: bool = True) -> Cookies: