redis-langchain-airbnb.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "Yk3jkeRp3Dd-",
    "outputId": "2b352d3a-1598-4727-b25e-714fc705d0f1",
    "tags": []
   },
   "outputs": [],
   "source": [
    "import numpy as np\n",
    "import pandas as pd\n",
    "import time\n",
    "import redis\n",
    "import time\n",
    "\n",
    "from redis.commands.search.field import VectorField\n",
    "from redis.commands.search.field import TextField\n",
    "from redis.commands.search.field import TagField\n",
    "from redis.commands.search.field import NumericField\n",
    "from redis.commands.search.query import Query"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "redis_conn = redis.Redis(\n",
    "  host='localhost',\n",
    "  port=6379,\n",
    "  password=''\n",
    ")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 669
    },
    "id": "00_n4VWH7FoB",
    "outputId": "f26daa8c-4af9-4def-d5ab-3197777fe2f9",
    "tags": []
   },
   "outputs": [],
   "source": [
    "MAX_TEXT_LENGTH=512\n",
    "\n",
    "def auto_truncate(val):\n",
    "    return str(val)[:MAX_TEXT_LENGTH]\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "room_data = pd.read_csv(\n",
    "    'app/data/tripath.csv', delimiter='\\t', encoding_errors='ignore', on_bad_lines='skip',\n",
    "    converters={'bullet_point': auto_truncate,'amenities':auto_truncate,'name':auto_truncate}\n",
    ")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "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>listing_id</th>\n",
       "      <th>city</th>\n",
       "      <th>Owner</th>\n",
       "      <th>Tipología</th>\n",
       "      <th>ID</th>\n",
       "      <th>NEIGHBOURHOOD</th>\n",
       "      <th>Address</th>\n",
       "      <th>Nº HABIT</th>\n",
       "      <th>description</th>\n",
       "      <th>Square Metres Room</th>\n",
       "      <th>...</th>\n",
       "      <th>AGE</th>\n",
       "      <th>PASSPORT/DNI/NIE</th>\n",
       "      <th>Identification Number</th>\n",
       "      <th>Sex</th>\n",
       "      <th>Occupation</th>\n",
       "      <th>Nivel</th>\n",
       "      <th>University</th>\n",
       "      <th></th>\n",
       "      <th>Telephone</th>\n",
       "      <th>Country</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>5011</td>\n",
       "      <td>Madrid</td>\n",
       "      <td>PRIVATE INDIVIDUAL</td>\n",
       "      <td>A</td>\n",
       "      <td>2</td>\n",
       "      <td>CHAMBERÍ</td>\n",
       "      <td>Vallehermoso 69, 3C</td>\n",
       "      <td>1</td>\n",
       "      <td>Do you want to live in one of the most charmin...</td>\n",
       "      <td>9 m2</td>\n",
       "      <td>...</td>\n",
       "      <td>22</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>MAN</td>\n",
       "      <td>STUDY</td>\n",
       "      <td>ERASMUS</td>\n",
       "      <td>UNIVERSIDAD POLITÉCNICA</td>\n",
       "      <td>nicolin.herault@gmail.com</td>\n",
       "      <td>3.378953e+10</td>\n",
       "      <td>FRANCE</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>5224</td>\n",
       "      <td>Madrid</td>\n",
       "      <td>COMPANY</td>\n",
       "      <td>A</td>\n",
       "      <td>7</td>\n",
       "      <td>CARABANCHEL</td>\n",
       "      <td>Oca 105, 2º1</td>\n",
       "      <td>1</td>\n",
       "      <td>This room is ideal for students or young profe...</td>\n",
       "      <td>7.35</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>5244</td>\n",
       "      <td>Madrid</td>\n",
       "      <td>COMPANY</td>\n",
       "      <td>A</td>\n",
       "      <td>5</td>\n",
       "      <td>GUINDALERA</td>\n",
       "      <td>Cartagena 10, 1ºA</td>\n",
       "      <td>1</td>\n",
       "      <td>If you are looking for a cozy and comfortable ...</td>\n",
       "      <td>16m2</td>\n",
       "      <td>...</td>\n",
       "      <td>22</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>MAN</td>\n",
       "      <td>STUDY</td>\n",
       "      <td>ERASMUS</td>\n",
       "      <td>UNIVERSIDAD POLITÉCNICA</td>\n",
       "      <td>nicolin.herault@gmail.com</td>\n",
       "      <td>3.378953e+10</td>\n",
       "      <td>FRANCE</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>5254</td>\n",
       "      <td>Alicante</td>\n",
       "      <td>PRIVATE INDIVIDUAL</td>\n",
       "      <td>A</td>\n",
       "      <td>4</td>\n",
       "      <td>CENTRO</td>\n",
       "      <td>Avenida Aguilera 18, 3ºB</td>\n",
       "      <td>1</td>\n",
       "      <td>This exterior room is perfect for solo travele...</td>\n",
       "      <td>18m2</td>\n",
       "      <td>...</td>\n",
       "      <td>23</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>MAN</td>\n",
       "      <td>STUDY</td>\n",
       "      <td>ERASMUS</td>\n",
       "      <td>UNIVERSIDAD FRANCISCO DE VITORIA</td>\n",
       "      <td>thomasl1997@hotmail.fr</td>\n",
       "      <td>3.363209e+10</td>\n",
       "      <td>FRANCE</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>5269</td>\n",
       "      <td>Madrid</td>\n",
       "      <td>COMPANY</td>\n",
       "      <td>A</td>\n",
       "      <td>7</td>\n",
       "      <td>CARABANCHEL</td>\n",
       "      <td>Oca 105, 2º1</td>\n",
       "      <td>2</td>\n",
       "      <td>This room is ideal for students or young profe...</td>\n",
       "      <td>7.87</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>5275</td>\n",
       "      <td>Madrid</td>\n",
       "      <td>COMPANY</td>\n",
       "      <td>A</td>\n",
       "      <td>7</td>\n",
       "      <td>CARABANCHEL</td>\n",
       "      <td>Oca 105, 2º1</td>\n",
       "      <td>3</td>\n",
       "      <td>This room is ideal for students or young profe...</td>\n",
       "      <td>6.92</td>\n",
       "      <td>...</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>5286</td>\n",
       "      <td>Alicante</td>\n",
       "      <td>PRIVATE INDIVIDUAL</td>\n",
       "      <td>A</td>\n",
       "      <td>4</td>\n",
       "      <td>CENTRO</td>\n",
       "      <td>Avenida Aguilera 18, 3ºB</td>\n",
       "      <td>2</td>\n",
       "      <td>This exterior room is ideal for solo travelers...</td>\n",
       "      <td>12m2</td>\n",
       "      <td>...</td>\n",
       "      <td>20</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>WOMAN</td>\n",
       "      <td>WORK</td>\n",
       "      <td>INTERNSHIPS</td>\n",
       "      <td>UNIVERSIDAD COMPLUTENSE</td>\n",
       "      <td>marielaurent.private@gmail.com</td>\n",
       "      <td>3.306515e+11</td>\n",
       "      <td>FRANCE</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>5294</td>\n",
       "      <td>Alicante</td>\n",
       "      <td>PRIVATE INDIVIDUAL</td>\n",
       "      <td>A</td>\n",
       "      <td>4</td>\n",
       "      <td>CENTRO</td>\n",
       "      <td>Avenida Aguilera 18, 3ºB</td>\n",
       "      <td>5</td>\n",
       "      <td>This room is ideal for solo travelers who want...</td>\n",
       "      <td>6,5m2</td>\n",
       "      <td>...</td>\n",
       "      <td>18/07/1997</td>\n",
       "      <td>PASAPORTE</td>\n",
       "      <td>U5528887</td>\n",
       "      <td>WOMAN</td>\n",
       "      <td>STUDY</td>\n",
       "      <td>GRADUATE</td>\n",
       "      <td>NaN</td>\n",
       "      <td>corinwtagg@gmail.com</td>\n",
       "      <td>3.465090e+10</td>\n",
       "      <td>ARGENTINA</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>5303</td>\n",
       "      <td>Alicante</td>\n",
       "      <td>PRIVATE INDIVIDUAL</td>\n",
       "      <td>A</td>\n",
       "      <td>4</td>\n",
       "      <td>CENTRO</td>\n",
       "      <td>Avenida Aguilera 18, 3ºB</td>\n",
       "      <td>3</td>\n",
       "      <td>This room is perfect for solo travelers who wa...</td>\n",
       "      <td>9m2</td>\n",
       "      <td>...</td>\n",
       "      <td>24</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>MAN</td>\n",
       "      <td>STUDY</td>\n",
       "      <td>GRADUATE</td>\n",
       "      <td>ESIC</td>\n",
       "      <td>elmansouri.ismail2@gmail.com</td>\n",
       "      <td>3.365897e+10</td>\n",
       "      <td>FRANCE</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>5312</td>\n",
       "      <td>Alicante</td>\n",
       "      <td>PRIVATE INDIVIDUAL</td>\n",
       "      <td>A</td>\n",
       "      <td>4</td>\n",
       "      <td>CENTRO</td>\n",
       "      <td>Avenida Aguilera 18, 3ºB</td>\n",
       "      <td>4</td>\n",
       "      <td>This room is ideal for solo travelers who want...</td>\n",
       "      <td>8m2</td>\n",
       "      <td>...</td>\n",
       "      <td>20</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>MAN</td>\n",
       "      <td>STUDY</td>\n",
       "      <td>GRADUATE</td>\n",
       "      <td>UNIVERSIDAD COMPLUTENSE</td>\n",
       "      <td>sim.pulcini@stud.uniroma3.it</td>\n",
       "      <td>3.933961e+11</td>\n",
       "      <td>ITALY</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>10 rows × 37 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "   listing_id      city               Owner Tipología  ID NEIGHBOURHOOD  \\\n",
       "0        5011    Madrid  PRIVATE INDIVIDUAL         A   2      CHAMBERÍ   \n",
       "1        5224    Madrid             COMPANY         A   7   CARABANCHEL   \n",
       "2        5244    Madrid             COMPANY         A   5    GUINDALERA   \n",
       "3        5254  Alicante  PRIVATE INDIVIDUAL         A   4        CENTRO   \n",
       "4        5269    Madrid             COMPANY         A   7   CARABANCHEL   \n",
       "5        5275    Madrid             COMPANY         A   7   CARABANCHEL   \n",
       "6        5286  Alicante  PRIVATE INDIVIDUAL         A   4        CENTRO   \n",
       "7        5294  Alicante  PRIVATE INDIVIDUAL         A   4        CENTRO   \n",
       "8        5303  Alicante  PRIVATE INDIVIDUAL         A   4        CENTRO   \n",
       "9        5312  Alicante  PRIVATE INDIVIDUAL         A   4        CENTRO   \n",
       "\n",
       "                    Address  Nº HABIT  \\\n",
       "0       Vallehermoso 69, 3C         1   \n",
       "1              Oca 105, 2º1         1   \n",
       "2         Cartagena 10, 1ºA         1   \n",
       "3  Avenida Aguilera 18, 3ºB         1   \n",
       "4              Oca 105, 2º1         2   \n",
       "5              Oca 105, 2º1         3   \n",
       "6  Avenida Aguilera 18, 3ºB         2   \n",
       "7  Avenida Aguilera 18, 3ºB         5   \n",
       "8  Avenida Aguilera 18, 3ºB         3   \n",
       "9  Avenida Aguilera 18, 3ºB         4   \n",
       "\n",
       "                                         description Square Metres Room  ...  \\\n",
       "0  Do you want to live in one of the most charmin...               9 m2  ...   \n",
       "1  This room is ideal for students or young profe...               7.35  ...   \n",
       "2  If you are looking for a cozy and comfortable ...               16m2  ...   \n",
       "3  This exterior room is perfect for solo travele...               18m2  ...   \n",
       "4  This room is ideal for students or young profe...               7.87  ...   \n",
       "5  This room is ideal for students or young profe...               6.92  ...   \n",
       "6  This exterior room is ideal for solo travelers...               12m2  ...   \n",
       "7  This room is ideal for solo travelers who want...              6,5m2  ...   \n",
       "8  This room is perfect for solo travelers who wa...                9m2  ...   \n",
       "9  This room is ideal for solo travelers who want...                8m2  ...   \n",
       "\n",
       "          AGE PASSPORT/DNI/NIE Identification Number    Sex  Occupation  \\\n",
       "0          22              NaN                   NaN    MAN       STUDY   \n",
       "1         NaN              NaN                   NaN    NaN         NaN   \n",
       "2          22              NaN                   NaN    MAN       STUDY   \n",
       "3          23              NaN                   NaN    MAN       STUDY   \n",
       "4         NaN              NaN                   NaN    NaN         NaN   \n",
       "5         NaN              NaN                   NaN    NaN         NaN   \n",
       "6          20              NaN                   NaN  WOMAN        WORK   \n",
       "7  18/07/1997        PASAPORTE              U5528887  WOMAN       STUDY   \n",
       "8          24              NaN                   NaN    MAN       STUDY   \n",
       "9          20              NaN                   NaN    MAN       STUDY   \n",
       "\n",
       "         Nivel                        University  \\\n",
       "0      ERASMUS           UNIVERSIDAD POLITÉCNICA   \n",
       "1          NaN                               NaN   \n",
       "2      ERASMUS           UNIVERSIDAD POLITÉCNICA   \n",
       "3      ERASMUS  UNIVERSIDAD FRANCISCO DE VITORIA   \n",
       "4          NaN                               NaN   \n",
       "5          NaN                               NaN   \n",
       "6  INTERNSHIPS           UNIVERSIDAD COMPLUTENSE   \n",
       "7     GRADUATE                               NaN   \n",
       "8     GRADUATE                              ESIC   \n",
       "9     GRADUATE           UNIVERSIDAD COMPLUTENSE   \n",
       "\n",
       "                                     Telephone     Country  \n",
       "0       nicolin.herault@gmail.com  3.378953e+10     FRANCE  \n",
       "1                             NaN           NaN        NaN  \n",
       "2       nicolin.herault@gmail.com  3.378953e+10     FRANCE  \n",
       "3          thomasl1997@hotmail.fr  3.363209e+10     FRANCE  \n",
       "4                             NaN           NaN        NaN  \n",
       "5                             NaN           NaN        NaN  \n",
       "6  marielaurent.private@gmail.com  3.306515e+11     FRANCE  \n",
       "7            corinwtagg@gmail.com  3.465090e+10  ARGENTINA  \n",
       "8    elmansouri.ismail2@gmail.com  3.365897e+10     FRANCE  \n",
       "9    sim.pulcini@stud.uniroma3.it  3.933961e+11      ITALY  \n",
       "\n",
       "[10 rows x 37 columns]"
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "room_data"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'Alicante': 0, 'Madrid': 1}"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "room_data['city_id'] = room_data.city.astype('category').cat.codes\n",
    "room_data = room_data.groupby(\n",
    "    'city', group_keys=False\n",
    ").apply(lambda x: x.sample(min(len(x), 2000))).reset_index(drop=True)\n",
    "city_mapping = {\n",
    "    city_item[0][0]: city_item[0][1] for city_item in room_data[['city', 'city_id']].value_counts().items()\n",
    "}\n",
    "room_data =  room_data[\n",
    "    [\n",
    "        'listing_id', 'description', 'city', 'city_id', 'status', 'CHECK-IN', 'CHECK-OUT', 'price'\n",
    "    ]\n",
    "]\n",
    "city_mapping"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "Alicante    5\n",
       "Madrid      5\n",
       "Name: city, dtype: int64"
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "room_data['city'].value_counts()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "room_data['city_id'] = room_data.city.astype('category').cat.codes"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "city_mapping = {city_item[0][0]:city_item[0][1] for city_item in room_data[['city', 'city_id']].value_counts().items()}"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "Alicante    5\n",
       "Madrid      5\n",
       "Name: city, dtype: int64"
      ]
     },
     "execution_count": 10,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "room_data.groupby('city', group_keys=False).apply(lambda x: x.sample(min(len(x), 2000))).city.value_counts()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "room_data = room_data.groupby('city', group_keys=False).apply(lambda x: x.sample(min(len(x), 2000))).reset_index(drop=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "10"
      ]
     },
     "execution_count": 12,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "NUMBER_PRODUCTS = room_data.shape[0]\n",
    "room_data.shape[0]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "#get the first 1000 products with non-empty item keywords\n",
    "rooms_metadata = room_data.to_dict(orient='index')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{0, 1}"
      ]
     },
     "execution_count": 14,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "set([rooms_metadata[key]['city_id'] for key in rooms_metadata])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {
    "id": "gjouwTqF7ftf",
    "tags": []
   },
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "/home/erik/hackatons/chatgpt/open_3.10/lib/python3.10/site-packages/tqdm/auto.py:21: TqdmWarning: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html\n",
      "  from .autonotebook import tqdm as notebook_tqdm\n"
     ]
    }
   ],
   "source": [
    "from sentence_transformers import SentenceTransformer\n",
    "#from .autonotebook import tqdm as notebook_tqdm\n",
    "\n",
    "model = SentenceTransformer('sentence-transformers/all-distilroberta-v1')\n",
    "\n",
    "rooms_metadata = room_data.to_dict(orient='index')\n",
    "item_keywords = [\n",
    "    ', '.join(auto_truncate(rooms_metadata[i][key]) for key in [\n",
    "        'city', 'description', 'price'\n",
    "        ]\n",
    "    )\n",
    "    for i in rooms_metadata.keys()\n",
    "]\n",
    "item_keywords_vectors = [model.encode(sentence) for sentence in item_keywords]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'Alicante, This exterior room is perfect for solo travelers who want to enjoy the city center of Alicante. It is located in Avenida Aguilera 18, a lively and well-connected area with many facilities nearby.  The room also has a smart TV and a mini fridge for your entertainment and convenience. The room has a double bed, a wardrobe, a desk and a chair. You will also have access to a shared bathroom, kitchen and living room with other guests. The room does not include bed linens, towels, toiletries or complimentary refr, 400 €'"
      ]
     },
     "execution_count": 16,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "item_keywords[0]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {
    "id": "Iw7rlppY8f3a",
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'listing_id': 5286,\n",
       " 'description': 'This exterior room is ideal for solo travelers who want to relax and enjoy the view of Alicante. It is located in Avenida Aguilera 18, a vibrant and convenient area with many amenities nearby.  The room also has a smart TV and a mini fridge for your entertainment and convenience. The room has a single bed, a wardrobe, a desk and a chair. You will also have access to a shared bathroom, kitchen and living room with other guests. The room does not include bed linens, towels, toiletries or complimentary refreshments, so please bring your own or buy them locally. The room has a balcony where you can admire the cityscape and breathe some fresh air. The room is also very close to the Alicante Renfe station, which is only a few minutes away by walking or by bus. This makes it easy to travel to other destinations by train. Book your stay now and discover the charm of Alicante!',\n",
       " 'city': 'Alicante',\n",
       " 'city_id': 0,\n",
       " 'status': 'BUSY',\n",
       " 'CHECK-IN': '1/3/2022',\n",
       " 'CHECK-OUT': '30/6/2022',\n",
       " 'price': '380 €'}"
      ]
     },
     "execution_count": 17,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "len(item_keywords_vectors)\n",
    "len(rooms_metadata)\n",
    "# Check one of the products\n",
    "rooms_metadata[2]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {
    "id": "lcjo2rXb8pT9",
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Loading and Indexing + 19940 products\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "True"
      ]
     },
     "execution_count": 18,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "\n",
    "ITEM_KEYWORD_EMBEDDING_FIELD='item_keyword_vector'\n",
    "TEXT_EMBEDDING_DIMENSION=768\n",
    "NUMBER_PRODUCTS=19940\n",
    "\n",
    "print ('Loading and Indexing + ' +  str(NUMBER_PRODUCTS) + ' products')\n",
    "\n",
    "#flush all data\n",
    "redis_conn.flushall()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "/home/erik/hackatons/chatgpt/open_3.10/lib/python3.10/site-packages/langchain/llms/openai.py:169: UserWarning: You are trying to use a chat model. This way of initializing it is no longer supported. Instead, please use: `from langchain.chat_models import ChatOpenAI`\n",
      "  warnings.warn(\n",
      "/home/erik/hackatons/chatgpt/open_3.10/lib/python3.10/site-packages/langchain/llms/openai.py:608: UserWarning: You are trying to use a chat model. This way of initializing it is no longer supported. Instead, please use: `from langchain.chat_models import ChatOpenAI`\n",
      "  warnings.warn(\n"
     ]
    }
   ],
   "source": [
    "import openai\n",
    "import logging\n",
    "\n",
    "from langchain.prompts import PromptTemplate\n",
    "from langchain.llms import OpenAI\n",
    "from langchain.chains import LLMChain\n",
    "\n",
    "MODEL = 'gpt-3.5-turbo'\n",
    "openai_api_key = \"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\"\n",
    "openai.api_key = \"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\"\n",
    "\n",
    "llm = OpenAI(\n",
    "    model_name=MODEL,\n",
    "    temperature=0.3,\n",
    "    openai_api_key=openai.api_key\n",
    ")\n",
    "\n",
    "booking_prompt = PromptTemplate(\n",
    "    input_variables=[\"product_description\"],\n",
    "    template=\"Create comma seperated product keywords to perform a query on a airbnb dataset for this user input: \"\n",
    "             \"{product_description}\",\n",
    ")\n",
    "\n",
    "\n",
    "def get_booking_chain():\n",
    "    logging.info('Building the chatgpt booking assistant object.')\n",
    "    return LLMChain(llm=llm, prompt=booking_prompt)\n",
    "\n",
    "\n",
    "booking_chain = get_booking_chain()\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 27,
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "import logging\n",
    "\n",
    "import numpy as np\n",
    "import redis\n",
    "\n",
    "from redis.commands.search.field import TextField, VectorField, NumericField\n",
    "from redis.commands.search.query import Query, NumericFilter\n",
    "\n",
    "\n",
    "def get_redis_connector(host='localhost', port=6379, password=''):\n",
    "    logging.warning('Connecting to redis...')\n",
    "    try:\n",
    "        return redis.Redis(\n",
    "            host=host,\n",
    "            port=port,\n",
    "            password=password\n",
    "        )\n",
    "    except Exception as e:\n",
    "        logging.error(e, exc_info=True)\n",
    "\n",
    "\n",
    "def get_booking_query(topK, city_code):\n",
    "    logging.warning('Querying redis...')\n",
    "    if city_code:\n",
    "        city_code = int(city_code)\n",
    "        return Query(\n",
    "            f'*=>[KNN {topK} @item_keyword_vector $vec_param AS vector_score]'\n",
    "        ).add_filter(\n",
    "            NumericFilter('city_id', city_code, city_code),\n",
    "        ).sort_by('vector_score').paging(0, topK).return_fields(\n",
    "            'vector_score', 'description', 'city', 'city_id', 'price'\n",
    "        ).dialect(2)\n",
    "    else:\n",
    "        return Query(\n",
    "            f'*=>[KNN {topK} @item_keyword_vector $vec_param AS vector_score]'\n",
    "        ).sort_by('vector_score').paging(0, topK).return_fields(\n",
    "            'vector_score', 'description', 'city', 'city_id', 'price'\n",
    "        ).dialect(2)\n",
    "\n",
    "\n",
    "def create_flat_index(redis_conn, vector_field_name, number_of_vectors, vector_dimensions=512, distance_metric='L2'):\n",
    "    logging.warning(f'Generating vector index for `{number_of_vectors}` records.')\n",
    "    redis_conn.ft().create_index([\n",
    "        VectorField(\n",
    "            vector_field_name, \"FLAT\",\n",
    "            {\n",
    "                \"TYPE\": \"FLOAT32\",\n",
    "                \"DIM\": vector_dimensions,\n",
    "                \"DISTANCE_METRIC\": distance_metric,\n",
    "                \"INITIAL_CAP\": number_of_vectors,\n",
    "                \"BLOCK_SIZE\": number_of_vectors\n",
    "            }\n",
    "        ),\n",
    "        TextField(\"description\", as_name='description'),\n",
    "        TextField(\"city\", as_name='city'),\n",
    "        NumericField(\"city_id\", as_name='city_id'),\n",
    "        TextField(\"price\", as_name='price')\n",
    "    ])\n",
    "\n",
    "\n",
    "def load_vectors(client, product_metadata, vector_dict, vector_field_name):\n",
    "    logging.warning('Loading vectors to redis...')\n",
    "    p = client.pipeline(transaction=False)\n",
    "    for index in product_metadata.keys():\n",
    "        # hash key\n",
    "        key = 'listing_id: ' + str(product_metadata[index]['listing_id'])\n",
    "\n",
    "        # hash values\n",
    "        item_metadata = product_metadata[index]\n",
    "        item_keywords_vector = vector_dict[index].astype(np.float32).tobytes()\n",
    "        item_metadata[vector_field_name] = item_keywords_vector\n",
    "\n",
    "        # HSET\n",
    "        p.hset(key, mapping=item_metadata)\n",
    "\n",
    "    p.execute()\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 28,
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "WARNING:root:Generating vector index for `10` records.\n",
      "WARNING:root:Loading vectors to redis...\n"
     ]
    }
   ],
   "source": [
    "#create flat index & load vectors\n",
    "create_flat_index(\n",
    "    redis_conn, ITEM_KEYWORD_EMBEDDING_FIELD,room_data.shape[0],TEXT_EMBEDDING_DIMENSION,'COSINE'\n",
    ")\n",
    "load_vectors(redis_conn,rooms_metadata,item_keywords_vectors,ITEM_KEYWORD_EMBEDDING_FIELD)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 29,
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{0, 1}"
      ]
     },
     "execution_count": 29,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "set([rooms_metadata[key]['city_id'] for key in rooms_metadata])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 30,
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'Alicante': 0, 'Madrid': 1}"
      ]
     },
     "execution_count": 30,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "city_mapping"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 32,
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "WARNING:root:Querying redis...\n",
      "WARNING:root:Result{3 total, docs: [Document {'id': 'listing_id: 5312', 'payload': None, 'vector_score': '0.357102572918', 'description': 'This room is ideal for solo travelers who want to stay in a simple and affordable space. It is located in Avenida Aguilera 18, a lively and well-connected area with many facilities nearby. The room has a single bed, a wardrobe, a desk and a chair. You will also have access to a shared bathroom, kitchen and living room with other guests. The room does not include bed linens, towels, toiletries or complimentary refreshments, so please bring your own or buy them locally. The room is interior and has window to the atrium, but it is clean and cozy.  The room also has a smart TV and a mini fridge for your entertainment and convenience. The room is also very close to the Alicante Renfe station, which is only a few minutes away by walking or by bus. This makes it easy to travel to other destinations by train. Book your stay now and enjoy the best of Alicante!', 'city': 'Alicante', 'city_id': '0', 'price': '330 €'}, Document {'id': 'listing_id: 5294', 'payload': None, 'vector_score': '0.359502673149', 'description': 'This room is ideal for solo travelers who want to stay in a simple and affordable space. It is located in Avenida Aguilera 18, a lively and well-connected area with many facilities nearby. The room has a single bed, a wardrobe, a desk and a chair. You will also have access to a shared bathroom, kitchen and living room with other guests. The room does not include bed linens, towels, toiletries or complimentary refreshments, so please bring your own or buy them locally. The room is interior and has window to the atrium, but it is clean and cozy.  The room also has a smart TV and a mini fridge for your entertainment and convenience. The room is also very close to the Alicante Renfe station, which is only a few minutes away by walking or by bus. This makes it easy to travel to other destinations by train. Book your stay now and enjoy the best of Alicante!', 'city': 'Alicante', 'city_id': '0', 'price': '300 €'}, Document {'id': 'listing_id: 5286', 'payload': None, 'vector_score': '0.363385975361', 'description': 'This exterior room is ideal for solo travelers who want to relax and enjoy the view of Alicante. It is located in Avenida Aguilera 18, a vibrant and convenient area with many amenities nearby.  The room also has a smart TV and a mini fridge for your entertainment and convenience. The room has a single bed, a wardrobe, a desk and a chair. You will also have access to a shared bathroom, kitchen and living room with other guests. The room does not include bed linens, towels, toiletries or complimentary refreshments, so please bring your own or buy them locally. The room has a balcony where you can admire the cityscape and breathe some fresh air. The room is also very close to the Alicante Renfe station, which is only a few minutes away by walking or by bus. This makes it easy to travel to other destinations by train. Book your stay now and discover the charm of Alicante!', 'city': 'Alicante', 'city_id': '0', 'price': '380 €'}]}\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "*=>[KNN 3 @item_keyword_vector $vec_param AS vector_score]\n"
     ]
    }
   ],
   "source": [
    "#keywords = booking_chain.run(\n",
    "#    user_input + ', in city: ' + context_entities['CITY'] + ', these are reservation specification: ' + str(context_entities)\n",
    "#)\n",
    "#logging.warning(user_input + ', in city: ' + context_entities['CITY'] + ', these are reservation specification: ' + str(context_entities))\n",
    "keywords = 'Alicante, accommodation, balcony,budget Cheap'\n",
    "\n",
    "top_k = 3\n",
    "# vectorize the query\n",
    "query_vector = model.encode(keywords).astype(np.float32).tobytes()\n",
    "params_dict = {\"vec_param\": query_vector}\n",
    "\n",
    "query = get_booking_query(top_k, 0)\n",
    "print(query.query_string())\n",
    "\n",
    "# Execute the query\n",
    "results = redis_conn.ft().search(query, query_params=params_dict)\n",
    "logging.warning(results)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 33,
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "Result{3 total, docs: [Document {'id': 'listing_id: 5312', 'payload': None, 'vector_score': '0.357102572918', 'description': 'This room is ideal for solo travelers who want to stay in a simple and affordable space. It is located in Avenida Aguilera 18, a lively and well-connected area with many facilities nearby. The room has a single bed, a wardrobe, a desk and a chair. You will also have access to a shared bathroom, kitchen and living room with other guests. The room does not include bed linens, towels, toiletries or complimentary refreshments, so please bring your own or buy them locally. The room is interior and has window to the atrium, but it is clean and cozy.  The room also has a smart TV and a mini fridge for your entertainment and convenience. The room is also very close to the Alicante Renfe station, which is only a few minutes away by walking or by bus. This makes it easy to travel to other destinations by train. Book your stay now and enjoy the best of Alicante!', 'city': 'Alicante', 'city_id': '0', 'price': '330 €'}, Document {'id': 'listing_id: 5294', 'payload': None, 'vector_score': '0.359502673149', 'description': 'This room is ideal for solo travelers who want to stay in a simple and affordable space. It is located in Avenida Aguilera 18, a lively and well-connected area with many facilities nearby. The room has a single bed, a wardrobe, a desk and a chair. You will also have access to a shared bathroom, kitchen and living room with other guests. The room does not include bed linens, towels, toiletries or complimentary refreshments, so please bring your own or buy them locally. The room is interior and has window to the atrium, but it is clean and cozy.  The room also has a smart TV and a mini fridge for your entertainment and convenience. The room is also very close to the Alicante Renfe station, which is only a few minutes away by walking or by bus. This makes it easy to travel to other destinations by train. Book your stay now and enjoy the best of Alicante!', 'city': 'Alicante', 'city_id': '0', 'price': '300 €'}, Document {'id': 'listing_id: 5286', 'payload': None, 'vector_score': '0.363385975361', 'description': 'This exterior room is ideal for solo travelers who want to relax and enjoy the view of Alicante. It is located in Avenida Aguilera 18, a vibrant and convenient area with many amenities nearby.  The room also has a smart TV and a mini fridge for your entertainment and convenience. The room has a single bed, a wardrobe, a desk and a chair. You will also have access to a shared bathroom, kitchen and living room with other guests. The room does not include bed linens, towels, toiletries or complimentary refreshments, so please bring your own or buy them locally. The room has a balcony where you can admire the cityscape and breathe some fresh air. The room is also very close to the Alicante Renfe station, which is only a few minutes away by walking or by bus. This makes it easy to travel to other destinations by train. Book your stay now and discover the charm of Alicante!', 'city': 'Alicante', 'city_id': '0', 'price': '380 €'}]}"
      ]
     },
     "execution_count": 33,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "results"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 118,
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "['*=>[KNN 3 @item_keyword_vector $vec_param AS vector_score]',\n",
       " 'RETURN',\n",
       " 5,\n",
       " 'vector_score',\n",
       " 'property_type',\n",
       " 'name',\n",
       " 'amenities',\n",
       " 'city',\n",
       " 'SORTBY',\n",
       " 'vector_score',\n",
       " 'ASC',\n",
       " 'DIALECT',\n",
       " 2,\n",
       " 'LIMIT',\n",
       " 0,\n",
       " 3]"
      ]
     },
     "execution_count": 118,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "q.get_args()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 53,
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "full_result_string = ''\n",
    "for product in results.docs:\n",
    "    full_result_string += ' '.join(\n",
    "        [\n",
    "            product.property_type, product.name, f\", amenities are:\", product.amenities,\" Located in city:\", product.city,\n",
    "            'ID of this booking is:', product.id,\n",
    "            \"\\n\\n\\n\"\n",
    "        ]\n",
    "    )\n",
    "\n",
    "\n",
    "from langchain.memory import ConversationBufferMemory\n",
    "\n",
    "BOOKING_AGENT_TEMPLATE = \"\"\"\n",
    "You are a room booking assistant. Be kind, detailed and try to sell the booking of the apartment to me. \n",
    "Present the three given queried search result in a nice way as answer to the user input.\n",
    "\n",
    "Collect entities after user confirms booking choice, we need these entities: booking_date_start, booking_date_end, name, surname. \n",
    "\n",
    "dont ask questions back! \n",
    "\n",
    "ALWAYS PROVIDE RESPONSE AS JSON STRING with keys:\n",
    "ANSWER, BOOKING_CONFIRMED, CONFIRMED_BOOKING_DATES, NAME and the ID of this.\n",
    "\n",
    "PLEASE ALWAYS HAVE THE JSON STRING PART\n",
    "\n",
    "\n",
    "\n",
    "{chat_history}\n",
    "Human: {user_msg}\n",
    "Chatbot:\"\"\"\n",
    "\n",
    "\n",
    "BOOKING_PROMPT = PromptTemplate(\n",
    "    input_variables=[\"chat_history\", \"user_msg\"],\n",
    "    template=BOOKING_AGENT_TEMPLATE\n",
    ")\n",
    "\n",
    "memory = ConversationBufferMemory(memory_key=\"chat_history\")\n",
    "\n",
    "llm_chain = LLMChain(\n",
    "    llm=OpenAI(model_name=\"gpt-3.5-turbo\", temperature=0.8, openai_api_key=openai_api_key), \n",
    "    prompt=BOOKING_PROMPT, \n",
    "    verbose=False, \n",
    "    memory=memory,\n",
    ")\n",
    "\n",
    "\n",
    "answer = llm_chain.predict_and_parse(user_msg= f\"{full_result_string} ---\\n\\n {userinput}\")\n",
    "print(\"Bot:\", answer)\n",
    "time.sleep(0.5)\n",
    "\n",
    "while True:\n",
    "    follow_up = input(\"\")\n",
    "    print(\"User:\", follow_up)\n",
    "    answer = llm_chain.predict(\n",
    "        user_msg=follow_up\n",
    "    )\n",
    "    print(\"Bot:\", answer)\n",
    "    time.sleep(0.5)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "answer"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "print(re.findall(r'{.+}',answer)[0])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "import re"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "json_string = re"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "json.loads(json_string)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "BOOKING_AGENT_GET_INFO_TEMPLATE = \"\"\"\n",
    "You are a room booking assistant. Be nice and helpful and get these informations from the customer:\n",
    "BOOKING_START, BOOKING_END, FULL_NAME, CITY, BUDGET, GUEST_COUNT\n",
    "\n",
    "Example of how such conversation may go, only generate responses to the users last question:\n",
    "Human: Hi, I'm looking to book a room in CITY.\n",
    "assistant: Great, I can help with that. What's your budget?\n",
    "Human: I'm looking to spend around BUDGET euros per month.\n",
    "assistant: Okay, I've found a few options that fit your budget. Would you prefer to be in the city center or in a quieter area?\n",
    "Human: I want to be close to the main attractions.\n",
    "assistant: Got it. How many people will be staying in the room?\n",
    "Human: It's just me.\n",
    "\n",
    "\n",
    "Collect this information and ALWAYS ADD JSON STRING AT THE END OF THE RESPONSE AS JSON STRING with keys:\n",
    "BOOKING_START, BOOKING_END, FULL_NAME, CITY, BUDGET, GUEST_COUNT\n",
    "\n",
    "PLEASE ALWAYS HAVE THE JSON STRING PART\n",
    "\n",
    "{chat_history}\n",
    "Human: {user_msg}\n",
    "Chatbot:\"\"\"\n",
    "\n",
    "BOOKING_AGENT_TEMPLATE = \"\"\"\n",
    "You are a room booking assistant. Be kind, detailed and try to sell the booking of the apartment to me. \n",
    "Present the three given queried search result in a nice way as answer to the user input.\n",
    "\n",
    "Collect entities when user confirms booking choice, we need these entities: CONFIRMED_BOOKING_START, CONFIRMED_BOOKING_END, FULL_NAME\n",
    "\n",
    "ALWAYS ADD JSON STRING AT THE END OF THE RESPONSE, ALWAYS PROVIDE RESPONSE AS JSON STRING with keys:\n",
    "ANSWER, BOOKING_CONFIRMED, CONFIRMED_BOOKING_START, CONFIRMED_BOOKING_END, NAME and the ID of this.\n",
    "\n",
    "PLEASE ALWAYS HAVE THE JSON STRING PART\n",
    "\n",
    "\n",
    "{chat_history}\n",
    "Human: {user_msg}\n",
    "Chatbot:\"\"\"\n",
    "\n",
    "BOOKING_PROMPT_SELL = PromptTemplate(\n",
    "    input_variables=[\"chat_history\", \"user_msg\"],\n",
    "    template=BOOKING_AGENT_TEMPLATE\n",
    ")\n",
    "\n",
    "BOOKING_PROMPT_ASK = PromptTemplate(\n",
    "    input_variables=[\"chat_history\", \"user_msg\"],\n",
    "    template=BOOKING_AGENT_GET_INFO_TEMPLATE\n",
    ")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "def booking_get_requirement_info(userinput):\n",
    "    memory = ConversationBufferMemory(memory_key=\"chat_history\")\n",
    "    llm_chain = LLMChain(\n",
    "        llm=OpenAI(model_name=\"gpt-3.5-turbo\", temperature=0.8,\n",
    "                   openai_api_key=openai_api_key),\n",
    "        prompt=BOOKING_PROMPT_ASK,\n",
    "        verbose=False,\n",
    "        memory=memory,\n",
    "    )\n",
    "\n",
    "    answer = llm_chain.predict(user_msg=f\"{userinput}\")\n",
    "    return \"Bot:\", answer"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "booking_get_requirement_info('I wanna book a room in Madrid')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "colab": {
   "provenance": []
  },
  "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.10.6"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}