Base_embedder
This module contains functionality related to the the base_embedder
module for embedding.embedders
.
Base_embedder
BaseEmbedder
Bases: ABC
Abstract base class for text node embedding operations.
This class provides core functionality for embedding text nodes, with derived classes implementing specific embedding strategies.
Source code in src/embedding/embedders/base_embedder.py
13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
|
__init__(configuration, embedding_model, vector_store)
Initialize embedder with configuration, model and storage.
Parameters: |
|
---|
Source code in src/embedding/embedders/base_embedder.py
20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
|
embed(nodes)
abstractmethod
Generate embeddings for text nodes using batch processing.
This method should implement a strategy for processing the provided nodes, potentially splitting them into batches for efficient embedding generation.
Parameters: |
|
---|
Note
Implementation should modify nodes in-place by adding embeddings
Source code in src/embedding/embedders/base_embedder.py
38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
|
embed_flush()
abstractmethod
Process and generate embeddings for any remaining nodes.
This method should handle any nodes that remain in the buffer, ensuring all nodes receive embeddings.
Note
Should be called at the end of processing to ensure no nodes remain unembedded in the buffer
Source code in src/embedding/embedders/base_embedder.py
53 54 55 56 57 58 59 60 61 62 63 |
|