From eb0756f3eebed169eb90a297d3392a7561acee0d Mon Sep 17 00:00:00 2001
From: Akash A Desai <62583018+akashAD98@users.noreply.github.com>
Date: Tue, 5 Mar 2024 08:26:16 +0530
Subject: [PATCH] templates: fix rag-lancedb template (#18551)

---
 templates/rag-lancedb/rag_lancedb/chain.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/templates/rag-lancedb/rag_lancedb/chain.py b/templates/rag-lancedb/rag_lancedb/chain.py
index 9ba07e3664f..856d65f0f22 100644
--- a/templates/rag-lancedb/rag_lancedb/chain.py
+++ b/templates/rag-lancedb/rag_lancedb/chain.py
@@ -19,10 +19,8 @@ text_splitter = RecursiveCharacterTextSplitter(chunk_size=500, chunk_overlap=0)
 all_splits = text_splitter.split_documents(data)
 
 # Add to vectorDB
-vectorstore = LanceDB.from_documents(documents=all_splits, 
-                                    collection_name="rag-chroma",
-                                    embedding=OpenAIEmbeddings(),
-                                    )
+vectorstore = LanceDB.from_documents(documents=all_splits,
+                             embedding=OpenAIEmbeddings())
 retriever = vectorstore.as_retriever()
 """