rag/generator/prompt.py


1
2
3
4
5
6
7
8
9
10
11
12
13
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]