Spaces:
Runtime error
Runtime error
Alexandre
commited on
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 |
|