Skip to content

Commit 4cf9697

Browse files
committed
update Gemini default model name
1 parent b962973 commit 4cf9697

File tree

4 files changed

+12
-8
lines changed

4 files changed

+12
-8
lines changed

src/ai/gemini/mod.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ impl LiveClient {
6363

6464
pub async fn receive(&mut self) -> anyhow::Result<types::ServerContent> {
6565
if let Some(msg) = self.ws.next().await {
66-
log::debug!("Received message: {:?}", msg);
66+
log::trace!("Received message: {:?}", msg);
6767
let msg = msg?;
6868
match msg {
6969
Message::Text(text) => {
@@ -79,7 +79,7 @@ impl LiveClient {
7979
serde_json::from_slice(&bin).map_err(|e| {
8080
anyhow::anyhow!("Failed to parse binary message: {} {:?}", e, bin)
8181
})?;
82-
log::debug!("Parsed binary message: {:?}", server_content);
82+
log::trace!("Parsed binary message: {:?}", server_content);
8383
Ok(server_content.server_content)
8484
}
8585
Message::Close { code, reason } => Err(anyhow::anyhow!(
@@ -113,7 +113,7 @@ mod test {
113113
cfg.response_modalities = Some(vec![types::Modality::TEXT]);
114114

115115
let setup = types::Setup {
116-
model: "models/gemini-2.0-flash-live-001".to_string(),
116+
model: "models/gemini-2.0-flash-exp".to_string(),
117117
generation_config: Some(cfg),
118118
system_instruction: Some(types::Content {
119119
parts: vec![types::Parts::Text(

src/ai/mod.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -647,7 +647,7 @@ impl ChatSession {
647647
gemini
648648
.model
649649
.clone()
650-
.unwrap_or("models/gemini-2.0-flash-live-001".to_string()),
650+
.unwrap_or("models/gemini-2.0-flash-exp".to_string()),
651651
None,
652652
20,
653653
tools,

src/config.rs

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,10 +47,14 @@ pub struct LLMConfig {
4747
pub extra: Option<serde_json::Value>,
4848
}
4949

50+
fn default_gemini_model() -> Option<String> {
51+
Some("models/gemini-2.0-flash-exp".to_string())
52+
}
53+
5054
#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
5155
pub struct GeminiConfig {
5256
pub api_key: String,
53-
#[serde(default)]
57+
#[serde(default = "default_gemini_model")]
5458
pub model: Option<String>,
5559
#[serde(default)]
5660
pub sys_prompts: Vec<Content>,

src/services/ws.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -859,7 +859,7 @@ async fn submit_to_gemini(
859859
let mut buff = Vec::with_capacity(5 * 1600 * 2);
860860

861861
loop {
862-
log::info!("waiting gemini response");
862+
log::debug!("waiting gemini response");
863863
match client.receive().await? {
864864
gemini::types::ServerContent::ModelTurn(turn) => {
865865
for item in turn.parts {
@@ -1033,7 +1033,7 @@ async fn handle_audio(
10331033
let model = gemini
10341034
.model
10351035
.clone()
1036-
.unwrap_or("models/gemini-2.0-flash-live-001".to_string());
1036+
.unwrap_or("models/gemini-2.0-flash-exp".to_string());
10371037

10381038
let mut generation_config = GenerationConfig::default();
10391039
generation_config.response_modalities = Some(vec![gemini::types::Modality::TEXT]);
@@ -1061,7 +1061,7 @@ async fn handle_audio(
10611061
let model = gemini
10621062
.model
10631063
.clone()
1064-
.unwrap_or("models/gemini-2.0-flash-live-001".to_string());
1064+
.unwrap_or("models/gemini-2.0-flash-exp".to_string());
10651065

10661066
let mut generation_config = GenerationConfig::default();
10671067
generation_config.response_modalities = Some(vec![gemini::types::Modality::AUDIO]);

0 commit comments

Comments
 (0)