matrix_chatgpt_bot/main.py

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())