Commit ·
16ab916
1
Parent(s): 959b1b0
[NOTICKET][doc] correct metadata key path in _format_context
Browse files- src/api/v1/chat.py +3 -2
src/api/v1/chat.py
CHANGED
|
@@ -52,8 +52,9 @@ def _format_context(results: List[RetrievalResult]) -> str:
|
|
| 52 |
"""Format retrieval results as context string for the LLM."""
|
| 53 |
lines = []
|
| 54 |
for result in results:
|
| 55 |
-
|
| 56 |
-
|
|
|
|
| 57 |
source_label = f"{filename}, p.{page}" if page else filename
|
| 58 |
lines.append(f"[Source: {source_label}]\n{result.content}\n")
|
| 59 |
return "\n".join(lines)
|
|
|
|
| 52 |
"""Format retrieval results as context string for the LLM."""
|
| 53 |
lines = []
|
| 54 |
for result in results:
|
| 55 |
+
data = result.metadata.get("data", {})
|
| 56 |
+
filename = data.get("filename", "Unknown")
|
| 57 |
+
page = data.get("page_label")
|
| 58 |
source_label = f"{filename}, p.{page}" if page else filename
|
| 59 |
lines.append(f"[Source: {source_label}]\n{result.content}\n")
|
| 60 |
return "\n".join(lines)
|