Made Changes based on review comments

This commit is contained in:
Divya 2025-04-29 22:29:12 -07:00
parent 2c682b086a
commit 67db749d15

View file

@ -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}"}