Skip to content

Base embedding model driver

BaseEmbeddingModelDriver

Bases: ABC

Source code in griptape/drivers/embedding_model/base_embedding_model_driver.py
@define
class BaseEmbeddingModelDriver(ABC):
    @abstractmethod
    def chunk_to_model_params(self, chunk: str) -> dict: ...

    @abstractmethod
    def process_output(self, output: dict) -> list[float]: ...

chunk_to_model_params(chunk) abstractmethod

Source code in griptape/drivers/embedding_model/base_embedding_model_driver.py
@abstractmethod
def chunk_to_model_params(self, chunk: str) -> dict: ...

process_output(output) abstractmethod

Source code in griptape/drivers/embedding_model/base_embedding_model_driver.py
@abstractmethod
def process_output(self, output: dict) -> list[float]: ...