Merge pull request #46 from scottleibrand/patch-1

Only get_embedding once

examples/Code_search.ipynb

@@ -84,13 +84,6 @@
     "print(\"Total number of functions extracted:\", len(all_funcs))"
    ]
   },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "For code search models we use code-search-{model}-code to obtain embeddings for code snippets, and code-search-{model}-text to embed natural language queries."
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": 2,

examples/Obtain_dataset.ipynb

@@ -162,7 +162,7 @@
     "\n",
     "# This will take just between 5 and 10 minutes\n",
     "df['ada_similarity'] = df.combined.apply(lambda x: get_embedding(x, engine='text-embedding-ada-002'))\n",
-    "df['ada_search'] = df.combined.apply(lambda x: get_embedding(x, engine='text-embedding-ada-002'))\n",
+    "df['ada_search'] = df['ada_similarity']\n",
     "df.to_csv('data/fine_food_reviews_with_embeddings_1k.csv')"
    ]
   }