Skip to content

Commit 6dc8ef1

Browse files
committed
add llama and mixtral models
1 parent 28185c5 commit 6dc8ef1

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

src/config.rs

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,10 +5,15 @@ use serde::{Deserialize, Serialize};
55

66
#[derive(Debug, Clone, Serialize, Deserialize)]
77
pub enum Model {
8+
// outdated
89
Claude12,
910
GPT35,
11+
12+
// current
1013
Claude,
1114
GPT3,
15+
Llama,
16+
Mixtral
1217
}
1318

1419
impl ToString for Model {
@@ -19,6 +24,8 @@ impl ToString for Model {
1924

2025
Self::Claude => String::from("claude-3-haiku-20240307"),
2126
Self::GPT3 => String::from("gpt-3.5-turbo-0125"),
27+
Self::Llama => String::from("meta-llama/Llama-3-70b-chat-hf"),
28+
Self::Mixtral => String::from("mistralai/Mixtral-8x7B-Instruct-v0.1")
2229
}
2330
}
2431
}

0 commit comments

Comments
 (0)