rag/generator/prompt.py


1
2
3
4
5
6
7
8
9
10

from dataclasses import dataclass
from typing import List

from rag.retriever.vector import Document


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