fix: Cohere inconsistent embeddings and documents lengths (#284)

* Don't double length of embeddings vs documents * Fix issue * Fix failing tests * Fix pylint * Pylint * Undo test fixes * Fix linter
deepset-ai · Feb 6, 2024 · 2fc2676 · 2fc2676
1 parent 32b6cc6
commit 2fc2676
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/integrations/cohere/src/haystack_integrations/components/embedders/cohere/utils.py b/integrations/cohere/src/haystack_integrations/components/embedders/cohere/utils.py
@@ -46,8 +46,6 @@ def get_response(
             response = cohere_client.embed(batch, model=model_name, input_type=input_type, truncate=truncate)
             for emb in response.embeddings:
                 all_embeddings.append(emb)
-            embeddings = [list(map(float, emb)) for emb in response.embeddings]
-            all_embeddings.extend(embeddings)
             if response.meta is not None:
                 metadata = response.meta