Skip to content

RAG

Auxiliary types for RAG System

SourceNode

Bases: BaseModel

Source code in src/fed_rag/data_structures/rag.py 
10
11
12
13
14
15
16
class SourceNode(BaseModel):
    score: float
    node: KnowledgeNode

    def __getattr__(self, __name: str) -> Any:
        """Convenient wrapper on getattr of associated node."""
        return getattr(self.node, __name)

__getattr__ 

__getattr__(__name)

Convenient wrapper on getattr of associated node.

Source code in src/fed_rag/data_structures/rag.py 
14
15
16
def __getattr__(self, __name: str) -> Any:
    """Convenient wrapper on getattr of associated node."""
    return getattr(self.node, __name)

RAGResponse

Bases: BaseModel

Response class returned by querying RAG systems.

Source code in src/fed_rag/data_structures/rag.py 
19
20
21
22
23
24
25
26
27
class RAGResponse(BaseModel):
    """Response class returned by querying RAG systems."""

    response: str
    raw_response: str | None = None
    source_nodes: list[SourceNode]

    def __str__(self) -> str:
        return self.response