feat: 调整结构

This commit is contained in:
李如威 2025-08-07 15:56:14 +08:00
parent 148474de9b
commit 5b101f26c6
1 changed files with 4 additions and 3 deletions

View File

@ -173,13 +173,14 @@ class ModelManager:
class BaseRAG(ABC):
def __init__(
self,
vector_store_name: str = "default",
embedding_config: Optional[Dict] = None,
retriever_top_k: int = 3,
llm: Optional[BaseLLM] = None,
persist_directory: str = "./chroma_db",
llm: Optional[BaseLLM] = None,
embedding_config: Optional[Dict] = None,
rerank_config: Optional[Dict] = None,
):
"""