Skip to content

Base embedding model driver

BaseEmbeddingModelDriver

Bases: ABC

Source code in griptape/drivers/embedding_model/base_embedding_model_driver.py 
 5
 6
 7
 8
 9
10
11
12
13
@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 
7
8
9
@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 
11
12
13
@abstractmethod
def process_output(self, output: dict) -> list[float]:
    ...