.env.example

# UltimateRAG 环境变量配置模板
# 复制此文件为 .env 并填入实际值

# ===============================
# API 配置
# ===============================

# OpenAI API Key（必填）
# 支持 OpenAI 官方 API 或兼容接口（如 DeepSeek）
OPENAI_API_KEY=your_api_key_here

# API Base URL（可选）
# 如果使用 DeepSeek 或其他兼容 API，请设置此项
# DeepSeek: https://api.deepseek.com/v1
# OpenAI: https://api.openai.com/v1
OPENAI_BASE_URL=https://api.openai.com/v1

# ===============================
# 模型配置
# ===============================

# LLM 模型名称
# OpenAI: gpt-4o, gpt-4-turbo, gpt-3.5-turbo
# DeepSeek: deepseek-chat, deepseek-coder
MODEL_NAME=gpt-4o

# Embedding 模型名称
# OpenAI: text-embedding-3-small, text-embedding-3-large
# DeepSeek: 使用 OpenAI 兼容的模型名称
EMBEDDING_MODEL=text-embedding-3-small

# ===============================
# 分块配置
# ===============================

# 每个块的最大字符数
CHUNK_SIZE=512

# 块之间的重叠字符数
CHUNK_OVERLAP=50

# ===============================
# 检索配置
# ===============================

# 检索返回的文档数量
TOP_K=3

# ===============================
# 存储配置
# ===============================

# ChromaDB 持久化目录
CHROMA_PERSIST_DIR=./data/chroma_db

# GraphRAG 配置
GRAPH_STORE_TYPE=memory  # 或 neo4j
NEO4J_URI=bolt://localhost:7687
NEO4J_USERNAME=neo4j
NEO4J_PASSWORD=your_password

# 微调配置
EMBEDDING_FINETUNE_MODEL=BAAI/bge-base-zh-v1.5