fabricatio_rag.actions.db

Database storage actions for RAG document models.

Classes

StoreTextFile

Ingest text files, chunk them, and store in the vector database.

StoreDocuments

Store pre-built document model instances directly into the vector database.

Module Contents

class fabricatio_rag.actions.db.StoreTextFile[STD: fabricatio_rag.models.document.StoredDocumentModel, SRD: fabricatio_rag.models.document.SearchedDocumentModel, AC: fabricatio_rag.capabilities.rag.RAGConfigBase, FC: fabricatio_rag.capabilities.rag.RAGConfigBase](/, **data: Any)

Bases: fabricatio_core.models.action.Action, fabricatio_rag.capabilities.rag.RAG[STD, SRD, AC, FC], abc.ABC

Ingest text files, chunk them, and store in the vector database.

store_model: Type[STD]
store_config: AC | None = None
chunk_size: int = 512
chunk_overlap_ratio: float = 0.3
ctx_override: ClassVar[bool] = True

Whether to override the instance attr by the context variable.

class fabricatio_rag.actions.db.StoreDocuments[STD: fabricatio_rag.models.document.StoredDocumentModel, SRD: fabricatio_rag.models.document.SearchedDocumentModel, AC: fabricatio_rag.capabilities.rag.RAGConfigBase, FC: fabricatio_rag.capabilities.rag.RAGConfigBase](/, **data: Any)

Bases: fabricatio_core.models.action.Action, fabricatio_rag.capabilities.rag.RAG[STD, SRD, AC, FC], abc.ABC

Store pre-built document model instances directly into the vector database.

Unlike StoreTextFile, this action does NOT ingest file paths or chunk text. The caller is responsible for constructing the model instances beforehand.

store_config: AC | None = None
ctx_override: ClassVar[bool] = True

Whether to override the instance attr by the context variable.