allow commands in image caption, add LFM model

DuckyBlender · Oct 16, 2024 · ccd5429 · ccd5429
1 parent 43dd1d1
commit ccd5429
Show file tree

Hide file tree

Showing 2 changed files with 173 additions and 170 deletions.
diff --git a/src/apis/openai.rs b/src/apis/openai.rs
@@ -35,6 +35,7 @@ impl OpenAIClient {
 
     fn get_model_and_provider(model: &BotCommand, image: bool) -> (String, Providers) {
         match model {
+            BotCommand::Lfm => ("liquid/lfm-40b:free".to_string(), Providers::OpenRouter),
             BotCommand::Llama => {
                 if image {
                     (
@@ -79,7 +80,7 @@ impl OpenAIClient {
     fn get_system_prompt(model: &BotCommand) -> Option<&'static str> {
         match model {
             BotCommand::Caveman => Some("You are a caveman. Speak like a caveman would. All caps, simple words, grammar mistakes etc."),
-            BotCommand::Llama | BotCommand::Lobotomy | BotCommand::Llama405 => Some("Be concise and precise. Don't be verbose. Answer in the user's language."),
+            BotCommand::Llama | BotCommand::Lobotomy | BotCommand::Llama405 | BotCommand::Lfm => Some("Be concise and precise. Don't be verbose. Answer in the user's language."),
             BotCommand::Help | BotCommand::Start | BotCommand::Flux => unreachable!(),
         }
     }