sofhiaazzhr commited on
Commit
16ab916
·
1 Parent(s): 959b1b0

[NOTICKET][doc] correct metadata key path in _format_context

Browse files
Files changed (1) hide show
  1. 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
- filename = result.metadata.get("filename", "Unknown")
56
- page = result.metadata.get("page_label")
 
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)