a

pinecone_func.py

@@ -26,14 +26,24 @@ def upsert_texts(texts, ids):
         )
         embeddings.append(response.data[0].embedding)
 
-    vectors = [(id, emb) for id, emb in zip(ids, embeddings)]
+    pc = Pinecone(api_key=os.environ["PINECONE_API_KEY"])
+
+    index = pc.Index("your-index")
+
+    vectors = []
+    for id, emb, text in zip(ids, embeddings, texts):
+        vectors.append((id, emb, {"text": text}))
+
     index.upsert(vectors)
 
 def query_text(query, top_k=5):
-    response = client.embeddings.create(
+    pc = Pinecone(api_key=os.environ["PINECONE_API_KEY"])
+    index = pc.Index('your-index')
+
+    embedding = client.embeddings.create(
         model="text-embedding-3-large",
         input=query
-    )
-    embedding = response.data[0].embedding
+    ).data[0].embedding
+
     results = index.query(vector=embedding, top_k=top_k, include_metadata=True)
     return results