url fix

UIUC-Chatbot · Nov 8, 2023 · 88004f3 · 88004f3
1 parent 86e721d
commit 88004f3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ai_ta_backend/vector_database.py b/ai_ta_backend/vector_database.py
@@ -1117,10 +1117,11 @@ def context_padding(self, found_docs, search_query, course_name):
             'readable_filename': doc.metadata['readable_filename'],
             'course_name': course_name,
             's3_path': doc.metadata['s3_path'],
-            'url': doc.metadata['url'],
             'base_url':doc.metadata['base_url']
           }
-
+          if 'url' in doc.metadata.keys():
+            context_dict['url'] = doc.metadata['url']
+
           result_contexts.append(context_dict)
           print("len of result contexts after qdrant append: ", len(result_contexts))