rag/generator/prompt.py


1
2
3
4
5
6
7
8
9
10
11

from dataclasses import dataclass
from typing import List


from rag.retriever.vector import Document


@dataclass
class Prompt:
    query: str
    documents: List[Document]