58 lines
2.3 KiB
Python
58 lines
2.3 KiB
Python
import asyncio
|
|
import json
|
|
import os
|
|
import signal
|
|
from functools import partial
|
|
from bot import Bot
|
|
from log import getlogger
|
|
|
|
logger = getlogger()
|
|
|
|
|
|
async def main():
|
|
|
|
if os.path.exists('config.json'):
|
|
fp = open('config.json', 'r', encoding="utf8")
|
|
config = json.load(fp)
|
|
|
|
matrix_bot = Bot(homeserver=config.get('homeserver'),
|
|
user_id=config.get('user_id'),
|
|
password=config.get('password'),
|
|
device_id=config.get('device_id'),
|
|
room_id=config.get('room_id'),
|
|
api_key=config.get('api_key'),
|
|
bing_api_endpoint=config.get('bing_api_endpoint'),
|
|
access_token=config.get('access_token'),
|
|
bard_token=config.get('bard_token'),
|
|
jailbreakEnabled=config.get('jailbreakEnabled'),
|
|
bing_auth_cookie=config.get('bing_auth_cookie'),
|
|
markdown_formatted=config.get('markdown_formatted'),
|
|
)
|
|
|
|
else:
|
|
matrix_bot = Bot(homeserver=os.environ.get('HOMESERVER'),
|
|
user_id=os.environ.get('USER_ID'),
|
|
password=os.environ.get('PASSWORD'),
|
|
device_id=os.environ.get("DEVICE_ID"),
|
|
room_id=os.environ.get("ROOM_ID"),
|
|
api_key=os.environ.get("OPENAI_API_KEY"),
|
|
bing_api_endpoint=os.environ.get("BING_API_ENDPOINT"),
|
|
access_token=os.environ.get("ACCESS_TOKEN"),
|
|
bard_token=os.environ.get("BARD_TOKEN"),
|
|
jailbreakEnabled=os.environ.get(
|
|
"JAILBREAKENABLED", "false").lower() in ('true', '1', 't'),
|
|
bing_auth_cookie=os.environ.get("BING_AUTH_COOKIE"),
|
|
markdown_formatted=os.environ.get(
|
|
"MARKDOWN_FORMATTED", "false").lower() in ('true', '1', 't'),
|
|
)
|
|
|
|
await matrix_bot.login()
|
|
|
|
await matrix_bot.sync_forever(timeout=30000, full_state=True)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
logger.info("matrix chatgpt bot start.....")
|
|
print("matrix chatgpt bot start.....")
|
|
asyncio.run(main())
|
|
|