Spaces:
Running
on
Zero
Running
on
Zero
feat: improve metadata handling
Browse files
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":
|
58 |
-
file_sources.append(meta
|
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):
|