From 5b754915e4191f982b14ab0572f4502f64d8d3d2 Mon Sep 17 00:00:00 2001
From: Nathan Sobo <nathan@zed.dev>
Date: Sun, 23 Jun 2024 14:10:31 -0600
Subject: [PATCH] Respect requested gemini model (#13427)

Previously, we always went to gemini-pro.

Release Notes:

- N/A
---
 crates/collab/src/rpc.rs          | 1 +
 crates/google_ai/src/google_ai.rs | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/crates/collab/src/rpc.rs b/crates/collab/src/rpc.rs
index 45a7052f0a..af2a0f2553 100644
--- a/crates/collab/src/rpc.rs
+++ b/crates/collab/src/rpc.rs
@@ -4503,6 +4503,7 @@ async fn complete_with_google_ai(
         session.http_client.clone(),
         google_ai::API_URL,
         api_key.as_ref(),
+        &request.model.clone(),
         crate::ai::language_model_request_to_google_ai(request)?,
     )
     .await
diff --git a/crates/google_ai/src/google_ai.rs b/crates/google_ai/src/google_ai.rs
index 6081de11b9..6a9b37cb43 100644
--- a/crates/google_ai/src/google_ai.rs
+++ b/crates/google_ai/src/google_ai.rs
@@ -11,10 +11,11 @@ pub async fn stream_generate_content(
     client: Arc<dyn HttpClient>,
     api_url: &str,
     api_key: &str,
+    model: &str,
     request: GenerateContentRequest,
 ) -> Result<BoxStream<'static, Result<GenerateContentResponse>>> {
     let uri = format!(
-        "{}/v1beta/models/gemini-pro:streamGenerateContent?alt=sse&key={}",
+        "{}/v1beta/models/{model}:streamGenerateContent?alt=sse&key={}",
         api_url, api_key
     );