From 91ddb3672e514fa9824609ff047d7cab0c65631a Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Tue, 9 Apr 2024 00:14:00 +0200 Subject: Refactor --- rag/generator/prompt.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) create mode 100644 rag/generator/prompt.py (limited to 'rag/generator/prompt.py') diff --git a/rag/generator/prompt.py b/rag/generator/prompt.py new file mode 100644 index 0000000..ed372c9 --- /dev/null +++ b/rag/generator/prompt.py @@ -0,0 +1,14 @@ +from dataclasses import dataclass +from typing import List + + +try: + from rag.retriever.vector import Document +except ModuleNotFoundError: + from retriever.vector import Document + + +@dataclass +class Prompt: + query: str + documents: List[Document] -- cgit v1.2.3-70-g09d2