46 lines
1.7 KiB
Python
46 lines
1.7 KiB
Python
|
import aiohttp
|
||
|
|
import json
|
||
|
|
import asyncio
|
||
|
|
from log import getlogger
|
||
|
|
# api_endpoint = "http://localhost:3000/conversation"
|
||
|
|
logger = getlogger()
|
||
|
|
python_boolean_to_json = {
|
||
|
|
"true": True,
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
class BingBot:
|
||
|
|
def __init__(self, bing_api_endpoint: str):
|
||
|
|
self.data = {
|
||
|
|
'jailbreakConversationId': json.dumps(python_boolean_to_json['true']),
|
||
|
|
'clientOptions.clientToUse': 'bing'
|
||
|
|
}
|
||
|
|
self.bing_api_endpoint = bing_api_endpoint
|
||
|
|
|
||
|
|
async def ask_bing(self, prompt) -> str:
|
||
|
|
self.data['message'] = prompt
|
||
|
|
async with aiohttp.ClientSession() as session:
|
||
|
|
max_try = 5
|
||
|
|
while max_try > 0:
|
||
|
|
try:
|
||
|
|
resp = await session.post(url=self.bing_api_endpoint, json=self.data)
|
||
|
|
status_code = resp.status
|
||
|
|
body = await resp.read()
|
||
|
|
if not status_code == 200:
|
||
|
|
# print failed reason
|
||
|
|
logger.warning(str(resp.reason))
|
||
|
|
max_try = max_try - 1
|
||
|
|
# print(await resp.text())
|
||
|
|
await asyncio.sleep(2)
|
||
|
|
continue
|
||
|
|
json_body = json.loads(body)
|
||
|
|
# print(json_body['jailbreakConversationId'])
|
||
|
|
self.data['jailbreakConversationId'] = json_body['jailbreakConversationId']
|
||
|
|
self.data['parentMessageId'] = json_body['messageId']
|
||
|
|
return json_body['response']
|
||
|
|
except Exception as e:
|
||
|
|
logger.error("Error Exception", exc_info=True)
|
||
|
|
print(f"Error: {e}")
|
||
|
|
pass
|
||
|
|
return "Error, please retry"
|