logo
Browse Source

support pymilvus 2.0.2

main
junjie.jiang 2 years ago
parent
commit
024e938c87
  1. 2
      README.md
  2. 2
      milvus_client.py

2
README.md

@ -24,7 +24,7 @@ import towhee
from towhee import ops from towhee import ops
p = ( 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')) .map('vec', (), ops.ann_insert.milvus_client(host='127.0.0.1', port='19530', collection_name='test_collection'))
.output() .output()
) )

2
milvus_client.py

@ -43,7 +43,7 @@ class MilvusClient(PyOperator):
else: else:
row.append([item]) row.append([item])
mr = self._collection.insert(row) mr = self._collection.insert(row)
if mr.err_count > 0:
if mr.insert_count != len(row[0]):
raise RuntimeError("Insert to milvus failed") raise RuntimeError("Insert to milvus failed")
return mr return mr

Loading…
Cancel
Save