AI/orchai
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: remove dead code warnings in agent defaults

Browse source
This commit is contained in:
thibaud-leclere 2026-04-14 16:10:50 +02:00
parent 3cdd880344
commit dd9708bd6e
1 changed files with 16 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
src-tauri/src/models/agent.rs
View file

@ -1,4 +1,4 @@
use rusqlite::{params, Connection, OptionalExtension, Result}; use rusqlite::{params, Connection, Result};
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use uuid::Uuid; use uuid::Uuid;
@ -142,12 +142,17 @@ impl Agent {
} }
pub fn get_default_by_role(conn: &Connection, role: AgentRole) -> Result<Agent> { pub fn get_default_by_role(conn: &Connection, role: AgentRole) -> Result<Agent> {
let default_id = match role {
AgentRole::Analyst => DEFAULT_ANALYST_AGENT_ID,
AgentRole::Developer => DEFAULT_DEVELOPER_AGENT_ID,
};
conn.query_row( conn.query_row(
"SELECT id, name, role, tool, custom_prompt, is_default, created_at, updated_at "SELECT id, name, role, tool, custom_prompt, is_default, created_at, updated_at
FROM agents FROM agents
WHERE role = ?1 AND is_default = 1 WHERE id = ?1 AND role = ?2 AND is_default = 1
LIMIT 1", LIMIT 1",
params![role.as_str()], params![default_id, role.as_str()],
from_row, from_row,
) )
} }
@ -198,31 +203,24 @@ impl Agent {
)); ));
} }
let default_agent_id: String = conn let default_agent = Self::get_default_by_role(conn, agent.role.clone()).map_err(|_| {
.query_row( rusqlite::Error::InvalidParameterName(format!(
"SELECT id FROM agents WHERE role = ?1 AND is_default = 1 LIMIT 1", "No default agent found for role '{}'",
params![agent.role.as_str()], agent.role.as_str()
|row| row.get(0), ))
) })?;
.optional()?
.ok_or_else(|| {
rusqlite::Error::InvalidParameterName(format!(
"No default agent found for role '{}'",
agent.role.as_str()
))
})?;
match agent.role { match agent.role {
AgentRole::Analyst => { AgentRole::Analyst => {
conn.execute( conn.execute(
"UPDATE watched_trackers SET analyst_agent_id = ?1 WHERE analyst_agent_id = ?2", "UPDATE watched_trackers SET analyst_agent_id = ?1 WHERE analyst_agent_id = ?2",
params![default_agent_id, id], params![default_agent.id, id],
)?; )?;
} }
AgentRole::Developer => { AgentRole::Developer => {
conn.execute( conn.execute(
"UPDATE watched_trackers SET developer_agent_id = ?1 WHERE developer_agent_id = ?2", "UPDATE watched_trackers SET developer_agent_id = ?1 WHERE developer_agent_id = ?2",
params![default_agent_id, id], params![default_agent.id, id],
)?; )?;
} }
} }