timeki commited on
Commit
31d501c
·
1 Parent(s): f49cc87

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