bug

run-llama · Feb 28, 2024 · a25b14a · a25b14a
1 parent f6bb399
commit a25b14a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...x-integrations/embeddings/llama-index-embeddings-nemo/llama_index/embeddings/nemo/base.py b/...x-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")