fixed a stupid bug of llm document reference
Browse files
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 =
|
|
|
|
|
|
|
| 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 = [
|