feat(rig-807): support Vertex AI

Cargo.toml

@@ -16,7 +16,7 @@ members = [
     "rig-bedrock",
     "rig-milvus",
     "rig-wasm",
-    "rig-s3vectors", "rig-helixdb",
+    "rig-s3vectors", "rig-helixdb", "rig-vertex",
 ]
 
 [profile.release]

rig-core/examples/agent_with_vertex.rs

@@ -0,0 +1,34 @@
+use rig::prelude::*;
+
+use rig::providers::gemini::completion::gemini_api_types::{
+    AdditionalParameters, GenerationConfig,
+};
+use rig::{completion::Prompt, providers};
+
+#[tokio::main]
+async fn main() -> Result<(), anyhow::Error> {
+    // Create OpenAI client
+    let client = providers::vertex::Client::from_env();
+
+    let gen_cfg = GenerationConfig {
+        top_k: Some(1),
+        top_p: Some(0.95),
+        candidate_count: Some(1),
+        ..Default::default()
+    };
+    let cfg = AdditionalParameters::default().with_config(gen_cfg);
+
+    // Create agent with a single context prompt
+    let comedian_agent = client
+        .agent("gemini-2.5-flash")
+        .preamble("You are a comedian here to entertain the user using humour and jokes.")
+        .additional_params(serde_json::to_value(cfg).unwrap())
+        .build();
+
+    // Prompt the agent and print the response
+    let response = comedian_agent.prompt("Entertain me!").await?;
+
+    println!("{response}");
+
+    Ok(())
+}

rig-core/src/providers/mod.rs

@@ -62,5 +62,6 @@
 pub mod openrouter;
 pub mod perplexity;
 pub mod together;
+pub mod vertex;
 pub mod voyageai;
 pub mod xai;

rig-vertex/Cargo.toml

@@ -0,0 +1,22 @@
+[package]
+name = "rig-vertex"
+version = "0.1.0"
+edition = { workspace = true }
+license = "MIT"
+readme = "README.md"
+description = "Google Vertex model provider Rig integration."
+repository = "https://github.com/0xPlaygrounds/rig"
+
+[dependencies]
+rig-core = { path = "../rig-core", version = "0.21.0" }
+reqwest = { workspace = true, features = ["json"] }
+serde = { workspace = true, features = ["derive"] }
+serde_json = { workspace = true }
+tracing = { workspace = true }
+schemars = { workspace = true }
+async-stream = { workspace = true }
+
+[dev-dependencies]
+anyhow = { workspace = true }
+tokio = { workspace = true, features = ["full"] }
+tracing-subscriber = { workspace = true }