Update BingCreateImages.py

This commit is contained in:
H Lohaus 2024-05-07 07:37:23 +02:00 committed by GitHub
parent a9672bead0
commit 03e5f10992
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -51,7 +51,7 @@ class BingCreateImages(AsyncGeneratorProvider, ProviderModelMixin):
Generator[str, None, None]: The final output as markdown formatted string with images. Generator[str, None, None]: The final output as markdown formatted string with images.
""" """
cookies = self.cookies or get_cookies(".bing.com", False) cookies = self.cookies or get_cookies(".bing.com", False)
if "_U" not in cookies: if cookies is None or "_U" not in cookies:
login_url = os.environ.get("G4F_LOGIN_URL") login_url = os.environ.get("G4F_LOGIN_URL")
if login_url: if login_url:
yield f"Please login: [Bing]({login_url})\n\n" yield f"Please login: [Bing]({login_url})\n\n"
@ -72,7 +72,7 @@ class BingCreateImages(AsyncGeneratorProvider, ProviderModelMixin):
str: Markdown formatted string with images. str: Markdown formatted string with images.
""" """
cookies = self.cookies or get_cookies(".bing.com", False) cookies = self.cookies or get_cookies(".bing.com", False)
if "_U" not in cookies: if cookies is None or "_U" not in cookies:
raise MissingAuthError('Missing "_U" cookie') raise MissingAuthError('Missing "_U" cookie')
proxy = self.proxy or os.environ.get("G4F_PROXY") proxy = self.proxy or os.environ.get("G4F_PROXY")
async with create_session(cookies, proxy) as session: async with create_session(cookies, proxy) as session: