diff options
| author | sigoden <sigoden@gmail.com> | 2025-02-26 21:06:11 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-02-26 21:06:11 +0800 |
| commit | cb54f8643a965fbc35b1f07eb3ea026692eb8091 (patch) | |
| tree | 719617569d379e4c34badbff95a77d0771df7213 /src/client/vertexai.rs | |
| parent | 6b28b1c4fcf57edc4375703bca34eeafa597c905 (diff) | |
| download | aichat-cb54f8643a965fbc35b1f07eb3ea026692eb8091.tar.gz | |
refactor: several improvements (#1203)
Diffstat (limited to 'src/client/vertexai.rs')
| -rw-r--r-- | src/client/vertexai.rs | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/client/vertexai.rs b/src/client/vertexai.rs index 7cf5a1f..730c5d8 100644 --- a/src/client/vertexai.rs +++ b/src/client/vertexai.rs @@ -154,7 +154,10 @@ fn prepare_embeddings(self_: &VertexAIClient, data: &EmbeddingsData) -> Result<R 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 url = format!("{base_url}/google/models/{}:predict", self_.model.real_name()); + let url = format!( + "{base_url}/google/models/{}:predict", + self_.model.real_name() + ); let instances: Vec<_> = data.texts.iter().map(|v| json!({"content": v})).collect(); |