Browse Source
update
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
main
1 changed files with
6 additions and
3 deletions
-
milvus_client.py
|
|
@ -22,7 +22,7 @@ class MilvusClient(PyOperator): |
|
|
|
|
|
|
|
|
def __call__(self, *data): |
|
|
def __call__(self, *data): |
|
|
""" |
|
|
""" |
|
|
Insert data to Milvus. |
|
|
|
|
|
|
|
|
Insert one row to Milvus. |
|
|
|
|
|
|
|
|
Args: |
|
|
Args: |
|
|
data (`list`): |
|
|
data (`list`): |
|
|
@ -34,6 +34,9 @@ class MilvusClient(PyOperator): |
|
|
""" |
|
|
""" |
|
|
row = [] |
|
|
row = [] |
|
|
for item in data: |
|
|
for item in data: |
|
|
|
|
|
if isinstance(item, list): |
|
|
|
|
|
row.extend([[i] for i in item]) |
|
|
|
|
|
else: |
|
|
row.append([item]) |
|
|
row.append([item]) |
|
|
mr = self._collection.insert(row) |
|
|
mr = self._collection.insert(row) |
|
|
if mr.err_count > 0: |
|
|
if mr.err_count > 0: |
|
|
|
