Donot retry on RESOURCE_EXHAUSTED

See also: milvus-io/milvus#25695 Signed-off-by: yangxuan <[email protected]>
milvus-io · Jul 24, 2023 · 0719a39 · 0719a39
1 parent 76c6326
commit 0719a39
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/pymilvus/client/grpc_handler.py b/pymilvus/client/grpc_handler.py
@@ -721,7 +721,7 @@ def create_index(
             if field_name != fields["name"]:
                 continue
             valid_field = True
-            if fields["type"] != DataType.FLOAT_VECTOR and fields["type"] != DataType.BINARY_VECTOR:
+            if fields["type"] not in {DataType.FLOAT_VECTOR, DataType.BINARY_VECTOR}:
                 break
 
         if not valid_field:

diff --git a/pymilvus/decorators.py b/pymilvus/decorators.py
@@ -74,6 +74,7 @@ def timeout(start_time: Optional[float] = None) -> bool:
                         grpc.StatusCode.UNAUTHENTICATED,
                         grpc.StatusCode.INVALID_ARGUMENT,
                         grpc.StatusCode.ALREADY_EXISTS,
+                        grpc.StatusCode.RESOURCE_EXHAUSTED,
                     ):
                         raise MilvusException(message=str(e)) from e
                     if timeout(start_time):