From 67db749d15c7b8e2c3c401e632cbe3cd6cf1537e Mon Sep 17 00:00:00 2001 From: Divya Date: Tue, 29 Apr 2025 22:29:12 -0700 Subject: [PATCH] Made Changes based on review comments --- .../providers/remote/vector_io/milvus/config.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) 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}"}