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")