fabricatio_lancedb.capabilities.lancedb

This module contains the capabilities for the lancedb.

Classes

LancedbAddRAGConfig	LanceDB-specific RAG configuration.
LancedbFetchRAGConfig	LanceDB-specific RAG configuration.
LancedbRAG	LanceDB-specific RAG capability extending the base RAG class.

Module Contents

class fabricatio_lancedb.capabilities.lancedb.LancedbAddRAGConfig(/, **data: Any)

Bases: fabricatio_rag.capabilities.rag.RAGConfigBase

LanceDB-specific RAG configuration.

table_name: str

embedding_batch_size: int = 10

embedding_parallel_size: int = 10

rebuild_index: bool = False

class fabricatio_lancedb.capabilities.lancedb.LancedbFetchRAGConfig[D: fabricatio_lancedb.models.lancedb.LancedbDocumentModel](/, **data: Any)

Bases: fabricatio_rag.capabilities.rag.RAGConfigBase

LanceDB-specific RAG configuration.

document_model: Type[D] | None = None

limit: int = 15

table_name: str

class fabricatio_lancedb.capabilities.lancedb.LancedbRAG[D: fabricatio_lancedb.models.lancedb.LancedbDocumentModel, AC: LancedbAddRAGConfig, FC: LancedbFetchRAGConfig](/, **data: Any)

Bases: fabricatio_rag.capabilities.rag.RAG[D, D, AC, FC]

LanceDB-specific RAG capability extending the base RAG class.

async add_document(data: D | List[D], config: AC | None = None) → Self

Add a document to the LanceDB collection.

async afetch_document(query: str | List[str], config: FC | None = None) → List[D]

Fetch documents from the LanceDB collection.

async rebuild_index(table_name: str | None = None) → Self

Rebuild the index of the given table.