fix config to use the correct sigil for env var replacement

This commit is contained in:
Ashwin Bharambe 2025-03-06 20:49:34 -08:00
parent 2f45b9f2f7
commit 2313b1bab5
2 changed files with 4 additions and 4 deletions

View file

@ -16,5 +16,5 @@ class MilvusVectorIOConfig(BaseModel):
db_path: str db_path: str
@classmethod @classmethod
def sample_run_config(cls) -> Dict[str, Any]: def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> Dict[str, Any]:
return {"db_path": "{env.MILVUS_ENDPOINT}"} return {"db_path": "${env.MILVUS_DB_PATH}"}

View file

@ -18,5 +18,5 @@ class MilvusVectorIOConfig(BaseModel):
consistency_level: str = "Strong" consistency_level: str = "Strong"
@classmethod @classmethod
def sample_run_config(cls) -> Dict[str, Any]: def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> Dict[str, Any]:
return {"uri": "{env.MILVUS_ENDPOINT}", "token": "{env.MILVUS_TOKEN}"} return {"uri": "${env.MILVUS_ENDPOINT}", "token": "${env.MILVUS_TOKEN}"}