Yehowshua/ZIm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Respect requested gemini model (#13427)

Browse source 
Previously, we always went to gemini-pro.

Release Notes:

- N/A
This commit is contained in:
Nathan Sobo 2024-06-23 14:10:31 -06:00 • committed by GitHub
parent 9298d3b525
commit 5b754915e4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
crates/collab/src/rpc.rs
View file

@ -4503,6 +4503,7 @@ async fn complete_with_google_ai(
session.http_client.clone(), session.http_client.clone(),
google_ai::API_URL, google_ai::API_URL,
api_key.as_ref(), api_key.as_ref(),
&request.model.clone(),
crate::ai::language_model_request_to_google_ai(request)?, crate::ai::language_model_request_to_google_ai(request)?,
) )
.await .await

3
crates/google_ai/src/google_ai.rs
View file

@ -11,10 +11,11 @@ pub async fn stream_generate_content(
client: Arc<dyn HttpClient>, client: Arc<dyn HttpClient>,
api_url: &str, api_url: &str,
api_key: &str, api_key: &str,
model: &str,
request: GenerateContentRequest, request: GenerateContentRequest,
) -> Result<BoxStream<'static, Result<GenerateContentResponse>>> { ) -> Result<BoxStream<'static, Result<GenerateContentResponse>>> {
let uri = format!( let uri = format!(
"{}/v1beta/models/gemini-pro:streamGenerateContent?alt=sse&key={}", "{}/v1beta/models/{model}:streamGenerateContent?alt=sse&key={}",
api_url, api_key api_url, api_key
); );