Spaces:

MKCL
/

Freeekyyy-chatBot

Sleeping

MKCL commited on Apr 29

Commit

5ae483b

verified ·

1 Parent(s): 4e32bcc

Update rag.py

Files changed (1) hide show

rag.py CHANGED Viewed

@@ -25,4 +25,4 @@ def transform( ingested_docs):
 def vectorStoreAndEmbeddings(docs, query):
     embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
     db = Chroma.from_documents(documents=docs, embedding=embeddings)
-    return db.similarity_search(query)[0].page_content

 def vectorStoreAndEmbeddings(docs, query):
     embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
     db = Chroma.from_documents(documents=docs, embedding=embeddings)
+    return db.similarity_search(query)