hanzla commited on
Commit
a127868
·
1 Parent(s): d9640db
Files changed (1) hide show
  1. src/pdfchatbot.py +1 -1
src/pdfchatbot.py CHANGED
@@ -138,7 +138,7 @@ class PDFChatBot:
138
  )
139
 
140
  def get_organic_context(self, query):
141
- documents = self.vectordb.similarity_search_with_relevance_scores(query, k=self.k)
142
  context = self.format_seperator.join([doc.page_content for doc, score in documents])
143
  self.current_context = context
144
  print(self.current_context)
 
138
  )
139
 
140
  def get_organic_context(self, query):
141
+ documents = self.vectordb.similarity_search_with_relevance_scores(query, k=3)
142
  context = self.format_seperator.join([doc.page_content for doc, score in documents])
143
  self.current_context = context
144
  print(self.current_context)