From 6dc8ef16b2da25a3b0f0a73ee5b9ab8ca30e65c2 Mon Sep 17 00:00:00 2001
From: b1ek <me@blek.codes>
Date: Wed, 5 Jun 2024 09:36:33 +1000
Subject: [PATCH] add llama and mixtral models

---
 src/config.rs | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/config.rs b/src/config.rs
index 0a9af9a..9f7d486 100644
--- 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")
         }
     }
 }