fix assistant and senator

Signed-off-by: salmonik <salmonik@tuta.io>

1 files changed, 3 insertions, 15 deletions

diff --git a/src/commands/ai.rs b/src/commands/ai.rs
index a75f164..5ab077e 100644
--- a/src/commands/ai.rs
+++ b/src/commands/ai.rs
@@ -4,7 +4,7 @@ use crate::lang;
 
 type Context<'a> = poise::Context<'a, Data, Error>;
 
-/// Query the AI senator via OpenRouter
+/// Query the AI senator via OpenRouter or local LLM
 #[poise::command(slash_command)]
 pub async fn senator(
     ctx: Context<'_>,
@@ -13,20 +13,8 @@ pub async fn senator(
 ) -> crate::error::Result<()> {
     ctx.defer().await?;
 
-    let api_key = ctx
-        .data()
-        .config
-        .openrouter_api_key
-        .as_ref()
-        .ok_or_else(|| Error::Custom(lang::L_OPENROUTER_NO_KEY.into()))?;
-    let model = ctx
-        .data()
-        .config
-        .openrouter_model
-        .as_deref()
-        .unwrap_or("cognitivecomputations/dolphin-mistral-24b-venice-edition:free");
-
-    let client = crate::openrouter::OpenRouterClient::new(api_key.clone(), model.to_string());
+    let (api_url, api_key, model) = ctx.data().config.resolve_llm();
+    let client = crate::openrouter::OpenRouterClient::new(api_url, api_key, model);
     let response = client.chat(lang::SENATOR_SYSTEM_PROMPT, &text).await?;
 
     if tts_vc.unwrap_or(false) {