[Feature] Add support for weaviate vector db (#782)

This commit is contained in:
Rupesh Bansal
2023-10-18 10:48:53 +05:30
committed by GitHub
parent e8a2846449
commit cdfd6519c8
9 changed files with 581 additions and 5 deletions

View File

@@ -0,0 +1,16 @@
from typing import Dict, Optional
from embedchain.config.vectordb.base import BaseVectorDbConfig
from embedchain.helper.json_serializable import register_deserializable
@register_deserializable
class WeaviateDBConfig(BaseVectorDbConfig):
def __init__(
self,
collection_name: Optional[str] = None,
dir: Optional[str] = None,
**extra_params: Dict[str, any],
):
self.extra_params = extra_params
super().__init__(collection_name=collection_name, dir=dir)