catface_backend_go/app/model_res/rag_websocket.go

28 lines
443 B
Go
Raw Normal View History

package model_res
2024-11-19 03:21:28 +08:00
import (
"catface/app/global/consts"
"encoding/json"
)
2024-11-19 03:21:28 +08:00
func CreateNlpWebSocketResult(t string, data any) *NlpWebSocketResult {
if t == "" {
t = consts.AiMessageTypeText
2024-11-19 03:21:28 +08:00
}
return &NlpWebSocketResult{
Type: t,
Data: data,
}
}
type NlpWebSocketResult struct {
Type string `json:"type"`
Data any `json:"data"`
}
func (n *NlpWebSocketResult) JsonMarshal() []byte {
data, _ := json.Marshal(n)
return data
}