From 3626d07d833a4ee78a7246aff916dff827de3de5 Mon Sep 17 00:00:00 2001
From: b1ek <me@blek.codes>
Date: Sun, 5 May 2024 16:38:28 +1000
Subject: [PATCH] modify config to not include model versions

---
 src/config.rs | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/src/config.rs b/src/config.rs
index d656dfc..0a9af9a 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -5,17 +5,20 @@ use serde::{Deserialize, Serialize};
 
 #[derive(Debug, Clone, Serialize, Deserialize)]
 pub enum Model {
-    #[serde(rename = "claude-instant-1.2")]
     Claude12,
-    #[serde(rename = "gpt-3.5-turbo-0125")]
     GPT35,
+    Claude,
+    GPT3,
 }
 
 impl ToString for Model {
     fn to_string(&self) -> String {
         match self {
-            Self::Claude12 => String::from("claude-instant-1.2"),
-            Self::GPT35 => String::from("gpt-3.5-turbo-0125"),
+            Self::Claude12 => panic!("Your config is outdated! Please delete your ~/.config/hey directory"),
+            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 {
     fn default() -> Self {
         Self {
-            model: Model::Claude12,
+            model: Model::Claude,
             tos: false
         }
     }
@@ -76,7 +79,10 @@ impl Config {
         if ! file_path.is_file() {
             Ok(Self::default())
         } 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)
         }
     }
 }
\ No newline at end of file