UPDATE
This commit is contained in:
@@ -15,10 +15,15 @@ class EmbeddingConfig:
|
||||
provider: str
|
||||
model: str
|
||||
batch_size: int = 32
|
||||
api_key: str | None = None
|
||||
api_key_env: str | None = None
|
||||
base_url: str | None = None
|
||||
avalai_api_key: str | None = None
|
||||
avalai_base_url: str | None = None
|
||||
avalai_api_key_env: str | None = None
|
||||
arvancloud_api_key: str | None = None
|
||||
arvancloud_base_url: str | None = None
|
||||
arvancloud_api_key_env: str | None = None
|
||||
|
||||
|
||||
@dataclass
|
||||
@@ -116,10 +121,15 @@ def load_rag_config(config_path: str | Path | None = None) -> RAGConfig:
|
||||
provider=emb.get("provider", "sentence_transformers"),
|
||||
model=emb.get("model", "text-embedding-3-small"),
|
||||
batch_size=emb.get("batch_size", 32),
|
||||
api_key=emb.get("api_key"),
|
||||
api_key_env=emb.get("api_key_env"),
|
||||
base_url=emb.get("base_url"),
|
||||
avalai_api_key=emb.get("avalai_api_key"),
|
||||
avalai_base_url=emb.get("avalai_base_url"),
|
||||
avalai_api_key_env=emb.get("avalai_api_key_env"),
|
||||
arvancloud_api_key=emb.get("arvancloud_api_key"),
|
||||
arvancloud_base_url=emb.get("arvancloud_base_url"),
|
||||
arvancloud_api_key_env=emb.get("arvancloud_api_key_env"),
|
||||
)
|
||||
|
||||
qd = data.get("qdrant", {})
|
||||
|
||||
Reference in New Issue
Block a user