oa-embedding.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "5dbe607b-3a6b-47bb-b8f0-d7764fef9b62",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "import os\n",
    "import openai\n",
    "\n",
    "openai.api_key = os.getenv(\"OPENAI_API_KEY\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "77884f56-164a-4b13-b944-afd745d88c3c",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "prompt = '''Only answer the question below if you have 100% certainty of the facts.\n",
    "Context: {context}\n",
    "Q: What does the start-up company Pentera do and who invested in it?\n",
    "A:'''"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "28b606d9-fca7-4e52-940e-d3b8108ebbde",
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "bde438f7-03c2-45b3-a18d-d83467e21377",
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "3546d8a2-6537-4d87-b607-f4fa4eb50273",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "import pandas as pd"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "a0631b7a-e384-416c-8e01-bcf1b1317434",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "df = pd.read_csv(\"unicorns.csv\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "9eb91f5e-afa6-4fe0-8f84-581713727abe",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Updated at</th>\n",
       "      <th>Company</th>\n",
       "      <th>Crunchbase Url</th>\n",
       "      <th>Last Valuation (Billion $)</th>\n",
       "      <th>Date Joined</th>\n",
       "      <th>Year Joined</th>\n",
       "      <th>City</th>\n",
       "      <th>Country</th>\n",
       "      <th>Industry</th>\n",
       "      <th>Investors</th>\n",
       "      <th>Company Website</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>10/31/2022, 2:37:05 AM</td>\n",
       "      <td>Esusu</td>\n",
       "      <td>https://www.cbinsights.com/company/esusu</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1/27/2022</td>\n",
       "      <td>2022</td>\n",
       "      <td>New York</td>\n",
       "      <td>United States</td>\n",
       "      <td>Fintech</td>\n",
       "      <td>[\"Next Play Ventures\",\"Zeal Capital Partners\",...</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>10/31/2022, 2:37:05 AM</td>\n",
       "      <td>Fever Labs</td>\n",
       "      <td>https://www.cbinsights.com/company/fever-labs</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1/26/2022</td>\n",
       "      <td>2022</td>\n",
       "      <td>New York</td>\n",
       "      <td>United States</td>\n",
       "      <td>Internet software &amp; services</td>\n",
       "      <td>[\"Accel\",\"14W\",\"GS Growth\"]</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>10/31/2022, 2:37:04 AM</td>\n",
       "      <td>Minio</td>\n",
       "      <td>https://www.cbinsights.com/company/minio</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1/26/2022</td>\n",
       "      <td>2022</td>\n",
       "      <td>Palo Alto</td>\n",
       "      <td>United States</td>\n",
       "      <td>Data management &amp; analytics</td>\n",
       "      <td>[\"General Catalyst\",\"Nexus Venture Partners\",\"...</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>10/31/2022, 2:37:04 AM</td>\n",
       "      <td>Darwinbox</td>\n",
       "      <td>https://www.cbinsights.com/company/darwinbox</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1/25/2022</td>\n",
       "      <td>2022</td>\n",
       "      <td>Hyderabad</td>\n",
       "      <td>India</td>\n",
       "      <td>Internet software &amp; services</td>\n",
       "      <td>[\"Lightspeed India Partners\",\"Sequoia Capital ...</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>10/31/2022, 2:37:04 AM</td>\n",
       "      <td>Pentera</td>\n",
       "      <td>https://www.cbinsights.com/company/pcysys</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1/11/2022</td>\n",
       "      <td>2022</td>\n",
       "      <td>Petah Tikva</td>\n",
       "      <td>Israel</td>\n",
       "      <td>Cybersecurity</td>\n",
       "      <td>[\"AWZ Ventures\",\"Blackstone\",\"Insight Partners\"]</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "               Updated at     Company  \\\n",
       "0  10/31/2022, 2:37:05 AM       Esusu   \n",
       "1  10/31/2022, 2:37:05 AM  Fever Labs   \n",
       "2  10/31/2022, 2:37:04 AM       Minio   \n",
       "3  10/31/2022, 2:37:04 AM   Darwinbox   \n",
       "4  10/31/2022, 2:37:04 AM     Pentera   \n",
       "\n",
       "                                  Crunchbase Url  Last Valuation (Billion $)  \\\n",
       "0       https://www.cbinsights.com/company/esusu                         1.0   \n",
       "1  https://www.cbinsights.com/company/fever-labs                         1.0   \n",
       "2       https://www.cbinsights.com/company/minio                         1.0   \n",
       "3   https://www.cbinsights.com/company/darwinbox                         1.0   \n",
       "4      https://www.cbinsights.com/company/pcysys                         1.0   \n",
       "\n",
       "  Date Joined  Year Joined         City        Country  \\\n",
       "0   1/27/2022         2022     New York  United States   \n",
       "1   1/26/2022         2022     New York  United States   \n",
       "2   1/26/2022         2022    Palo Alto  United States   \n",
       "3   1/25/2022         2022    Hyderabad          India   \n",
       "4   1/11/2022         2022  Petah Tikva         Israel   \n",
       "\n",
       "                       Industry  \\\n",
       "0                       Fintech   \n",
       "1  Internet software & services   \n",
       "2   Data management & analytics   \n",
       "3  Internet software & services   \n",
       "4                Cybersecurity    \n",
       "\n",
       "                                           Investors  Company Website  \n",
       "0  [\"Next Play Ventures\",\"Zeal Capital Partners\",...              NaN  \n",
       "1                        [\"Accel\",\"14W\",\"GS Growth\"]              NaN  \n",
       "2  [\"General Catalyst\",\"Nexus Venture Partners\",\"...              NaN  \n",
       "3  [\"Lightspeed India Partners\",\"Sequoia Capital ...              NaN  \n",
       "4   [\"AWZ Ventures\",\"Blackstone\",\"Insight Partners\"]              NaN  "
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "a9c4a9a1-2761-4494-9cca-b99247e830ff",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "import ast"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "id": "2915f87a-9112-4827-9510-7ce14646303f",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "import ast \n",
    "def summary(company,crunchbase_url,city,country,industry,investor_list):\n",
    "    investors = 'The investors in the company are'\n",
    "     \n",
    "    for investor in ast.literal_eval(investor_list):\n",
    "        investors += f\" {investor}, \"\n",
    "\n",
    "    text = f\"{company} has headquarters in {city} in {country} and is in the field of {industry}. {investors}. You can find more information at {crunchbase_url}\"\n",
    "\n",
    "    return text "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "id": "c4dbf88f-abf8-479d-b8eb-c793b8d4c1e9",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "df['summary'] = df.apply(lambda df: summary(df['Company'],df['Crunchbase Url'],df['City'],df['Country'],df['Industry'],df['Investors']),axis=1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "c0e20f60-98ad-4fd6-84f3-50ccdfad3df7",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'Esusu has headquarters in New York in United States and is in the field of Fintech. The investors in the company are Next Play Ventures,  Zeal Capital Partners,  SoftBank Group, . You can find more information at https://www.cbinsights.com/company/esusu'"
      ]
     },
     "execution_count": 11,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df['summary'][0]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "id": "73f1ce96-fe73-4ece-b7e6-f6cb97b89d0f",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "# let's get a token count"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "id": "2617fe65-2e71-46ca-a552-214e21710ca4",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "import tiktoken"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "id": "6d453e34-c168-4e4d-bfa8-275d8902425d",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "def num_tokens_from_string(string,encoding_name):\n",
    "    encoding = tiktoken.get_encoding(encoding_name)\n",
    "    num_tokens = len(encoding.encode(string))\n",
    "    return num_tokens"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "id": "e53709af-016f-4ea6-8175-44d89975806d",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "58"
      ]
     },
     "execution_count": 15,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "num_tokens_from_string(df['summary'][0],\"cl100k_base\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "id": "f6f59515-42be-435f-80e7-6ba686c1d4a0",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "df['token_count'] = df['summary'].apply(lambda text:num_tokens_from_string(text,\"cl100k_base\"))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "id": "d1842eb1-8077-4512-a6db-4d656c271bf4",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Updated at</th>\n",
       "      <th>Company</th>\n",
       "      <th>Crunchbase Url</th>\n",
       "      <th>Last Valuation (Billion $)</th>\n",
       "      <th>Date Joined</th>\n",
       "      <th>Year Joined</th>\n",
       "      <th>City</th>\n",
       "      <th>Country</th>\n",
       "      <th>Industry</th>\n",
       "      <th>Investors</th>\n",
       "      <th>Company Website</th>\n",
       "      <th>summary</th>\n",
       "      <th>token_count</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>10/31/2022, 2:37:05 AM</td>\n",
       "      <td>Esusu</td>\n",
       "      <td>https://www.cbinsights.com/company/esusu</td>\n",
       "      <td>1.00</td>\n",
       "      <td>1/27/2022</td>\n",
       "      <td>2022</td>\n",
       "      <td>New York</td>\n",
       "      <td>United States</td>\n",
       "      <td>Fintech</td>\n",
       "      <td>[\"Next Play Ventures\",\"Zeal Capital Partners\",...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Esusu has headquarters in New York in United S...</td>\n",
       "      <td>58</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>10/31/2022, 2:37:05 AM</td>\n",
       "      <td>Fever Labs</td>\n",
       "      <td>https://www.cbinsights.com/company/fever-labs</td>\n",
       "      <td>1.00</td>\n",
       "      <td>1/26/2022</td>\n",
       "      <td>2022</td>\n",
       "      <td>New York</td>\n",
       "      <td>United States</td>\n",
       "      <td>Internet software &amp; services</td>\n",
       "      <td>[\"Accel\",\"14W\",\"GS Growth\"]</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Fever Labs has headquarters in New York in Uni...</td>\n",
       "      <td>60</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>10/31/2022, 2:37:04 AM</td>\n",
       "      <td>Minio</td>\n",
       "      <td>https://www.cbinsights.com/company/minio</td>\n",
       "      <td>1.00</td>\n",
       "      <td>1/26/2022</td>\n",
       "      <td>2022</td>\n",
       "      <td>Palo Alto</td>\n",
       "      <td>United States</td>\n",
       "      <td>Data management &amp; analytics</td>\n",
       "      <td>[\"General Catalyst\",\"Nexus Venture Partners\",\"...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Minio has headquarters in Palo Alto in United ...</td>\n",
       "      <td>57</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>10/31/2022, 2:37:04 AM</td>\n",
       "      <td>Darwinbox</td>\n",
       "      <td>https://www.cbinsights.com/company/darwinbox</td>\n",
       "      <td>1.00</td>\n",
       "      <td>1/25/2022</td>\n",
       "      <td>2022</td>\n",
       "      <td>Hyderabad</td>\n",
       "      <td>India</td>\n",
       "      <td>Internet software &amp; services</td>\n",
       "      <td>[\"Lightspeed India Partners\",\"Sequoia Capital ...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Darwinbox has headquarters in Hyderabad in Ind...</td>\n",
       "      <td>62</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>10/31/2022, 2:37:04 AM</td>\n",
       "      <td>Pentera</td>\n",
       "      <td>https://www.cbinsights.com/company/pcysys</td>\n",
       "      <td>1.00</td>\n",
       "      <td>1/11/2022</td>\n",
       "      <td>2022</td>\n",
       "      <td>Petah Tikva</td>\n",
       "      <td>Israel</td>\n",
       "      <td>Cybersecurity</td>\n",
       "      <td>[\"AWZ Ventures\",\"Blackstone\",\"Insight Partners\"]</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Pentera has headquarters in Petah Tikva in Isr...</td>\n",
       "      <td>58</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1194</th>\n",
       "      <td>10/31/2022, 2:32:42 AM</td>\n",
       "      <td>Fanatics</td>\n",
       "      <td>https://www.cbinsights.com/company/fanatics</td>\n",
       "      <td>27.00</td>\n",
       "      <td>6/6/2012</td>\n",
       "      <td>2012</td>\n",
       "      <td>Jacksonville</td>\n",
       "      <td>United States</td>\n",
       "      <td>E-commerce &amp; direct-to-consumer</td>\n",
       "      <td>[\"SoftBank Group\",\"Andreessen Horowitz\",\"Temas...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Fanatics has headquarters in Jacksonville in U...</td>\n",
       "      <td>63</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1195</th>\n",
       "      <td>10/31/2022, 2:32:38 AM</td>\n",
       "      <td>SpaceX</td>\n",
       "      <td>https://www.cbinsights.com/company/space-explo...</td>\n",
       "      <td>127.00</td>\n",
       "      <td>12/1/2012</td>\n",
       "      <td>2012</td>\n",
       "      <td>Hawthorne</td>\n",
       "      <td>United States</td>\n",
       "      <td>Other</td>\n",
       "      <td>[\"Founders Fund\",\"Draper Fisher Jurvetson\",\"Ro...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>SpaceX has headquarters in Hawthorne in United...</td>\n",
       "      <td>65</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1196</th>\n",
       "      <td>10/31/2022, 2:33:10 AM</td>\n",
       "      <td>Vice Media</td>\n",
       "      <td>https://www.cbinsights.com/company/vice-media</td>\n",
       "      <td>5.70</td>\n",
       "      <td>4/2/2011</td>\n",
       "      <td>2011</td>\n",
       "      <td>Brooklyn</td>\n",
       "      <td>United States</td>\n",
       "      <td>Internet software &amp; services</td>\n",
       "      <td>[\"Technology Crossover Ventures\",\"A&amp;E Televisi...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Vice Media has headquarters in Brooklyn in Uni...</td>\n",
       "      <td>56</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1197</th>\n",
       "      <td>10/31/2022, 2:33:05 AM</td>\n",
       "      <td>Klarna</td>\n",
       "      <td>https://www.cbinsights.com/company/klarna</td>\n",
       "      <td>6.70</td>\n",
       "      <td>12/12/2011</td>\n",
       "      <td>2011</td>\n",
       "      <td>Stockholm</td>\n",
       "      <td>Sweden</td>\n",
       "      <td>Fintech</td>\n",
       "      <td>[\"Institutional Venture Partners\",\"Sequoia Cap...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Klarna has headquarters in Stockholm in Sweden...</td>\n",
       "      <td>57</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1198</th>\n",
       "      <td>10/31/2022, 2:35:27 AM</td>\n",
       "      <td>Veepee</td>\n",
       "      <td>https://www.cbinsights.com/company/vente-privee</td>\n",
       "      <td>1.38</td>\n",
       "      <td>7/2/2007</td>\n",
       "      <td>2007</td>\n",
       "      <td>La Plaine Saint-Denis</td>\n",
       "      <td>France</td>\n",
       "      <td>E-commerce &amp; direct-to-consumer</td>\n",
       "      <td>[\"Summit Partners\",\"Qatar Holding\"]</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Veepee has headquarters in La Plaine Saint-Den...</td>\n",
       "      <td>62</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>1199 rows × 13 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "                  Updated at     Company  \\\n",
       "0     10/31/2022, 2:37:05 AM       Esusu   \n",
       "1     10/31/2022, 2:37:05 AM  Fever Labs   \n",
       "2     10/31/2022, 2:37:04 AM       Minio   \n",
       "3     10/31/2022, 2:37:04 AM   Darwinbox   \n",
       "4     10/31/2022, 2:37:04 AM     Pentera   \n",
       "...                      ...         ...   \n",
       "1194  10/31/2022, 2:32:42 AM    Fanatics   \n",
       "1195  10/31/2022, 2:32:38 AM      SpaceX   \n",
       "1196  10/31/2022, 2:33:10 AM  Vice Media   \n",
       "1197  10/31/2022, 2:33:05 AM      Klarna   \n",
       "1198  10/31/2022, 2:35:27 AM      Veepee   \n",
       "\n",
       "                                         Crunchbase Url  \\\n",
       "0              https://www.cbinsights.com/company/esusu   \n",
       "1         https://www.cbinsights.com/company/fever-labs   \n",
       "2              https://www.cbinsights.com/company/minio   \n",
       "3          https://www.cbinsights.com/company/darwinbox   \n",
       "4             https://www.cbinsights.com/company/pcysys   \n",
       "...                                                 ...   \n",
       "1194        https://www.cbinsights.com/company/fanatics   \n",
       "1195  https://www.cbinsights.com/company/space-explo...   \n",
       "1196      https://www.cbinsights.com/company/vice-media   \n",
       "1197          https://www.cbinsights.com/company/klarna   \n",
       "1198    https://www.cbinsights.com/company/vente-privee   \n",
       "\n",
       "      Last Valuation (Billion $) Date Joined  Year Joined  \\\n",
       "0                           1.00   1/27/2022         2022   \n",
       "1                           1.00   1/26/2022         2022   \n",
       "2                           1.00   1/26/2022         2022   \n",
       "3                           1.00   1/25/2022         2022   \n",
       "4                           1.00   1/11/2022         2022   \n",
       "...                          ...         ...          ...   \n",
       "1194                       27.00    6/6/2012         2012   \n",
       "1195                      127.00   12/1/2012         2012   \n",
       "1196                        5.70    4/2/2011         2011   \n",
       "1197                        6.70  12/12/2011         2011   \n",
       "1198                        1.38    7/2/2007         2007   \n",
       "\n",
       "                       City        Country                         Industry  \\\n",
       "0                  New York  United States                          Fintech   \n",
       "1                  New York  United States     Internet software & services   \n",
       "2                 Palo Alto  United States      Data management & analytics   \n",
       "3                 Hyderabad          India     Internet software & services   \n",
       "4               Petah Tikva         Israel                   Cybersecurity    \n",
       "...                     ...            ...                              ...   \n",
       "1194           Jacksonville  United States  E-commerce & direct-to-consumer   \n",
       "1195              Hawthorne  United States                            Other   \n",
       "1196               Brooklyn  United States     Internet software & services   \n",
       "1197              Stockholm         Sweden                          Fintech   \n",
       "1198  La Plaine Saint-Denis         France  E-commerce & direct-to-consumer   \n",
       "\n",
       "                                              Investors  Company Website  \\\n",
       "0     [\"Next Play Ventures\",\"Zeal Capital Partners\",...              NaN   \n",
       "1                           [\"Accel\",\"14W\",\"GS Growth\"]              NaN   \n",
       "2     [\"General Catalyst\",\"Nexus Venture Partners\",\"...              NaN   \n",
       "3     [\"Lightspeed India Partners\",\"Sequoia Capital ...              NaN   \n",
       "4      [\"AWZ Ventures\",\"Blackstone\",\"Insight Partners\"]              NaN   \n",
       "...                                                 ...              ...   \n",
       "1194  [\"SoftBank Group\",\"Andreessen Horowitz\",\"Temas...              NaN   \n",
       "1195  [\"Founders Fund\",\"Draper Fisher Jurvetson\",\"Ro...              NaN   \n",
       "1196  [\"Technology Crossover Ventures\",\"A&E Televisi...              NaN   \n",
       "1197  [\"Institutional Venture Partners\",\"Sequoia Cap...              NaN   \n",
       "1198                [\"Summit Partners\",\"Qatar Holding\"]              NaN   \n",
       "\n",
       "                                                summary  token_count  \n",
       "0     Esusu has headquarters in New York in United S...           58  \n",
       "1     Fever Labs has headquarters in New York in Uni...           60  \n",
       "2     Minio has headquarters in Palo Alto in United ...           57  \n",
       "3     Darwinbox has headquarters in Hyderabad in Ind...           62  \n",
       "4     Pentera has headquarters in Petah Tikva in Isr...           58  \n",
       "...                                                 ...          ...  \n",
       "1194  Fanatics has headquarters in Jacksonville in U...           63  \n",
       "1195  SpaceX has headquarters in Hawthorne in United...           65  \n",
       "1196  Vice Media has headquarters in Brooklyn in Uni...           56  \n",
       "1197  Klarna has headquarters in Stockholm in Sweden...           57  \n",
       "1198  Veepee has headquarters in La Plaine Saint-Den...           62  \n",
       "\n",
       "[1199 rows x 13 columns]"
      ]
     },
     "execution_count": 17,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "id": "8751e5a1-1bc3-4ca7-b3bd-be3336b62c1a",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "0.028168400000000003"
      ]
     },
     "execution_count": 18,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df['token_count'].sum() * 0.0004 / 1000"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "id": "aa1a5b50-b22f-44c4-9482-e3a673fd85a5",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "def get_embedding(text):\n",
    "    result = openai.Embedding.create(\n",
    "            model='text-embedding-ada-002',\n",
    "            input=text\n",
    "    )\n",
    "    return result['data'][0]['embedding']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "id": "9f4e81ca-f4ac-4012-a592-2872eb9fec45",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[0.011981870047748089,\n",
       " -0.017718156799674034,\n",
       " -0.022288423031568527,\n",
       " -0.03439091518521309,\n",
       " -0.013777810148894787,\n",
       " 0.010226137936115265,\n",
       " -0.016270682215690613,\n",
       " 0.02547822706401348,\n",
       " 0.0037393085658550262,\n",
       " -0.022047176957130432,\n",
       " 0.01929965615272522,\n",
       " -0.013161293230950832,\n",
       " 0.005350963678210974,\n",
       " -0.01687379740178585,\n",
       " 0.010661720298230648,\n",
       " -0.0236956886947155,\n",
       " -0.0007991263992153108,\n",
       " -0.0235750675201416,\n",
       " 0.01706143282353878,\n",
       " 0.0025749627966433764,\n",
       " -0.02321319840848446,\n",
       " -0.015814997255802155,\n",
       " -0.009911177679896355,\n",
       " 0.021470867097377777,\n",
       " -0.022114189341664314,\n",
       " -0.0009163985960185528,\n",
       " -0.011881351470947266,\n",
       " -0.0014164808671921492,\n",
       " -0.011593196541070938,\n",
       " -0.011331846937537193,\n",
       " 0.012028778903186321,\n",
       " -0.003712503472343087,\n",
       " -0.01907181367278099,\n",
       " -0.0041648391634225845,\n",
       " -0.012832931242883205,\n",
       " 0.0060110390186309814,\n",
       " -0.02235543541610241,\n",
       " 0.014179886318743229,\n",
       " 0.019366668537259102,\n",
       " -0.011800935491919518,\n",
       " 0.03181762993335724,\n",
       " 0.008644637651741505,\n",
       " -0.014019056223332882,\n",
       " 0.002127653220668435,\n",
       " -0.02827935852110386,\n",
       " 0.013764407485723495,\n",
       " -0.005260496865957975,\n",
       " 0.0022667045705020428,\n",
       " 0.0053911712020635605,\n",
       " 0.03642810136079788,\n",
       " -0.0101725272834301,\n",
       " 0.027434997260570526,\n",
       " -0.01955430395901203,\n",
       " 0.01149267703294754,\n",
       " -0.00728427991271019,\n",
       " -0.009958086535334587,\n",
       " -0.004989095032215118,\n",
       " -0.02254307083785534,\n",
       " 0.00855081994086504,\n",
       " -0.009495698846876621,\n",
       " 0.006148414686322212,\n",
       " 0.006681165657937527,\n",
       " -0.022234812378883362,\n",
       " -0.01836147904396057,\n",
       " -0.006088103633373976,\n",
       " 0.00771986274048686,\n",
       " -0.0253710076212883,\n",
       " -0.00369239947758615,\n",
       " 0.004114579409360886,\n",
       " 0.006248933728784323,\n",
       " 0.019272850826382637,\n",
       " 0.021846139803528786,\n",
       " -0.0019249396864324808,\n",
       " 0.008993104100227356,\n",
       " 0.004942186176776886,\n",
       " -0.0014508249005302787,\n",
       " -0.024767892435193062,\n",
       " -0.009951385669410229,\n",
       " 0.01415308192372322,\n",
       " -0.0004669947375077754,\n",
       " -0.004138033837080002,\n",
       " -0.01074883621186018,\n",
       " -0.03420327976346016,\n",
       " -0.007022930774837732,\n",
       " 0.013442747294902802,\n",
       " 0.020519288256764412,\n",
       " -0.003017246723175049,\n",
       " 0.023749299347400665,\n",
       " -0.023561663925647736,\n",
       " -0.0005055270157754421,\n",
       " 0.028440188616514206,\n",
       " 0.025062749162316322,\n",
       " -0.004184942692518234,\n",
       " -0.008879181928932667,\n",
       " -0.04578307271003723,\n",
       " 0.03624046593904495,\n",
       " -0.019688330590724945,\n",
       " 0.01018592994660139,\n",
       " -0.01791919581592083,\n",
       " -0.006325998343527317,\n",
       " 0.024003947153687477,\n",
       " 0.009535906836390495,\n",
       " -0.009227648377418518,\n",
       " -0.007920900359749794,\n",
       " -0.020814143121242523,\n",
       " 0.0045133051462471485,\n",
       " -0.009582815691828728,\n",
       " -0.01251797191798687,\n",
       " 0.03095986507833004,\n",
       " -0.00043893317342735827,\n",
       " -0.011821039952337742,\n",
       " 0.015600555576384068,\n",
       " -0.019715135917067528,\n",
       " 0.008289470337331295,\n",
       " 0.02913711965084076,\n",
       " -0.01510466169565916,\n",
       " 0.01582839898765087,\n",
       " 0.010487486608326435,\n",
       " -0.016337694600224495,\n",
       " -0.00869824830442667,\n",
       " 0.02681848034262657,\n",
       " 0.024821503087878227,\n",
       " 0.025156566873192787,\n",
       " -0.009582815691828728,\n",
       " 0.019045008346438408,\n",
       " 0.006071350071579218,\n",
       " -0.013938641175627708,\n",
       " -0.01862953044474125,\n",
       " -0.0027491957880556583,\n",
       " -0.022703900933265686,\n",
       " 0.024231791496276855,\n",
       " -0.020907960832118988,\n",
       " -0.007210566196590662,\n",
       " 0.01814703829586506,\n",
       " -0.004828264936804771,\n",
       " 0.010594706982374191,\n",
       " -0.017316080629825592,\n",
       " 0.017986208200454712,\n",
       " -0.0011082225246354938,\n",
       " -0.018495503813028336,\n",
       " 0.03342593461275101,\n",
       " 0.00042971892980858684,\n",
       " 0.0012321959948167205,\n",
       " 0.019942978397011757,\n",
       " -0.0030222726054489613,\n",
       " 0.0024426127783954144,\n",
       " 0.013288618065416813,\n",
       " 0.01988936774432659,\n",
       " 0.007847187109291553,\n",
       " -0.0023521457333117723,\n",
       " 0.014890220947563648,\n",
       " -0.03200526535511017,\n",
       " -0.005260496865957975,\n",
       " -0.004385980777442455,\n",
       " 0.006798438262194395,\n",
       " -0.0006852048099972308,\n",
       " 0.03307746723294258,\n",
       " 0.01959451287984848,\n",
       " -0.01380461547523737,\n",
       " -0.01070862915366888,\n",
       " -0.03184443339705467,\n",
       " 0.009710139594972134,\n",
       " -0.004690888803452253,\n",
       " -0.012980359606444836,\n",
       " 0.009375075809657574,\n",
       " 0.02708653174340725,\n",
       " 0.00981735996901989,\n",
       " -0.014126276597380638,\n",
       " 0.00037694640923291445,\n",
       " -0.003910190891474485,\n",
       " -0.027207154780626297,\n",
       " 0.02455345168709755,\n",
       " -0.013677291572093964,\n",
       " 0.001755732693709433,\n",
       " 0.00012208874977659434,\n",
       " 0.010889562778174877,\n",
       " 0.004677486140280962,\n",
       " 0.022449253126978874,\n",
       " -0.02704632468521595,\n",
       " -0.02120281755924225,\n",
       " 0.04508614167571068,\n",
       " 0.013704096898436546,\n",
       " 0.025639057159423828,\n",
       " 0.017383093014359474,\n",
       " 0.008644637651741505,\n",
       " -0.013328825123608112,\n",
       " 0.014220094308257103,\n",
       " 0.0017959402175620198,\n",
       " 0.02652362547814846,\n",
       " -0.016699563711881638,\n",
       " 0.020331652835011482,\n",
       " 0.011191120371222496,\n",
       " 0.0023069120943546295,\n",
       " -0.015895411372184753,\n",
       " -0.6313132047653198,\n",
       " -0.016297487542033195,\n",
       " -0.0007777661085128784,\n",
       " 0.01271901000291109,\n",
       " 0.00010313672828488052,\n",
       " 0.02265029214322567,\n",
       " 0.02715354412794113,\n",
       " 0.008637936785817146,\n",
       " 0.003658893285319209,\n",
       " -0.0027089882642030716,\n",
       " -0.014595365151762962,\n",
       " -0.015734581276774406,\n",
       " -0.015144869685173035,\n",
       " -0.013590174727141857,\n",
       " -0.019460486248135567,\n",
       " -0.014488144777715206,\n",
       " -0.009006506763398647,\n",
       " -0.05130492150783539,\n",
       " 0.002641975646838546,\n",
       " -0.010849355719983578,\n",
       " -0.003933645319193602,\n",
       " 0.018348075449466705,\n",
       " 0.004664083477109671,\n",
       " -0.016042839735746384,\n",
       " 0.02924434095621109,\n",
       " 0.011566391214728355,\n",
       " 0.0025213526096194983,\n",
       " -0.021591490134596825,\n",
       " -0.01829446665942669,\n",
       " 0.01419328898191452,\n",
       " -0.026349391788244247,\n",
       " 0.04227161034941673,\n",
       " 0.026952506974339485,\n",
       " 0.02499573491513729,\n",
       " 0.04101176932454109,\n",
       " 0.011693715117871761,\n",
       " -0.01806662231683731,\n",
       " 0.014019056223332882,\n",
       " -0.0059607792645692825,\n",
       " 0.023789506405591965,\n",
       " -0.0008770286804065108,\n",
       " -0.024312205612659454,\n",
       " 0.011961766518652439,\n",
       " -0.013683992438018322,\n",
       " 0.01129163894802332,\n",
       " 0.038867365568876266,\n",
       " 0.027019519358873367,\n",
       " -0.04154787212610245,\n",
       " 0.009100324474275112,\n",
       " 0.0092544537037611,\n",
       " 0.003893437562510371,\n",
       " -0.027957696467638016,\n",
       " -0.011747325770556927,\n",
       " 0.009656529873609543,\n",
       " 0.02731437422335148,\n",
       " 0.0006077213911339641,\n",
       " -0.006289141718298197,\n",
       " -0.014059263281524181,\n",
       " 0.013147891499102116,\n",
       " -0.014220094308257103,\n",
       " -0.014756196178495884,\n",
       " -0.0059808832593262196,\n",
       " -0.009502400644123554,\n",
       " -0.024499841034412384,\n",
       " -0.03680337220430374,\n",
       " 0.01164010539650917,\n",
       " -0.01825425773859024,\n",
       " 0.003240063786506653,\n",
       " 0.030906256288290024,\n",
       " -0.009147233329713345,\n",
       " 0.0034746082965284586,\n",
       " 0.018723348155617714,\n",
       " 0.004365877248346806,\n",
       " 0.019085215404629707,\n",
       " 0.0068252431228756905,\n",
       " 0.020519288256764412,\n",
       " 0.027247361838817596,\n",
       " -0.011184419505298138,\n",
       " -0.01201537624001503,\n",
       " 0.005505092907696962,\n",
       " -0.01184784434735775,\n",
       " -0.013107683509588242,\n",
       " 0.0008661390747874975,\n",
       " -0.02723396010696888,\n",
       " 0.0019651472102850676,\n",
       " -0.005783195607364178,\n",
       " -0.012283427640795708,\n",
       " -0.012068986892700195,\n",
       " -0.002254977123811841,\n",
       " -0.010232838802039623,\n",
       " 0.015144869685173035,\n",
       " 0.007156956009566784,\n",
       " -0.011412261985242367,\n",
       " 0.0009189115953631699,\n",
       " -0.0025414566043764353,\n",
       " 0.02078733779489994,\n",
       " 0.0023554961662739515,\n",
       " 0.011211223900318146,\n",
       " 0.02145746536552906,\n",
       " -0.006138362921774387,\n",
       " 0.0022147695999592543,\n",
       " -0.0038565807044506073,\n",
       " 0.0030641555786132812,\n",
       " 0.001305909943766892,\n",
       " 0.018455296754837036,\n",
       " 0.02298535406589508,\n",
       " 0.0003409271012060344,\n",
       " -0.012135999277234077,\n",
       " 0.03280941769480705,\n",
       " -0.060633089393377304,\n",
       " -0.011184419505298138,\n",
       " -0.029566001147031784,\n",
       " -0.02078733779489994,\n",
       " -0.0020120562985539436,\n",
       " -0.004868472460657358,\n",
       " -0.044201575219631195,\n",
       " 0.0010227812454104424,\n",
       " -0.008075029589235783,\n",
       " 0.005565404426306486,\n",
       " -0.003551672911271453,\n",
       " -0.0019232643535360694,\n",
       " -0.0160160344094038,\n",
       " 0.005846857558935881,\n",
       " -0.011398859322071075,\n",
       " 0.015185076743364334,\n",
       " 0.004737797658890486,\n",
       " -0.009602919220924377,\n",
       " -0.005535248667001724,\n",
       " -0.013033969327807426,\n",
       " 0.0010814174311235547,\n",
       " -0.01216280460357666,\n",
       " -0.04666764289140701,\n",
       " 0.0019400175660848618,\n",
       " -0.0022264968138188124,\n",
       " 0.004342422820627689,\n",
       " -0.004700940568000078,\n",
       " 0.008818870410323143,\n",
       " -0.007056436967104673,\n",
       " 0.03125471994280815,\n",
       " -0.02126982994377613,\n",
       " -0.024834904819726944,\n",
       " -0.009582815691828728,\n",
       " 0.01543972548097372,\n",
       " -0.010346760042011738,\n",
       " 0.0007786037749610841,\n",
       " -0.02019762620329857,\n",
       " 0.019098618999123573,\n",
       " -0.012430855073034763,\n",
       " -0.008517313748598099,\n",
       " -0.014019056223332882,\n",
       " -0.002797780092805624,\n",
       " -0.005890415981411934,\n",
       " 0.0002458528324496001,\n",
       " 0.015774788334965706,\n",
       " 0.013228306546807289,\n",
       " -0.012591686099767685,\n",
       " -0.015305699780583382,\n",
       " -0.01758413203060627,\n",
       " -0.01419328898191452,\n",
       " -0.014796403236687183,\n",
       " 0.006731425411999226,\n",
       " 0.01900480128824711,\n",
       " -0.033881619572639465,\n",
       " -0.01529229711741209,\n",
       " -0.021323440596461296,\n",
       " 0.009656529873609543,\n",
       " 0.001876355498097837,\n",
       " 0.0024576906580477953,\n",
       " -0.022114189341664314,\n",
       " 0.010125618427991867,\n",
       " -0.01529229711741209,\n",
       " 0.010266344994306564,\n",
       " -0.0033690633717924356,\n",
       " 0.016592344269156456,\n",
       " 0.03575797379016876,\n",
       " 0.015855204313993454,\n",
       " -0.009080220013856888,\n",
       " -0.00019695449736900628,\n",
       " -0.0041112289763987064,\n",
       " -0.0011031965259462595,\n",
       " 0.01802641525864601,\n",
       " 0.00326016778126359,\n",
       " -0.015613958239555359,\n",
       " 0.01240404974669218,\n",
       " 0.01944708451628685,\n",
       " -0.0011635079281404614,\n",
       " 0.0118277408182621,\n",
       " 0.019058410078287125,\n",
       " 0.005350963678210974,\n",
       " -0.007425006944686174,\n",
       " -0.01635109819471836,\n",
       " 0.012980359606444836,\n",
       " -0.023038964718580246,\n",
       " 0.015345907770097256,\n",
       " 0.01019263081252575,\n",
       " 0.0211626086384058,\n",
       " 0.010025099851191044,\n",
       " 0.004908679984509945,\n",
       " 0.004402734339237213,\n",
       " 0.007217267528176308,\n",
       " 0.019420279189944267,\n",
       " 0.0013754356186836958,\n",
       " 0.02608134225010872,\n",
       " -0.01821405068039894,\n",
       " -0.00821575615555048,\n",
       " -0.02738138847053051,\n",
       " -0.0003176820755470544,\n",
       " -0.01543972548097372,\n",
       " 0.0160562414675951,\n",
       " 0.02629578299820423,\n",
       " 0.030128909274935722,\n",
       " -0.02346784621477127,\n",
       " -0.0008397528436034918,\n",
       " 0.004416136536747217,\n",
       " 0.020479079335927963,\n",
       " 0.008631234988570213,\n",
       " 0.03543631359934807,\n",
       " -0.01166020892560482,\n",
       " -0.01290664542466402,\n",
       " 0.010990082286298275,\n",
       " -0.005739637650549412,\n",
       " -0.012826230376958847,\n",
       " 0.017570728436112404,\n",
       " -0.015533543191850185,\n",
       " -0.0114189637824893,\n",
       " 0.021846139803528786,\n",
       " 0.04530058428645134,\n",
       " 0.026161756366491318,\n",
       " 0.008845675736665726,\n",
       " -0.0019768746569752693,\n",
       " -0.005518495570868254,\n",
       " 0.027957696467638016,\n",
       " -0.016940809786319733,\n",
       " 0.003933645319193602,\n",
       " 0.00481486227363348,\n",
       " -0.022556474432349205,\n",
       " 0.03784877061843872,\n",
       " -0.03929624706506729,\n",
       " 0.010655018500983715,\n",
       " -0.013516460545361042,\n",
       " 0.027770061045885086,\n",
       " 0.003990605939179659,\n",
       " 0.04227161034941673,\n",
       " -0.013442747294902802,\n",
       " 0.0373394750058651,\n",
       " 0.0061919731087982655,\n",
       " 0.02574627846479416,\n",
       " 0.037098228931427,\n",
       " -0.03326510265469551,\n",
       " 0.004540110006928444,\n",
       " -0.018267661333084106,\n",
       " 0.008845675736665726,\n",
       " 0.003173051169142127,\n",
       " -0.012605087831616402,\n",
       " 0.01649852655827999,\n",
       " -0.01869654282927513,\n",
       " 4.2851479520322755e-05,\n",
       " 0.017262469977140427,\n",
       " 0.019648123532533646,\n",
       " 0.005719533655792475,\n",
       " 0.002161159412935376,\n",
       " -0.00782708264887333,\n",
       " 0.014488144777715206,\n",
       " 0.026537027209997177,\n",
       " 0.007773472927510738,\n",
       " -0.018093427643179893,\n",
       " -0.01848210208117962,\n",
       " -0.02924434095621109,\n",
       " -0.016739770770072937,\n",
       " -0.03050418011844158,\n",
       " -0.014300509355962276,\n",
       " -0.01415308192372322,\n",
       " 0.00043642017408274114,\n",
       " 0.007753368932753801,\n",
       " 0.020479079335927963,\n",
       " -0.015064454637467861,\n",
       " 0.014796403236687183,\n",
       " -0.022636888548731804,\n",
       " 0.015386114828288555,\n",
       " -0.028332967311143875,\n",
       " -0.000993463210761547,\n",
       " -0.040368448942899704,\n",
       " 0.013369033113121986,\n",
       " 0.01616346277296543,\n",
       " 0.023869922384619713,\n",
       " -0.00948229618370533,\n",
       " 0.015667568892240524,\n",
       " 0.002362197497859597,\n",
       " -0.017517119646072388,\n",
       " 0.03956429660320282,\n",
       " 0.009321466088294983,\n",
       " 0.0012858060654252768,\n",
       " -0.029110316187143326,\n",
       " -0.005444781389087439,\n",
       " -0.010018398053944111,\n",
       " 0.01179423462599516,\n",
       " 0.04004678875207901,\n",
       " -0.02187294326722622,\n",
       " -0.00040647387504577637,\n",
       " -0.01817384362220764,\n",
       " 0.024124570190906525,\n",
       " -0.016042839735746384,\n",
       " -0.020586300641298294,\n",
       " 0.0005109718185849488,\n",
       " 0.05666593462228775,\n",
       " 0.0152386873960495,\n",
       " 0.005836805794388056,\n",
       " 0.006587347947061062,\n",
       " 0.007056436967104673,\n",
       " -0.02183273620903492,\n",
       " 0.000417154049500823,\n",
       " -0.01795940287411213,\n",
       " -0.04597070813179016,\n",
       " 0.026175159960985184,\n",
       " -0.0035416209138929844,\n",
       " 0.007324487902224064,\n",
       " -0.01415308192372322,\n",
       " -0.007398201618343592,\n",
       " 0.04149426147341728,\n",
       " 0.00886578019708395,\n",
       " 0.00014261138858273625,\n",
       " -0.034176476299762726,\n",
       " -0.028842264786362648,\n",
       " 0.019701732322573662,\n",
       " 0.11365353316068649,\n",
       " 0.006315946578979492,\n",
       " -0.00873845536261797,\n",
       " 0.017637740820646286,\n",
       " -0.0036119844298809767,\n",
       " 0.009180739521980286,\n",
       " -0.0178119745105505,\n",
       " -0.0556473433971405,\n",
       " -0.012243219651281834,\n",
       " 0.007170358672738075,\n",
       " -0.03095986507833004,\n",
       " 0.0023554961662739515,\n",
       " 0.004520006477832794,\n",
       " -0.011036991141736507,\n",
       " 0.002315288642421365,\n",
       " 0.002797780092805624,\n",
       " 0.00011842400272144005,\n",
       " -0.017450105398893356,\n",
       " -2.947511529782787e-05,\n",
       " -0.006691217888146639,\n",
       " -0.011130808852612972,\n",
       " 0.0017842130037024617,\n",
       " 0.026537027209997177,\n",
       " 0.03658893331885338,\n",
       " 0.03337232396006584,\n",
       " 0.01783877983689308,\n",
       " 0.018803762272000313,\n",
       " 0.03334551677107811,\n",
       " 0.005072860978543758,\n",
       " -0.014930428937077522,\n",
       " 0.012129298411309719,\n",
       " -0.0020405366085469723,\n",
       " 0.011874649673700333,\n",
       " 0.02055949531495571,\n",
       " -0.04570265859365463,\n",
       " -0.0002389421424595639,\n",
       " 0.017570728436112404,\n",
       " 0.01456855982542038,\n",
       " 0.007351292762905359,\n",
       " 0.007210566196590662,\n",
       " -0.002662079408764839,\n",
       " 0.02254307083785534,\n",
       " -0.001341929193586111,\n",
       " -0.013080878183245659,\n",
       " 0.00010360791202401742,\n",
       " -0.027502011507749557,\n",
       " -0.0037292565684765577,\n",
       " 0.015493335202336311,\n",
       " 0.007914199493825436,\n",
       " 0.0019701733253896236,\n",
       " 0.019125424325466156,\n",
       " -0.04318298026919365,\n",
       " -0.03165679797530174,\n",
       " -0.017865585163235664,\n",
       " 0.0011040341341868043,\n",
       " -0.0177315603941679,\n",
       " 0.0040274630300700665,\n",
       " -0.01680678501725197,\n",
       " -0.023977143689990044,\n",
       " -0.04272729530930519,\n",
       " -0.02313278242945671,\n",
       " -0.012189609929919243,\n",
       " -0.007304383907467127,\n",
       " -0.01735628768801689,\n",
       " 0.012410751543939114,\n",
       " -0.033693984150886536,\n",
       " -0.013436045497655869,\n",
       " -0.0005490852636285126,\n",
       " -0.022958550602197647,\n",
       " -0.020733729004859924,\n",
       " 0.026175159960985184,\n",
       " -0.012839633040130138,\n",
       " -0.015332505106925964,\n",
       " 0.01836147904396057,\n",
       " 0.05304725095629692,\n",
       " -0.0044797989539802074,\n",
       " -0.0032434144522994757,\n",
       " 0.0017507066950201988,\n",
       " 0.005565404426306486,\n",
       " 0.0355435349047184,\n",
       " -0.00905341561883688,\n",
       " 0.005873662885278463,\n",
       " -0.005665923468768597,\n",
       " -0.020626507699489594,\n",
       " -0.0018244206439703703,\n",
       " 0.0026939103845506907,\n",
       " 0.024767892435193062,\n",
       " -0.001332715037278831,\n",
       " -0.007398201618343592,\n",
       " 0.009455491788685322,\n",
       " -0.00699612544849515,\n",
       " -0.014514950104057789,\n",
       " 0.004315617494285107,\n",
       " -0.004928783979266882,\n",
       " 0.00707654096186161,\n",
       " -0.00836318451911211,\n",
       " 0.01996978372335434,\n",
       " -0.007706460077315569,\n",
       " -0.008430196903645992,\n",
       " -0.013704096898436546,\n",
       " 0.0006106531945988536,\n",
       " -0.02529059164226055,\n",
       " 0.012792724184691906,\n",
       " 0.004168189596384764,\n",
       " 0.021095596253871918,\n",
       " 0.03514145687222481,\n",
       " 0.0034511538688093424,\n",
       " -0.009602919220924377,\n",
       " -0.008631234988570213,\n",
       " 0.02801130712032318,\n",
       " -0.0019148878054693341,\n",
       " -0.021028583869338036,\n",
       " 0.009174037724733353,\n",
       " 0.00034720954135991633,\n",
       " -0.008242561481893063,\n",
       " 0.011727222241461277,\n",
       " 0.0067615811713039875,\n",
       " 0.021108999848365784,\n",
       " 0.00332047906704247,\n",
       " 0.012270024977624416,\n",
       " 0.013764407485723495,\n",
       " -0.03715183958411217,\n",
       " 0.02421838790178299,\n",
       " 0.01657894067466259,\n",
       " 0.01873674988746643,\n",
       " -0.014434535056352615,\n",
       " -0.0010588006116449833,\n",
       " -0.015211882069706917,\n",
       " -0.005079562310129404,\n",
       " 0.02413797378540039,\n",
       " -0.03495382145047188,\n",
       " 0.024312205612659454,\n",
       " -0.0043323710560798645,\n",
       " 0.007331188768148422,\n",
       " -0.02354826219379902,\n",
       " -0.013409240171313286,\n",
       " 0.004322318825870752,\n",
       " -0.003923593554645777,\n",
       " 0.013087579980492592,\n",
       " -0.0114591708406806,\n",
       " -0.004355825483798981,\n",
       " -0.017718156799674034,\n",
       " -0.00510301673784852,\n",
       " -0.01877695694565773,\n",
       " -0.012290128506720066,\n",
       " -0.03173721209168434,\n",
       " 0.00032249861396849155,\n",
       " 0.009991592727601528,\n",
       " 0.01680678501725197,\n",
       " 0.005257145967334509,\n",
       " -0.007538928184658289,\n",
       " 0.01817384362220764,\n",
       " -0.0054313791915774345,\n",
       " 0.0017976155504584312,\n",
       " -0.007364695426076651,\n",
       " -0.02980724722146988,\n",
       " -0.019286254420876503,\n",
       " -0.009562712162733078,\n",
       " 0.028332967311143875,\n",
       " 0.02298535406589508,\n",
       " 0.037232253700494766,\n",
       " 0.016176864504814148,\n",
       " 0.0012271699961274862,\n",
       " -0.002732442691922188,\n",
       " 0.010795745067298412,\n",
       " -0.004201696254312992,\n",
       " -0.01505105197429657,\n",
       " 0.017530521377921104,\n",
       " 0.009830762632191181,\n",
       " 0.016297487542033195,\n",
       " 0.022342033684253693,\n",
       " 0.0025113008450716734,\n",
       " 0.022864732891321182,\n",
       " -0.03160318732261658,\n",
       " -0.029512392356991768,\n",
       " 0.02153788134455681,\n",
       " -0.011894753202795982,\n",
       " -0.013496357016265392,\n",
       " -0.03420327976346016,\n",
       " -0.04471087083220482,\n",
       " -0.022851329296827316,\n",
       " -0.006895606406033039,\n",
       " -0.014809805899858475,\n",
       " 0.008959597907960415,\n",
       " -0.050795622169971466,\n",
       " -0.005900467745959759,\n",
       " 0.002690559718757868,\n",
       " 0.003136194311082363,\n",
       " 0.009924580343067646,\n",
       " -0.010072008706629276,\n",
       " 0.01959451287984848,\n",
       " -0.01649852655827999,\n",
       " 0.030450569465756416,\n",
       " 0.00699612544849515,\n",
       " 0.006131661590188742,\n",
       " 0.002368898829445243,\n",
       " -0.004282111302018166,\n",
       " -0.020854352042078972,\n",
       " -0.0069090090692043304,\n",
       " 0.02734117954969406,\n",
       " 0.018428491428494453,\n",
       " 0.006838645786046982,\n",
       " 0.01529229711741209,\n",
       " 0.006718022748827934,\n",
       " -0.0009055090486072004,\n",
       " 0.02819894254207611,\n",
       " -0.02325340546667576,\n",
       " -0.0029937922954559326,\n",
       " 0.03594560921192169,\n",
       " -0.01743670366704464,\n",
       " -0.011224626563489437,\n",
       " 0.0003578896867111325,\n",
       " -0.017409898340702057,\n",
       " -0.029941271990537643,\n",
       " 0.01240404974669218,\n",
       " 0.0058234031312167645,\n",
       " -0.01568097062408924,\n",
       " 0.01844189316034317,\n",
       " -0.04932134598493576,\n",
       " -5.1961014833068475e-05,\n",
       " -0.007760070264339447,\n",
       " 0.004757901653647423,\n",
       " 0.021323440596461296,\n",
       " 0.008329678326845169,\n",
       " 0.02905670553445816,\n",
       " 0.03162999451160431,\n",
       " 0.018307868391275406,\n",
       " -0.05331530049443245,\n",
       " -0.01806662231683731,\n",
       " -0.001044560456648469,\n",
       " -0.0007890745182521641,\n",
       " 0.01529229711741209,\n",
       " 0.011204523034393787,\n",
       " -0.013697395101189613,\n",
       " -0.004245254211127758,\n",
       " 0.015024246647953987,\n",
       " 0.014206691645085812,\n",
       " -0.018120232969522476,\n",
       " 0.0030541035812348127,\n",
       " 0.005381119437515736,\n",
       " -0.009187440387904644,\n",
       " 0.01836147904396057,\n",
       " -0.011579793877899647,\n",
       " -0.01873674988746643,\n",
       " -0.015962423756718636,\n",
       " 0.006362855434417725,\n",
       " -0.021738918498158455,\n",
       " 6.43426610622555e-05,\n",
       " -0.010547798126935959,\n",
       " -0.013791212812066078,\n",
       " -0.006339401006698608,\n",
       " 0.013328825123608112,\n",
       " -0.005890415981411934,\n",
       " 0.02082754671573639,\n",
       " 0.004868472460657358,\n",
       " -0.004848368465900421,\n",
       " -0.016739770770072937,\n",
       " -0.026255574077367783,\n",
       " -0.0010127293644472957,\n",
       " -0.025303993374109268,\n",
       " -0.013952043838799,\n",
       " -0.006114908494055271,\n",
       " -0.0014223444741219282,\n",
       " 0.009475595317780972,\n",
       " 0.006751528941094875,\n",
       " 0.007934303022921085,\n",
       " -0.014648975804448128,\n",
       " 0.003940346650779247,\n",
       " 0.004231852013617754,\n",
       " 0.026925701647996902,\n",
       " -0.003662243951112032,\n",
       " -0.032353729009628296,\n",
       " 0.021591490134596825,\n",
       " -0.008095134049654007,\n",
       " 0.009636425413191319,\n",
       " -0.02395033836364746,\n",
       " -0.021484270691871643,\n",
       " -0.013127787038683891,\n",
       " -0.004516655579209328,\n",
       " -0.013925238512456417,\n",
       " 0.02105538919568062,\n",
       " 0.01543972548097372,\n",
       " -0.018468698486685753,\n",
       " -0.0012388972099870443,\n",
       " -0.0009096973808482289,\n",
       " -0.020170820876955986,\n",
       " -0.01921924203634262,\n",
       " 0.0054950411431491375,\n",
       " 0.018709944561123848,\n",
       " 0.0063293492421507835,\n",
       " -0.011184419505298138,\n",
       " -0.011036991141736507,\n",
       " 0.006687866989523172,\n",
       " 0.01806662231683731,\n",
       " -0.0030758827924728394,\n",
       " 0.02529059164226055,\n",
       " 0.007914199493825436,\n",
       " -0.0013201500987634063,\n",
       " -0.004399383440613747,\n",
       " 0.011707117781043053,\n",
       " -0.013885030522942543,\n",
       " 0.01948729157447815,\n",
       " -0.009897775016725063,\n",
       " 0.025049345567822456,\n",
       " -0.007726564072072506,\n",
       " -0.015788191929459572,\n",
       " 0.014541755430400372,\n",
       " -0.017785169184207916,\n",
       " 0.0056960792280733585,\n",
       " 0.007853887975215912,\n",
       " 0.008584326133131981,\n",
       " 0.0010018397588282824,\n",
       " 0.011861247010529041,\n",
       " 0.003169700503349304,\n",
       " -0.0013955393806099892,\n",
       " -0.0032735702116042376,\n",
       " 0.0055587030947208405,\n",
       " -0.0030272984877228737,\n",
       " -0.0033221545163542032,\n",
       " 0.015493335202336311,\n",
       " -0.003263518214225769,\n",
       " -0.022382240742444992,\n",
       " 0.003886736463755369,\n",
       " -0.006071350071579218,\n",
       " 0.016672758385539055,\n",
       " 0.004952238406985998,\n",
       " 0.014796403236687183,\n",
       " 0.002501248847693205,\n",
       " -0.012631893157958984,\n",
       " -0.03294344246387482,\n",
       " -0.006252284627407789,\n",
       " -0.0089260907843709,\n",
       " -0.021712113171815872,\n",
       " -0.005803299602121115,\n",
       " -0.00608140230178833,\n",
       " -0.0211224015802145,\n",
       " -0.006922411732375622,\n",
       " -0.0033690633717924356,\n",
       " 0.032166093587875366,\n",
       " -0.0147695979103446,\n",
       " -0.004181592259556055,\n",
       " 0.016672758385539055,\n",
       " 0.017892390489578247,\n",
       " 0.0027575725689530373,\n",
       " 0.003977203741669655,\n",
       " -0.020023394376039505,\n",
       " -0.03385481610894203,\n",
       " -0.022717304527759552,\n",
       " -0.029297951608896255,\n",
       " -0.001936666900292039,\n",
       " -6.685563857899979e-05,\n",
       " 0.05015230178833008,\n",
       " 0.006855398882180452,\n",
       " -0.02309257537126541,\n",
       " -0.03457855060696602,\n",
       " 0.004007359035313129,\n",
       " -0.060150597244501114,\n",
       " 0.0076796552166342735,\n",
       " -0.003846528707072139,\n",
       " 0.03782196715474129,\n",
       " 0.03141555190086365,\n",
       " 0.008544119074940681,\n",
       " -0.009100324474275112,\n",
       " -0.009348271414637566,\n",
       " 0.007445110473781824,\n",
       " 0.02011721208691597,\n",
       " -0.044094353914260864,\n",
       " 0.003343933494761586,\n",
       " 0.021618295460939407,\n",
       " -0.03280941769480705,\n",
       " 0.005099666304886341,\n",
       " -0.018937788903713226,\n",
       " -0.03307746723294258,\n",
       " -0.0022667045705020428,\n",
       " 0.018642932176589966,\n",
       " 5.049511310062371e-05,\n",
       " 0.011780831962823868,\n",
       " 0.022744109854102135,\n",
       " -0.012142700143158436,\n",
       " 0.006500231567770243,\n",
       " 0.00912712886929512,\n",
       " -0.0009314764756709337,\n",
       " -0.02771645039319992,\n",
       " 0.002983740298077464,\n",
       " 0.01955430395901203,\n",
       " -0.043424226343631744,\n",
       " 0.003628737526014447,\n",
       " 0.03018251806497574,\n",
       " -0.0203048475086689,\n",
       " -0.02801130712032318,\n",
       " 0.01649852655827999,\n",
       " -0.0130741773173213,\n",
       " -0.00143155874684453,\n",
       " 0.005826754029840231,\n",
       " -0.007934303022921085,\n",
       " 0.005451482720673084,\n",
       " -0.034551747143268585,\n",
       " -0.005511794239282608,\n",
       " 0.011740623973309994,\n",
       " -0.01873674988746643,\n",
       " 0.009227648377418518,\n",
       " 0.023159587755799294,\n",
       " 0.0068587493151426315,\n",
       " -0.015882009640336037,\n",
       " -0.032246507704257965,\n",
       " 0.04374588653445244,\n",
       " -0.02126982994377613,\n",
       " 0.006406413856893778,\n",
       " -0.0018478750716894865,\n",
       " -0.011552988551557064,\n",
       " 0.004215098451822996,\n",
       " -0.010628214105963707,\n",
       " -0.003752710996195674,\n",
       " 0.008168847300112247,\n",
       " -0.018374880775809288,\n",
       " -0.015600555576384068,\n",
       " 0.0021243025548756123,\n",
       " -0.016324292868375778,\n",
       " 0.008537417277693748,\n",
       " 0.0031244668643921614,\n",
       " -0.0177717674523592,\n",
       " -0.02693910337984562,\n",
       " 0.02480809949338436,\n",
       " 0.01125143188983202,\n",
       " -0.019648123532533646,\n",
       " -0.01237054355442524,\n",
       " 0.01052769459784031,\n",
       " -0.020278042182326317,\n",
       " -0.0043122670613229275,\n",
       " -0.010460682213306427,\n",
       " 0.007867290638387203,\n",
       " -0.029753636568784714,\n",
       " -0.004218449350446463,\n",
       " 0.010722031816840172,\n",
       " -0.02626897767186165,\n",
       " 0.0338280089199543,\n",
       " 0.19492653012275696,\n",
       " 0.014394327066838741,\n",
       " 0.0071435533463954926,\n",
       " 0.03658893331885338,\n",
       " -0.0024995736312121153,\n",
       " 0.008222457952797413,\n",
       " 0.01988936774432659,\n",
       " 0.005639118608087301,\n",
       " -0.019688330590724945,\n",
       " -0.004657382611185312,\n",
       " -0.03728586435317993,\n",
       " -0.012236518785357475,\n",
       " 0.004456344526261091,\n",
       " -0.0006592373829334974,\n",
       " 0.00628579081967473,\n",
       " -0.024392621591687202,\n",
       " -0.02838657796382904,\n",
       " -0.015694374218583107,\n",
       " -0.00983746349811554,\n",
       " 0.028788654133677483,\n",
       " 0.002258327789604664,\n",
       " -0.004667434375733137,\n",
       " 0.008430196903645992,\n",
       " -0.02328021079301834,\n",
       " 0.024714281782507896,\n",
       " 0.004322318825870752,\n",
       " 0.004915381316095591,\n",
       " 0.0028379876166582108,\n",
       " 0.036562126129865646,\n",
       " 0.019621318206191063,\n",
       " -0.010949874296784401,\n",
       " 0.02168530784547329,\n",
       " 0.016846992075443268,\n",
       " 0.01687379740178585,\n",
       " -0.0025732875801622868,\n",
       " -0.02857421338558197,\n",
       " 0.009931281208992004,\n",
       " -0.014514950104057789,\n",
       " 0.02313278242945671,\n",
       " 0.008972999639809132,\n",
       " 0.001453337841667235,\n",
       " -0.013308721594512463,\n",
       " -0.027930891141295433,\n",
       " 0.007465214468538761,\n",
       " -0.01529229711741209,\n",
       " 0.014072665944695473,\n",
       " ...]"
      ]
     },
     "execution_count": 20,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "get_embedding('Esusu has headquarters in New York in United States and is in the field of Fintech. The investors in the company are Next Play Ventures,  Zeal Capital Partners,  SoftBank Group, . You can find more information at https://www.cbinsights.com/company/esusu')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "id": "2f09bb88-7d86-41b1-af20-61ca2128b327",
   "metadata": {},
   "outputs": [],
   "source": [
    "# this will take a while\n",
    "# there's a premade file in the course materials if you don't want to pay to run this\n",
    "# df['embedding'] = df['summary'].apply(get_embedding)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "4efeec41-dd99-431f-b3de-5c45b74a2d58",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.11.2"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}