define err

Signed-off-by: bigsheeper <[email protected]>
milvus-io · Mar 18, 2024 · ecbd6ad · ecbd6ad
1 parent f798a4b
commit ecbd6ad
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/client/data.go b/client/data.go
@@ -25,6 +25,7 @@ import (
 	"github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 	"github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 	"github.com/milvus-io/milvus-sdk-go/v2/entity"
+	"github.com/milvus-io/milvus-sdk-go/v2/merr"
 )
 
 const (
@@ -80,7 +81,7 @@ func (c *GrpcClient) HybridSearch(ctx context.Context, collName string, partitio
 
 	r, err := RetryOnMilvusErrors(ctx, func() (interface{}, error) {
 		return c.Service.HybridSearch(ctx, req)
-	}, OnMerrCodes(2200))
+	}, OnMerrCodes(merr.Code(merr.ErrInconsistentRequery)))
 	if err != nil {
 		return nil, err
 	}
@@ -122,7 +123,7 @@ func (c *GrpcClient) Search(ctx context.Context, collName string, partitions []s
 
 	r, err := RetryOnMilvusErrors(ctx, func() (interface{}, error) {
 		return c.Service.Search(ctx, req)
-	}, OnMerrCodes(2200))
+	}, OnMerrCodes(merr.Code(merr.ErrInconsistentRequery)))
 	if err != nil {
 		return nil, err
 	}

diff --git a/merr/errors.go b/merr/errors.go
@@ -144,6 +144,9 @@ var (
 
 	// import
 	ErrImportFailed = newMilvusError("importing data failed", 2100, false)
+
+	// Search/Query related
+	ErrInconsistentRequery = newMilvusError("inconsistent requery result", 2200, true)
 )
 
 type milvusError struct {