Use GPT-4o-mini, rejig structs

Author: matt-thomson

clippy/rust/src/chat.rs

@@ -1,4 +1,4 @@
-use crate::openai::{completions, Message};
+use crate::openai::{completions, Message, Role};
 
 #[derive(Default)]
 pub struct Chat {
@@ -8,14 +8,14 @@ pub struct Chat {
 impl Chat {
     pub fn send(&mut self, input: &str) -> Result<&str, ureq::Error> {
         self.messages.push(Message {
-            role: "user".to_string(),
+            role: Role::User,
             content: input.to_string(),
         });
 
         let completions = completions(&self.messages)?;
 
         self.messages.push(Message {
-            role: "assistant".to_string(),
+            role: Role::Assistant,
             content: completions.choices[0].message.content.to_string(),
         });
 

clippy/rust/src/openai.rs

@@ -4,7 +4,7 @@ const OPENAI_TOKEN: &str = include_str!("../../../.openai-token");
 
 #[derive(Serialize)]
 pub struct Request<'a> {
-    pub model: String,
+    pub model: &'static str,
     pub messages: &'a [Message],
     pub temperature: f32,
 }
@@ -19,15 +19,23 @@ pub struct Choice {
     pub message: Message,
 }
 
+#[derive(Serialize, Deserialize)]
+#[serde(rename_all = "lowercase")]
+pub enum Role {
+    System,
+    User,
+    Assistant,
+}
+
 #[derive(Serialize, Deserialize)]
 pub struct Message {
-    pub role: String,
+    pub role: Role,
     pub content: String,
 }
 
 pub fn completions(messages: &[Message]) -> Result<Response, ureq::Error> {
     let request = Request {
-        model: "gpt-3.5-turbo".to_string(),
+        model: "gpt-4o-mini",
         messages,
         temperature: 0.7,
     };