add llama and mixtral models

b1ek · Jun 4, 2024 · 6dc8ef1 · 6dc8ef1
1 parent 28185c5
commit 6dc8ef1
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/config.rs b/src/config.rs
@@ -5,10 +5,15 @@ use serde::{Deserialize, Serialize};
 
 #[derive(Debug, Clone, Serialize, Deserialize)]
 pub enum Model {
+    // outdated
     Claude12,
     GPT35,
+
+    // current
     Claude,
     GPT3,
+    Llama,
+    Mixtral
 }
 
 impl ToString for Model {
@@ -19,6 +24,8 @@ impl ToString for Model {
 
             Self::Claude => String::from("claude-3-haiku-20240307"),
             Self::GPT3 => String::from("gpt-3.5-turbo-0125"),
+            Self::Llama => String::from("meta-llama/Llama-3-70b-chat-hf"),
+            Self::Mixtral => String::from("mistralai/Mixtral-8x7B-Instruct-v0.1")
         }
     }
 }