Skip to content

feat: Created DocEmbedder class#5973

Open
patelchaitany wants to merge 11 commits intofeast-dev:masterfrom
patelchaitany:fet-DocEmbedder
Open

feat: Created DocEmbedder class#5973
patelchaitany wants to merge 11 commits intofeast-dev:masterfrom
patelchaitany:fet-DocEmbedder

Conversation

@patelchaitany
Copy link

@patelchaitany patelchaitany commented Feb 16, 2026

What this PR does / why we need it:

This PR adds a Document Embedder capability to Feast, allowing users to go from raw documents to embeddings stored in the online vector store in a single step. It handles chunking, embedding generation, and writing the results to the online store — providing an end-to-end ingestion pipeline for RAG workflows within Feast.

What changed:

sdk/python/feast/chunker.py

Defines the document chunking layer. Provides:

  • A BaseChunker abstract class that users can extend with custom chunking strategies
  • A built-in TextChunker that splits plain text by word count with configurable size, overlap, and thresholds
  • A chunk_dataframe() convenience method to chunk all documents in a DataFrame

Currently only basic text chunking is implemented. There is room for improvement — future iterations can support more advanced strategies like semantic chunking, sentence-aware splitting, or format-specific chunkers (PDF, HTML, etc.).

sdk/python/feast/embedder.py

Defines the embedding generation layer. Provides:

  • A BaseEmbedder abstract class with a modality-routing registry for extensibility
  • A MultiModalEmbedder with built-in support for text (via sentence-transformers) and image (via CLIP) embeddings
  • Lazy model loading and configurable batch processing

sdk/python/feast/doc_embedder.py

The high-level orchestrator that coordinates chunking, embedding, and storage. Provides:

  • A DocEmbedder class that runs the full pipeline: chunk -> embed -> logical layer -> write to online store
  • A user-defined logical layer function to map output to the FeatureView schema
  • Auto-generation of a default FeatureView if the user does not define their own (controlled by create_feature_view). This is a basic implementation intended as a starting point

sdk/python/feast/init.py

Updated to export DocEmbedder, LogicalLayerFn, BaseChunker, TextChunker, ChunkingConfig, BaseEmbedder, MultiModalEmbedder, and EmbeddingConfig as part of Feast's public API.

Which issue(s) this PR fixes:

Create DocEmbedder class along with RAGRetriever #5426

Misc


Open with Devin

patelchaitany and others added 6 commits February 16, 2026 16:51
Signed-off-by: Chaitany patel <patelchaitany93@gmail.com>
Signed-off-by: Chaitany patel <patelchaitany93@gmail.com>
…embedder.py

Signed-off-by: Chaitany patel <patelchaitany93@gmail.com>
…r.py

Signed-off-by: Chaitany patel <patelchaitany93@gmail.com>
Co-authored-by: devin-ai-integration[bot] <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Signed-off-by: Chaitany patel <patelchaitany93@gmail.com>
Signed-off-by: Chaitany patel <patelchaitany93@gmail.com>
@patelchaitany patelchaitany requested a review from a team as a code owner February 16, 2026 11:28
devin-ai-integration[bot]

This comment was marked as resolved.

Signed-off-by: Chaitany patel <patelchaitany93@gmail.com>
devin-ai-integration[bot]

This comment was marked as resolved.

Signed-off-by: Chaitany patel <patelchaitany93@gmail.com>
@patelchaitany
Copy link
Author

@ntkathole @jyejare can You pls review this PR and let me know if any changes is needed.

@patelchaitany patelchaitany changed the title feat: Created DocEmbedder class #5972 feat: Created DocEmbedder class Feb 16, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant