changing pinecone dims
Browse files- src/pinecone_store.py +4 -4
src/pinecone_store.py
CHANGED
|
@@ -8,15 +8,15 @@ class PineconeStore:
|
|
| 8 |
def __init__(self):
|
| 9 |
self.pc = Pinecone(api_key=os.getenv("PINECONE_API_KEY"))
|
| 10 |
|
| 11 |
-
if 'blip-soccer-highlights' not in self.pc.list_indexes().names():
|
| 12 |
self.pc.create_index(
|
| 13 |
-
name='blip-soccer-highlights',
|
| 14 |
-
dimension=
|
| 15 |
metric='cosine',
|
| 16 |
spec=ServerlessSpec(cloud='aws', region=os.getenv("PINECONE_ENV", "us-east-1"))
|
| 17 |
)
|
| 18 |
|
| 19 |
-
self.index = self.pc.Index('blip-soccer-highlights')
|
| 20 |
|
| 21 |
def upsert(self, id, vector, metadata):
|
| 22 |
self.index.upsert([(id, vector.tolist(), metadata)])
|
|
|
|
| 8 |
def __init__(self):
|
| 9 |
self.pc = Pinecone(api_key=os.getenv("PINECONE_API_KEY"))
|
| 10 |
|
| 11 |
+
if 'blip-soccer-highlights-index' not in self.pc.list_indexes().names():
|
| 12 |
self.pc.create_index(
|
| 13 |
+
name='blip-soccer-highlights-index',
|
| 14 |
+
dimension=768, # This must match InternVL2_5-8B-MPO
|
| 15 |
metric='cosine',
|
| 16 |
spec=ServerlessSpec(cloud='aws', region=os.getenv("PINECONE_ENV", "us-east-1"))
|
| 17 |
)
|
| 18 |
|
| 19 |
+
self.index = self.pc.Index('blip-soccer-highlights-index')
|
| 20 |
|
| 21 |
def upsert(self, id, vector, metadata):
|
| 22 |
self.index.upsert([(id, vector.tolist(), metadata)])
|