Refactoring vectordb naming convention in embedchain.config (#1469)

This commit is contained in:
Vatsal Rathod
2024-07-08 19:01:17 -04:00
committed by GitHub
parent 1a5d0d236a
commit 83e8c97295
20 changed files with 124 additions and 124 deletions

View File

@@ -0,0 +1,18 @@
from typing import Optional
from embedchain.config.vector_db.base import BaseVectorDbConfig
from embedchain.helpers.json_serializable import register_deserializable
@register_deserializable
class WeaviateDBConfig(BaseVectorDbConfig):
def __init__(
self,
collection_name: Optional[str] = None,
dir: Optional[str] = None,
batch_size: Optional[int] = 100,
**extra_params: dict[str, any],
):
self.batch_size = batch_size
self.extra_params = extra_params
super().__init__(collection_name=collection_name, dir=dir)