[Feature] Pinecone Vector DB support (#723)

This commit is contained in:
Rupesh Bansal
2023-10-15 14:24:07 +05:30
committed by GitHub
parent 5ec12212e4
commit a7a61fae1d
6 changed files with 308 additions and 6 deletions

View File

@@ -0,0 +1,18 @@
from typing import Optional
from embedchain.config.vectordb.base import BaseVectorDbConfig
from embedchain.helper.json_serializable import register_deserializable
@register_deserializable
class PineconeDbConfig(BaseVectorDbConfig):
def __init__(
self,
collection_name: Optional[str] = None,
dir: Optional[str] = None,
dimension: Optional[int] = 1536,
metric: Optional[str] = "cosine",
):
self.dimension = dimension
self.metric = metric
super().__init__(collection_name=collection_name, dir=dir)