mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-26 01:33:05 +00:00
59 lines
1.3 KiB
Python
59 lines
1.3 KiB
Python
# coding=utf-8
|
||
"""
|
||
@project: MaxKB
|
||
@Author：虎虎
|
||
@file： web.py
|
||
@date：2025/11/5 15:14
|
||
@desc:
|
||
"""
|
||
import builtins
|
||
import os
|
||
import sys
|
||
|
||
from django.core.wsgi import get_wsgi_application
|
||
|
||
|
||
class TorchBlocker:
|
||
def __init__(self):
|
||
self.original_import = builtins.__import__
|
||
|
||
def __call__(self, name, *args, **kwargs):
|
||
if len([True for i in
|
||
['torch']
|
||
if
|
||
i in name.lower()]) > 0:
|
||
print(f"Disable package is being imported: 【{name}】", file=sys.stderr)
|
||
pass
|
||
else:
|
||
return self.original_import(name, *args, **kwargs)
|
||
|
||
|
||
# 安装导入拦截器
|
||
builtins.__import__ = TorchBlocker()
|
||
|
||
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'maxkb.settings')
|
||
os.environ['TIKTOKEN_CACHE_DIR'] = '/opt/maxkb-app/model/tokenizer/openai-tiktoken-cl100k-base'
|
||
application = get_wsgi_application()
|
||
|
||
|
||
def post_handler():
|
||
from common.database_model_manage.database_model_manage import DatabaseModelManage
|
||
from common import event
|
||
|
||
event.run()
|
||
DatabaseModelManage.init()
|
||
|
||
|
||
def post_scheduler_handler():
|
||
from common import job
|
||
|
||
job.run()
|
||
|
||
|
||
# 启动后处理函数
|
||
post_handler()
|
||
|
||
# 仅在scheduler中启动定时任务，dev local_model celery 不需要
|
||
if os.environ.get('ENABLE_SCHEDULER') == '1':
|
||
post_scheduler_handler()
|