blek
/
hey
2
3
Fork 0
Code Issues 2 Pull Requests Packages Projects Releases 4 Wiki Activity

add llama and mixtral models

This commit is contained in:
b1ek 2024-06-05 09:36:33 +10:00
parent 28185c57a2
commit 6dc8ef16b2
Signed by: blek
GPG Key ID: 14546221E3595D0C
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/config.rs
View File

@ -5,10 +5,15 @@ use serde::{Deserialize, Serialize};
#[derive(Debug, Clone, Serialize, Deserialize)] #[derive(Debug, Clone, Serialize, Deserialize)]
pub enum Model { pub enum Model {
// outdated
Claude12, Claude12,
GPT35, GPT35,
// current
Claude, Claude,
GPT3, GPT3,
Llama,
Mixtral
} }
impl ToString for Model { impl ToString for Model {
@ -19,6 +24,8 @@ impl ToString for Model {
Self::Claude => String::from("claude-3-haiku-20240307"), Self::Claude => String::from("claude-3-haiku-20240307"),
Self::GPT3 => String::from("gpt-3.5-turbo-0125"), 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")
} }
} }
} }