DylanonWic commited on
Commit
ab4ba0d
1 Parent(s): c2c08c4

Update tools.py

Browse files
Files changed (1) hide show
  1. tools.py +1 -1
tools.py CHANGED
@@ -166,7 +166,7 @@ def get_retriver_from_docs(docs):
166
  splits = text_splitter.split_documents(docs)
167
 
168
  # Text Vectorization.
169
- vectorstore = Chroma.from_documents(documents=splits)
170
 
171
  # Retrieve and generate using the relevant snippets of the blog.
172
  retriever = vectorstore.as_retriever()
 
166
  splits = text_splitter.split_documents(docs)
167
 
168
  # Text Vectorization.
169
+ vectorstore = Chroma.from_documents(documents=splits, embedding=OpenAIEmbeddings())
170
 
171
  # Retrieve and generate using the relevant snippets of the blog.
172
  retriever = vectorstore.as_retriever()