From 96f77878c11dd7fb1e42c86cbd7d7280716dde2f Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 20 Jun 2024 18:23:31 -0300 Subject: [PATCH] refactor: Update CohereEmbeddingsComponent to use type hinting for CohereEmbeddings return value --- .../base/langflow/components/embeddings/CohereEmbeddings.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/backend/base/langflow/components/embeddings/CohereEmbeddings.py b/src/backend/base/langflow/components/embeddings/CohereEmbeddings.py index 7f4f6a513..7c4062c99 100644 --- a/src/backend/base/langflow/components/embeddings/CohereEmbeddings.py +++ b/src/backend/base/langflow/components/embeddings/CohereEmbeddings.py @@ -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,