Respect requested gemini model (#13427)

Nathan Sobo created 2 years ago

Previously, we always went to gemini-pro.

Release Notes:

- N/A

Change summary

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

Detailed changes

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

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
     );