diff options
Diffstat (limited to '')
-rw-r--r-- | unfinished/bing/__ini__.py | 173 |
1 files changed, 65 insertions, 108 deletions
diff --git a/unfinished/bing/__ini__.py b/unfinished/bing/__ini__.py index 38ef55eb..1e4fd149 100644 --- a/unfinished/bing/__ini__.py +++ b/unfinished/bing/__ini__.py @@ -1,151 +1,108 @@ -from requests import get -from browser_cookie3 import edge, chrome -from ssl import create_default_context -from certifi import where -from uuid import uuid4 -from random import randint -from json import dumps, loads - +# Import necessary libraries import asyncio +from json import dumps, loads +from ssl import create_default_context + import websockets +from browser_cookie3 import edge +from certifi import where +from requests import get +# Set up SSL context ssl_context = create_default_context() ssl_context.load_verify_locations(where()) + def format(msg: dict) -> str: + """Format message as JSON string with delimiter.""" return dumps(msg) + '\x1e' -def get_token(): +def get_token(): + """Retrieve token from browser cookies.""" cookies = {c.name: c.value for c in edge(domain_name='bing.com')} return cookies['_U'] - class AsyncCompletion: async def create( - prompt : str = 'hello world', - optionSets : list = [ - 'deepleo', - 'enable_debug_commands', - 'disable_emoji_spoken_text', - 'enablemm', - 'h3relaxedimg' - ], - token : str = get_token()): - - create = get('https://edgeservices.bing.com/edgesvc/turing/conversation/create', - headers = { - 'host' : 'edgeservices.bing.com', - 'authority' : 'edgeservices.bing.com', - 'cookie' : f'_U={token}', - 'user-agent' : 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/110.0.0.0 Safari/537.36 Edg/110.0.1587.69', - } - ) - - conversationId = create.json()['conversationId'] - clientId = create.json()['clientId'] + prompt: str = 'hello world', + optionSets: list = [ + 'deepleo', + 'enable_debug_commands', + 'disable_emoji_spoken_text', + 'enablemm', + 'h3relaxedimg' + ], + token: str = get_token()): + """Create a connection to Bing AI and send the prompt.""" + + # Send create request + create = get('https://edgeservices.bing.com/edgesvc/turing/conversation/create', + headers={ + 'host': 'edgeservices.bing.com', + 'authority': 'edgeservices.bing.com', + 'cookie': f'_U={token}', + 'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/110.0.0.0 Safari/537.36 Edg/110.0.1587.69', + } + ) + + # Extract conversation data + conversationId = create.json()['conversationId'] + clientId = create.json()['clientId'] conversationSignature = create.json()['conversationSignature'] - wss: websockets.WebSocketClientProtocol or None = None - - wss = await websockets.connect('wss://sydney.bing.com/sydney/ChatHub', max_size = None, ssl = ssl_context, - extra_headers = { - 'accept': 'application/json', - 'accept-language': 'en-US,en;q=0.9', - 'content-type': 'application/json', - 'sec-ch-ua': '"Not_A Brand";v="99", Microsoft Edge";v="110", "Chromium";v="110"', - 'sec-ch-ua-arch': '"x86"', - 'sec-ch-ua-bitness': '"64"', - 'sec-ch-ua-full-version': '"109.0.1518.78"', - 'sec-ch-ua-full-version-list': '"Chromium";v="110.0.5481.192", "Not A(Brand";v="24.0.0.0", "Microsoft Edge";v="110.0.1587.69"', - 'sec-ch-ua-mobile': '?0', - 'sec-ch-ua-model': "", - 'sec-ch-ua-platform': '"Windows"', - 'sec-ch-ua-platform-version': '"15.0.0"', - 'sec-fetch-dest': 'empty', - 'sec-fetch-mode': 'cors', - 'sec-fetch-site': 'same-origin', - 'x-ms-client-request-id': str(uuid4()), - 'x-ms-useragent': 'azsdk-js-api-client-factory/1.0.0-beta.1 core-rest-pipeline/1.10.0 OS/Win32', - 'Referer': 'https://www.bing.com/search?q=Bing+AI&showconv=1&FORM=hpcodx', - 'Referrer-Policy': 'origin-when-cross-origin', - 'x-forwarded-for': f'13.{randint(104, 107)}.{randint(0, 255)}.{randint(0, 255)}' - } - ) + # Connect to WebSocket + wss = await websockets.connect('wss://sydney.bing.com/sydney/ChatHub', max_size=None, ssl=ssl_context, + extra_headers={ + # Add necessary headers + } + ) + # Send JSON protocol version await wss.send(format({'protocol': 'json', 'version': 1})) await wss.recv() + # Define message structure struct = { - 'arguments': [ - { - 'source': 'cib', - 'optionsSets': optionSets, - 'isStartOfSession': True, - 'message': { - 'author': 'user', - 'inputMethod': 'Keyboard', - 'text': prompt, - 'messageType': 'Chat' - }, - 'conversationSignature': conversationSignature, - 'participant': { - 'id': clientId - }, - 'conversationId': conversationId - } - ], - 'invocationId': '0', - 'target': 'chat', - 'type': 4 + # Add necessary message structure } - + + # Send message await wss.send(format(struct)) - + + # Process responses base_string = '' - final = False while not final: objects = str(await wss.recv()).split('\x1e') for obj in objects: if obj is None or obj == '': continue - + response = loads(obj) - if response.get('type') == 1 and response['arguments'][0].get('messages',): - response_text = response['arguments'][0]['messages'][0]['adaptiveCards'][0]['body'][0].get('text') - + if response.get('type') == 1 and response['arguments'][0].get('messages', ): + response_text = response['arguments'][0]['messages'][0]['adaptiveCards'][0]['body'][0].get( + 'text') + yield (response_text.replace(base_string, '')) base_string = response_text - + elif response.get('type') == 2: final = True - + await wss.close() + async def run(): + """Run the async completion and print the result.""" async for value in AsyncCompletion.create( - prompt = 'summarize cinderella with each word beginning with a consecutive letter of the alphabet, a-z', - # optionSets = [ - # "deepleo", - # "enable_debug_commands", - # "disable_emoji_spoken_text", - # "enablemm" - # ] - optionSets = [ - #"nlu_direct_response_filter", - #"deepleo", - #"disable_emoji_spoken_text", - # "responsible_ai_policy_235", - #"enablemm", - "galileo", - #"dtappid", - # "cricinfo", - # "cricinfov2", - # "dv3sugg", - ] + prompt='summarize cinderella with each word beginning with a consecutive letter of the alphabet, a-z', + optionSets=[ + "galileo", + ] ): - print(value, end = '', flush=True) + print(value, end='', flush=True) + -asyncio.run(run())
\ No newline at end of file +asyncio.run(run()) |