File tree Expand file tree Collapse file tree 3 files changed +10
-6
lines changed
test_elasticsearch/test_server/test_vectorstore Expand file tree Collapse file tree 3 files changed +10
-6
lines changed Original file line number Diff line number Diff line change @@ -20,9 +20,15 @@ services:
20
20
retries : 60
21
21
22
22
# Currently fails on Mac: https://github.com/elastic/elasticsearch/issues/106206
23
- elasticsearch-with-models :
23
+ elasticsearch-with-model :
24
24
image : docker.elastic.co/eland/eland
25
25
depends_on :
26
26
- elasticsearch
27
27
restart : no
28
- command : sh -c "sleep 10 && eland_import_hub_model --url http://elasticsearch:9200 --hub-model-id sentence-transformers/msmarco-minilm-l-12-v3 --start"
28
+ command : sh -c "
29
+ sleep 10 &&
30
+ eland_import_hub_model \
31
+ --hub-model-id sentence-transformers/all-minilm-l6-v2 \
32
+ --url http://elasticsearch:9200 \
33
+ --start
34
+ "
Original file line number Diff line number Diff line change 26
26
27
27
# deployed with
28
28
# https://www.elastic.co/guide/en/machine-learning/current/ml-nlp-text-emb-vector-search-example.html
29
- MODEL_ID = os .getenv ("MODEL_ID" , "sentence-transformers__msmarco -minilm-l-12-v3 " )
29
+ MODEL_ID = os .getenv ("MODEL_ID" , "sentence-transformers__all -minilm-l6-v2 " )
30
30
NUM_DIMENSIONS = int (os .getenv ("NUM_DIMENSIONS" , "384" ))
31
31
32
32
Original file line number Diff line number Diff line change @@ -525,9 +525,7 @@ def test_search_with_knn_infer_instack(
525
525
526
526
store = VectorStore (
527
527
index = index ,
528
- retrieval_strategy = DenseVectorStrategy (
529
- model_id = "sentence-transformers__all-minilm-l6-v2"
530
- ),
528
+ retrieval_strategy = DenseVectorStrategy (model_id = TRANSFORMER_MODEL_ID ),
531
529
client = sync_client ,
532
530
)
533
531
You can’t perform that action at this time.
0 commit comments