changes for breaking api changes to embed jobs

Author: mahjongmen

notebooks/Embed_Jobs_Semantic_Search.ipynb

@@ -145,7 +145,8 @@
         "job = co.create_embed_job(\n",
         "    dataset_id=ds.id,\n",
         "    input_type='search_document' ,\n",
-        "    model='embed-english-v3.0')\n",
+        "    model='embed-english-v3.0', \n",
+        "    embeddings_types=['float'])\n",
         "\n",
         "job.wait() # poll the server until the job is completed "
       ]
@@ -228,7 +229,7 @@
         "embeddings=[]\n",
         "texts=[]\n",
         "for record in output_dataset:\n",
-        "  embeddings.append(record['embedding'])\n",
+        "  embeddings.append(record['embeddings']['float'])\n",
         "  texts.append(record['text'])"
       ]
     },

notebooks/Embed_Jobs_Serverless_Pinecone_Semantic_Search.ipynb

@@ -146,7 +146,8 @@
         "# Dataset has been uploaded, create an embed job and specify the input type as \"search document\" since this will live in your Pinecone DB\n",
         "job = co.create_embed_job(dataset_id=ds.id,\n",
         "                          input_type='search_document',\n",
-        "                          model='embed-english-v3.0')\n",
+        "                          model='embed-english-v3.0',\n",
+        "                          embeddings_types=['float'])\n",
         "\n",
         "job.wait() # poll the server until the job is completed "
       ]
@@ -220,7 +221,7 @@
         "# Take the output and format it in the shape for upserting into Pinecone's DB\n",
         "ids = [str(i) for i in range(len(data_array))]\n",
         "meta = [{'text':str(data_array[i]['text'])} for i in range(len(data_array))]\n",
-        "embeds=[np.float32(data_array[i]['embedding']) for i in range(len(data_array))]\n",
+        "embeds=[np.float32(data_array[i]['embeddings']['float']) for i in range(len(data_array))]\n",
         "\n",
         "to_upsert = list(zip(ids, embeds, meta))"
       ]