Merge pull request #372 from ShravanSunder/pinecone-allow-objects

Allows objects to be uploaded to pinecone metadata
Ironclad · Mar 12, 2024 · 82d45e4 · 82d45e4
2 parents 6235566 + 638727a
commit 82d45e4
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/packages/core/src/plugins/pinecone/PineconeVectorDatabase.ts b/packages/core/src/plugins/pinecone/PineconeVectorDatabase.ts
@@ -23,16 +23,22 @@ export class PineconeVectorDatabase implements VectorDatabase {
       id = CryptoJS.SHA256(vector.value.join(',')).toString(CryptoJS.enc.Hex);
     }
 
+    let metadata: Record<string, unknown> = {}
+    if (data.type === 'object') {
+      metadata = data.value;
+    }
+    else {
+      metadata = { data: data.value };
+    }
+
     const response = await fetch(`${collectionDetails.host}/vectors/upsert`, {
       method: 'POST',
       body: JSON.stringify({
         vectors: [
           {
             id,
             values: vector.value,
-            metadata: {
-              data: data.value,
-            },
+            metadata,
           },
         ],
         ...collectionDetails.options,