crodri commited on
Commit
f0a0564
·
verified ·
1 Parent(s): 212976b

Update rag.py

Browse files
Files changed (1) hide show
  1. rag.py +2 -2
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 = "index-BAAI_bge-m3-1500-200-recursive_splitter-CA_ES_UE"
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["Títol de la norma"] + "\n\n"
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