diff --git a/test_elasticsearch/test_server/test_vectorstore/docker-compose.yml b/test_elasticsearch/test_server/test_vectorstore/docker-compose.yml index b13520e06..009c427a8 100644 --- a/test_elasticsearch/test_server/test_vectorstore/docker-compose.yml +++ b/test_elasticsearch/test_server/test_vectorstore/docker-compose.yml @@ -20,9 +20,15 @@ services: retries: 60 # Currently fails on Mac: https://github.com/elastic/elasticsearch/issues/106206 - elasticsearch-with-models: + elasticsearch-with-model: image: docker.elastic.co/eland/eland depends_on: - elasticsearch restart: no - command: sh -c "sleep 10 && eland_import_hub_model --url http://elasticsearch:9200 --hub-model-id sentence-transformers/msmarco-minilm-l-12-v3 --start" + command: sh -c " + sleep 10 && + eland_import_hub_model \ + --hub-model-id sentence-transformers/all-minilm-l6-v2 \ + --url http://elasticsearch:9200 \ + --start + " diff --git a/test_elasticsearch/test_server/test_vectorstore/test_embedding_service.py b/test_elasticsearch/test_server/test_vectorstore/test_embedding_service.py index c667a8a38..1f9b196a7 100644 --- a/test_elasticsearch/test_server/test_vectorstore/test_embedding_service.py +++ b/test_elasticsearch/test_server/test_vectorstore/test_embedding_service.py @@ -26,7 +26,7 @@ # deployed with # https://www.elastic.co/guide/en/machine-learning/current/ml-nlp-text-emb-vector-search-example.html -MODEL_ID = os.getenv("MODEL_ID", "sentence-transformers__msmarco-minilm-l-12-v3") +MODEL_ID = os.getenv("MODEL_ID", "sentence-transformers__all-minilm-l6-v2") NUM_DIMENSIONS = int(os.getenv("NUM_DIMENSIONS", "384")) diff --git a/test_elasticsearch/test_server/test_vectorstore/test_vectorstore.py b/test_elasticsearch/test_server/test_vectorstore/test_vectorstore.py index 2dedc61c3..bb15d3dc7 100644 --- a/test_elasticsearch/test_server/test_vectorstore/test_vectorstore.py +++ b/test_elasticsearch/test_server/test_vectorstore/test_vectorstore.py @@ -525,9 +525,7 @@ def test_search_with_knn_infer_instack( store = VectorStore( index=index, - retrieval_strategy=DenseVectorStrategy( - model_id="sentence-transformers__all-minilm-l6-v2" - ), + retrieval_strategy=DenseVectorStrategy(model_id=TRANSFORMER_MODEL_ID), client=sync_client, )