diff --git a/milvus_client.py b/milvus_client.py
index 637e5ca..2c8d248 100644
--- a/milvus_client.py
+++ b/milvus_client.py
@@ -12,16 +12,23 @@ class MilvusClient(PyOperator):
     Milvus ANN index class.
     """
 
-    def __init__(self, host: str, port: int, collection_name: str, user: str = None, password: str = None):
+    def __init__(
+        self, host: str = 'localhost', port: int = 19530, collection_name: str = None,  uri: str = None, user: str = None, password: str = None, token: str = None
+    ):
         self._host = host
         self._port = port
+        self._uri = uri
         self._collection_name = collection_name
         self._connect_name = uuid.uuid4().hex
-        if None in [user, password]:
-            connections.connect(alias=self._connect_name, host=self._host, port=self._port)
-        else:
+        if uri and token:
+            connections.connect(alias=self._connect_name, uri=self._uri, token=token, secure=True)
+        elif user and password:
             connections.connect(alias=self._connect_name, host=self._host, port=self._port,
                                 user=user, password=password, secure=True)
+        else:
+            connections.connect(alias=self._connect_name, host=self._host, port=self._port)
+        
+    
         self._collection = Collection(self._collection_name, using=self._connect_name)
 
     def __call__(self, *data):