diff --git a/llama_stack/providers/remote/vector_io/milvus/config.py b/llama_stack/providers/remote/vector_io/milvus/config.py index 191aa34b8..09387dd79 100644 --- a/llama_stack/providers/remote/vector_io/milvus/config.py +++ b/llama_stack/providers/remote/vector_io/milvus/config.py @@ -16,12 +16,13 @@ class MilvusVectorIOConfig(BaseModel): uri: str token: str | None = None consistency_level: str = "Strong" - # TLS-related options - secure: bool = False - server_pem_path: Optional[str] = None + + class Config: + extra = "allow" + @classmethod def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> Dict[str, Any]: - return {"uri": "${env.MILVUS_ENDPOINT}", "token": "${env.MILVUS_TOKEN}","secure": True, - "server_pem_path": "/path/to/ca.crt"} + + return {"uri": "${env.MILVUS_ENDPOINT}", "token": "${env.MILVUS_TOKEN}"}