templates: fix rag-lancedb template (#18551)

3 months ago · eb0756f3ee
parent 25c7d52140
commit eb0756f3ee
1 changed files with 2 additions and 4 deletions
--- 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()
 """