logo
Browse Source

Support uri and token

Signed-off-by: Kaiyuan.Hu <kaiyuan.hu@zilliz.com>
main
Kaiyuan.Hu 2 years ago
parent
commit
c96e1a5d5b
  1. 15
      milvus_client.py

15
milvus_client.py

@ -12,16 +12,23 @@ class MilvusClient(PyOperator):
Milvus ANN index class. 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._host = host
self._port = port self._port = port
self._uri = uri
self._collection_name = collection_name self._collection_name = collection_name
self._connect_name = uuid.uuid4().hex 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, connections.connect(alias=self._connect_name, host=self._host, port=self._port,
user=user, password=password, secure=True) 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) self._collection = Collection(self._collection_name, using=self._connect_name)
def __call__(self, *data): def __call__(self, *data):

Loading…
Cancel
Save