Julia Ostheimer commited on
Commit
a685766
·
1 Parent(s): 163f7aa

Fix page number and show in German

Browse files
Files changed (1) hide show
  1. conversation/citation_utils.py +8 -1
conversation/citation_utils.py CHANGED
@@ -42,6 +42,13 @@ def format_artifacts_to_string(artifacts: List[dict]) -> str:
42
  )
43
  return "\n\n".join(parts)
44
 
 
 
 
 
 
 
 
45
  def embed_references(result: dict) -> str:
46
  """
47
  Given an LLM result dict with keys:
@@ -64,6 +71,6 @@ def embed_references(result: dict) -> str:
64
  refs = ["", "**Referenzen:**"]
65
  for src in sources:
66
  idx = src.source_number
67
- refs.append(f"[{idx}] {src.file}, page {src.page}")
68
 
69
  return answer + "\n\n" + "\n\n".join(refs)
 
42
  )
43
  return "\n\n".join(parts)
44
 
45
+ def fix_page_number(incorrect_page_number: str) -> str:
46
+ if incorrect_page_number == "No page number":
47
+ return incorrect_page_number
48
+ else:
49
+ correct_page_number = int(incorrect_page_number) + 1
50
+ return str(correct_page_number)
51
+
52
  def embed_references(result: dict) -> str:
53
  """
54
  Given an LLM result dict with keys:
 
71
  refs = ["", "**Referenzen:**"]
72
  for src in sources:
73
  idx = src.source_number
74
+ refs.append(f"[{idx}] {src.file}, Seite {fix_page_number(src.page)}")
75
 
76
  return answer + "\n\n" + "\n\n".join(refs)