Update rag.py
Browse files
rag.py
CHANGED
@@ -15,7 +15,7 @@ class RAG:
|
|
15 |
|
16 |
#vectorstore = "index-intfloat_multilingual-e5-small-500-100-CA-ES" # mixed
|
17 |
#vectorstore = "vectorestore" # CA only
|
18 |
-
vectorstore = "
|
19 |
|
20 |
def __init__(self, hf_token, embeddings_model, model_name, rerank_model, rerank_number_contexts):
|
21 |
|
@@ -139,7 +139,7 @@ class RAG:
|
|
139 |
for doc in docs:
|
140 |
text_context += doc[0].page_content
|
141 |
full_context += doc[0].page_content + "\n"
|
142 |
-
full_context += doc[0].metadata["
|
143 |
full_context += doc[0].metadata["url"] + "\n\n"
|
144 |
source_context.append(doc[0].metadata["url"])
|
145 |
|
|
|
15 |
|
16 |
#vectorstore = "index-intfloat_multilingual-e5-small-500-100-CA-ES" # mixed
|
17 |
#vectorstore = "vectorestore" # CA only
|
18 |
+
vectorstore = " faiss_index.bin"
|
19 |
|
20 |
def __init__(self, hf_token, embeddings_model, model_name, rerank_model, rerank_number_contexts):
|
21 |
|
|
|
139 |
for doc in docs:
|
140 |
text_context += doc[0].page_content
|
141 |
full_context += doc[0].page_content + "\n"
|
142 |
+
full_context += doc[0].metadata["Fuente"] + "\n\n"
|
143 |
full_context += doc[0].metadata["url"] + "\n\n"
|
144 |
source_context.append(doc[0].metadata["url"])
|
145 |
|