fix: Fix compactibility of exception with Milvus 2.2.x

Signed-off-by: bigsheeper <[email protected]>
milvus-io · Dec 2, 2023 · 1fbaac4 · 1fbaac4
1 parent d58447a
commit 1fbaac4
Show file tree

Hide file tree

Showing 2 changed files with 86 additions and 86 deletions.
diff --git a/pymilvus/client/asynch.py b/pymilvus/client/asynch.py
@@ -162,7 +162,7 @@ def exception(self):
 
 class SearchFuture(Future):
     def on_response(self, response: milvus_pb2.SearchResults):
-        if response.status.code == 0:
+        if response.status.code == 0 and response.status.error_code == 0:
             return SearchResult(response.results)
 
         status = response.status
@@ -172,7 +172,7 @@ def on_response(self, response: milvus_pb2.SearchResults):
 class MutationFuture(Future):
     def on_response(self, response: Any):
         status = response.status
-        if status.code == 0:
+        if status.code == 0 and status.error_code == 0:
             return MutationResult(response)
 
         status = response.status
@@ -181,7 +181,7 @@ def on_response(self, response: Any):
 
 class CreateIndexFuture(Future):
     def on_response(self, response: Any):
-        if response.code != 0:
+        if response.code != 0 or response.error_code != 0:
             raise MilvusException(response.code, response.reason, response.error_code)
 
         return Status(response.code, response.reason)
@@ -244,19 +244,19 @@ def exception(self):
 
 class FlushFuture(Future):
     def on_response(self, response: Any):
-        if response.status.code != 0:
+        if response.status.code != 0 or response.status.error_code != 0:
             raise MilvusException(
                 response.status.code, response.status.reason, response.status.error_code
             )
 
 
 class LoadCollectionFuture(Future):
     def on_response(self, response: Any):
-        if response.code != 0:
+        if response.code != 0 or response.error_code != 0:
             raise MilvusException(response.code, response.reason, response.error_code)
 
 
 class LoadPartitionsFuture(Future):
     def on_response(self, response: Any):
-        if response.code != 0:
+        if response.code != 0 or response.error_code != 0:
             raise MilvusException(response.code, response.reason, response.error_code)