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

modify config to not include model versions

This commit is contained in:
b1ek 2024-05-05 16:38:28 +10:00
parent 063b3317e7
commit 3626d07d83
Signed by: blek
GPG Key ID: 14546221E3595D0C
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/config.rs
View File

@ -5,17 +5,20 @@ use serde::{Deserialize, Serialize};
#[derive(Debug, Clone, Serialize, Deserialize)] #[derive(Debug, Clone, Serialize, Deserialize)]
pub enum Model { pub enum Model {
#[serde(rename = "claude-instant-1.2")]
Claude12, Claude12,
#[serde(rename = "gpt-3.5-turbo-0125")]
GPT35, GPT35,
Claude,
GPT3,
} }
impl ToString for Model { impl ToString for Model {
fn to_string(&self) -> String { fn to_string(&self) -> String {
match self { match self {
Self::Claude12 => String::from("claude-instant-1.2"), Self::Claude12 => panic!("Your config is outdated! Please delete your ~/.config/hey directory"),
Self::GPT35 => String::from("gpt-3.5-turbo-0125"), Self::GPT35 => panic!("Your config is outdated! Please delete your ~/.config/hey directory"),
Self::Claude => String::from("claude-3-haiku-20240307"),
Self::GPT3 => String::from("gpt-3.5-turbo-0125"),
} }
} }
} }
@ -29,7 +32,7 @@ pub struct Config {
impl Default for Config { impl Default for Config {
fn default() -> Self { fn default() -> Self {
Self { Self {
model: Model::Claude12, model: Model::Claude,
tos: false tos: false
} }
} }
@ -76,7 +79,10 @@ impl Config {
if ! file_path.is_file() { if ! file_path.is_file() {
Ok(Self::default()) Ok(Self::default())
} else { } else {
Ok(toml::from_str(&fs::read_to_string(file_path)?)?) let conf: Config = toml::from_str(&fs::read_to_string(file_path)?)?;
conf.model.to_string(); // so that it would panic if the config is outdated
Ok(conf)
} }
} }
} }