pinecone.py 570 B

123456789101112131415161718
  1. from typing import Optional
  2. from embedchain.config.vectordb.base import BaseVectorDbConfig
  3. from embedchain.helper.json_serializable import register_deserializable
  4. @register_deserializable
  5. class PineconeDbConfig(BaseVectorDbConfig):
  6. def __init__(
  7. self,
  8. collection_name: Optional[str] = None,
  9. dir: Optional[str] = None,
  10. dimension: Optional[int] = 1536,
  11. metric: Optional[str] = "cosine",
  12. ):
  13. self.dimension = dimension
  14. self.metric = metric
  15. super().__init__(collection_name=collection_name, dir=dir)