klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

enable using custom Pinecone index name (#1172)

Browse Source 
Co-authored-by: Deven Patel <deven298@yahoo.com>
This commit is contained in:
Peter Jausovec
2024-01-25 00:00:10 -08:00
committed by GitHub
parent b7d365119c
commit 446d0975aa
4 changed files with 49 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
embedchain/config/vectordb/pinecone.py
View File

@@ -9,6 +9,7 @@ class PineconeDBConfig(BaseVectorDbConfig):
def __init__(
self,
collection_name: Optional[str] = None,
index_name: Optional[str] = None,
dir: Optional[str] = None,
vector_dimension: int = 1536,
metric: Optional[str] = "cosine",
@@ -17,4 +18,5 @@ class PineconeDBConfig(BaseVectorDbConfig):
self.metric = metric
self.vector_dimension = vector_dimension
self.extra_params = extra_params
self.index_name = index_name or f"{collection_name}-{vector_dimension}".lower().replace("_", "-")
super().__init__(collection_name=collection_name, dir=dir)