5e5524293f
- Updated streamChat method to utilize content.Type for function handling, enhancing the interaction with function calls. - Improved the String method in Content struct to handle function arguments more robustly, including JSON unmarshalling for completed content. - Set default message type to "text" in NewOpenAI function, ensuring consistent message processing. These changes enhance the flexibility and maintainability of the Neo API, paving the way for improved assistant functionalities and message management.
94 lines
1.8 KiB
Go
94 lines
1.8 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",
|
|
"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
|
|
}
|