templates: fix rag-lancedb template (#18551)

This commit is contained in:
Akash A Desai 2024-03-05 08:26:16 +05:30 committed by GitHub
parent 25c7d52140
commit eb0756f3ee
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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()
"""