crodri commited on
Commit
35ed935
·
verified ·
1 Parent(s): 207ffeb

Update rag.py

Browse files
Files changed (1) hide show
  1. rag.py +3 -3
rag.py CHANGED
@@ -139,9 +139,9 @@ 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["Fuente"] + "\n\n"
143
- full_context += doc[0].metadata["url"] + "\n\n"
144
- source_context.append(doc[0].metadata["url"])
145
 
146
  return text_context, full_context, source_context
147
 
 
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["source"] + "\n\n"
143
+ #full_context += doc[0].metadata["url"] + "\n\n"
144
+ source_context.append(doc[0].metadata["source"])
145
 
146
  return text_context, full_context, source_context
147