rag_server/vector_app.py

from quart import Quart

from applications.config import LOCAL_MODEL_CONFIG, DEFAULT_MODEL
from applications.config import ES_HOSTS, ES_PASSWORD, ELASTIC_SEARCH_INDEX
from applications.config import MILVUS_CONFIG
from applications.resource import init_resource_manager

app = Quart(__name__)

# 初始化
MODEL_PATH = LOCAL_MODEL_CONFIG[DEFAULT_MODEL]

resource_manager = init_resource_manager(
    es_hosts=ES_HOSTS,
    es_index=ELASTIC_SEARCH_INDEX,
    es_password=ES_PASSWORD,
    milvus_config=MILVUS_CONFIG,
)


@app.before_serving
async def startup():
    await resource_manager.startup()
    print("Resource manager is ready.")
    print("Jieba dictionary loaded successfully")


@app.after_serving
async def shutdown():
    await resource_manager.shutdown()
    print("Resource manager is Down.")


# 注册路由
from routes import server_bp

app.config["MAX_CONTENT_LENGTH"] = 100 * 1024 * 1024
app.register_blueprint(server_bp)