windmill_api/models/
memory_message.rs1use crate::models;
12use serde::{Deserialize, Serialize};
13
14#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize)]
16pub struct MemoryMessage {
17 #[serde(rename = "role")]
18 pub role: Role,
19 #[serde(rename = "content")]
20 pub content: String,
21}
22
23impl MemoryMessage {
24 pub fn new(role: Role, content: String) -> MemoryMessage {
26 MemoryMessage {
27 role,
28 content,
29 }
30 }
31}
32#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
34pub enum Role {
35 #[serde(rename = "user")]
36 User,
37 #[serde(rename = "assistant")]
38 Assistant,
39 #[serde(rename = "system")]
40 System,
41}
42
43impl Default for Role {
44 fn default() -> Role {
45 Self::User
46 }
47}
48