diff --git a/api/ragflow_server.py b/api/ragflow_server.py index 9bc217f..e8598a7 100644 --- a/api/ragflow_server.py +++ b/api/ragflow_server.py @@ -14,6 +14,7 @@ from werkzeug.serving import run_simple from api import settings, utils from api.apps import app from api.db.db_models import init_database_tables as init_web_db +from api.db.init_data import init_llm_factory from api.db.runtime_config import RuntimeConfig from api.db.services.document_service import DocumentService from api.root_path import get_root_folder @@ -70,6 +71,8 @@ if __name__ == "__main__": # init db init_web_db() + # init llm_factory + init_llm_factory() # init runtime config import argparse diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml index 795b4a7..35ea81a 100644 --- a/docker/docker-compose.yml +++ b/docker/docker-compose.yml @@ -17,6 +17,7 @@ services: - ./nginx/ragflow.conf:/etc/nginx/conf.d/ragflow.conf - ./nginx/proxy.conf:/etc/nginx/proxy.conf - ./nginx/nginx.conf:/etc/nginx/nginx.conf + - ../api/ragflow_server.py:/ragflow/api/ragflow_server.py env_file: .env environment: - TZ=${TIMEZONE} @@ -58,7 +59,7 @@ services: - ./ragflow-plus-logs:/app/logs - ./magic-pdf.json:/root/magic-pdf.json - ../management/server/services/knowledgebases/document_parser.py:/app/services/knowledgebases/document_parser.py - depends_on: + depends_on: mysql: condition: service_healthy es01: