Update query.py

query.py

@@ -102,9 +102,10 @@ class VectaraQuery():
             doc_num = responses[response_num-1]['documentIndex']
             metadata = {item['name']: item['value'] for item in docs[doc_num]['metadata']}
             text = extract_between_tags(responses[response_num-1]['text'], start_tag, end_tag)
-            url = f"{metadata['url']}#:~:text={quote(text)}"
-            if url not in refs:
-                refs.append(url)
+            if 'url' in metadata.keys():
+                url = f"{metadata['url']}#:~:text={quote(text)}"
+                if url not in refs:
+                    refs.append(url)
 
         # replace references with markdown links
         refs_dict = {url:(inx+1) for inx,url in enumerate(refs)}
@@ -118,5 +119,4 @@ class VectaraQuery():
             citation_inx = refs_dict[url]
             summary = summary[:start] + f'[\[{citation_inx}\]]({url})' + summary[end:]
 
-        return summary
-        
+        return summary