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