Replaces session manager with message history (#94)

Author: justin-cechmanek

python-recipes/RAG/01_redisvl.ipynb

@@ -1869,14 +1869,14 @@
    "outputs": [],
    "source": [
     "from functools import wraps\n",
-    "from redisvl.extensions.session_manager import StandardSessionManager\n",
+    "from redisvl.extensions.message_history import MessageHistory\n",
     "\n",
     "\n",
     "class ChatBot:\n",
     "    def __init__(self, index: AsyncSearchIndex, vectorizer: BaseVectorizer, user: str):\n",
     "        self.index = index\n",
     "        self.vectorizer = vectorizer\n",
-    "        self.session_manager = StandardSessionManager(\n",
+    "        self.history = MessageHistory(\n",
     "            name=f\"chat_session_{user}\",\n",
     "            session_tag=user,\n",
     "            redis_url=REDIS_URL,\n",
@@ -1915,7 +1915,7 @@
     "\n",
     "    async def clear_history(self):\n",
     "        \"\"\"Clear session chat\"\"\"\n",
-    "        self.session_manager.clear()\n",
+    "        self.history.clear()\n",
     "\n",
     "    async def answer_question(self, query: str):\n",
     "        \"\"\"Answer the user's question with historical context and caching baked-in\"\"\"\n",
@@ -1933,7 +1933,7 @@
     "            answer = result[0]['response']\n",
     "        else:\n",
     "            context = await self.retrieve_context(query_vector)\n",
-    "            session = self.session_manager.messages\n",
+    "            session = self.history.messages\n",
     "            messages = (\n",
     "                [{\"role\": \"system\", \"content\": SYSTEM_PROMPT}] +\n",
     "                session +\n",
@@ -1950,7 +1950,7 @@
     "            llmcache.store(query, answer, query_vector)\n",
     "\n",
     "        # Add message history\n",
-    "        self.session_manager.add_messages([\n",
+    "        self.history.add_messages([\n",
     "            {\"role\": \"user\", \"content\": query},\n",
     "            {\"role\": \"assistant\", \"content\": answer}\n",
     "        ])\n",
@@ -2037,7 +2037,7 @@
    ],
    "source": [
     "# NBVAL_SKIP\n",
-    "chat.session_manager.messages"
+    "chat.history.messages"
    ]
   },
   {
@@ -2102,7 +2102,7 @@
    "provenance": []
   },
   "kernelspec": {
-   "display_name": "env",
+   "display_name": "redis-ai-res",
    "language": "python",
    "name": "python3"
   },
@@ -2116,7 +2116,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.11"
+   "version": "3.11.9"
   }
  },
  "nbformat": 4,