Fix bug

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
1 year ago · 332d50c42d
1 changed files with 4 additions and 0 deletions
--- a/milvus_client.py
+++ b/milvus_client.py
@ -38,6 +38,10 @@ class MilvusClient(PyOperator):
        else:
            self._connection_args['secure'] = False

+        self._connection_args['user'] = user
+        self._connection_args['password'] = password
+        self._connection_args['token'] = token
+
        connections.connect(**self._connection_args)

    def __call__(self, collection_name: str, *data):