bobmaster/matrix_chatgpt_bot
1
0
Fork 0
Code Issues 5 Pull requests 2 Projects Releases 10 Packages Wiki Activity

Add bing ai

Browse source
This commit is contained in:
hibobmaster 2023-03-10 23:45:38 +08:00
parent 69dd8ff81c
commit 942348e9d0
Signed by: bobmaster
GPG key ID: 316B77D7914D713C
7 changed files with 123 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
README.md
View file

@ -46,4 +46,8 @@ To interact with the bot, simply send a message to the bot in the Matrix room wi
- `!chat` To chat using official api with context associated support
```
!chat Can you tell me a joke?
```
```
## Bing AI
https://github.com/waylaidwanderer/node-chatgpt-api
![](https://i.imgur.com/KuYddd5.jpg)

45
bing.py Normal file
View file

@ -0,0 +1,45 @@
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"

50
bot.py
View file

@ -9,10 +9,11 @@ from ask_gpt import ask
from send_message import send_room_message
from v3 import Chatbot
from log import getlogger
from bing import BingBot
"""
free api_endpoint from https://github.com/ayaka14732/ChatGPTAPIFree
"""
api_endpoint_list = {
chatgpt_api_endpoint_list = {
"free": "https://chatgpt-api.shn.hk/v1/",
"paid": "https://api.openai.com/v1/chat/completions"
}
@ -21,13 +22,14 @@ logger = getlogger()
class Bot:
def __init__(
self,
homeserver: str,
user_id: str,
password: str,
device_id: str,
api_key: Optional[str] = "",
room_id: Optional[str] = '',
self,
homeserver: str,
user_id: str,
password: str,
device_id: str,
api_key: Optional[str] = "",
room_id: Optional[str] = '',
bing_api_endpoint: Optional[str] = '',
):
self.homeserver = homeserver
self.user_id = user_id
@ -35,6 +37,7 @@ class Bot:
self.device_id = device_id
self.room_id = room_id
self.api_key = api_key
self.bing_api_endpoint = bing_api_endpoint
# initialize AsyncClient object
self.store_path = os.getcwd()
self.config = AsyncClientConfig(store=SqliteStore,
@ -46,21 +49,27 @@ class Bot:
# regular expression to match keyword [!gpt {prompt}] [!chat {prompt}]
self.gpt_prog = re.compile(r"^\s*!gpt\s*(.+)$")
self.chat_prog = re.compile(r"^\s*!chat\s*(.+)$")
# initialize chatbot and api_endpoint
self.bing_prog = re.compile(r"^\s*!bing\s*(.+)$")
# initialize chatbot and chatgpt_api_endpoint
if self.api_key != '':
self.chatbot = Chatbot(api_key=self.api_key)
self.api_endpoint = api_endpoint_list['paid']
self.chatgpt_api_endpoint = chatgpt_api_endpoint_list['paid']
# request header for !gpt command
self.headers = {
"Content-Type": "application/json",
"Authorization": "Bearer " + self.api_key,
}
else:
self.chatgpt_api_endpoint = chatgpt_api_endpoint_list['free']
self.headers = {
"Content-Type": "application/json",
}
# initialize bingbot
if self.bing_api_endpoint != '':
self.bingbot = BingBot(bing_api_endpoint)
# message_callback event
async def message_callback(self, room: MatrixRoom, event: RoomMessageText) -> None:
# print info to console
@ -75,6 +84,7 @@ class Bot:
room_id = room.room_id
else:
room_id = self.room_id
# chatgpt
n = self.chat_prog.match(event.body)
if n:
@ -101,8 +111,8 @@ class Bot:
await self.client.room_typing(room_id)
prompt = m.group(1)
try:
# 默认等待30s
text = await asyncio.wait_for(ask(prompt, self.api_endpoint, self.headers), timeout=30)
# timeout 30s
text = await asyncio.wait_for(ask(prompt, self.chatgpt_api_endpoint, self.headers), timeout=30)
except TimeoutError:
logger.error("timeoutException", exc_info=True)
text = "Timeout error"
@ -110,6 +120,22 @@ class Bot:
text = text.strip()
await send_room_message(self.client, room_id, send_text=text)
# bing ai
if self.bing_api_endpoint != '':
b = self.bing_prog.match(event.body)
if b:
# sending typing state
await self.client.room_typing(room_id)
prompt = b.group(1)
try:
# timeout 30s
text = await asyncio.wait_for(self.bingbot.ask_bing(prompt), timeout=30)
except TimeoutError:
logger.error("timeoutException", exc_info=True)
text = "Timeout error"
text = text.strip()
await send_room_message(self.client, room_id, send_text=text)
# bot login
async def login(self) -> None:
resp = await self.client.login(password=self.password)

15
compose.yaml
View file

@ -7,5 +7,18 @@ services:
# context: .
# dockerfile: ./Dockerfile
volumes:
- ./config.json:/app/config.json
- ./app:/app
networks:
- matrix_network
# api:
# # bing api
# image: hibobmaster/node-chatgpt-api:latest
# container_name: node-chatgpt-api
# restart: always
# volumes:
# - ./settings.js:/var/chatgpt-api/settings.js
# networks:
# - matrix_network
networks:
matrix_network:

3
config.json.sample
View file

@ -4,5 +4,6 @@
"password": "xxxxxxxxxxxxxxxxxx",
"device_id": "ECYEOKVPLG",
"room_id": "!FYCmBSkCRUNvZDBaDQ:matrix.qqs.tw",
"api_key": "xxxxxxxxxxxxxxxxxxxxxxxx"
"api_key": "xxxxxxxxxxxxxxxxxxxxxxxx",
"bing_api_endpoint": ""
}

1
main.py
View file

@ -13,6 +13,7 @@ async def main():
device_id=config['device_id'],
room_id=config.get('room_id', ''), # provide a default value when the key does not exist
api_key=config.get('api_key', ''),
bing_api_endpoint=config.get('bing_api_endpoint', ''),
)
await matrix_bot.login()
await matrix_bot.sync_forever()

18
test.py
View file

@ -2,9 +2,12 @@ from v3 import Chatbot
import asyncio
from ask_gpt import ask
import json
from bing import BingBot
fp = open("config.json", "r")
config = json.load(fp)
api_key = config.get('api_key', '')
bing_api_endpoint = config.get('bing_api_endpoint', '')
api_endpoint_list = {
"free": "https://chatgpt-api.shn.hk/v1/",
"paid": "https://api.openai.com/v1/chat/completions"
@ -34,7 +37,22 @@ async def test_ask_gpt_free(prompt: str):
api_endpoint = api_endpoint_list['free']
print(await ask(prompt, api_endpoint, headers))
async def test_bingbot():
if bing_api_endpoint != '':
bingbot = BingBot(bing_api_endpoint)
prompt1 = "Hello World"
prompt2 = "Do you know Victor Marie Hugo"
prompt3 = "Can you tell me something about him?"
resp1 = await bingbot.ask_bing(prompt1)
resp2 = await bingbot.ask_bing(prompt2)
resp3 = await bingbot.ask_bing(prompt3)
print(resp1)
print(resp2)
print(resp3)
if __name__ == "__main__":
test_v3("Hello World")
asyncio.run(test_ask_gpt_paid("Hello World"))
asyncio.run(test_ask_gpt_free("Hello World"))
asyncio.run(test_bingbot())