You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
40 lines
1.1 KiB
40 lines
1.1 KiB
package doubao
|
|
|
|
import "context"
|
|
|
|
type (
|
|
Message struct {
|
|
Role string `json:"role"`
|
|
Content string `json:"content"`
|
|
}
|
|
//回应流式切片
|
|
ChatResponseChoice struct {
|
|
Role string `json:"role"`
|
|
Content string `json:"content,omitempty"`
|
|
Meta map[string]interface{} `json:"meta,omitempty"`
|
|
}
|
|
ISys interface {
|
|
Chat(ctx context.Context, msgs []Message) (result *ChatResponseChoice, err error)
|
|
ChatForSteams(ctx context.Context, msgs []Message, choiceChan chan *ChatResponseChoice) (err error)
|
|
}
|
|
)
|
|
|
|
var defsys ISys
|
|
|
|
func OnInit(config map[string]interface{}, option ...Option) (err error) {
|
|
defsys, err = newSys(newOptions(config, option...))
|
|
return
|
|
}
|
|
|
|
func NewSys(option ...Option) (sys ISys, err error) {
|
|
sys, err = newSys(newOptionsByOption(option...))
|
|
return
|
|
}
|
|
|
|
func Chat(ctx context.Context, msgs []Message) (result *ChatResponseChoice, err error) {
|
|
return defsys.Chat(ctx, msgs)
|
|
}
|
|
|
|
func ChatForSteams(ctx context.Context, msgs []Message, choiceChan chan *ChatResponseChoice) (err error) {
|
|
return defsys.ChatForSteams(ctx, msgs, choiceChan)
|
|
}
|
|
|