acumplido commited on
Commit
708527c
1 Parent(s): 2daa8fa

Increased number of contexts

Browse files
Files changed (1) hide show
  1. rag.py +1 -1
rag.py CHANGED
@@ -27,7 +27,7 @@ class RAG:
27
 
28
  logging.info("RAG loaded!")
29
 
30
- def get_context(self, instruction, number_of_contexts=1):
31
 
32
  documentos = self.vectore_store.similarity_search_with_score(instruction, k=number_of_contexts)
33
 
 
27
 
28
  logging.info("RAG loaded!")
29
 
30
+ def get_context(self, instruction, number_of_contexts=2):
31
 
32
  documentos = self.vectore_store.similarity_search_with_score(instruction, k=number_of_contexts)
33