11 lines
445 B
Python
11 lines
445 B
Python
from pydantic_settings import BaseSettings
|
|
|
|
class Settings(BaseSettings):
|
|
EMBEDDING_DIM: int = 768 # 嵌入维度（取决于模型）
|
|
TOP_K: int = 5 # 默认检索 top K 个段落
|
|
DOC_PATH: str = "docs/" # 默认文档根目录
|
|
DEVICE: str = "cuda" # 可改为 "cuda" 使用 GPU
|
|
MODEL_NAME: str = "BAAI/bge-m3" # 多语种语义嵌入模型
|
|
|
|
settings = Settings()
|