ann-insert
copied
support pymilvus 2.0.2
@ -24,7 +24,7 @@ import towhee
from towhee import ops
p = (
towhee.pipe.inputs('vec')
towhee.pipe.input('vec')
.map('vec', (), ops.ann_insert.milvus_client(host='127.0.0.1', port='19530', collection_name='test_collection'))
.output()
)
@ -43,7 +43,7 @@ class MilvusClient(PyOperator):
else:
row.append([item])
mr = self._collection.insert(row)
if mr.err_count > 0:
if mr.insert_count != len(row[0]):
raise RuntimeError("Insert to milvus failed")
return mr