From 638727a730ce670f239c014a87d250e4aa66761b Mon Sep 17 00:00:00 2001
From: Shravan Sunder <ShravanSunder@users.noreply.github.com>
Date: Tue, 12 Mar 2024 13:54:17 -0400
Subject: [PATCH] Allows objects to be uploaded to pinecone metadata

---
 .../src/plugins/pinecone/PineconeVectorDatabase.ts   | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/packages/core/src/plugins/pinecone/PineconeVectorDatabase.ts b/packages/core/src/plugins/pinecone/PineconeVectorDatabase.ts
index 2b73afc28..cd3d382bf 100644
--- a/packages/core/src/plugins/pinecone/PineconeVectorDatabase.ts
+++ b/packages/core/src/plugins/pinecone/PineconeVectorDatabase.ts
@@ -23,6 +23,14 @@ 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({
@@ -30,9 +38,7 @@ export class PineconeVectorDatabase implements VectorDatabase {
           {
             id,
             values: vector.value,
-            metadata: {
-              data: data.value,
-            },
+            metadata,
           },
         ],
         ...collectionDetails.options,