Skip to content

Commit 0d93175

Browse files
committed
fix field usage instead of adding a new field
1 parent 4cbf3a2 commit 0d93175

2 files changed

Lines changed: 1 addition & 4 deletions

File tree

sdks/python/apache_beam/examples/snippets/transforms/elementwise/enrichment_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -380,7 +380,7 @@ def pre_milvus_enrichment() -> VectorDBContainerInfo:
380380
uri=db.uri,
381381
user=db.user,
382382
password=db.password,
383-
db_id=db.id,
383+
db_name=db.id,
384384
token=db.token)
385385
collection_name = MilvusTestHelpers.initialize_db_with_data(
386386
connection_params)

sdks/python/apache_beam/ml/rag/utils.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -62,13 +62,10 @@ class MilvusConnectionParameters:
6262
token: str = field(default_factory=str)
6363
timeout: Optional[float] = None
6464
kwargs: dict[str, Any] = field(default_factory=dict)
65-
db_id: Optional[str] = None
6665

6766
def __post_init__(self):
6867
if not self.uri:
6968
raise ValueError("URI must be provided for Milvus connection")
70-
if self.db_id is not None and self.db_name == "default":
71-
self.db_name = self.db_id
7269

7370
# Generate unique alias if not provided. One-to-one mapping between alias
7471
# and connection - each alias represents exactly one Milvus connection.

0 commit comments

Comments
 (0)