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
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
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
SourceNode
¶
Bases: BaseModel
Source code in src/fed_rag/data_structures/rag.py
RAGResponse
¶
Bases: BaseModel
Response class returned by querying RAG systems.