Update app.py
Browse files
app.py
CHANGED
@@ -104,12 +104,12 @@ async def main(message):
|
|
104 |
# --documents, user session
|
105 |
docs = res.get("source_documents", [])
|
106 |
metadatas = [doc.metadata for doc in docs]
|
107 |
-
all_sources = [m["source"] for m in metadatas]
|
108 |
|
109 |
# --append source(s), specific rows only
|
110 |
for doc, metadata in zip(docs, metadatas):
|
111 |
-
|
112 |
-
|
113 |
|
114 |
if row_index in [2, 8, 14] and source and source not in visited_sources:
|
115 |
visited_sources.add(source)
|
|
|
104 |
# --documents, user session
|
105 |
docs = res.get("source_documents", [])
|
106 |
metadatas = [doc.metadata for doc in docs]
|
107 |
+
#all_sources = [m["source"] for m in metadatas]
|
108 |
|
109 |
# --append source(s), specific rows only
|
110 |
for doc, metadata in zip(docs, metadatas):
|
111 |
+
row_index = metadata.get("row_index", -1)
|
112 |
+
source = metadata.get("source", "")
|
113 |
|
114 |
if row_index in [2, 8, 14] and source and source not in visited_sources:
|
115 |
visited_sources.add(source)
|