mirror of
https://github.com/hibobmaster/matrix_chatgpt_bot.git
synced 2024-10-01 05:35:36 -04:00
75 lines
3.4 KiB
Python
75 lines
3.4 KiB
Python
import asyncio
|
|
import json
|
|
import os
|
|
from bot import Bot
|
|
from log import getlogger
|
|
|
|
logger = getlogger()
|
|
|
|
|
|
async def main():
|
|
need_import_keys = False
|
|
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'),
|
|
output_four_images=config.get('output_four_images'),
|
|
import_keys_path=config.get('import_keys_path'),
|
|
import_keys_password=config.get(
|
|
'import_keys_password'),
|
|
)
|
|
if config.get('import_keys_path') and \
|
|
config.get('import_keys_password') is not None:
|
|
need_import_keys = True
|
|
|
|
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'),
|
|
output_four_images=os.environ.get(
|
|
"OUTPUT_FOUR_IMAGES", "false").lower() \
|
|
in ('true', '1', 't'),
|
|
import_keys_path=os.environ.get("IMPORT_KEYS_PATH"),
|
|
import_keys_password=os.environ.get(
|
|
"IMPORT_KEYS_PASSWORD"),
|
|
)
|
|
if os.environ.get("IMPORT_KEYS_PATH") \
|
|
and os.environ.get("IMPORT_KEYS_PASSWORD") is not None:
|
|
need_import_keys = True
|
|
|
|
await matrix_bot.login()
|
|
if need_import_keys:
|
|
logger.info("start import_keys process, this may take a while...")
|
|
await matrix_bot.import_keys()
|
|
await matrix_bot.sync_forever(timeout=30000, full_state=True)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
logger.info("matrix chatgpt bot start.....")
|
|
asyncio.run(main())
|