Skip to content

Commit b7c064f

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

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

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

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -455,6 +455,13 @@ 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+
}, 'data': {
463+
'dir': '/var/lib/milvus/etcd'
464+
}
458465
}
459466
}
460467

@@ -481,10 +488,11 @@ class TestMilvusSearchEnrichment(unittest.TestCase):
481488
"""Tests for search functionality across all search strategies"""
482489

483490
_db: MilvusDBContainerInfo
491+
_version = "milvusdb/milvus:v2.5.10"
484492

485493
@classmethod
486494
def setUpClass(cls):
487-
cls._db = MilvusEnrichmentTestHelper.start_db_container()
495+
cls._db = MilvusEnrichmentTestHelper.start_db_container(cls._version)
488496
cls._connection_params = MilvusConnectionParameters(
489497
uri=cls._db.uri,
490498
user=cls._db.user,

0 commit comments

Comments
 (0)