Added comments to notebook

notebooks/04-RAG_with_VectorStore.ipynb

@@ -333,7 +333,7 @@
       "outputs": [],
       "source": [
         "from langchain.schema.document import Document\n",
-        "# Convert the chunks to Document objects so the LlamaIndex framework can process them.\n",
+        "# Convert the chunks to Document objects so the LangChain framework can process them.\n",
         "documents = [Document(page_content=t) for t in chunks]"
       ]
     },
@@ -356,9 +356,7 @@
       "source": [
         "from langchain_chroma import Chroma\n",
         "from langchain_openai import OpenAIEmbeddings\n",
-        "# create client and a new collection\n",
-        "# chromadb.EphemeralClient saves data in-memory.\n",
-        "# Add the documents to the database and create Index / embeddings\n",
+        "# Add the documents to chroma DB and create Index / embeddings\n",
         "\n",
         "embeddings = OpenAIEmbeddings(model=\"text-embedding-ada-002\")\n",
         "chroma_db = Chroma.from_documents(\n",
@@ -387,9 +385,7 @@
       "outputs": [],
       "source": [
         "from langchain_openai import ChatOpenAI\n",
-        "# Define a query engine that is responsible for retrieving related pieces of text,\n",
-        "# and using a LLM to formulate the final answer.\n",
-        "\n",
+        "# Initializing the LLM model\n",
         "llm = ChatOpenAI(temperature=0, model=\"gpt-3.5-turbo-0125\", max_tokens=512)"
       ]
     },
@@ -416,6 +412,8 @@
         "from langchain.chains import RetrievalQA\n",
         "query = \"How many parameters LLaMA2 model has?\"\n",
         "retriever = chroma_db.as_retriever(search_kwargs={\"k\": 2})\n",
+        "# Define a RetrievalQA chain that is responsible for retrieving related pieces of text,\n",
+        "# and using a LLM to formulate the final answer.\n",
         "chain = RetrievalQA.from_chain_type(llm=llm,\n",
         "                                    chain_type=\"stuff\",\n",
         "                                    retriever=retriever)\n",