diff --git a/hindsight-api/hindsight_api/engine/embeddings.py b/hindsight-api/hindsight_api/engine/embeddings.py index d943591c..7ba9b1bd 100644 --- a/hindsight-api/hindsight_api/engine/embeddings.py +++ b/hindsight-api/hindsight_api/engine/embeddings.py @@ -794,6 +794,7 @@ class LiteLLMSDKEmbeddings(Embeddings): "model": self.model, "input": ["test"], "api_key": self.api_key, + "encoding_format": "float", } if self.api_base: embed_kwargs["api_base"] = self.api_base @@ -840,6 +841,7 @@ class LiteLLMSDKEmbeddings(Embeddings): "model": self.model, "input": batch, "api_key": self.api_key, + "encoding_format": "float", } if self.api_base: embed_kwargs["api_base"] = self.api_base diff --git a/hindsight-api/tests/test_litellm_sdk_embeddings.py b/hindsight-api/tests/test_litellm_sdk_embeddings.py index 857769c7..256662bc 100644 --- a/hindsight-api/tests/test_litellm_sdk_embeddings.py +++ b/hindsight-api/tests/test_litellm_sdk_embeddings.py @@ -85,6 +85,7 @@ class TestLiteLLMSDKEmbeddings: model="cohere/embed-english-v3.0", input=["test"], api_key="test_key", + encoding_format="float", ) async def test_initialization_missing_package(self): @@ -137,6 +138,7 @@ class TestLiteLLMSDKEmbeddings: model="cohere/embed-english-v3.0", input=["Hello world"], api_key="test_key", + encoding_format="float", ) async def test_encode_multiple_texts(self, embeddings, mock_litellm):