Update query.py

query.py

@@ -43,9 +43,12 @@ class CitationNormalizer():
             doc_num = self.responses[response_num-1]['documentIndex']
             metadata = {item['name']: item['value'] for item in self.docs[doc_num]['metadata']}
             text = extract_between_tags(self.responses[response_num-1]['text'], start_tag, end_tag)
-            url = f"{metadata['url']}#:~:text={quote(text)}"
-            citation_inx = refs_dict[url]
-            summary = summary[:start] + f'[\[{citation_inx}\]]({url})' + summary[end:]
+            if 'url' in metadata.keys():
+                url = f"{metadata['url']}#:~:text={quote(text)}"
+                citation_inx = refs_dict[url]
+                summary = summary[:start] + f'[\[{citation_inx}\]]({url})' + summary[end:]
+            else:
+                summary = summary[:start] + summary[end:]
 
         return summary
 
@@ -192,5 +195,4 @@ class VectaraQuery():
             chunks.append(accumulated_text)
             yield accumulated_text        
         
-        return ''.join(chunks)
-        
+        return ''.join(chunks)