33
44package software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat
55
6+ import com.fasterxml.jackson.annotation.JsonProperty
7+
68data class ChatMessage (
79 val type : MessageType ? = MessageType .ANSWER ,
810 val header : MessageHeader ? = null ,
@@ -68,20 +70,34 @@ data class Changes(
6870)
6971
7072enum class IconType {
73+ @JsonProperty(" file" )
7174 FILE ,
75+
76+ @JsonProperty(" folder" )
7277 FOLDER ,
78+
79+ @JsonProperty(" code-block" )
7380 CODE_BLOCK ,
81+
82+ @JsonProperty(" list-add" )
7483 LIST_ADD ,
84+
85+ @JsonProperty(" magic" )
7586 MAGIC ,
87+
88+ @JsonProperty(" help" )
7689 HELP ,
90+
91+ @JsonProperty(" trash" )
7792 TRASH ,
93+
94+ @JsonProperty(" search" )
7895 SEARCH ,
96+
97+ @JsonProperty(" calendar" )
7998 CALENDAR ,
8099 ;
81100
82- val value: String
83- get() = name.lowercase().replace(' _' , ' -' )
84-
85101 companion object {
86102 private val stringToEnum: Map <String , IconType > = entries.associateBy { it.name.lowercase() }
87103
@@ -90,38 +106,55 @@ enum class IconType {
90106}
91107
92108enum class Status {
109+ @JsonProperty(" info" )
93110 INFO ,
111+
112+ @JsonProperty(" success" )
94113 SUCCESS ,
114+
115+ @JsonProperty(" warning" )
95116 WARNING ,
96- ERROR ,
97- ;
98117
99- val value : String
100- get() = name.lowercase()
118+ @JsonProperty( " error " )
119+ ERROR ,
101120}
102121
103122enum class ButtonStatus {
123+ @JsonProperty(" main" )
104124 MAIN ,
125+
126+ @JsonProperty(" primary" )
105127 PRIMARY ,
128+
129+ @JsonProperty(" clear" )
106130 CLEAR ,
131+
132+ @JsonProperty(" info" )
107133 INFO ,
134+
135+ @JsonProperty(" success" )
108136 SUCCESS ,
137+
138+ @JsonProperty(" warning" )
109139 WARNING ,
110- ERROR ,
111- ;
112140
113- val value : String
114- get() = name.lowercase()
141+ @JsonProperty( " error " )
142+ ERROR ,
115143}
116144
117145enum class MessageType {
146+ @JsonProperty(" answer" )
118147 ANSWER ,
148+
149+ @JsonProperty(" prompt" )
119150 PROMPT ,
151+
152+ @JsonProperty(" system-prompt" )
120153 SYSTEM_PROMPT ,
154+
155+ @JsonProperty(" directive" )
121156 DIRECTIVE ,
122- TOOL ,
123- ;
124157
125- val value : String
126- get() = name.lowercase().replace( ' _ ' , ' - ' )
158+ @JsonProperty( " tool " )
159+ TOOL ,
127160}
0 commit comments