jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-20 18:23:31 -03:00
commit 96f77878c1
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/backend/base/langflow/components/embeddings/CohereEmbeddings.py
View file

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