joortif commited on
Commit
946d2c8
·
verified ·
1 Parent(s): b9eeb67

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -50,13 +50,14 @@ def test_rag_reranking(query, ranker):
50
 
51
  print(f"📚 Contextos pasados al ranker: {len(context)}")
52
 
53
- # Rankeamos (CORREGIDO: argumentos posicionales)
54
- reranked = ranker.rank(query, context, 1)
 
55
  print(f"🏅 Resultado del reranker: {reranked}")
56
 
57
- # Usa 'text' o 'content' según lo que haya
58
- best_context = reranked[0].get("text", reranked[0].get("content", ""))
59
- print(f"🧠 Contexto elegido: {best_context[:300]}...") # Muestra solo los primeros 300 caracteres
60
 
61
  respuesta = generate_text(best_context, query)
62
  print(f"💬 Respuesta generada: {respuesta}")
@@ -64,6 +65,7 @@ def test_rag_reranking(query, ranker):
64
  return respuesta
65
 
66
 
 
67
  def responder_chat(message, history):
68
  respuesta = test_rag_reranking(message, ranker)
69
  return respuesta
 
50
 
51
  print(f"📚 Contextos pasados al ranker: {len(context)}")
52
 
53
+ # Corregido: pasar solo lista de strings
54
+ context_strings = [str(c) for c in context]
55
+ reranked = ranker.rank(query, context_strings, 1)
56
  print(f"🏅 Resultado del reranker: {reranked}")
57
 
58
+ # Seguridad en el acceso al mejor contexto
59
+ best_context = reranked[0].get("text", reranked[0].get("content", context_strings[0]))
60
+ print(f"🧠 Contexto elegido: {best_context[:300]}...")
61
 
62
  respuesta = generate_text(best_context, query)
63
  print(f"💬 Respuesta generada: {respuesta}")
 
65
  return respuesta
66
 
67
 
68
+
69
  def responder_chat(message, history):
70
  respuesta = test_rag_reranking(message, ranker)
71
  return respuesta