alexandraroze commited on
Commit
6155057
·
1 Parent(s): 191c2df

fixed docs

Browse files
Files changed (1) hide show
  1. src/rag.py +2 -1
src/rag.py CHANGED
@@ -146,7 +146,8 @@ class FaissDB:
146
  def similarity_search(self, query: str, k: int = 5):
147
  if self.index is None:
148
  raise ValueError("Index is not initialized")
149
- return self.index.similarity_search(query, k)
 
150
 
151
 
152
  class AICompletion:
 
146
  def similarity_search(self, query: str, k: int = 5):
147
  if self.index is None:
148
  raise ValueError("Index is not initialized")
149
+ documents = self.index.similarity_search(query, k)
150
+ return [doc.page_content for doc in documents]
151
 
152
 
153
  class AICompletion: