feat: 调整结构
This commit is contained in:
parent
148474de9b
commit
5b101f26c6
|
@ -173,13 +173,14 @@ class ModelManager:
|
||||||
|
|
||||||
|
|
||||||
class BaseRAG(ABC):
|
class BaseRAG(ABC):
|
||||||
|
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
vector_store_name: str = "default",
|
vector_store_name: str = "default",
|
||||||
embedding_config: Optional[Dict] = None,
|
|
||||||
retriever_top_k: int = 3,
|
retriever_top_k: int = 3,
|
||||||
llm: Optional[BaseLLM] = None,
|
|
||||||
persist_directory: str = "./chroma_db",
|
persist_directory: str = "./chroma_db",
|
||||||
|
llm: Optional[BaseLLM] = None,
|
||||||
|
embedding_config: Optional[Dict] = None,
|
||||||
rerank_config: Optional[Dict] = None,
|
rerank_config: Optional[Dict] = None,
|
||||||
):
|
):
|
||||||
"""
|
"""
|
||||||
|
|
Loading…
Reference in New Issue