templates: fix rag-lancedb template (#18551)

langchain-ai · Mar 5, 2024 · eb0756f · eb0756f
1 parent 25c7d52
commit eb0756f
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/templates/rag-lancedb/rag_lancedb/chain.py b/templates/rag-lancedb/rag_lancedb/chain.py
@@ -19,10 +19,8 @@
 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()
 """