acumplido commited on
Commit
015c03e
1 Parent(s): 43728a5

Update rag.py

Browse files

Reduced number of context

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