From 1d9a05fb51ea540c1f6358a25b62f83dec43f07e Mon Sep 17 00:00:00 2001 From: "Ahmad-A0 (aider)" <71089234+Ahmad-A0@users.noreply.github.com> Date: Sun, 4 Aug 2024 01:41:15 +0100 Subject: [PATCH] feat: Add gpt-4o-mini model to config --- src/config.rs | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/config.rs b/src/config.rs index 9f7d486..3ada810 100644 --- a/src/config.rs +++ b/src/config.rs @@ -13,7 +13,8 @@ pub enum Model { Claude, GPT3, Llama, - Mixtral + Mixtral, + GPT4OMini } impl ToString for Model { @@ -25,7 +26,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") + Self::Mixtral => String::from("mistralai/Mixtral-8x7B-Instruct-v0.1"), + Self::GPT4OMini => String::from("gpt-4-0125-preview") } } } @@ -92,4 +94,4 @@ impl Config { Ok(conf) } } -} \ No newline at end of file +}