From 2313b1bab56a8d7bf27fa8be4e16849a6bba6f57 Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Thu, 6 Mar 2025 20:49:34 -0800 Subject: [PATCH] fix config to use the correct sigil for env var replacement --- llama_stack/providers/inline/vector_io/milvus/config.py | 4 ++-- llama_stack/providers/remote/vector_io/milvus/config.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/llama_stack/providers/inline/vector_io/milvus/config.py b/llama_stack/providers/inline/vector_io/milvus/config.py index 59fbbd6e9..0e11d8c7c 100644 --- a/llama_stack/providers/inline/vector_io/milvus/config.py +++ b/llama_stack/providers/inline/vector_io/milvus/config.py @@ -16,5 +16,5 @@ class MilvusVectorIOConfig(BaseModel): db_path: str @classmethod - def sample_run_config(cls) -> Dict[str, Any]: - return {"db_path": "{env.MILVUS_ENDPOINT}"} + def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> Dict[str, Any]: + return {"db_path": "${env.MILVUS_DB_PATH}"} diff --git a/llama_stack/providers/remote/vector_io/milvus/config.py b/llama_stack/providers/remote/vector_io/milvus/config.py index 1fe16c2ec..17da6b23d 100644 --- a/llama_stack/providers/remote/vector_io/milvus/config.py +++ b/llama_stack/providers/remote/vector_io/milvus/config.py @@ -18,5 +18,5 @@ class MilvusVectorIOConfig(BaseModel): consistency_level: str = "Strong" @classmethod - def sample_run_config(cls) -> Dict[str, Any]: - return {"uri": "{env.MILVUS_ENDPOINT}", "token": "{env.MILVUS_TOKEN}"} + def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> Dict[str, Any]: + return {"uri": "${env.MILVUS_ENDPOINT}", "token": "${env.MILVUS_TOKEN}"}