From 2c682b086a829dcab2206aae78f9ddead42eb5fc Mon Sep 17 00:00:00 2001 From: Divya Date: Wed, 23 Apr 2025 04:09:49 -0700 Subject: [PATCH] Adding TLS support one-way --- .../providers/remote/vector_io/milvus/config.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/llama_stack/providers/remote/vector_io/milvus/config.py b/llama_stack/providers/remote/vector_io/milvus/config.py index 3d25e9c49..191aa34b8 100644 --- a/llama_stack/providers/remote/vector_io/milvus/config.py +++ b/llama_stack/providers/remote/vector_io/milvus/config.py @@ -16,7 +16,12 @@ 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 @classmethod - def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> 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}","secure": True, + "server_pem_path": "/path/to/ca.crt"} +