style(gemini): apply rustfmt to oauth endpoint patch
This commit is contained in:
parent
1d8e57d388
commit
9e9a4a53ab
1 changed files with 26 additions and 9 deletions
|
|
@ -268,9 +268,7 @@ impl GeminiProvider {
|
||||||
}
|
}
|
||||||
_ => {
|
_ => {
|
||||||
let model_name = Self::format_model_name(model);
|
let model_name = Self::format_model_name(model);
|
||||||
let base_url = format!(
|
let base_url = format!("{PUBLIC_API_ENDPOINT}/{model_name}:generateContent");
|
||||||
"{PUBLIC_API_ENDPOINT}/{model_name}:generateContent"
|
|
||||||
);
|
|
||||||
|
|
||||||
if auth.is_api_key() {
|
if auth.is_api_key() {
|
||||||
format!("{base_url}?key={}", auth.credential())
|
format!("{base_url}?key={}", auth.credential())
|
||||||
|
|
@ -294,16 +292,35 @@ impl GeminiProvider {
|
||||||
let internal_request = InternalGenerateContentRequest {
|
let internal_request = InternalGenerateContentRequest {
|
||||||
model: Self::format_model_name(model),
|
model: Self::format_model_name(model),
|
||||||
generation_config: request.generation_config.clone(),
|
generation_config: request.generation_config.clone(),
|
||||||
contents: request.contents.iter().map(|c| Content {
|
contents: request
|
||||||
|
.contents
|
||||||
|
.iter()
|
||||||
|
.map(|c| Content {
|
||||||
role: c.role.clone(),
|
role: c.role.clone(),
|
||||||
parts: c.parts.iter().map(|p| Part { text: p.text.clone() }).collect(),
|
parts: c
|
||||||
}).collect(),
|
.parts
|
||||||
|
.iter()
|
||||||
|
.map(|p| Part {
|
||||||
|
text: p.text.clone(),
|
||||||
|
})
|
||||||
|
.collect(),
|
||||||
|
})
|
||||||
|
.collect(),
|
||||||
system_instruction: request.system_instruction.as_ref().map(|si| Content {
|
system_instruction: request.system_instruction.as_ref().map(|si| Content {
|
||||||
role: si.role.clone(),
|
role: si.role.clone(),
|
||||||
parts: si.parts.iter().map(|p| Part { text: p.text.clone() }).collect(),
|
parts: si
|
||||||
|
.parts
|
||||||
|
.iter()
|
||||||
|
.map(|p| Part {
|
||||||
|
text: p.text.clone(),
|
||||||
|
})
|
||||||
|
.collect(),
|
||||||
}),
|
}),
|
||||||
};
|
};
|
||||||
self.client.post(url).json(&internal_request).bearer_auth(token)
|
self.client
|
||||||
|
.post(url)
|
||||||
|
.json(&internal_request)
|
||||||
|
.bearer_auth(token)
|
||||||
}
|
}
|
||||||
_ => self.client.post(url).json(request),
|
_ => self.client.post(url).json(request),
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue