autorag.vectordb package¶
Submodules¶
autorag.vectordb.base module¶
- class autorag.vectordb.base.BaseVectorStore(embedding_model: str, similarity_metric: str = 'cosine', embedding_batch: int = 100)[source]¶
Bases:
object
- abstract async is_exist(ids: List[str]) List[bool] [source]¶
Check if the ids exist in the Vector DB.
- abstract async query(queries: List[str], top_k: int, **kwargs) Tuple[List[List[str]], List[List[float]]] [source]¶
- support_similarity_metrics = ['l2', 'ip', 'cosine']¶
autorag.vectordb.chroma module¶
- class autorag.vectordb.chroma.Chroma(embedding_model: str, collection_name: str, embedding_batch: int = 100, client_type: str = 'persistent', similarity_metric: str = 'cosine', path: str | None = None, host: str = 'localhost', port: int = 8000, ssl: bool = False, headers: Dict[str, str] | None = None, api_key: str | None = None, tenant: str = 'default_tenant', database: str = 'default_database')[source]¶
Bases:
BaseVectorStore
autorag.vectordb.milvus module¶
- class autorag.vectordb.milvus.Milvus(embedding_model: str, collection_name: str, embedding_batch: int = 100, similarity_metric: str = 'cosine', uri: str = 'http://localhost:19530', db_name: str = '', token: str = '', user: str = '', password: str = '', timeout: float | None = None)[source]¶
Bases:
BaseVectorStore
Module contents¶
- autorag.vectordb.load_all_vectordb_from_yaml(yaml_path: str, project_dir: str) List[BaseVectorStore] [source]¶