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)])
|