e874e8dfb4
- Added nil check for script context in the call method to prevent potential nil pointer dereference. - Updated the String method in the Content struct to include additional properties (text and props) for better message representation. These changes improve the robustness and maintainability of the Neo API, paving the way for enhanced message handling and assistant functionalities.
100 lines
1.9 KiB
Go
100 lines
1.9 KiB
Go
package message
|
|
|
|
import (
|
|
jsoniter "github.com/json-iterator/go"
|
|
)
|
|
|
|
const (
|
|
// ContentStatusPending the content status pending
|
|
ContentStatusPending = iota
|
|
// ContentStatusDone the content status done
|
|
ContentStatusDone
|
|
// ContentStatusError the content status error
|
|
ContentStatusError
|
|
)
|
|
|
|
// Content the content
|
|
type Content struct {
|
|
ID string `json:"id"`
|
|
Name string `json:"name"`
|
|
Bytes []byte `json:"bytes"`
|
|
Type string `json:"type"` // text, function, error
|
|
Status uint8 `json:"status"` // 0: pending, 1: done
|
|
}
|
|
|
|
// NewContent create a new content
|
|
func NewContent(typ string) *Content {
|
|
if typ == "" {
|
|
typ = "text"
|
|
}
|
|
|
|
return &Content{
|
|
Bytes: []byte{},
|
|
Type: typ,
|
|
Status: ContentStatusPending,
|
|
}
|
|
}
|
|
|
|
// String the content string
|
|
func (c *Content) String() string {
|
|
if c.Type == "function" {
|
|
|
|
var arguments interface{} = string(c.Bytes)
|
|
if c.Status == ContentStatusDone {
|
|
var vv interface{} = nil
|
|
err := jsoniter.Unmarshal(c.Bytes, &vv)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
arguments = vv
|
|
}
|
|
|
|
data := map[string]interface{}{
|
|
"id": c.ID,
|
|
"type": "function",
|
|
"text": c.Name,
|
|
"props": map[string]interface{}{
|
|
"id": c.ID,
|
|
"name": c.Name,
|
|
"arguments": arguments,
|
|
},
|
|
"function": map[string]interface{}{
|
|
"name": c.Name,
|
|
"arguments": arguments,
|
|
},
|
|
}
|
|
|
|
raw, err := jsoniter.MarshalToString(data)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
|
|
return raw
|
|
}
|
|
return string(c.Bytes)
|
|
}
|
|
|
|
// SetID set the content id
|
|
func (c *Content) SetID(id string) {
|
|
c.ID = id
|
|
}
|
|
|
|
// SetName set the content name
|
|
func (c *Content) SetName(name string) {
|
|
c.Name = name
|
|
}
|
|
|
|
// SetType set the content type
|
|
func (c *Content) SetType(typ string) {
|
|
c.Type = typ
|
|
}
|
|
|
|
// Append append the content
|
|
func (c *Content) Append(data string) {
|
|
c.Bytes = append(c.Bytes, []byte(data)...)
|
|
}
|
|
|
|
// SetStatus set the content status
|
|
func (c *Content) SetStatus(status uint8) {
|
|
c.Status = status
|
|
}
|