diff --git a/docker-compose.yml b/docker-compose.yml index 7d05ab1..e5a404c 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -8,6 +8,7 @@ services: SECRET_KEY: ${SECRET_KEY:?请设置SECRET_KEY} ENVIRONMENT: ${ENVIRONMENT:-production} DEBUG: ${DEBUG:-false} + LOGGING_LEVEL: ${LOGGING_LEVEL:-INFO} DB_PROVIDER: ${DB_PROVIDER:-sqlite} @@ -26,11 +27,26 @@ services: OPENAI_MODEL_NAME: ${OPENAI_MODEL_NAME:-gpt-3.5-turbo} WRITER_CHAPTER_VERSION_COUNT: ${WRITER_CHAPTER_VERSION_COUNT:-2} + EMBEDDING_PROVIDER: ${EMBEDDING_PROVIDER:-openai} + EMBEDDING_BASE_URL: ${EMBEDDING_BASE_URL:-https://api.openai.com/v1} + EMBEDDING_API_KEY: ${EMBEDDING_API_KEY:-${OPENAI_API_KEY}} + EMBEDDING_MODEL: ${EMBEDDING_MODEL:-text-embedding-3-large} + EMBEDDING_MODEL_VECTOR_SIZE: ${EMBEDDING_MODEL_VECTOR_SIZE:-3072} + OLLAMA_EMBEDDING_BASE_URL: ${OLLAMA_EMBEDDING_BASE_URL:-http://localhost:11434} + OLLAMA_EMBEDDING_MODEL: ${OLLAMA_EMBEDDING_MODEL:-nomic-embed-text:latest} + + VECTOR_DB_URL: ${VECTOR_DB_URL:-file:./storage/rag_vectors.db} + VECTOR_DB_AUTH_TOKEN: ${VECTOR_DB_AUTH_TOKEN:-} + VECTOR_TOP_K_CHUNKS: ${VECTOR_TOP_K_CHUNKS:-5} + VECTOR_TOP_K_SUMMARIES: ${VECTOR_TOP_K_SUMMARIES:-3} + VECTOR_CHUNK_SIZE: ${VECTOR_CHUNK_SIZE:-480} + VECTOR_CHUNK_OVERLAP: ${VECTOR_CHUNK_OVERLAP:-120} + SMTP_SERVER: ${SMTP_SERVER:-smtp.example.com} SMTP_PORT: ${SMTP_PORT:-465} SMTP_USERNAME: ${SMTP_USERNAME:-no-reply@example.com} SMTP_PASSWORD: ${SMTP_PASSWORD} - EMAIL_FROM: ${EMAIL_FROM:-小说生成器} + EMAIL_FROM: ${EMAIL_FROM:-Arboris} ALLOW_USER_REGISTRATION: ${ALLOW_USER_REGISTRATION:-true} ENABLE_LINUXDO_LOGIN: ${ENABLE_LINUXDO_LOGIN:-false} @@ -67,7 +83,6 @@ services: TZ: Asia/Shanghai volumes: - mysql-data:/var/lib/mysql - - ./backend/db/init.sql:/docker-entrypoint-initdb.d/init.sql:ro restart: unless-stopped networks: - app-network