weaviate.py 505 B

12345678910111213141516
  1. from typing import Optional
  2. from embedchain.config.vectordb.base import BaseVectorDbConfig
  3. from embedchain.helpers.json_serializable import register_deserializable
  4. @register_deserializable
  5. class WeaviateDBConfig(BaseVectorDbConfig):
  6. def __init__(
  7. self,
  8. collection_name: Optional[str] = None,
  9. dir: Optional[str] = None,
  10. **extra_params: dict[str, any],
  11. ):
  12. self.extra_params = extra_params
  13. super().__init__(collection_name=collection_name, dir=dir)