hibobmaster
600c5e161a
Bump requests from 2.28.2 to 2.31.0 Bumps [requests](https://github.com/psf/requests) from 2.28.2 to 2.31.0. - [Release notes](https://github.com/psf/requests/releases) - [Changelog](https://github.com/psf/requests/blob/main/HISTORY.md) - [Commits](https://github.com/psf/requests/compare/v2.28.2...v2.31.0) --- updated-dependencies: - dependency-name: requests dependency-type: direct:production ... Signed-off-by: dependabot[bot] <support@github.com>
78 lines
3.7 KiB
Python
78 lines
3.7 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'),
|
|
flowise_api_url=config.get('flowise_api_url'),
|
|
flowise_api_key=config.get('flowise_api_key'),
|
|
)
|
|
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"),
|
|
flowise_api_url=os.environ.get("FLOWISE_API_URL"),
|
|
flowise_api_key=os.environ.get("FLOWISE_API_KEY"),
|
|
)
|
|
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())
|