From 64f573c57aff8ee93346e3da505de434c9586d78 Mon Sep 17 00:00:00 2001
From: Andrew Grigorev <andrew@ei-grad.ru>
Date: Mon, 21 Jul 2025 01:51:39 +0300
Subject: feat: vertexai supports `global` location (#1354)

---
 src/client/vertexai.rs | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/client/vertexai.rs b/src/client/vertexai.rs
index 730c5d8..16b55d1 100644
--- a/src/client/vertexai.rs
+++ b/src/client/vertexai.rs
@@ -98,7 +98,11 @@ fn prepare_chat_completions(
     let location = self_.get_location()?;
     let access_token = get_access_token(self_.name())?;
 
-    let base_url = format!("https://{location}-aiplatform.googleapis.com/v1/projects/{project_id}/locations/{location}/publishers");
+    let base_url = if location == "global" {
+        format!("https://aiplatform.googleapis.com/v1/projects/{project_id}/locations/global/publishers")
+    } else {
+        format!("https://{location}-aiplatform.googleapis.com/v1/projects/{project_id}/locations/{location}/publishers")
+    };
 
     let model_name = self_.model.real_name();
 
@@ -153,7 +157,11 @@ fn prepare_embeddings(self_: &VertexAIClient, data: &EmbeddingsData) -> Result<R
     let location = self_.get_location()?;
     let access_token = get_access_token(self_.name())?;
 
-    let base_url = format!("https://{location}-aiplatform.googleapis.com/v1/projects/{project_id}/locations/{location}/publishers");
+    let base_url = if location == "global" {
+        format!("https://aiplatform.googleapis.com/v1/projects/{project_id}/locations/global/publishers")
+    } else {
+        format!("https://{location}-aiplatform.googleapis.com/v1/projects/{project_id}/locations/{location}/publishers")
+    };
     let url = format!(
         "{base_url}/google/models/{}:predict",
         self_.model.real_name()
-- 
cgit v1.2.3