Arno' Francesco (GDS DS&G) commited on
Commit
53820cc
·
1 Parent(s): d6b8788
Files changed (1) hide show
  1. retriever.py +5 -3
retriever.py CHANGED
@@ -34,11 +34,13 @@ class GuestInfoRetrieverTool(Tool):
34
 
35
  #self.model = model
36
 
37
- docs_for_embeddings = ['passage: '+d.page_content for d in docs]
38
-
 
 
39
  #embeddings = self.model.encode(docs_for_embeddings, normalize_embeddings=True)
40
 
41
- self.vector_store = FAISS.from_documents(docs_for_embeddings, hf)
42
  self.faiss_retriever = self.vector_store.as_retriever(search_kwargs={"k": 5})
43
 
44
 
 
34
 
35
  #self.model = model
36
 
37
+ #docs_for_embeddings = ['passage: '+d.page_content for d in docs]
38
+
39
+ for d in docs:
40
+ d.page_content = 'passage: '+d.page_content
41
  #embeddings = self.model.encode(docs_for_embeddings, normalize_embeddings=True)
42
 
43
+ self.vector_store = FAISS.from_documents(docs, hf)
44
  self.faiss_retriever = self.vector_store.as_retriever(search_kwargs={"k": 5})
45
 
46