From 2e85325639ce3827cc2eb32f9750dfa873e3a480 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Wed, 24 Apr 2024 01:51:29 +0200 Subject: Experimenting with prompt --- rag/generator/ollama.py | 12 +++++++----- rag/generator/prompt.py | 6 +++--- 2 files changed, 10 insertions(+), 8 deletions(-) (limited to 'rag') diff --git a/rag/generator/ollama.py b/rag/generator/ollama.py index 9118906..52521ca 100644 --- a/rag/generator/ollama.py +++ b/rag/generator/ollama.py @@ -24,12 +24,14 @@ class Ollama(metaclass=AbstractGenerator): def __metaprompt(self, prompt: Prompt) -> str: metaprompt = ( - f"{ANSWER_INSTRUCTION}" - "Only the information between ... should be used to answer the question.\n" - f"Question: {prompt.query.strip()}\n\n" - "\n" + "Context information is below.\n" + "---------------------\n" f"{self.__context(prompt.documents)}\n\n" - "\n" + "---------------------\n" + f"{ANSWER_INSTRUCTION}" + "Do not attempt to answer the query without relevant context and do not use" + " prior knowledge or training data!\n" + f"Query: {prompt.query.strip()}\n\n" "Answer:" ) return metaprompt diff --git a/rag/generator/prompt.py b/rag/generator/prompt.py index f607122..10afe7b 100644 --- a/rag/generator/prompt.py +++ b/rag/generator/prompt.py @@ -4,9 +4,9 @@ from typing import List from rag.retriever.vector import Document ANSWER_INSTRUCTION = ( - "Given the context information and not prior knowledge, answer the question." - "If the context is irrelevant to the question or empty, then do not attempt to answer " - "the question, just reply that you do not know based on the context provided.\n" + "Given the context information and not prior knowledge, answer the query." + "If the context is irrelevant to the query or empty, then do not attempt to answer " + "the query, just reply that you do not know based on the context provided.\n" ) -- cgit v1.2.3-70-g09d2