diff --git a/app/pipeline/prompts/citation_prompt.txt b/app/pipeline/prompts/citation_prompt.txt index f69e885..1d785b0 100644 --- a/app/pipeline/prompts/citation_prompt.txt +++ b/app/pipeline/prompts/citation_prompt.txt @@ -15,6 +15,11 @@ Lorem ipsum dolor sit amet, consectetur adipiscing elit [1]. Ded do eiusmod temp [2] Lecture 2, Unit B, page 5,25. " +Note: If there is no link available, please do not include the link in the citation. For example, if citation 1 does not have a link, it should look like this: +[1] "Lecture title", "Lecture unit title", "page number" +but if citation 2 has a link, it should look like this: +[2] "Lecture title", "Lecture unit title", "page number" + Here are the answer and the paragraphs: Answer without citations: diff --git a/app/pipeline/shared/citation_pipeline.py b/app/pipeline/shared/citation_pipeline.py index 01e96e2..22e1336 100644 --- a/app/pipeline/shared/citation_pipeline.py +++ b/app/pipeline/shared/citation_pipeline.py @@ -57,7 +57,7 @@ def create_formatted_string(self, paragraphs): paragraph.get(LectureSchema.LECTURE_NAME.value), paragraph.get(LectureSchema.LECTURE_UNIT_NAME.value), paragraph.get(LectureSchema.PAGE_NUMBER.value), - paragraph.get(LectureSchema.LECTURE_UNIT_LINK.value), + paragraph.get(LectureSchema.LECTURE_UNIT_LINK.value) or "No link available", paragraph.get(LectureSchema.PAGE_TEXT_CONTENT.value), ) formatted_string += lct