mirror of
https://github.com/langbot-app/LangBot.git
synced 2025-11-25 03:15:06 +08:00
74c3a77ed1
* debug: print detailed make connection failure error * perf: active heartbeat to plugin runtime * feat: add `--debug` arg
118 lines
3.5 KiB
Python
118 lines
3.5 KiB
Python
import asyncio
|
||
import argparse
|
||
# LangBot 终端启动入口
|
||
# 在此层级解决依赖项检查。
|
||
# LangBot/main.py
|
||
|
||
asciiart = r"""
|
||
_ ___ _
|
||
| | __ _ _ _ __ _| _ ) ___| |_
|
||
| |__/ _` | ' \/ _` | _ \/ _ \ _|
|
||
|____\__,_|_||_\__, |___/\___/\__|
|
||
|___/
|
||
|
||
⭐️ Open Source 开源地址: https://github.com/langbot-app/LangBot
|
||
📖 Documentation 文档地址: https://docs.langbot.app
|
||
"""
|
||
|
||
|
||
async def main_entry(loop: asyncio.AbstractEventLoop):
|
||
parser = argparse.ArgumentParser(description='LangBot')
|
||
parser.add_argument(
|
||
'--standalone-runtime',
|
||
action='store_true',
|
||
help='Use standalone plugin runtime / 使用独立插件运行时',
|
||
default=False,
|
||
)
|
||
parser.add_argument('--debug', action='store_true', help='Debug mode / 调试模式', default=False)
|
||
args = parser.parse_args()
|
||
|
||
if args.standalone_runtime:
|
||
from pkg.utils import platform
|
||
|
||
platform.standalone_runtime = True
|
||
|
||
if args.debug:
|
||
from pkg.utils import constants
|
||
|
||
constants.debug_mode = True
|
||
|
||
print(asciiart)
|
||
|
||
import sys
|
||
|
||
# 检查依赖
|
||
|
||
from pkg.core.bootutils import deps
|
||
|
||
missing_deps = await deps.check_deps()
|
||
|
||
if missing_deps:
|
||
print('以下依赖包未安装，将自动安装，请完成后重启程序：')
|
||
print(
|
||
'These dependencies are missing, they will be installed automatically, please restart the program after completion:'
|
||
)
|
||
for dep in missing_deps:
|
||
print('-', dep)
|
||
await deps.install_deps(missing_deps)
|
||
print('已自动安装缺失的依赖包，请重启程序。')
|
||
print('The missing dependencies have been installed automatically, please restart the program.')
|
||
sys.exit(0)
|
||
|
||
# # 检查pydantic版本，如果没有 pydantic.v1，则把 pydantic 映射为 v1
|
||
# import pydantic.version
|
||
|
||
# if pydantic.version.VERSION < '2.0':
|
||
# import pydantic
|
||
|
||
# sys.modules['pydantic.v1'] = pydantic
|
||
|
||
# 检查配置文件
|
||
|
||
from pkg.core.bootutils import files
|
||
|
||
generated_files = await files.generate_files()
|
||
|
||
if generated_files:
|
||
print('以下文件不存在，已自动生成：')
|
||
print('Following files do not exist and have been automatically generated:')
|
||
for file in generated_files:
|
||
print('-', file)
|
||
|
||
from pkg.core import boot
|
||
|
||
await boot.main(loop)
|
||
|
||
|
||
if __name__ == '__main__':
|
||
import os
|
||
import sys
|
||
|
||
# 必须大于 3.10.1
|
||
if sys.version_info < (3, 10, 1):
|
||
print('需要 Python 3.10.1 及以上版本，当前 Python 版本为：', sys.version)
|
||
input('按任意键退出...')
|
||
print('Your Python version is not supported. Please exit the program by pressing any key.')
|
||
exit(1)
|
||
|
||
# Check if the current directory is the LangBot project root directory
|
||
invalid_pwd = False
|
||
|
||
if not os.path.exists('main.py'):
|
||
invalid_pwd = True
|
||
else:
|
||
with open('main.py', 'r', encoding='utf-8') as f:
|
||
content = f.read()
|
||
if 'LangBot/main.py' not in content:
|
||
invalid_pwd = True
|
||
if invalid_pwd:
|
||
print('请在 LangBot 项目根目录下以命令形式运行此程序。')
|
||
input('按任意键退出...')
|
||
print('Please run this program in the LangBot project root directory in command form.')
|
||
print('Press any key to exit...')
|
||
exit(1)
|
||
|
||
loop = asyncio.new_event_loop()
|
||
|
||
loop.run_until_complete(main_entry(loop))
|