Merged
Conversation
natoverse
approved these changes
Feb 20, 2026
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Streaming Performance Improvement (with .txt):
run time: -16.46%
peak memory: -55.57%
memory delta: -81.58%
This pull request introduces a major refactor of the text embedding workflow to enable streaming, table-based processing and improve modularity and maintainability. The changes replace the previous DataFrame-based approach with a streaming Table API, introduce per-field embedding configuration, and simplify the workflow logic. Additionally, the embedding operation is now more efficient and robust, supporting output snapshots and row-level transformation.
Key changes include:
Streaming Embedding Workflow and Table API
generate_text_embeddingsworkflow to operate on streamingTableobjects instead of Pandas DataFrames, allowing for more scalable and memory-efficient processing. The workflow now iterates over configured embedding fields, applies optional row transformations, and writes embeddings to vector stores and optional output tables. [1] [2]embed_textoperation to accept a streamingTableas input, process rows in batches, and flush results to the vector store and optional output tables, improving efficiency and supporting large datasets. [1] [2] [3] [4]Embedding Field Configuration and Row Transformation
EmbeddingFieldConfigdataclass and theEMBEDDING_FIELDSmapping, enabling per-field configuration of source tables, columns, and optional row transformers for embedding generation.transform_entity_row_for_embedding, to concatenate entity title and description for embedding.Workflow and API Simplification
generate_text_embeddingsandupdate_text_embeddingsworkflows, removing DataFrame manipulation and centralizing configuration and iteration logic. The update workflow now delegates directly to the streaming embedding generator. [1] [2]Licensing and Metadata