diff --git a/haystack/utils/doc_store.py b/haystack/utils/doc_store.py index ccdb71554..5000efe4a 100644 --- a/haystack/utils/doc_store.py +++ b/haystack/utils/doc_store.py @@ -37,7 +37,7 @@ def launch_es(sleep=15, delete_existing=False): time.sleep(sleep) -def launch_opensearch(sleep=15, delete_existing=False): +def launch_opensearch(sleep=15, delete_existing=False, local_port=9200): """ Start an OpenSearch server via Docker. """ @@ -48,7 +48,7 @@ def launch_opensearch(sleep=15, delete_existing=False): _ = subprocess.run([f"docker rm --force {OPENSEARCH_CONTAINER_NAME}"], shell=True, stdout=subprocess.DEVNULL) status = subprocess.run( [ - f'docker start {OPENSEARCH_CONTAINER_NAME} > /dev/null 2>&1 || docker run -d -p 9201:9200 -p 9600:9600 -e "discovery.type=single-node" --name {OPENSEARCH_CONTAINER_NAME} opensearchproject/opensearch:1.3.5' + f'docker start {OPENSEARCH_CONTAINER_NAME} > /dev/null 2>&1 || docker run -d -p {local_port}:9200 -p 9600:9600 -e "discovery.type=single-node" --name {OPENSEARCH_CONTAINER_NAME} opensearchproject/opensearch:1.3.5' ], shell=True, ) diff --git a/test/benchmarks/utils.py b/test/benchmarks/utils.py index 0625e9bad..1e3bfbe02 100644 --- a/test/benchmarks/utils.py +++ b/test/benchmarks/utils.py @@ -93,7 +93,7 @@ def get_document_store(document_store_type, similarity="dot_product", index="doc ) assert document_store.get_document_count() == 0 elif document_store_type in ("opensearch_flat", "opensearch_hnsw"): - launch_opensearch() + launch_opensearch(local_port=9201) if document_store_type == "opensearch_flat": index_type = "flat" elif document_store_type == "opensearch_hnsw":