Spaces:
Sleeping
Sleeping
Update rag.py
Browse filesReduced number of context
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=
|
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 |
|