update

Author: TheovanKraay

Python/CosmosDB-NoSQL_VectorSearch/CosmosDB-NoSQL-Quickstart-RAG-Chatbot.ipynb

@@ -471,26 +471,23 @@
     "    return response.model_dump()\n",
     "\n",
     "def chat_completion(cache_container, movies_container, user_input):\n",
-    "\n",
+    "    print(\"starting completion\")\n",
     "    # Generate embeddings from the user input\n",
     "    user_embeddings = generate_embeddings(user_input)\n",
     "    # Query the chat history cache first to see if this question has been asked before\n",
-    "    cache_results = vector_search(container = cache_container, vectors = user_embeddings, similarity_score=0.99, num_results=1)\n",
-    "\n",
+    "    cache_results = get_cache(container = cache_container, vectors = user_embeddings, similarity_score=0.99, num_results=1)\n",
     "    if len(cache_results) > 0:\n",
-    "        print(\"\\n Cached Result\\n\")\n",
-    "        return cache_results[0]['document']['completion']\n",
+    "        print(\"Cached Result\\n\")\n",
+    "        return cache_results[0]['completion'], True\n",
     "        \n",
     "    else:\n",
-    "    \n",
     "        #perform vector search on the movie collection\n",
-    "        print(\"\\n New result\\n\")\n",
+    "        print(\"New result\\n\")\n",
     "        search_results = vector_search(movies_container, user_embeddings)\n",
     "\n",
     "        print(\"Getting Chat History\\n\")\n",
     "        #chat history\n",
     "        chat_history = get_chat_history(cache_container, 3)\n",
-    "\n",
     "        #generate the completion\n",
     "        print(\"Generating completions \\n\")\n",
     "        completions_results = generate_completion(user_input, search_results, chat_history)\n",
@@ -501,7 +498,7 @@
     "\n",
     "        print(\"\\n\")\n",
     "        # Return the generated LLM completion\n",
-    "        return completions_results['choices'][0]['message']['content'] "
+    "        return completions_results['choices'][0]['message']['content'], False"
    ]
   },
   {