Skip to content

Commit 14ad6ea

Browse files
ml/rag: enforce running etcd in milvus itests in standalone mode
1 parent c764314 commit 14ad6ea

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

sdks/python/apache_beam/ml/rag/enrichment/milvus_search_it_test.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -455,6 +455,14 @@ def create_user_yaml(service_port: int, max_vector_field_num=5):
455455
user_config = {
456456
'proxy': {
457457
'maxVectorFieldNum': max_vector_field_num, 'port': service_port
458+
},
459+
'etcd': {
460+
'use': {
461+
'embed': True
462+
},
463+
'data': {
464+
'dir': '/var/lib/milvus/etcd'
465+
}
458466
}
459467
}
460468

@@ -481,10 +489,12 @@ class TestMilvusSearchEnrichment(unittest.TestCase):
481489
"""Tests for search functionality across all search strategies"""
482490

483491
_db: MilvusDBContainerInfo
492+
_version = "milvusdb/milvus:v2.5.10"
493+
484494

485495
@classmethod
486496
def setUpClass(cls):
487-
cls._db = MilvusEnrichmentTestHelper.start_db_container()
497+
cls._db = MilvusEnrichmentTestHelper.start_db_container(cls._version)
488498
cls._connection_params = MilvusConnectionParameters(
489499
uri=cls._db.uri,
490500
user=cls._db.user,

0 commit comments

Comments
 (0)