diff --git a/src/providers/llm/litellm_provider.rs b/src/providers/llm/litellm_provider.rs index f9d71bb..d8b0343 100644 --- a/src/providers/llm/litellm_provider.rs +++ b/src/providers/llm/litellm_provider.rs @@ -142,17 +142,20 @@ impl AiProvider for LiteLLMProvider { async fn complete_as( &self, - model: &str, + _model: &str, system_prompt: &str, user_prompt: &str, ) -> Result { - self.chat(&self.api_key, model, system_prompt, user_prompt).await + // LiteLLM routes everything through the configured model (e.g. askash-main). + // Logical model names like "jd-generator" are internal identifiers only. + let model = self.model.clone(); + self.chat(&self.api_key, &model, system_prompt, user_prompt).await } async fn complete_as_user( &self, user_id: Option<&str>, - model: &str, + _model: &str, system_prompt: &str, user_prompt: &str, ) -> Result { @@ -160,7 +163,8 @@ impl AiProvider for LiteLLMProvider { (Some(uid), Some(client)) => client.get_key(uid).await, _ => None, }; - self.chat(key.as_deref().unwrap_or(&self.api_key), model, system_prompt, user_prompt) + let model = self.model.clone(); + self.chat(key.as_deref().unwrap_or(&self.api_key), &model, system_prompt, user_prompt) .await } }