diff --git a/go.mod b/go.mod
index ebca7d78..afed64c8 100644
--- a/go.mod
+++ b/go.mod
@@ -31,6 +31,7 @@ require (
 	github.com/spf13/pflag v1.0.5
 	github.com/spf13/viper v1.18.2
 	github.com/stretchr/testify v1.9.0
+	github.com/tidbcloud/tidbcloud-cli/pkg v0.0.1
 	github.com/xo/usql v0.19.2
 	github.com/zalando/go-keyring v0.2.3
 	go.einride.tech/aip v0.67.1
@@ -118,3 +119,5 @@ require (
 	gopkg.in/ini.v1 v1.67.0 // indirect
 	gopkg.in/natefinch/lumberjack.v2 v2.0.0 // indirect
 )
+
+replace github.com/tidbcloud/tidbcloud-cli/pkg => ./pkg
diff --git a/pkg/go.mod b/pkg/go.mod
new file mode 100644
index 00000000..6e85ac08
--- /dev/null
+++ b/pkg/go.mod
@@ -0,0 +1,3 @@
+module github.com/tidbcloud/tidbcloud-cli/pkg
+
+go 1.22