From 14cb2b95c6d8fec2128a9108a5d90ad7b467ecf9 Mon Sep 17 00:00:00 2001
From: dinhkarate <dinhkarat@gmail.com>
Date: Thu, 29 Jan 2026 13:29:55 +0700
Subject: [PATCH] feat(vertex): add --vertex-import-prefix flag for model
 namespacing

---
 cmd/server/main.go            | 4 +++-
 internal/cmd/vertex_import.go | 3 ++-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/cmd/server/main.go b/cmd/server/main.go
index 385d7cfa..740a7511 100644
--- a/cmd/server/main.go
+++ b/cmd/server/main.go
@@ -65,6 +65,7 @@ func main() {
 	var antigravityLogin bool
 	var projectID string
 	var vertexImport string
+	var vertexImportPrefix string
 	var configPath string
 	var password string
 
@@ -81,6 +82,7 @@ func main() {
 	flag.StringVar(&projectID, "project_id", "", "Project ID (Gemini only, not required)")
 	flag.StringVar(&configPath, "config", DefaultConfigPath, "Configure File Path")
 	flag.StringVar(&vertexImport, "vertex-import", "", "Import Vertex service account key JSON file")
+	flag.StringVar(&vertexImportPrefix, "vertex-import-prefix", "", "Prefix for Vertex model namespacing (use with -vertex-import)")
 	flag.StringVar(&password, "password", "", "")
 
 	flag.CommandLine.Usage = func() {
@@ -449,7 +451,7 @@ func main() {
 
 	if vertexImport != "" {
 		// Handle Vertex service account import
-		cmd.DoVertexImport(cfg, vertexImport)
+		cmd.DoVertexImport(cfg, vertexImport, vertexImportPrefix)
 	} else if login {
 		// Handle Google/Gemini login
 		cmd.DoLogin(cfg, projectID, options)
diff --git a/internal/cmd/vertex_import.go b/internal/cmd/vertex_import.go
index 32d782d8..034906ac 100644
--- a/internal/cmd/vertex_import.go
+++ b/internal/cmd/vertex_import.go
@@ -20,7 +20,7 @@ import (
 // DoVertexImport imports a Google Cloud service account key JSON and persists
 // it as a "vertex" provider credential. The file content is embedded in the auth
 // file to allow portable deployment across stores.
-func DoVertexImport(cfg *config.Config, keyPath string) {
+func DoVertexImport(cfg *config.Config, keyPath string, prefix string) {
 	if cfg == nil {
 		cfg = &config.Config{}
 	}
@@ -69,6 +69,7 @@ func DoVertexImport(cfg *config.Config, keyPath string) {
 		ProjectID:      projectID,
 		Email:          email,
 		Location:       location,
+		Prefix:         strings.TrimSpace(prefix),
 	}
 	metadata := map[string]any{
 		"service_account": sa,