diff --git a/llama_stack/providers/inline/vector_io/milvus/config.py b/llama_stack/providers/inline/vector_io/milvus/config.py index aa7ea6dcf..59fbbd6e9 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_config(cls) -> Dict[str, Any]: + def sample_run_config(cls) -> Dict[str, Any]: return {"db_path": "{env.MILVUS_ENDPOINT}"} diff --git a/llama_stack/providers/remote/vector_io/milvus/config.py b/llama_stack/providers/remote/vector_io/milvus/config.py index 34bd42987..1fe16c2ec 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_config(cls) -> Dict[str, Any]: + def sample_run_config(cls) -> Dict[str, Any]: return {"uri": "{env.MILVUS_ENDPOINT}", "token": "{env.MILVUS_TOKEN}"} diff --git a/tests/api/vector_io/test_vector_io.py b/tests/api/vector_io/test_vector_io.py deleted file mode 100644 index e69de29bb..000000000