Spaces:
Running
Running
Increased number of contexts
Browse files
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=
|
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 |
|