refactor: Update CohereEmbeddingsComponent to use type hinting for CohereEmbeddings return value

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-20 18:23:31 -03:00
commit 96f77878c1

View file

@ -34,7 +34,7 @@ class CohereEmbeddingsComponent(LCModelComponent):
]
def build_embeddings(self) -> Embeddings:
return CohereEmbeddings(
return CohereEmbeddings( # type: ignore
cohere_api_key=self.cohere_api_key,
model=self.model,
truncate=self.truncate,