arsiba commited on
Commit
c65b0b0
·
1 Parent(s): 099d11b

feat: improve metadata handling

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -54,8 +54,8 @@ def retrieve(q, k=3):
54
  for i in ids:
55
  chunk = chunks[i]
56
  meta = metadata_dict[i]
57
- docs.append({"title": chunk, "pages": chunk})
58
- file_sources.append(meta["source"])
59
  return docs, file_sources
60
 
61
  def make_prompt(q, docs, reasoning_mode):
 
54
  for i in ids:
55
  chunk = chunks[i]
56
  meta = metadata_dict[i]
57
+ docs.append({"title": meta, "pages": chunk})
58
+ file_sources.append(meta)
59
  return docs, file_sources
60
 
61
  def make_prompt(q, docs, reasoning_mode):