Skip to content

RAG

Auxiliary types for RAG System

Query

Bases: _MultiModalDataContainer

Query data structure.

This class represents a multimodal representation of a RAG query.

Attributes:

Name Type Description
text str

Text content of query.

images list[Image] | None

Images content of query.

audios list[Any] | None

Audios content of query.

videos list[Any] | None

Videos content of query.

Source code in src/fed_rag/data_structures/rag.py
class Query(_MultiModalDataContainer):
    """Query data structure.

    This class represents a multimodal representation of a RAG query.

    Attributes:
        text: Text content of query.
        images: Images content of query.
        audios: Audios content of query.
        videos: Videos content of query.
    """

    pass

Context

Bases: _MultiModalDataContainer

Context data structure.

This class represents a multimodal representation of RAG context.

Attributes:

Name Type Description
text str

Text content of query.

images list[Image] | None

Images content of query.

audios list[Any] | None

Audios content of query.

videos list[Any] | None

Videos content of query.

Source code in src/fed_rag/data_structures/rag.py
class Context(_MultiModalDataContainer):
    """Context data structure.

    This class represents a multimodal representation of RAG context.

    Attributes:
        text: Text content of query.
        images: Images content of query.
        audios: Audios content of query.
        videos: Videos content of query.
    """

    pass

Prompt

Bases: _MultiModalDataContainer

Prompt data structure.

This class represents a multimodal representation of a prompt given to a multi-modal LLM.

Attributes:

Name Type Description
text str

Text content of query.

images list[Image] | None

Images content of query.

audios list[Any] | None

Audios content of query.

videos list[Any] | None

Videos content of query.

Source code in src/fed_rag/data_structures/rag.py
class Prompt(_MultiModalDataContainer):
    """Prompt data structure.

    This class represents a multimodal representation of a prompt given to a
    multi-modal LLM.

    Attributes:
        text: Text content of query.
        images: Images content of query.
        audios: Audios content of query.
        videos: Videos content of query.
    """

    pass

SourceNode

Bases: BaseModel

Source code in src/fed_rag/data_structures/rag.py
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
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
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