Cudd1es commited on
Commit
8d407ac
·
1 Parent(s): 1ddbde6

fixed a stupid bug of llm document reference

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -86,7 +86,9 @@ def respond(
86
  total_reranked = reranked + reformulated_reranked
87
  deduped = dedup_by_chapter_event(total_reranked, max_per_group=1)
88
  expanded_results = expand_with_neighbors(deduped[:3], collection)
89
- context = expanded_results[0][0] if expanded_results else ""
 
 
90
 
91
  rag_prompt = build_rag_prompt(message, context, system_msg)
92
  messages = [
 
86
  total_reranked = reranked + reformulated_reranked
87
  deduped = dedup_by_chapter_event(total_reranked, max_per_group=1)
88
  expanded_results = expand_with_neighbors(deduped[:3], collection)
89
+ context = []
90
+ for idx, text in enumerate(expanded_results):
91
+ context.append(text[0]) if text else ""
92
 
93
  rag_prompt = build_rag_prompt(message, context, system_msg)
94
  messages = [