feat: chat typing

src/dto/api.rs

@@ -3,27 +3,6 @@ use utoipa::ToSchema;
 
 use crate::errors::OllamaError;
 
-#[derive(Debug, Serialize, Deserialize, ToSchema)]
-#[serde(rename_all = "lowercase")]
-pub enum Role {
-    System,
-    User,
-    Assistant,
-}
-
-#[derive(Serialize, Deserialize, ToSchema)]
-pub struct ChatResponse {
-    pub response: String,
-}
-
-#[derive(Debug, Serialize, Deserialize, ToSchema)]
-pub struct Message {
-    pub role: Role,
-    pub content: String,
-}
-
-// ---------------------------
-
 #[derive(Debug, Serialize, Deserialize, ToSchema)]
 pub struct ModelsResponse {
     pub models: Vec<ModelInfo>,
@@ -155,3 +134,62 @@ pub struct CompletionChunk {
     pub object: String,
     pub choices: Vec<Choice>,
 }
+
+#[derive(Debug, Deserialize, Serialize, ToSchema)]
+pub struct ChatRequest {
+    #[serde(flatten)]
+    pub base: BaseLLMRequest,
+
+    pub messages: Vec<Message>,
+}
+
+#[derive(Debug, Deserialize, Serialize, ToSchema)]
+pub struct Message {
+    pub role: Role,
+    pub content: String,
+}
+
+#[derive(Debug, Deserialize, Serialize, ToSchema)]
+#[serde(rename_all = "lowercase")]
+pub enum Role {
+    System,
+    User,
+    Assistant,
+}
+
+#[derive(Debug, Serialize, Deserialize, ToSchema)]
+pub struct ChatCompletionResponse {
+    pub id: String,
+    pub object: String,
+    pub created: u64,
+    pub model: String,
+    pub choices: Vec<ChatChoice>,
+    pub usage: Option<Usage>, // optional (Ollama may not always provide)
+}
+
+#[derive(Debug, Serialize, Deserialize, ToSchema)]
+pub struct ChatChoice {
+    pub index: u32,
+    pub message: Message,
+    pub finish_reason: FinishReason,
+}
+
+#[derive(Debug, Serialize)]
+pub struct ChatCompletionChunk {
+    pub id: String,
+    pub object: String,
+    pub choices: Vec<ChatChunkChoice>,
+}
+
+#[derive(Debug, Serialize)]
+pub struct ChatChunkChoice {
+    pub index: u32,
+    pub delta: ChatDelta,
+    pub finish_reason: Option<FinishReason>,
+}
+
+#[derive(Debug, Serialize)]
+pub struct ChatDelta {
+    pub role: Option<Role>,
+    pub content: Option<String>,
+}