add API response models

Author: maltsev-dev

src/models/general/llm.rs

@@ -1,14 +1,29 @@
 use serde::{Deserialize, Serialize};
 
-#[derive(Debug, Serialize, Deserialize, Clone)]
+#[derive(Debug, Serialize, Clone)]
 pub struct Message {
     pub role: String,
-    pub content: String
+    pub content: String,
 }
 
-#[derive(Debug, Serialize, Deserialize, Clone)]
+#[derive(Debug, Serialize, Clone)]
 pub struct ChatCompletion {
     pub model: String,
     pub messages: Vec<Message>,
-    pub temperature: f32
+    pub temperature: f32,
+}
+
+#[derive(Debug, Deserialize)]
+pub struct APIMessage {
+    pub content: String,
+}
+
+#[derive(Debug, Deserialize)]
+pub struct APIChoice {
+    pub message: APIMessage,
+}
+
+#[derive(Debug, Deserialize)]
+pub struct APIResponse {
+    pub choices: Vec<APIChoice>,
 }