fix search for images
Browse files
climateqa/engine/chains/retrieve_documents.py
CHANGED
@@ -108,7 +108,7 @@ def _add_metadata_and_score(docs: List) -> Document:
|
|
108 |
doc.page_content = doc.page_content.replace("\r\n"," ")
|
109 |
doc.metadata["similarity_score"] = score
|
110 |
doc.metadata["content"] = doc.page_content
|
111 |
-
if doc.metadata["page_number"] != "N/A":
|
112 |
doc.metadata["page_number"] = int(doc.metadata["page_number"]) + 1
|
113 |
else:
|
114 |
doc.metadata["page_number"] = 1
|
|
|
108 |
doc.page_content = doc.page_content.replace("\r\n"," ")
|
109 |
doc.metadata["similarity_score"] = score
|
110 |
doc.metadata["content"] = doc.page_content
|
111 |
+
if doc.metadata["page_number"] is not None and doc.metadata["page_number"] != "N/A":
|
112 |
doc.metadata["page_number"] = int(doc.metadata["page_number"]) + 1
|
113 |
else:
|
114 |
doc.metadata["page_number"] = 1
|