diff options
| author | Andrew Grigorev <andrew@ei-grad.ru> | 2025-07-21 01:51:39 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-07-21 06:51:39 +0800 |
| commit | 64f573c57aff8ee93346e3da505de434c9586d78 (patch) | |
| tree | 729d320a37d8670ffd0268e7f25850f63a5e2003 /src | |
| parent | 0cebc27a038bdeb77b591d9288b7e8a5a248230c (diff) | |
| download | aichat-64f573c57aff8ee93346e3da505de434c9586d78.tar.gz | |
feat: vertexai supports `global` location (#1354)
Diffstat (limited to 'src')
| -rw-r--r-- | src/client/vertexai.rs | 12 |
1 files 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() |