From 024e938c872edc48a2c5f689ccdb3b446a0ae547 Mon Sep 17 00:00:00 2001
From: "junjie.jiang" <junjie.jiang@zilliz.com>
Date: Fri, 21 Apr 2023 11:38:05 +0800
Subject: [PATCH] support pymilvus 2.0.2

---
 README.md        | 2 +-
 milvus_client.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index f70049e..b0d94f6 100644
--- a/README.md
+++ b/README.md
@@ -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()
         )
diff --git a/milvus_client.py b/milvus_client.py
index 4f07a85..637e5ca 100644
--- a/milvus_client.py
+++ b/milvus_client.py
@@ -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