From a25b14a5a14abfca7cd47cce4eba8854f375442e Mon Sep 17 00:00:00 2001 From: Logan Markewich Date: Wed, 28 Feb 2024 10:23:23 -0600 Subject: [PATCH] bug --- .../llama_index/embeddings/nemo/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama-index-integrations/embeddings/llama-index-embeddings-nemo/llama_index/embeddings/nemo/base.py b/llama-index-integrations/embeddings/llama-index-embeddings-nemo/llama_index/embeddings/nemo/base.py index 68c83b0e5675d..1e48899c67d5e 100644 --- a/llama-index-integrations/embeddings/llama-index-embeddings-nemo/llama_index/embeddings/nemo/base.py +++ b/llama-index-integrations/embeddings/llama-index-embeddings-nemo/llama_index/embeddings/nemo/base.py @@ -49,7 +49,7 @@ def _get_embedding(self, text: str, input_type: str): return response["data"][0]["embedding"] def _get_query_embedding(self, query: str) -> List[float]: - return self._get_embedding(text, input_type="query") + return self._get_embedding(query, input_type="query") def _get_text_embedding(self, text: str) -> List[float]: return self._get_embedding(text, input_type="passage")