Update Bing.py

Fixed some errors in my previously precipitated commit.
pull/851/head
Luneye 1 year ago committed by GitHub
parent 87c7db73a2
commit a3851150b7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -10,12 +10,13 @@ class Bing(AsyncGeneratorProvider):
supports_gpt_4 = True supports_gpt_4 = True
working=True working=True
supports_stream=True supports_stream=True
needs_auth=True
@staticmethod @staticmethod
def create_completion( def create_completion(
model: str, model: str,
messages: list[dict[str, str]], messages: list[dict[str, str]],
cookies: dict = None, cookies: dict,
**kwargs **kwargs
) -> AsyncGenerator: ) -> AsyncGenerator:
@ -26,11 +27,11 @@ class Bing(AsyncGeneratorProvider):
else: else:
prompt = messages[-1]["content"] prompt = messages[-1]["content"]
context = create_context(messages[:-1]) context = create_context(messages[:-1])
if cookies is None:
if cookies:
#TODO: Will implement proper cookie retrieval later and use a try-except mechanism in 'stream_generate' instead of defaulting the cookie value like this #TODO: Will implement proper cookie retrieval later and use a try-except mechanism in 'stream_generate' instead of defaulting the cookie value like this
#cookies = get_cookies(".bing.com") #cookies_dict = get_cookies(".bing.com")
cookies = cookies_dict = {
{
'MUID': '', 'MUID': '',
'BCP': '', 'BCP': '',
'MUIDB': '', 'MUIDB': '',
@ -55,8 +56,7 @@ class Bing(AsyncGeneratorProvider):
'SRCHHPGUSR': '', 'SRCHHPGUSR': '',
'ipv6': '', 'ipv6': '',
} }
return stream_generate(prompt, context, cookies_dict)
return stream_generate(prompt, context, cookies)
def create_context(messages: list[dict[str, str]]): def create_context(messages: list[dict[str, str]]):
context = "" context = ""

Loading…
Cancel
Save