Repository: milvus-io/milvus-sdk-go
Pull request #489 (2023-06-09 09:09:18)

Use milvus-proto/go-api/v2 for master
/kind improvement
See also milvus-io/milvus#24717

sre-ci-robot: [APPROVALNOTIFIER] This PR is **APPROVED**

This pull-request has been approved by: *<a href="https://github.com/milvus-io/milvus-sdk-go/pull/489#" title="Author self-approved">congqixia</a>*

The full list of commands accepted by this bot can be found [here](https://go.k8s.io/bot-commands?repo=milvus-io%2Fmilvus-sdk-go).

The pull request process is described [here](https://git.k8s.io/community/contributors/guide/owners.md#the-code-review-process)

<details >
Needs approval from an approver in each of these files:

- ~~[OWNERS](https://github.com/milvus-io/milvus-sdk-go/blob/master/OWNERS)~~ [congqixia]

Approvers can indicate their approval by writing `/approve` in a comment
Approvers can cancel approval by writing `/approve cancel` in a comment
</details>
<!-- META={"approvers":[]} -->
codecov[bot]: ## [Codecov](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/489?src=pr&el=h1&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io) Report
> Merging [#489](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/489?src=pr&el=desc&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io) (4153be2) into [master](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/commit/cc1c52d2a50ab6dd84053dad81397308dff9c164?el=desc&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io) (cc1c52d) will **not change** coverage.
> The diff coverage is `100.00%`.

[![Impacted file tree graph](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/489/graphs/tree.svg?width=650&height=150&src=pr&token=WJK43PV390&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io)](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/489?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io)

```diff
@@           Coverage Diff           @@
##           master     #489   +/-   ##
=======================================
  Coverage   84.09%   84.09%           
=======================================
  Files          36       36           
  Lines        3319     3319           
=======================================
  Hits         2791     2791           
  Misses        425      425           
  Partials      103      103           
```


| [Impacted Files](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/489?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io) | Coverage Δ | |
|---|---|---|
| [client/admin.go](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/489?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io#diff-Y2xpZW50L2FkbWluLmdv) | `83.33% <ø> (ø)` | |
| [client/alias.go](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/489?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io#diff-Y2xpZW50L2FsaWFzLmdv) | `100.00% <ø> (ø)` | |
| [client/authentication.go](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/489?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io#diff-Y2xpZW50L2F1dGhlbnRpY2F0aW9uLmdv) | `100.00% <ø> (ø)` | |
| [client/collection.go](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/489?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io#diff-Y2xpZW50L2NvbGxlY3Rpb24uZ28=) | `87.81% <ø> (ø)` | |
| [client/connect.go](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/489?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io#diff-Y2xpZW50L2Nvbm5lY3QuZ28=) | `80.51% <ø> (ø)` | |
| [client/data.go](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/489?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io#diff-Y2xpZW50L2RhdGEuZ28=) | `69.91% <ø> (ø)` | |
| [client/index.go](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/489?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io#diff-Y2xpZW50L2luZGV4Lmdv) | `77.95% <ø> (ø)` | |
| [client/insert.go](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/489?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io#diff-Y2xpZW50L2luc2VydC5nbw==) | `78.87% <ø> (ø)` | |
| [client/maintainance.go](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/489?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io#diff-Y2xpZW50L21haW50YWluYW5jZS5nbw==) | `95.55% <ø> (ø)` | |
| [client/options.go](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/489?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io#diff-Y2xpZW50L29wdGlvbnMuZ28=) | `68.00% <ø> (ø)` | |
| ... and [13 more](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/489?src=pr&el=tree-more&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io) | |

jiaoew1991: /lgtm

Patch:
diff --git a/Makefile b/Makefile
index 10a05d8..adb5f37 100644
--- a/Makefile
+++ b/Makefile
@@ -40,7 +40,7 @@ generate-proto: check-protoc-version
 
 generate-mockery: install-tool
 	@echo "generating mockery milvus service server"
-	@$(INSTALL_PATH)/mockery --srcpkg=github.com/milvus-io/milvus-proto/go-api/milvuspb --name=MilvusServiceServer --output=mocks --outpkg=mocks --with-expecter
+	@$(INSTALL_PATH)/mockery --srcpkg=github.com/milvus-io/milvus-proto/go-api/v2/milvuspb --name=MilvusServiceServer --output=mocks --outpkg=mocks --with-expecter
 
 static-check:
 	@echo "Running $@ check:"
diff --git a/client/admin.go b/client/admin.go
index 8f0dc6c..2fad21f 100644
--- a/client/admin.go
+++ b/client/admin.go
@@ -14,7 +14,7 @@ package client
 import (
 	"context"
 
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 )
 
 // GetVersion returns milvus server version information.
diff --git a/client/alias.go b/client/alias.go
index 9a29fd4..c0bd410 100644
--- a/client/alias.go
+++ b/client/alias.go
@@ -19,7 +19,7 @@ package client
 import (
 	"context"
 
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 )
 
 // CreateAlias creates an alias for collection
diff --git a/client/authentication.go b/client/authentication.go
index fc844ac..e407529 100644
--- a/client/authentication.go
+++ b/client/authentication.go
@@ -3,7 +3,7 @@ package client
 import (
 	"context"
 
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 	"github.com/milvus-io/milvus-sdk-go/v2/internal/utils/crypto"
 )
 
diff --git a/client/collection.go b/client/collection.go
index 0992457..74f22f8 100644
--- a/client/collection.go
+++ b/client/collection.go
@@ -21,8 +21,8 @@ import (
 	"github.com/golang/protobuf/proto"
 	"github.com/milvus-io/milvus-sdk-go/v2/entity"
 
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 )
 
 // handles response status
diff --git a/client/connect.go b/client/connect.go
index 03cad89..0420c42 100644
--- a/client/connect.go
+++ b/client/connect.go
@@ -22,8 +22,8 @@ import (
 
 	grpc_middleware "github.com/grpc-ecosystem/go-grpc-middleware"
 	grpc_retry "github.com/grpc-ecosystem/go-grpc-middleware/retry"
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/codes"
 	"google.golang.org/grpc/credentials"
diff --git a/client/data.go b/client/data.go
index 0e52dd3..147e8ba 100644
--- a/client/data.go
+++ b/client/data.go
@@ -21,9 +21,9 @@ import (
 
 	"github.com/cockroachdb/errors"
 
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
-	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
+	schema "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 	"github.com/milvus-io/milvus-sdk-go/v2/entity"
 )
 
diff --git a/client/index.go b/client/index.go
index 9c8fd4c..c7c0ac0 100644
--- a/client/index.go
+++ b/client/index.go
@@ -16,8 +16,8 @@ import (
 	"fmt"
 	"time"
 
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 	"github.com/milvus-io/milvus-sdk-go/v2/entity"
 )
 
diff --git a/client/insert.go b/client/insert.go
index c76562b..8f756c9 100644
--- a/client/insert.go
+++ b/client/insert.go
@@ -19,9 +19,9 @@ import (
 
 	"github.com/cockroachdb/errors"
 	"github.com/golang/protobuf/proto"
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
-	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
+	schema "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 
 	"github.com/milvus-io/milvus-sdk-go/v2/entity"
 )
diff --git a/client/maintainance.go b/client/maintainance.go
index 757638e..0af2735 100644
--- a/client/maintainance.go
+++ b/client/maintainance.go
@@ -20,7 +20,7 @@ import (
 	"context"
 	"time"
 
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 	"github.com/milvus-io/milvus-sdk-go/v2/entity"
 	"github.com/milvus-io/milvus-sdk-go/v2/internal/utils/tso"
 )
diff --git a/client/options.go b/client/options.go
index f992490..5eaa460 100644
--- a/client/options.go
+++ b/client/options.go
@@ -16,8 +16,8 @@ import (
 
 	"github.com/cockroachdb/errors"
 
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 	"github.com/milvus-io/milvus-sdk-go/v2/entity"
 )
 
diff --git a/client/partition.go b/client/partition.go
index 5cba323..d9cd338 100644
--- a/client/partition.go
+++ b/client/partition.go
@@ -18,8 +18,8 @@ import (
 
 	"github.com/cockroachdb/errors"
 
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 	"github.com/milvus-io/milvus-sdk-go/v2/entity"
 )
 
diff --git a/client/rate_limit_interceptor.go b/client/rate_limit_interceptor.go
index eeaeb4c..d59f847 100644
--- a/client/rate_limit_interceptor.go
+++ b/client/rate_limit_interceptor.go
@@ -25,8 +25,8 @@ import (
 	"google.golang.org/grpc/status"
 
 	grpc_retry "github.com/grpc-ecosystem/go-grpc-middleware/retry"
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 )
 
 // ref: https://github.com/grpc-ecosystem/go-grpc-middleware
diff --git a/client/rbac.go b/client/rbac.go
index 2dad952..70acb2a 100644
--- a/client/rbac.go
+++ b/client/rbac.go
@@ -19,8 +19,8 @@ package client
 import (
 	"context"
 
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 	"github.com/milvus-io/milvus-sdk-go/v2/entity"
 )
 
diff --git a/client/resource_group.go b/client/resource_group.go
index cb4d99f..b957d39 100644
--- a/client/resource_group.go
+++ b/client/resource_group.go
@@ -14,7 +14,7 @@ package client
 import (
 	"context"
 
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 	"github.com/milvus-io/milvus-sdk-go/v2/entity"
 )
 
diff --git a/client/row.go b/client/row.go
index 6145028..1a17060 100644
--- a/client/row.go
+++ b/client/row.go
@@ -8,8 +8,8 @@ import (
 	"github.com/cockroachdb/errors"
 
 	"github.com/golang/protobuf/proto"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
-	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
+	schema "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 	"github.com/milvus-io/milvus-sdk-go/v2/entity"
 )
 
diff --git a/entity/columns.go b/entity/columns.go
index 7ce8d50..d88fb3d 100644
--- a/entity/columns.go
+++ b/entity/columns.go
@@ -16,7 +16,7 @@ import (
 	"fmt"
 	"math"
 
-	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+	schema "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 
 	"github.com/cockroachdb/errors"
 )
diff --git a/entity/columns_json.go b/entity/columns_json.go
index 8cb2226..8b03518 100644
--- a/entity/columns_json.go
+++ b/entity/columns_json.go
@@ -4,7 +4,7 @@ import (
 	"fmt"
 
 	"github.com/cockroachdb/errors"
-	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+	schema "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 )
 
 var _ (Column) = (*ColumnJSONBytes)(nil)
diff --git a/entity/columns_scalar_gen.go b/entity/columns_scalar_gen.go
index 7378537..67d91f6 100755
--- a/entity/columns_scalar_gen.go
+++ b/entity/columns_scalar_gen.go
@@ -1,13 +1,13 @@
 // Code generated by go generate; DO NOT EDIT
-// This file is generated by go generate 
+// This file is generated by go generate
 
-package entity 
+package entity
 
 import (
 	"errors"
 	"fmt"
 
-	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+	schema "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 )
 
 // ColumnBool generated columns type for Bool
@@ -44,7 +44,7 @@ func (c *ColumnBool) Get(idx int) (interface{}, error) {
 // FieldData return column data mapped to schema.FieldData
 func (c *ColumnBool) FieldData() *schema.FieldData {
 	fd := &schema.FieldData{
-		Type: schema.DataType_Bool,
+		Type:      schema.DataType_Bool,
 		FieldName: c.name,
 	}
 	data := make([]bool, 0, c.Len())
@@ -74,7 +74,7 @@ func (c *ColumnBool) ValueByIdx(idx int) (bool, error) {
 }
 
 // AppendValue append value into column
-func(c *ColumnBool) AppendValue(i interface{}) error {
+func (c *ColumnBool) AppendValue(i interface{}) error {
 	v, ok := i.(bool)
 	if !ok {
 		return fmt.Errorf("invalid type, expected bool, got %T", i)
@@ -91,8 +91,8 @@ func (c *ColumnBool) Data() []bool {
 
 // NewColumnBool auto generated constructor
 func NewColumnBool(name string, values []bool) *ColumnBool {
-	return &ColumnBool {
-		name: name,
+	return &ColumnBool{
+		name:   name,
 		values: values,
 	}
 }
@@ -131,7 +131,7 @@ func (c *ColumnInt8) Get(idx int) (interface{}, error) {
 // FieldData return column data mapped to schema.FieldData
 func (c *ColumnInt8) FieldData() *schema.FieldData {
 	fd := &schema.FieldData{
-		Type: schema.DataType_Int8,
+		Type:      schema.DataType_Int8,
 		FieldName: c.name,
 	}
 	data := make([]int32, 0, c.Len())
@@ -161,7 +161,7 @@ func (c *ColumnInt8) ValueByIdx(idx int) (int8, error) {
 }
 
 // AppendValue append value into column
-func(c *ColumnInt8) AppendValue(i interface{}) error {
+func (c *ColumnInt8) AppendValue(i interface{}) error {
 	v, ok := i.(int8)
 	if !ok {
 		return fmt.Errorf("invalid type, expected int8, got %T", i)
@@ -178,8 +178,8 @@ func (c *ColumnInt8) Data() []int8 {
 
 // NewColumnInt8 auto generated constructor
 func NewColumnInt8(name string, values []int8) *ColumnInt8 {
-	return &ColumnInt8 {
-		name: name,
+	return &ColumnInt8{
+		name:   name,
 		values: values,
 	}
 }
@@ -218,7 +218,7 @@ func (c *ColumnInt16) Get(idx int) (interface{}, error) {
 // FieldData return column data mapped to schema.FieldData
 func (c *ColumnInt16) FieldData() *schema.FieldData {
 	fd := &schema.FieldData{
-		Type: schema.DataType_Int16,
+		Type:      schema.DataType_Int16,
 		FieldName: c.name,
 	}
 	data := make([]int32, 0, c.Len())
@@ -248,7 +248,7 @@ func (c *ColumnInt16) ValueByIdx(idx int) (int16, error) {
 }
 
 // AppendValue append value into column
-func(c *ColumnInt16) AppendValue(i interface{}) error {
+func (c *ColumnInt16) AppendValue(i interface{}) error {
 	v, ok := i.(int16)
 	if !ok {
 		return fmt.Errorf("invalid type, expected int16, got %T", i)
@@ -265,8 +265,8 @@ func (c *ColumnInt16) Data() []int16 {
 
 // NewColumnInt16 auto generated constructor
 func NewColumnInt16(name string, values []int16) *ColumnInt16 {
-	return &ColumnInt16 {
-		name: name,
+	return &ColumnInt16{
+		name:   name,
 		values: values,
 	}
 }
@@ -305,7 +305,7 @@ func (c *ColumnInt32) Get(idx int) (interface{}, error) {
 // FieldData return column data mapped to schema.FieldData
 func (c *ColumnInt32) FieldData() *schema.FieldData {
 	fd := &schema.FieldData{
-		Type: schema.DataType_Int32,
+		Type:      schema.DataType_Int32,
 		FieldName: c.name,
 	}
 	data := make([]int32, 0, c.Len())
@@ -335,7 +335,7 @@ func (c *ColumnInt32) ValueByIdx(idx int) (int32, error) {
 }
 
 // AppendValue append value into column
-func(c *ColumnInt32) AppendValue(i interface{}) error {
+func (c *ColumnInt32) AppendValue(i interface{}) error {
 	v, ok := i.(int32)
 	if !ok {
 		return fmt.Errorf("invalid type, expected int32, got %T", i)
@@ -352,8 +352,8 @@ func (c *ColumnInt32) Data() []int32 {
 
 // NewColumnInt32 auto generated constructor
 func NewColumnInt32(name string, values []int32) *ColumnInt32 {
-	return &ColumnInt32 {
-		name: name,
+	return &ColumnInt32{
+		name:   name,
 		values: values,
 	}
 }
@@ -392,7 +392,7 @@ func (c *ColumnInt64) Get(idx int) (interface{}, error) {
 // FieldData return column data mapped to schema.FieldData
 func (c *ColumnInt64) FieldData() *schema.FieldData {
 	fd := &schema.FieldData{
-		Type: schema.DataType_Int64,
+		Type:      schema.DataType_Int64,
 		FieldName: c.name,
 	}
 	data := make([]int64, 0, c.Len())
@@ -422,7 +422,7 @@ func (c *ColumnInt64) ValueByIdx(idx int) (int64, error) {
 }
 
 // AppendValue append value into column
-func(c *ColumnInt64) AppendValue(i interface{}) error {
+func (c *ColumnInt64) AppendValue(i interface{}) error {
 	v, ok := i.(int64)
 	if !ok {
 		return fmt.Errorf("invalid type, expected int64, got %T", i)
@@ -439,8 +439,8 @@ func (c *ColumnInt64) Data() []int64 {
 
 // NewColumnInt64 auto generated constructor
 func NewColumnInt64(name string, values []int64) *ColumnInt64 {
-	return &ColumnInt64 {
-		name: name,
+	return &ColumnInt64{
+		name:   name,
 		values: values,
 	}
 }
@@ -479,7 +479,7 @@ func (c *ColumnFloat) Get(idx int) (interface{}, error) {
 // FieldData return column data mapped to schema.FieldData
 func (c *ColumnFloat) FieldData() *schema.FieldData {
 	fd := &schema.FieldData{
-		Type: schema.DataType_Float,
+		Type:      schema.DataType_Float,
 		FieldName: c.name,
 	}
 	data := make([]float32, 0, c.Len())
@@ -509,7 +509,7 @@ func (c *ColumnFloat) ValueByIdx(idx int) (float32, error) {
 }
 
 // AppendValue append value into column
-func(c *ColumnFloat) AppendValue(i interface{}) error {
+func (c *ColumnFloat) AppendValue(i interface{}) error {
 	v, ok := i.(float32)
 	if !ok {
 		return fmt.Errorf("invalid type, expected float32, got %T", i)
@@ -526,8 +526,8 @@ func (c *ColumnFloat) Data() []float32 {
 
 // NewColumnFloat auto generated constructor
 func NewColumnFloat(name string, values []float32) *ColumnFloat {
-	return &ColumnFloat {
-		name: name,
+	return &ColumnFloat{
+		name:   name,
 		values: values,
 	}
 }
@@ -566,7 +566,7 @@ func (c *ColumnDouble) Get(idx int) (interface{}, error) {
 // FieldData return column data mapped to schema.FieldData
 func (c *ColumnDouble) FieldData() *schema.FieldData {
 	fd := &schema.FieldData{
-		Type: schema.DataType_Double,
+		Type:      schema.DataType_Double,
 		FieldName: c.name,
 	}
 	data := make([]float64, 0, c.Len())
@@ -596,7 +596,7 @@ func (c *ColumnDouble) ValueByIdx(idx int) (float64, error) {
 }
 
 // AppendValue append value into column
-func(c *ColumnDouble) AppendValue(i interface{}) error {
+func (c *ColumnDouble) AppendValue(i interface{}) error {
 	v, ok := i.(float64)
 	if !ok {
 		return fmt.Errorf("invalid type, expected float64, got %T", i)
@@ -613,8 +613,8 @@ func (c *ColumnDouble) Data() []float64 {
 
 // NewColumnDouble auto generated constructor
 func NewColumnDouble(name string, values []float64) *ColumnDouble {
-	return &ColumnDouble {
-		name: name,
+	return &ColumnDouble{
+		name:   name,
 		values: values,
 	}
 }
@@ -653,7 +653,7 @@ func (c *ColumnString) Get(idx int) (interface{}, error) {
 // FieldData return column data mapped to schema.FieldData
 func (c *ColumnString) FieldData() *schema.FieldData {
 	fd := &schema.FieldData{
-		Type: schema.DataType_String,
+		Type:      schema.DataType_String,
 		FieldName: c.name,
 	}
 	data := make([]string, 0, c.Len())
@@ -683,7 +683,7 @@ func (c *ColumnString) ValueByIdx(idx int) (string, error) {
 }
 
 // AppendValue append value into column
-func(c *ColumnString) AppendValue(i interface{}) error {
+func (c *ColumnString) AppendValue(i interface{}) error {
 	v, ok := i.(string)
 	if !ok {
 		return fmt.Errorf("invalid type, expected string, got %T", i)
@@ -700,9 +700,8 @@ func (c *ColumnString) Data() []string {
 
 // NewColumnString auto generated constructor
 func NewColumnString(name string, values []string) *ColumnString {
-	return &ColumnString {
-		name: name,
+	return &ColumnString{
+		name:   name,
 		values: values,
 	}
 }
-
diff --git a/entity/columns_varchar.go b/entity/columns_varchar.go
index 6873f80..5615647 100644
--- a/entity/columns_varchar.go
+++ b/entity/columns_varchar.go
@@ -4,7 +4,7 @@ import (
 	"errors"
 	"fmt"
 
-	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+	schema "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 )
 
 // ColumnVarChar generated columns type for VarChar
diff --git a/entity/columns_vector_gen.go b/entity/columns_vector_gen.go
index 048158c..87a874c 100755
--- a/entity/columns_vector_gen.go
+++ b/entity/columns_vector_gen.go
@@ -5,12 +5,11 @@ package entity
 import (
 	"fmt"
 
-	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+	schema "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 
 	"github.com/cockroachdb/errors"
 )
 
-
 // ColumnBinaryVector generated columns type for BinaryVector
 type ColumnBinaryVector struct {
 	ColumnBase
@@ -30,7 +29,7 @@ func (c *ColumnBinaryVector) Type() FieldType {
 }
 
 // Len returns column data length
-func (c * ColumnBinaryVector) Len() int {
+func (c *ColumnBinaryVector) Len() int {
 	return len(c.values)
 }
 
@@ -48,7 +47,7 @@ func (c *ColumnBinaryVector) Get(idx int) (interface{}, error) {
 }
 
 // AppendValue append value into column
-func(c *ColumnBinaryVector) AppendValue(i interface{}) error {
+func (c *ColumnBinaryVector) AppendValue(i interface{}) error {
 	v, ok := i.([]byte)
 	if !ok {
 		return fmt.Errorf("invalid type, expected []byte, got %T", i)
@@ -66,11 +65,11 @@ func (c *ColumnBinaryVector) Data() [][]byte {
 // FieldData return column data mapped to schema.FieldData
 func (c *ColumnBinaryVector) FieldData() *schema.FieldData {
 	fd := &schema.FieldData{
-		Type: schema.DataType_BinaryVector,
+		Type:      schema.DataType_BinaryVector,
 		FieldName: c.name,
 	}
 
-	data := make([]byte, 0, len(c.values)* c.dim)
+	data := make([]byte, 0, len(c.values)*c.dim)
 
 	for _, vector := range c.values {
 		data = append(data, vector...)
@@ -79,11 +78,10 @@ func (c *ColumnBinaryVector) FieldData() *schema.FieldData {
 	fd.Field = &schema.FieldData_Vectors{
 		Vectors: &schema.VectorField{
 			Dim: int64(c.dim),
-			
+
 			Data: &schema.VectorField_BinaryVector{
 				BinaryVector: data,
 			},
-			
 		},
 	}
 	return fd
@@ -91,7 +89,7 @@ func (c *ColumnBinaryVector) FieldData() *schema.FieldData {
 
 // NewColumnBinaryVector auto generated constructor
 func NewColumnBinaryVector(name string, dim int, values [][]byte) *ColumnBinaryVector {
-	return &ColumnBinaryVector {
+	return &ColumnBinaryVector{
 		name:   name,
 		dim:    dim,
 		values: values,
@@ -117,7 +115,7 @@ func (c *ColumnFloatVector) Type() FieldType {
 }
 
 // Len returns column data length
-func (c * ColumnFloatVector) Len() int {
+func (c *ColumnFloatVector) Len() int {
 	return len(c.values)
 }
 
@@ -135,7 +133,7 @@ func (c *ColumnFloatVector) Get(idx int) (interface{}, error) {
 }
 
 // AppendValue append value into column
-func(c *ColumnFloatVector) AppendValue(i interface{}) error {
+func (c *ColumnFloatVector) AppendValue(i interface{}) error {
 	v, ok := i.([]float32)
 	if !ok {
 		return fmt.Errorf("invalid type, expected []float32, got %T", i)
@@ -153,11 +151,11 @@ func (c *ColumnFloatVector) Data() [][]float32 {
 // FieldData return column data mapped to schema.FieldData
 func (c *ColumnFloatVector) FieldData() *schema.FieldData {
 	fd := &schema.FieldData{
-		Type: schema.DataType_FloatVector,
+		Type:      schema.DataType_FloatVector,
 		FieldName: c.name,
 	}
 
-	data := make([]float32, 0, len(c.values)* c.dim)
+	data := make([]float32, 0, len(c.values)*c.dim)
 
 	for _, vector := range c.values {
 		data = append(data, vector...)
@@ -166,13 +164,12 @@ func (c *ColumnFloatVector) FieldData() *schema.FieldData {
 	fd.Field = &schema.FieldData_Vectors{
 		Vectors: &schema.VectorField{
 			Dim: int64(c.dim),
-			
+
 			Data: &schema.VectorField_FloatVector{
 				FloatVector: &schema.FloatArray{
 					Data: data,
 				},
 			},
-			
 		},
 	}
 	return fd
@@ -180,10 +177,9 @@ func (c *ColumnFloatVector) FieldData() *schema.FieldData {
 
 // NewColumnFloatVector auto generated constructor
 func NewColumnFloatVector(name string, dim int, values [][]float32) *ColumnFloatVector {
-	return &ColumnFloatVector {
+	return &ColumnFloatVector{
 		name:   name,
 		dim:    dim,
 		values: values,
 	}
 }
-
diff --git a/entity/gen/gen.go b/entity/gen/gen.go
index 68510f5..1faef61 100644
--- a/entity/gen/gen.go
+++ b/entity/gen/gen.go
@@ -32,7 +32,7 @@ import (
 	"errors"
 	"fmt"
 
-	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+	schema "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 )
 {{ range .Types }}{{with .}}
 // Column{{.TypeName}} generated columns type for {{.TypeName}}
@@ -130,7 +130,7 @@ package entity
 import (
 	"fmt"
 
-	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+	schema "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 
 	"github.com/cockroachdb/errors"
 )
@@ -243,7 +243,7 @@ import (
 	"time"
 
 	"github.com/stretchr/testify/assert"
-	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+	schema "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 )
 {{ range .Types }}{{with.}}
 func TestColumn{{.TypeName}}(t *testing.T) {
@@ -364,7 +364,7 @@ import (
 	"testing"
 	"time"
 
-	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+	schema "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 	"github.com/stretchr/testify/assert"
 )
 {{range .Types}}{{with .}}
diff --git a/entity/index.go b/entity/index.go
index 71cfde7..04193de 100644
--- a/entity/index.go
+++ b/entity/index.go
@@ -11,7 +11,7 @@
 
 package entity
 
-import common "github.com/milvus-io/milvus-proto/go-api/commonpb"
+import common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
 
 //go:generate go run genidx/genidx.go
 
diff --git a/entity/rbac.go b/entity/rbac.go
index 0e1d14d..32819d9 100644
--- a/entity/rbac.go
+++ b/entity/rbac.go
@@ -1,6 +1,6 @@
 package entity
 
-import common "github.com/milvus-io/milvus-proto/go-api/commonpb"
+import common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
 
 // User is the model for RBAC user object.
 type User struct {
diff --git a/entity/schema.go b/entity/schema.go
index e46310d..9dc79ee 100644
--- a/entity/schema.go
+++ b/entity/schema.go
@@ -14,8 +14,8 @@ package entity
 import (
 	"strconv"
 
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	schema "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 )
 
 const (
diff --git a/entity/segment.go b/entity/segment.go
index fef5cb4..d920c80 100644
--- a/entity/segment.go
+++ b/entity/segment.go
@@ -12,7 +12,7 @@
 package entity
 
 import (
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
 )
 
 // Segment represent segment in milvus
diff --git a/go.mod b/go.mod
index 99df4dd..d302324 100644
--- a/go.mod
+++ b/go.mod
@@ -3,16 +3,17 @@ module github.com/milvus-io/milvus-sdk-go/v2
 go 1.17
 
 require (
+	github.com/cockroachdb/errors v1.9.1
 	github.com/golang/protobuf v1.5.2
 	github.com/grpc-ecosystem/go-grpc-middleware v1.3.0
-	github.com/milvus-io/milvus-proto/go-api v0.0.0-20230522080721-ef84459b8f87
+	github.com/milvus-io/milvus-proto/go-api/v2 v2.3.0-dev.1
 	github.com/stretchr/testify v1.8.1
+	github.com/tidwall/gjson v1.14.4
 	google.golang.org/grpc v1.48.0
 	google.golang.org/grpc/examples v0.0.0-20220617181431-3e7b97febc7f
 )
 
 require (
-	github.com/cockroachdb/errors v1.9.1 // indirect
 	github.com/cockroachdb/logtags v0.0.0-20211118104740-dabe8e521a4f // indirect
 	github.com/cockroachdb/redact v1.1.3 // indirect
 	github.com/davecgh/go-spew v1.1.1 // indirect
@@ -24,7 +25,6 @@ require (
 	github.com/pmezard/go-difflib v1.0.0 // indirect
 	github.com/rogpeppe/go-internal v1.8.1 // indirect
 	github.com/stretchr/objx v0.5.0 // indirect
-	github.com/tidwall/gjson v1.14.4 // indirect
 	github.com/tidwall/match v1.1.1 // indirect
 	github.com/tidwall/pretty v1.2.0 // indirect
 	golang.org/x/net v0.0.0-20220127200216-cd36cc0744dd // indirect
diff --git a/go.sum b/go.sum
index 803cb2b..be1aa44 100644
--- a/go.sum
+++ b/go.sum
@@ -57,6 +57,7 @@ github.com/ghodss/yaml v1.0.0/go.mod h1:4dBDuWmgqj2HViK6kFavaiC9ZROes6MMH2rRYeME
 github.com/gin-contrib/sse v0.0.0-20190301062529-5545eab6dad3/go.mod h1:VJ0WA2NBN22VlZ2dKZQPAPnyWw5XTlK1KymzLKsr59s=
 github.com/gin-gonic/gin v1.4.0/go.mod h1:OW2EZn3DO8Ln9oIKOvM++LBO+5UPHJJDH72/q/3rZdM=
 github.com/go-check/check v0.0.0-20180628173108-788fd7840127/go.mod h1:9ES+weclKsC9YodN5RgxqK/VD9HM9JsCSh7rNhMZE98=
+github.com/go-errors/errors v1.0.1 h1:LUHzmkK3GUKUrL/1gfBUxAHzcev3apQlezX/+O7ma6w=
 github.com/go-errors/errors v1.0.1/go.mod h1:f4zRHt4oKfwPJE5k8C9vpYG+aDHdBFUsgrm6/TyX73Q=
 github.com/go-kit/kit v0.9.0/go.mod h1:xBxKIO96dXMWWy0MnWVtmwkA9/13aqxPnvrjFYMA2as=
 github.com/go-logfmt/logfmt v0.4.0/go.mod h1:3RMwSq7FuexP4Kalkev3ejPJsZTpXXBr9+V4qmtdjCk=
@@ -154,17 +155,8 @@ github.com/mattn/go-isatty v0.0.14/go.mod h1:7GGIvUiUoEMVVmxf/4nioHXj79iQHKdU27k
 github.com/mattn/goveralls v0.0.2/go.mod h1:8d1ZMHsd7fW6IRPKQh46F2WRpyib5/X4FOpevwGNQEw=
 github.com/mediocregopher/radix/v3 v3.4.2/go.mod h1:8FL3F6UQRXHXIBSPUs5h0RybMF8i4n7wVopoX3x7Bv8=
 github.com/microcosm-cc/bluemonday v1.0.2/go.mod h1:iVP4YcDBq+n/5fb23BhYFvIMq/leAFZyRl6bYmGDlGc=
-github.com/milvus-io/milvus-proto v2.2.8+incompatible/go.mod h1:urRUo4TMLWgew3AYu4Ak1oYr7KcV6AO2momWggKrDIw=
-github.com/milvus-io/milvus-proto/go-api v0.0.0-20230105121931-9f9303dcc729 h1:hsb1ifdNe3qlXi1YY5dWPPzWMNZmnqe5uunYPYK3gd0=
-github.com/milvus-io/milvus-proto/go-api v0.0.0-20230105121931-9f9303dcc729/go.mod h1:148qnlmZ0Fdm1Fq+Mj/OW2uDoEP25g3mjh0vMGtkgmk=
-github.com/milvus-io/milvus-proto/go-api v0.0.0-20230303054144-16f081962572 h1:QOimLfT1VwjH9jUkq2SN3uT/WGfgcD4pYBoFOGUMwjM=
-github.com/milvus-io/milvus-proto/go-api v0.0.0-20230303054144-16f081962572/go.mod h1:148qnlmZ0Fdm1Fq+Mj/OW2uDoEP25g3mjh0vMGtkgmk=
-github.com/milvus-io/milvus-proto/go-api v0.0.0-20230418025426-3735aafecc6c h1:9BaqfI2UoA2OJY89ogizV+b6vDlcwTguZT3rwUJl7yY=
-github.com/milvus-io/milvus-proto/go-api v0.0.0-20230418025426-3735aafecc6c/go.mod h1:148qnlmZ0Fdm1Fq+Mj/OW2uDoEP25g3mjh0vMGtkgmk=
-github.com/milvus-io/milvus-proto/go-api v0.0.0-20230517025117-8ba62a3f3a63 h1:V962mSHjOFUbuMgAXziBdbYPOCVZmN1MkqEeKpME+MA=
-github.com/milvus-io/milvus-proto/go-api v0.0.0-20230517025117-8ba62a3f3a63/go.mod h1:148qnlmZ0Fdm1Fq+Mj/OW2uDoEP25g3mjh0vMGtkgmk=
-github.com/milvus-io/milvus-proto/go-api v0.0.0-20230522080721-ef84459b8f87 h1:LdDHjEjus1NdC9ELbpQa6DfUHJotJUW2kD4S+8nvjw4=
-github.com/milvus-io/milvus-proto/go-api v0.0.0-20230522080721-ef84459b8f87/go.mod h1:148qnlmZ0Fdm1Fq+Mj/OW2uDoEP25g3mjh0vMGtkgmk=
+github.com/milvus-io/milvus-proto/go-api/v2 v2.3.0-dev.1 h1:x6vhrVyK3wEuXIDHt0uk2l/UFPa/RRGWk1nkjgN5jkI=
+github.com/milvus-io/milvus-proto/go-api/v2 v2.3.0-dev.1/go.mod h1:1OIl0v5PQeNxIJhCvY+K55CBUOYDZevw9g9380u1Wek=
 github.com/mitchellh/go-homedir v1.1.0/go.mod h1:SfyaCUpYCn1Vlf4IUYiD9fPX4A5wJrkLzIz1N1q0pr0=
 github.com/mitchellh/mapstructure v1.1.2/go.mod h1:FVVH3fgwuzCH5S8UJGiWEs2h04kUh9fWfEaFds41c1Y=
 github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
@@ -181,6 +173,7 @@ github.com/onsi/ginkgo v1.10.3/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+
 github.com/onsi/gomega v1.7.1/go.mod h1:XdKZgCCFLUoM/7CFJVPcG8C1xQ1AJ0vpAezJrB7JYyY=
 github.com/opentracing/opentracing-go v1.1.0/go.mod h1:UkNAQd3GIcIGf0SeVgPpRdFStlNbqXla1AfSYxPUl2o=
 github.com/pelletier/go-toml v1.2.0/go.mod h1:5z9KED0ma1S8pY6P1sdut58dfprrGBbd/94hg7ilaic=
+github.com/pingcap/errors v0.11.4 h1:lFuQV/oaUMGcD2tqt+01ROSmJs75VG1ToEOkZIZ4nE4=
 github.com/pingcap/errors v0.11.4/go.mod h1:Oi8TUi2kEtXXLMJk9l1cGmz20kV3TaQ0usTwv5KuLY8=
 github.com/pkg/diff v0.0.0-20210226163009-20ebb0f2a09e/go.mod h1:pJLUxLENpZxwdsKMEsNbx1VGcRFpLqf3715MtcvvzbA=
 github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
@@ -248,7 +241,6 @@ github.com/yudai/pp v2.0.1+incompatible/go.mod h1:PuxR/8QJ7cyCkFp/aUDS+JY727OFEZ
 github.com/yuin/goldmark v1.1.27/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
 github.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
 github.com/yuin/goldmark v1.3.5/go.mod h1:mwnBkeHKe2W/ZEtQ+71ViKU8L12m81fl3OWwC1Zlc8k=
-github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
 go.opentelemetry.io/proto/otlp v0.7.0/go.mod h1:PqfVotwruBrMGOCsRd/89rSnXhoiJIqeYNgFYFoEGnI=
 go.uber.org/atomic v1.4.0/go.mod h1:gD2HeocX3+yG+ygLZcrzQJaqmWj9AIm7n08wl/qW/PE=
 go.uber.org/multierr v1.1.0/go.mod h1:wR5kodmAFQ0UK8QlbwjlSNy0Z68gJhDJUG5sjR94q/0=
@@ -270,7 +262,6 @@ golang.org/x/mod v0.1.1-0.20191105210325-c90efee705ee/go.mod h1:QqPTAvyqsEbceGzB
 golang.org/x/mod v0.2.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=
 golang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=
 golang.org/x/mod v0.4.2/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=
-golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
 golang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
 golang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
 golang.org/x/net v0.0.0-20180906233101-161cd47e91fd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
@@ -292,8 +283,6 @@ golang.org/x/net v0.0.0-20210405180319-a5a99cb37ef4/go.mod h1:p54w0d4576C0XHj96b
 golang.org/x/net v0.0.0-20211008194852-3b03d305991f/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
 golang.org/x/net v0.0.0-20220127200216-cd36cc0744dd h1:O7DYs+zxREGLKzKoMQrtrEacpb0ZVXA5rIwylE2Xchk=
 golang.org/x/net v0.0.0-20220127200216-cd36cc0744dd/go.mod h1:CfG3xpIq0wQ8r1q4Su4UZFWDARRcnwPjda9FqA0JpMk=
-golang.org/x/net v0.0.0-20220722155237-a158d28d115b h1:PxfKdU9lEEDYjdIzOtC4qFWgkU2rGHdKlKowJSMN9h0=
-golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
 golang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=
 golang.org/x/oauth2 v0.0.0-20200107190931-bf48bf16ab8d/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw=
 golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
@@ -304,7 +293,6 @@ golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJ
 golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sync v0.0.0-20210220032951-036812b2e83c/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
-golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sys v0.0.0-20180830151530-49385e6e1522/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
 golang.org/x/sys v0.0.0-20180909124046-d0be0721c37e/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
 golang.org/x/sys v0.0.0-20181205085412-a5c9d58dba9a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
@@ -328,14 +316,10 @@ golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBc
 golang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.0.0-20210927094055-39ccf1dd6fa6/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.0.0-20211007075335-d3039528d8ac/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
-golang.org/x/sys v0.0.0-20211216021012-1d35b9e2eb4e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.0.0-20220209214540-3681064d5158/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
-golang.org/x/sys v0.0.0-20220422013727-9388b58f7150/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
-golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f h1:v4INt8xihDGvnrfjMDVXGxw9wrfxYyCjk0KbXjhR55s=
 golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
-golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
 golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
 golang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk=
 golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
@@ -358,7 +342,6 @@ golang.org/x/tools v0.0.0-20200130002326-2f3ba24bd6e7/go.mod h1:TB2adYChydJhpapK
 golang.org/x/tools v0.0.0-20200619180055-7c47624df98f/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=
 golang.org/x/tools v0.0.0-20210106214847-113979e3529a/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=
 golang.org/x/tools v0.1.3/go.mod h1:o0xws9oXOQQZyjljx8fwUC0k7L1pTE6eaCbjGeHmOkk=
-golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
 golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
 golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
 golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
@@ -406,7 +389,6 @@ google.golang.org/protobuf v1.27.1/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQ
 google.golang.org/protobuf v1.28.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=
 google.golang.org/protobuf v1.28.1 h1:d0NfwRgPtno5B1Wa6L2DAG+KivqkdutMf1UhdNx175w=
 google.golang.org/protobuf v1.28.1/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=
-gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15 h1:YR8cESwS4TdDjEe65xsg0ogRM/Nc3DYOhEAlW+xobZo=
diff --git a/mocks/MilvusServiceServer.go b/mocks/MilvusServiceServer.go
index 26cc023..1a0f159 100644
--- a/mocks/MilvusServiceServer.go
+++ b/mocks/MilvusServiceServer.go
@@ -5,11 +5,11 @@ package mocks
 import (
 	context "context"
 
-	commonpb "github.com/milvus-io/milvus-proto/go-api/commonpb"
+	commonpb "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
 
-	federpb "github.com/milvus-io/milvus-proto/go-api/federpb"
+	federpb "github.com/milvus-io/milvus-proto/go-api/v2/federpb"
 
-	milvuspb "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	milvuspb "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 
 	mock "github.com/stretchr/testify/mock"
 )
@@ -403,6 +403,53 @@ func (_c *MilvusServiceServer_CreateCredential_Call) Return(_a0 *commonpb.Status
 	return _c
 }
 
+// CreateDatabase provides a mock function with given fields: _a0, _a1
+func (_m *MilvusServiceServer) CreateDatabase(_a0 context.Context, _a1 *milvuspb.CreateDatabaseRequest) (*commonpb.Status, error) {
+	ret := _m.Called(_a0, _a1)
+
+	var r0 *commonpb.Status
+	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateDatabaseRequest) *commonpb.Status); ok {
+		r0 = rf(_a0, _a1)
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).(*commonpb.Status)
+		}
+	}
+
+	var r1 error
+	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateDatabaseRequest) error); ok {
+		r1 = rf(_a0, _a1)
+	} else {
+		r1 = ret.Error(1)
+	}
+
+	return r0, r1
+}
+
+// MilvusServiceServer_CreateDatabase_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CreateDatabase'
+type MilvusServiceServer_CreateDatabase_Call struct {
+	*mock.Call
+}
+
+// CreateDatabase is a helper method to define mock.On call
+//  - _a0 context.Context
+//  - _a1 *milvuspb.CreateDatabaseRequest
+func (_e *MilvusServiceServer_Expecter) CreateDatabase(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_CreateDatabase_Call {
+	return &MilvusServiceServer_CreateDatabase_Call{Call: _e.mock.On("CreateDatabase", _a0, _a1)}
+}
+
+func (_c *MilvusServiceServer_CreateDatabase_Call) Run(run func(_a0 context.Context, _a1 *milvuspb.CreateDatabaseRequest)) *MilvusServiceServer_CreateDatabase_Call {
+	_c.Call.Run(func(args mock.Arguments) {
+		run(args[0].(context.Context), args[1].(*milvuspb.CreateDatabaseRequest))
+	})
+	return _c
+}
+
+func (_c *MilvusServiceServer_CreateDatabase_Call) Return(_a0 *commonpb.Status, _a1 error) *MilvusServiceServer_CreateDatabase_Call {
+	_c.Call.Return(_a0, _a1)
+	return _c
+}
+
 // CreateIndex provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) CreateIndex(_a0 context.Context, _a1 *milvuspb.CreateIndexRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
@@ -1014,6 +1061,53 @@ func (_c *MilvusServiceServer_DropCollection_Call) Return(_a0 *commonpb.Status,
 	return _c
 }
 
+// DropDatabase provides a mock function with given fields: _a0, _a1
+func (_m *MilvusServiceServer) DropDatabase(_a0 context.Context, _a1 *milvuspb.DropDatabaseRequest) (*commonpb.Status, error) {
+	ret := _m.Called(_a0, _a1)
+
+	var r0 *commonpb.Status
+	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropDatabaseRequest) *commonpb.Status); ok {
+		r0 = rf(_a0, _a1)
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).(*commonpb.Status)
+		}
+	}
+
+	var r1 error
+	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropDatabaseRequest) error); ok {
+		r1 = rf(_a0, _a1)
+	} else {
+		r1 = ret.Error(1)
+	}
+
+	return r0, r1
+}
+
+// MilvusServiceServer_DropDatabase_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DropDatabase'
+type MilvusServiceServer_DropDatabase_Call struct {
+	*mock.Call
+}
+
+// DropDatabase is a helper method to define mock.On call
+//  - _a0 context.Context
+//  - _a1 *milvuspb.DropDatabaseRequest
+func (_e *MilvusServiceServer_Expecter) DropDatabase(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_DropDatabase_Call {
+	return &MilvusServiceServer_DropDatabase_Call{Call: _e.mock.On("DropDatabase", _a0, _a1)}
+}
+
+func (_c *MilvusServiceServer_DropDatabase_Call) Run(run func(_a0 context.Context, _a1 *milvuspb.DropDatabaseRequest)) *MilvusServiceServer_DropDatabase_Call {
+	_c.Call.Run(func(args mock.Arguments) {
+		run(args[0].(context.Context), args[1].(*milvuspb.DropDatabaseRequest))
+	})
+	return _c
+}
+
+func (_c *MilvusServiceServer_DropDatabase_Call) Return(_a0 *commonpb.Status, _a1 error) *MilvusServiceServer_DropDatabase_Call {
+	_c.Call.Return(_a0, _a1)
+	return _c
+}
+
 // DropIndex provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) DropIndex(_a0 context.Context, _a1 *milvuspb.DropIndexRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
@@ -2471,6 +2565,53 @@ func (_c *MilvusServiceServer_ListCredUsers_Call) Return(_a0 *milvuspb.ListCredU
 	return _c
 }
 
+// ListDatabases provides a mock function with given fields: _a0, _a1
+func (_m *MilvusServiceServer) ListDatabases(_a0 context.Context, _a1 *milvuspb.ListDatabasesRequest) (*milvuspb.ListDatabasesResponse, error) {
+	ret := _m.Called(_a0, _a1)
+
+	var r0 *milvuspb.ListDatabasesResponse
+	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListDatabasesRequest) *milvuspb.ListDatabasesResponse); ok {
+		r0 = rf(_a0, _a1)
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).(*milvuspb.ListDatabasesResponse)
+		}
+	}
+
+	var r1 error
+	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ListDatabasesRequest) error); ok {
+		r1 = rf(_a0, _a1)
+	} else {
+		r1 = ret.Error(1)
+	}
+
+	return r0, r1
+}
+
+// MilvusServiceServer_ListDatabases_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ListDatabases'
+type MilvusServiceServer_ListDatabases_Call struct {
+	*mock.Call
+}
+
+// ListDatabases is a helper method to define mock.On call
+//  - _a0 context.Context
+//  - _a1 *milvuspb.ListDatabasesRequest
+func (_e *MilvusServiceServer_Expecter) ListDatabases(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_ListDatabases_Call {
+	return &MilvusServiceServer_ListDatabases_Call{Call: _e.mock.On("ListDatabases", _a0, _a1)}
+}
+
+func (_c *MilvusServiceServer_ListDatabases_Call) Run(run func(_a0 context.Context, _a1 *milvuspb.ListDatabasesRequest)) *MilvusServiceServer_ListDatabases_Call {
+	_c.Call.Run(func(args mock.Arguments) {
+		run(args[0].(context.Context), args[1].(*milvuspb.ListDatabasesRequest))
+	})
+	return _c
+}
+
+func (_c *MilvusServiceServer_ListDatabases_Call) Return(_a0 *milvuspb.ListDatabasesResponse, _a1 error) *MilvusServiceServer_ListDatabases_Call {
+	_c.Call.Return(_a0, _a1)
+	return _c
+}
+
 // ListImportTasks provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) ListImportTasks(_a0 context.Context, _a1 *milvuspb.ListImportTasksRequest) (*milvuspb.ListImportTasksResponse, error) {
 	ret := _m.Called(_a0, _a1)


Test patch:
diff --git a/client/alias_test.go b/client/alias_test.go
index f6f77b3..c3d6396 100644
--- a/client/alias_test.go
+++ b/client/alias_test.go
@@ -21,8 +21,8 @@ import (
 	"testing"
 
 	"github.com/golang/protobuf/proto"
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 	"github.com/stretchr/testify/assert"
 )
 
diff --git a/client/authentication_test.go b/client/authentication_test.go
index 1fbfd5e..2f75629 100644
--- a/client/authentication_test.go
+++ b/client/authentication_test.go
@@ -7,8 +7,8 @@ import (
 	"github.com/cockroachdb/errors"
 
 	"github.com/golang/protobuf/proto"
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 	"github.com/stretchr/testify/assert"
 )
 
diff --git a/client/client_mock_test.go b/client/client_mock_test.go
index 28f7472..e8b543d 100644
--- a/client/client_mock_test.go
+++ b/client/client_mock_test.go
@@ -8,11 +8,10 @@ import (
 	"github.com/cockroachdb/errors"
 
 	"github.com/golang/protobuf/proto"
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	"github.com/milvus-io/milvus-proto/go-api/federpb"
-	"github.com/milvus-io/milvus-proto/go-api/milvuspb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
-	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	"github.com/milvus-io/milvus-proto/go-api/v2/federpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
+	schema "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/mocks"
 	"github.com/stretchr/testify/mock"
@@ -139,7 +138,7 @@ func (s *MockSuiteBase) setupHasPartitionError(errorCode common.ErrorCode, err e
 func (s *MockSuiteBase) setupDescribeCollection(collName string, schema *entity.Schema) {
 	s.mock.EXPECT().DescribeCollection(mock.Anything, mock.AnythingOfType("*milvuspb.DescribeCollectionRequest")).
 		Call.Return(func(ctx context.Context, req *server.DescribeCollectionRequest) *server.DescribeCollectionResponse {
-		return &milvuspb.DescribeCollectionResponse{
+		return &server.DescribeCollectionResponse{
 			Status: &common.Status{ErrorCode: common.ErrorCode_Success},
 			Schema: schema.ProtoMessage(),
 		}
@@ -148,7 +147,7 @@ func (s *MockSuiteBase) setupDescribeCollection(collName string, schema *entity.
 
 func (s *MockSuiteBase) setupDescribeCollectionError(errorCode common.ErrorCode, err error) {
 	s.mock.EXPECT().DescribeCollection(mock.Anything, mock.AnythingOfType("*milvuspb.DescribeCollectionRequest")).
-		Return(&milvuspb.DescribeCollectionResponse{
+		Return(&server.DescribeCollectionResponse{
 			Status: &common.Status{ErrorCode: errorCode},
 		}, err)
 }
@@ -873,6 +872,18 @@ func (m *MockServer) GetIndexStatistics(_ context.Context, _ *server.GetIndexSta
 	panic("not implemented") // TODO: Implement
 }
 
+func (m *MockServer) CreateDatabase(_a0 context.Context, _a1 *server.CreateDatabaseRequest) (*common.Status, error) {
+	panic("not implemented") // TODO: Implement
+}
+
+func (m *MockServer) DropDatabase(_a0 context.Context, _a1 *server.DropDatabaseRequest) (*common.Status, error) {
+	panic("not implemented") // TODO: Implement
+}
+
+func (m *MockServer) ListDatabases(_a0 context.Context, _a1 *server.ListDatabasesRequest) (*server.ListDatabasesResponse, error) {
+	panic("not implemented") // TODO: Implement
+}
+
 func (m *MockServer) Connect(_ context.Context, _ *server.ConnectRequest) (*server.ConnectResponse, error) {
 	return &server.ConnectResponse{
 		Status:     &common.Status{ErrorCode: common.ErrorCode_Success},
diff --git a/client/client_test.go b/client/client_test.go
index 41e130e..cab3ff3 100644
--- a/client/client_test.go
+++ b/client/client_test.go
@@ -21,8 +21,8 @@ import (
 	"google.golang.org/grpc/status"
 	"google.golang.org/grpc/test/bufconn"
 
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 	"github.com/milvus-io/milvus-sdk-go/v2/entity"
 )
 
diff --git a/client/collection_test.go b/client/collection_test.go
index 2910446..74eef01 100644
--- a/client/collection_test.go
+++ b/client/collection_test.go
@@ -10,9 +10,9 @@ import (
 	"github.com/cockroachdb/errors"
 
 	"github.com/golang/protobuf/proto"
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
-	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
+	schema "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 	"github.com/milvus-io/milvus-sdk-go/v2/entity"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/mock"
diff --git a/client/data_test.go b/client/data_test.go
index ce6a154..f2bd43a 100644
--- a/client/data_test.go
+++ b/client/data_test.go
@@ -20,9 +20,9 @@ import (
 	"github.com/cockroachdb/errors"
 
 	"github.com/golang/protobuf/proto"
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
-	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
+	schema "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 	"github.com/milvus-io/milvus-sdk-go/v2/entity"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/mock"
diff --git a/client/index_test.go b/client/index_test.go
index e339fac..9882f0c 100644
--- a/client/index_test.go
+++ b/client/index_test.go
@@ -9,8 +9,8 @@ import (
 	"github.com/cockroachdb/errors"
 
 	"github.com/golang/protobuf/proto"
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 	"github.com/milvus-io/milvus-sdk-go/v2/entity"
 	"github.com/stretchr/testify/assert"
 )
diff --git a/client/insert_test.go b/client/insert_test.go
index dc3d02a..fc78d34 100644
--- a/client/insert_test.go
+++ b/client/insert_test.go
@@ -16,9 +16,9 @@ import (
 	"testing"
 
 	"github.com/cockroachdb/errors"
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
-	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
+	schema "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 	"github.com/milvus-io/milvus-sdk-go/v2/entity"
 	"github.com/stretchr/testify/mock"
 	"github.com/stretchr/testify/suite"
diff --git a/client/maintainance_test.go b/client/maintainance_test.go
index 356caf8..52bebd4 100644
--- a/client/maintainance_test.go
+++ b/client/maintainance_test.go
@@ -24,8 +24,8 @@ import (
 	"github.com/cockroachdb/errors"
 
 	"github.com/golang/protobuf/proto"
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 	"github.com/milvus-io/milvus-sdk-go/v2/entity"
 	"github.com/milvus-io/milvus-sdk-go/v2/internal/utils/tso"
 	"github.com/stretchr/testify/assert"
diff --git a/client/options_test.go b/client/options_test.go
index e7f8ea0..0a0bed1 100644
--- a/client/options_test.go
+++ b/client/options_test.go
@@ -15,8 +15,8 @@ import (
 	"math/rand"
 	"testing"
 
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 	"github.com/milvus-io/milvus-sdk-go/v2/entity"
 	"github.com/stretchr/testify/assert"
 )
diff --git a/client/partition_test.go b/client/partition_test.go
index 88fd422..5502308 100644
--- a/client/partition_test.go
+++ b/client/partition_test.go
@@ -10,8 +10,8 @@ import (
 	"github.com/cockroachdb/errors"
 
 	"github.com/golang/protobuf/proto"
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 	"github.com/milvus-io/milvus-sdk-go/v2/entity"
 	"github.com/stretchr/testify/assert"
 )
diff --git a/client/rate_limit_interceptor_test.go b/client/rate_limit_interceptor_test.go
index 65514b0..af5a949 100644
--- a/client/rate_limit_interceptor_test.go
+++ b/client/rate_limit_interceptor_test.go
@@ -25,7 +25,7 @@ import (
 	"github.com/stretchr/testify/assert"
 	"google.golang.org/grpc"
 
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
 )
 
 var mockInvokerError error
diff --git a/client/rbac_test.go b/client/rbac_test.go
index ea6b9e4..6bcb6e0 100644
--- a/client/rbac_test.go
+++ b/client/rbac_test.go
@@ -11,8 +11,8 @@ import (
 
 	"github.com/stretchr/testify/mock"
 
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 )
 
 type RBACSuite struct {
diff --git a/client/resource_group_test.go b/client/resource_group_test.go
index 78bc182..ff91ea1 100644
--- a/client/resource_group_test.go
+++ b/client/resource_group_test.go
@@ -21,8 +21,8 @@ import (
 	"github.com/stretchr/testify/mock"
 	"github.com/stretchr/testify/suite"
 
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
 )
 
 type ResourceGroupSuite struct {
diff --git a/client/row_test.go b/client/row_test.go
index 68ded83..89c7f20 100644
--- a/client/row_test.go
+++ b/client/row_test.go
@@ -9,9 +9,9 @@ import (
 	"github.com/cockroachdb/errors"
 
 	"github.com/golang/protobuf/proto"
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
-	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
-	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
+	server "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
+	schema "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 	"github.com/milvus-io/milvus-sdk-go/v2/entity"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/mock"
diff --git a/entity/columns_scalar_gen_test.go b/entity/columns_scalar_gen_test.go
index 214acfb..5366e40 100755
--- a/entity/columns_scalar_gen_test.go
+++ b/entity/columns_scalar_gen_test.go
@@ -1,7 +1,7 @@
 // Code generated by go generate; DO NOT EDIT
-// This file is generated by go generated 
+// This file is generated by go generated
 
-package entity 
+package entity
 
 import (
 	"fmt"
@@ -9,8 +9,8 @@ import (
 	"testing"
 	"time"
 
+	schema "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 	"github.com/stretchr/testify/assert"
-	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
 )
 
 func TestColumnBool(t *testing.T) {
@@ -60,7 +60,7 @@ func TestFieldDataBoolColumn(t *testing.T) {
 	len := rand.Intn(10) + 8
 	name := fmt.Sprintf("fd_Bool_%d", rand.Int())
 	fd := &schema.FieldData{
-		Type: schema.DataType_Bool,
+		Type:      schema.DataType_Bool,
 		FieldName: name,
 	}
 
@@ -77,7 +77,7 @@ func TestFieldDataBoolColumn(t *testing.T) {
 		column, err := FieldDataColumn(fd, 0, len)
 		assert.Nil(t, err)
 		assert.NotNil(t, column)
- 
+
 		assert.Equal(t, name, column.Name())
 		assert.Equal(t, len, column.Len())
 		assert.Equal(t, FieldTypeBool, column.Type())
@@ -92,13 +92,12 @@ func TestFieldDataBoolColumn(t *testing.T) {
 		assert.NotNil(t, err)
 	})
 
-	
 	t.Run("nil data", func(t *testing.T) {
 		fd.Field = nil
 		_, err := FieldDataColumn(fd, 0, len)
 		assert.NotNil(t, err)
 	})
-	
+
 	t.Run("get all data", func(t *testing.T) {
 		fd.Field = &schema.FieldData_Scalars{
 			Scalars: &schema.ScalarField{
@@ -112,7 +111,7 @@ func TestFieldDataBoolColumn(t *testing.T) {
 		column, err := FieldDataColumn(fd, 0, -1)
 		assert.Nil(t, err)
 		assert.NotNil(t, column)
-		
+
 		assert.Equal(t, name, column.Name())
 		assert.Equal(t, len, column.Len())
 		assert.Equal(t, FieldTypeBool, column.Type())
@@ -166,7 +165,7 @@ func TestFieldDataInt8Column(t *testing.T) {
 	len := rand.Intn(10) + 8
 	name := fmt.Sprintf("fd_Int8_%d", rand.Int())
 	fd := &schema.FieldData{
-		Type: schema.DataType_Int8,
+		Type:      schema.DataType_Int8,
 		FieldName: name,
 	}
 
@@ -183,7 +182,7 @@ func TestFieldDataInt8Column(t *testing.T) {
 		column, err := FieldDataColumn(fd, 0, len)
 		assert.Nil(t, err)
 		assert.NotNil(t, column)
- 
+
 		assert.Equal(t, name, column.Name())
 		assert.Equal(t, len, column.Len())
 		assert.Equal(t, FieldTypeInt8, column.Type())
@@ -198,13 +197,12 @@ func TestFieldDataInt8Column(t *testing.T) {
 		assert.NotNil(t, err)
 	})
 
-	
 	t.Run("nil data", func(t *testing.T) {
 		fd.Field = nil
 		_, err := FieldDataColumn(fd, 0, len)
 		assert.NotNil(t, err)
 	})
-	
+
 	t.Run("get all data", func(t *testing.T) {
 		fd.Field = &schema.FieldData_Scalars{
 			Scalars: &schema.ScalarField{
@@ -218,7 +216,7 @@ func TestFieldDataInt8Column(t *testing.T) {
 		column, err := FieldDataColumn(fd, 0, -1)
 		assert.Nil(t, err)
 		assert.NotNil(t, column)
-		
+
 		assert.Equal(t, name, column.Name())
 		assert.Equal(t, len, column.Len())
 		assert.Equal(t, FieldTypeInt8, column.Type())
@@ -272,7 +270,7 @@ func TestFieldDataInt16Column(t *testing.T) {
 	len := rand.Intn(10) + 8
 	name := fmt.Sprintf("fd_Int16_%d", rand.Int())
 	fd := &schema.FieldData{
-		Type: schema.DataType_Int16,
+		Type:      schema.DataType_Int16,
 		FieldName: name,
 	}
 
@@ -289,7 +287,7 @@ func TestFieldDataInt16Column(t *testing.T) {
 		column, err := FieldDataColumn(fd, 0, len)
 		assert.Nil(t, err)
 		assert.NotNil(t, column)
- 
+
 		assert.Equal(t, name, column.Name())
 		assert.Equal(t, len, column.Len())
 		assert.Equal(t, FieldTypeInt16, column.Type())
@@ -304,13 +302,12 @@ func TestFieldDataInt16Column(t *testing.T) {
 		assert.NotNil(t, err)
 	})
 
-	
 	t.Run("nil data", func(t *testing.T) {
 		fd.Field = nil
 		_, err := FieldDataColumn(fd, 0, len)
 		assert.NotNil(t, err)
 	})
-	
+
 	t.Run("get all data", func(t *testing.T) {
 		fd.Field = &schema.FieldData_Scalars{
 			Scalars: &schema.ScalarField{
@@ -324,7 +321,7 @@ func TestFieldDataInt16Column(t *testing.T) {
 		column, err := FieldDataColumn(fd, 0, -1)
 		assert.Nil(t, err)
 		assert.NotNil(t, column)
-		
+
 		assert.Equal(t, name, column.Name())
 		assert.Equal(t, len, column.Len())
 		assert.Equal(t, FieldTypeInt16, column.Type())
@@ -378,7 +375,7 @@ func TestFieldDataInt32Column(t *testing.T) {
 	len := rand.Intn(10) + 8
 	name := fmt.Sprintf("fd_Int32_%d", rand.Int())
 	fd := &schema.FieldData{
-		Type: schema.DataType_Int32,
+		Type:      schema.DataType_Int32,
 		FieldName: name,
 	}
 
@@ -395,7 +392,7 @@ func TestFieldDataInt32Column(t *testing.T) {
 		column, err := FieldDataColumn(fd, 0, len)
 		assert.Nil(t, err)
 		assert.NotNil(t, column)
- 
+
 		assert.Equal(t, name, column.Name())
 		assert.Equal(t, len, column.Len())
 		assert.Equal(t, FieldTypeInt32, column.Type())
@@ -410,13 +407,12 @@ func TestFieldDataInt32Column(t *testing.T) {
 		assert.NotNil(t, err)
 	})
 
-	
 	t.Run("nil data", func(t *testing.T) {
 		fd.Field = nil
 		_, err := FieldDataColumn(fd, 0, len)
 		assert.NotNil(t, err)
 	})
-	
+
 	t.Run("get all data", func(t *testing.T) {
 		fd.Field = &schema.FieldData_Scalars{
 			Scalars: &schema.ScalarField{
@@ -430,7 +426,7 @@ func TestFieldDataInt32Column(t *testing.T) {
 		column, err := FieldDataColumn(fd, 0, -1)
 		assert.Nil(t, err)
 		assert.NotNil(t, column)
-		
+
 		assert.Equal(t, name, column.Name())
 		assert.Equal(t, len, column.Len())
 		assert.Equal(t, FieldTypeInt32, column.Type())
@@ -484,7 +480,7 @@ func TestFieldDataInt64Column(t *testing.T) {
 	len := rand.Intn(10) + 8
 	name := fmt.Sprintf("fd_Int64_%d", rand.Int())
 	fd := &schema.FieldData{
-		Type: schema.DataType_Int64,
+		Type:      schema.DataType_Int64,
 		FieldName: name,
 	}
 
@@ -501,7 +497,7 @@ func TestFieldDataInt64Column(t *testing.T) {
 		column, err := FieldDataColumn(fd, 0, len)
 		assert.Nil(t, err)
 		assert.NotNil(t, column)
- 
+
 		assert.Equal(t, name, column.Name())
 		assert.Equal(t, len, column.Len())
 		assert.Equal(t, FieldTypeInt64, column.Type())
@@ -516,13 +512,12 @@ func TestFieldDataInt64Column(t *testing.T) {
 		assert.NotNil(t, err)
 	})
 
-	
 	t.Run("nil data", func(t *testing.T) {
 		fd.Field = nil
 		_, err := FieldDataColumn(fd, 0, len)
 		assert.NotNil(t, err)
 	})
-	
+
 	t.Run("get all data", func(t *testing.T) {
 		fd.Field = &schema.FieldData_Scalars{
 			Scalars: &schema.ScalarField{
@@ -536,7 +531,7 @@ func TestFieldDataInt64Column(t *testing.T) {
 		column, err := FieldDataColumn(fd, 0, -1)
 		assert.Nil(t, err)
 		assert.NotNil(t, column)
-		
+
 		assert.Equal(t, name, column.Name())
 		assert.Equal(t, len, column.Len())
 		assert.Equal(t, FieldTypeInt64, column.Type())
@@ -590,7 +585,7 @@ func TestFieldDataFloatColumn(t *testing.T) {
 	len := rand.Intn(10) + 8
 	name := fmt.Sprintf("fd_Float_%d", rand.Int())
 	fd := &schema.FieldData{
-		Type: schema.DataType_Float,
+		Type:      schema.DataType_Float,
 		FieldName: name,
 	}
 
@@ -607,7 +602,7 @@ func TestFieldDataFloatColumn(t *testing.T) {
 		column, err := FieldDataColumn(fd, 0, len)
 		assert.Nil(t, err)
 		assert.NotNil(t, column)
- 
+
 		assert.Equal(t, name, column.Name())
 		assert.Equal(t, len, column.Len())
 		assert.Equal(t, FieldTypeFloat, column.Type())
@@ -622,13 +617,12 @@ func TestFieldDataFloatColumn(t *testing.T) {
 		assert.NotNil(t, err)
 	})
 
-	
 	t.Run("nil data", func(t *testing.T) {
 		fd.Field = nil
 		_, err := FieldDataColumn(fd, 0, len)
 		assert.NotNil(t, err)
 	})
-	
+
 	t.Run("get all data", func(t *testing.T) {
 		fd.Field = &schema.FieldData_Scalars{
 			Scalars: &schema.ScalarField{
@@ -642,7 +636,7 @@ func TestFieldDataFloatColumn(t *testing.T) {
 		column, err := FieldDataColumn(fd, 0, -1)
 		assert.Nil(t, err)
 		assert.NotNil(t, column)
-		
+
 		assert.Equal(t, name, column.Name())
 		assert.Equal(t, len, column.Len())
 		assert.Equal(t, FieldTypeFloat, column.Type())
@@ -696,7 +690,7 @@ func TestFieldDataDoubleColumn(t *testing.T) {
 	len := rand.Intn(10) + 8
 	name := fmt.Sprintf("fd_Double_%d", rand.Int())
 	fd := &schema.FieldData{
-		Type: schema.DataType_Double,
+		Type:      schema.DataType_Double,
 		FieldName: name,
 	}
 
@@ -713,7 +707,7 @@ func TestFieldDataDoubleColumn(t *testing.T) {
 		column, err := FieldDataColumn(fd, 0, len)
 		assert.Nil(t, err)
 		assert.NotNil(t, column)
- 
+
 		assert.Equal(t, name, column.Name())
 		assert.Equal(t, len, column.Len())
 		assert.Equal(t, FieldTypeDouble, column.Type())
@@ -728,13 +722,12 @@ func TestFieldDataDoubleColumn(t *testing.T) {
 		assert.NotNil(t, err)
 	})
 
-	
 	t.Run("nil data", func(t *testing.T) {
 		fd.Field = nil
 		_, err := FieldDataColumn(fd, 0, len)
 		assert.NotNil(t, err)
 	})
-	
+
 	t.Run("get all data", func(t *testing.T) {
 		fd.Field = &schema.FieldData_Scalars{
 			Scalars: &schema.ScalarField{
@@ -748,7 +741,7 @@ func TestFieldDataDoubleColumn(t *testing.T) {
 		column, err := FieldDataColumn(fd, 0, -1)
 		assert.Nil(t, err)
 		assert.NotNil(t, column)
-		
+
 		assert.Equal(t, name, column.Name())
 		assert.Equal(t, len, column.Len())
 		assert.Equal(t, FieldTypeDouble, column.Type())
@@ -802,7 +795,7 @@ func TestFieldDataStringColumn(t *testing.T) {
 	len := rand.Intn(10) + 8
 	name := fmt.Sprintf("fd_String_%d", rand.Int())
 	fd := &schema.FieldData{
-		Type: schema.DataType_String,
+		Type:      schema.DataType_String,
 		FieldName: name,
 	}
 
@@ -819,7 +812,7 @@ func TestFieldDataStringColumn(t *testing.T) {
 		column, err := FieldDataColumn(fd, 0, len)
 		assert.Nil(t, err)
 		assert.NotNil(t, column)
- 
+
 		assert.Equal(t, name, column.Name())
 		assert.Equal(t, len, column.Len())
 		assert.Equal(t, FieldTypeString, column.Type())
@@ -834,13 +827,12 @@ func TestFieldDataStringColumn(t *testing.T) {
 		assert.NotNil(t, err)
 	})
 
-	
 	t.Run("nil data", func(t *testing.T) {
 		fd.Field = nil
 		_, err := FieldDataColumn(fd, 0, len)
 		assert.NotNil(t, err)
 	})
-	
+
 	t.Run("get all data", func(t *testing.T) {
 		fd.Field = &schema.FieldData_Scalars{
 			Scalars: &schema.ScalarField{
@@ -854,10 +846,9 @@ func TestFieldDataStringColumn(t *testing.T) {
 		column, err := FieldDataColumn(fd, 0, -1)
 		assert.Nil(t, err)
 		assert.NotNil(t, column)
-		
+
 		assert.Equal(t, name, column.Name())
 		assert.Equal(t, len, column.Len())
 		assert.Equal(t, FieldTypeString, column.Type())
 	})
 }
-
diff --git a/entity/columns_test.go b/entity/columns_test.go
index 1975c07..833d788 100644
--- a/entity/columns_test.go
+++ b/entity/columns_test.go
@@ -6,7 +6,7 @@ import (
 	"testing"
 	"time"
 
-	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+	schema "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 	"github.com/stretchr/testify/assert"
 )
 
diff --git a/entity/columns_varchar_test.go b/entity/columns_varchar_test.go
index c212b72..46a8835 100644
--- a/entity/columns_varchar_test.go
+++ b/entity/columns_varchar_test.go
@@ -6,7 +6,7 @@ import (
 	"testing"
 	"time"
 
-	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+	schema "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 	"github.com/stretchr/testify/assert"
 )
 
diff --git a/entity/columns_vector_gen_test.go b/entity/columns_vector_gen_test.go
index 612f002..d959e2c 100755
--- a/entity/columns_vector_gen_test.go
+++ b/entity/columns_vector_gen_test.go
@@ -1,7 +1,7 @@
 // Code generated by go generate; DO NOT EDIT
-// This file is generated by go generated 
+// This file is generated by go generated
 
-package entity 
+package entity
 
 import (
 	"fmt"
@@ -9,7 +9,7 @@ import (
 	"testing"
 	"time"
 
-	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+	schema "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 	"github.com/stretchr/testify/assert"
 )
 
@@ -19,16 +19,16 @@ func TestColumnBinaryVector(t *testing.T) {
 	columnLen := 12 + rand.Intn(10)
 	dim := ([]int{64, 128, 256, 512})[rand.Intn(4)]
 
-	v := make([][]byte,0, columnLen)
+	v := make([][]byte, 0, columnLen)
 	dlen := dim
 	dlen /= 8
-	
+
 	for i := 0; i < columnLen; i++ {
 		entry := make([]byte, dlen)
 		v = append(v, entry)
 	}
 	column := NewColumnBinaryVector(columnName, dim, v)
-	
+
 	t.Run("test meta", func(t *testing.T) {
 		ft := FieldTypeBinaryVector
 		assert.Equal(t, "BinaryVector", ft.Name())
@@ -43,13 +43,13 @@ func TestColumnBinaryVector(t *testing.T) {
 		assert.Equal(t, FieldTypeBinaryVector, column.Type())
 		assert.Equal(t, columnLen, column.Len())
 		assert.Equal(t, dim, column.Dim())
-		assert.Equal(t ,v, column.Data())
-		
+		assert.Equal(t, v, column.Data())
+
 		var ev []byte
 		err := column.AppendValue(ev)
 		assert.Equal(t, columnLen+1, column.Len())
 		assert.Nil(t, err)
-		
+
 		err = column.AppendValue(struct{}{})
 		assert.Equal(t, columnLen+1, column.Len())
 		assert.NotNil(t, err)
@@ -70,7 +70,7 @@ func TestColumnBinaryVector(t *testing.T) {
 			Type:      schema.DataType_BinaryVector,
 			FieldName: columnName,
 		}
-		_, err := FieldDataVector(fd) 
+		_, err := FieldDataVector(fd)
 		assert.Error(t, err)
 	})
 
@@ -82,16 +82,15 @@ func TestColumnFloatVector(t *testing.T) {
 	columnLen := 12 + rand.Intn(10)
 	dim := ([]int{64, 128, 256, 512})[rand.Intn(4)]
 
-	v := make([][]float32,0, columnLen)
+	v := make([][]float32, 0, columnLen)
 	dlen := dim
-	
-	
+
 	for i := 0; i < columnLen; i++ {
 		entry := make([]float32, dlen)
 		v = append(v, entry)
 	}
 	column := NewColumnFloatVector(columnName, dim, v)
-	
+
 	t.Run("test meta", func(t *testing.T) {
 		ft := FieldTypeFloatVector
 		assert.Equal(t, "FloatVector", ft.Name())
@@ -106,13 +105,13 @@ func TestColumnFloatVector(t *testing.T) {
 		assert.Equal(t, FieldTypeFloatVector, column.Type())
 		assert.Equal(t, columnLen, column.Len())
 		assert.Equal(t, dim, column.Dim())
-		assert.Equal(t ,v, column.Data())
-		
+		assert.Equal(t, v, column.Data())
+
 		var ev []float32
 		err := column.AppendValue(ev)
 		assert.Equal(t, columnLen+1, column.Len())
 		assert.Nil(t, err)
-		
+
 		err = column.AppendValue(struct{}{})
 		assert.Equal(t, columnLen+1, column.Len())
 		assert.NotNil(t, err)
@@ -133,9 +132,8 @@ func TestColumnFloatVector(t *testing.T) {
 			Type:      schema.DataType_FloatVector,
 			FieldName: columnName,
 		}
-		_, err := FieldDataVector(fd) 
+		_, err := FieldDataVector(fd)
 		assert.Error(t, err)
 	})
 
 }
-
diff --git a/entity/schema_test.go b/entity/schema_test.go
index f79f2da..81d3b08 100644
--- a/entity/schema_test.go
+++ b/entity/schema_test.go
@@ -3,7 +3,7 @@ package entity
 import (
 	"testing"
 
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/suite"
 )
diff --git a/entity/segment_test.go b/entity/segment_test.go
index ff5801d..28993e1 100644
--- a/entity/segment_test.go
+++ b/entity/segment_test.go
@@ -3,7 +3,7 @@ package entity
 import (
 	"testing"
 
-	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
+	common "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
 	"github.com/stretchr/testify/assert"
 )
 
diff --git a/test/testcases/search_test.go b/test/testcases/search_test.go
index 4e6dd96..fa4fca8 100644
--- a/test/testcases/search_test.go
+++ b/test/testcases/search_test.go
@@ -434,24 +434,27 @@ func TestSearchInvalidVectorField(t *testing.T) {
 		// non-vector field
 		{vectorField: common.DefaultIntFieldName, errMsg: fmt.Sprintf("failed to create query plan: field (%s) to search is not of vector data type", common.DefaultIntFieldName)},
 
+		// skip since 2.3 allows empty vector field name
 		// empty field name
-		{vectorField: "", errMsg: "failed to get field schema by name: fieldName() not found"},
+		//{vectorField: "", errMsg: "failed to get field schema by name: fieldName() not found"},
 	}
 
 	sp, _ := entity.NewIndexHNSWSearchParam(74)
 	for _, invalidVectorField := range invalidVectorFields {
-		_, errSearchNotExist := mc.Search(
-			ctx, collName,
-			[]string{},
-			"",
-			[]string{common.DefaultIntFieldName},
-			common.GenSearchVectors(common.DefaultNq, common.DefaultDim, entity.FieldTypeFloatVector),
-			invalidVectorField.vectorField,
-			entity.L2,
-			common.DefaultTopK,
-			sp,
-		)
-		common.CheckErr(t, errSearchNotExist, false, invalidVectorField.errMsg)
+		t.Run(invalidVectorField.vectorField, func(t *testing.T) {
+			_, errSearchNotExist := mc.Search(
+				ctx, collName,
+				[]string{},
+				"",
+				[]string{common.DefaultIntFieldName},
+				common.GenSearchVectors(common.DefaultNq, common.DefaultDim, entity.FieldTypeFloatVector),
+				invalidVectorField.vectorField,
+				entity.L2,
+				common.DefaultTopK,
+				sp,
+			)
+			common.CheckErr(t, errSearchNotExist, false, invalidVectorField.errMsg)
+		})
 	}
 }
 
@@ -530,7 +533,7 @@ func TestSearchNotMatchMetricType(t *testing.T) {
 		common.DefaultTopK,
 		sp,
 	)
-	common.CheckErr(t, errSearchEmpty, false, "Metric type of field index isn't the same with search info")
+	common.CheckErr(t, errSearchEmpty, false, "invalid parameter")
 }
 
 // test search with invalid topK -> error