Daniel00611 commited on
Commit
1bf817a
verified
1 Parent(s): 80b5efc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,10 +25,10 @@ vectorstore = Chroma(
25
  # Crear un retriever
26
  retriever = vectorstore.as_retriever()
27
 
 
28
  def obtener_extractos(pregunta):
29
- """Recupera documentos relevantes desde ChromaDB."""
30
  docs_relevantes = retriever.invoke(pregunta)
31
- return "\n".join([doc.page_content for doc in docs_relevantes]) if docs_relevantes else "No se encontraron documentos relevantes."
32
 
33
  def respond(message, history: list[tuple[str, str]], system_message, max_tokens, temperature, top_p):
34
  """Genera una respuesta basada en el historial y documentos relevantes."""
 
25
  # Crear un retriever
26
  retriever = vectorstore.as_retriever()
27
 
28
+ # Funci贸n para obtener extractos relevantes
29
  def obtener_extractos(pregunta):
 
30
  docs_relevantes = retriever.invoke(pregunta)
31
+ return [(doc.page_content, doc.metadata.get("url", "URL no disponible")) for doc in docs_relevantes]
32
 
33
  def respond(message, history: list[tuple[str, str]], system_message, max_tokens, temperature, top_p):
34
  """Genera una respuesta basada en el historial y documentos relevantes."""