mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-09-17 09:16:36 +08:00
27 lines
521 B
Go
27 lines
521 B
Go
package coze
|
|
|
|
import (
|
|
"github.com/songquanpeng/one-api/relay/adaptor/coze/constant/event"
|
|
"strings"
|
|
)
|
|
|
|
func event2StopReason(e *string) string {
|
|
if e == nil || *e == event.Message {
|
|
return ""
|
|
}
|
|
return "stop"
|
|
}
|
|
|
|
func splitOnDoubleNewline(data []byte, atEOF bool) (advance int, token []byte, err error) {
|
|
if atEOF && len(data) == 0 {
|
|
return 0, nil, nil
|
|
}
|
|
if i := strings.Index(string(data), "\n\n"); i >= 0 {
|
|
return i + 1, data[0:i], nil
|
|
}
|
|
if atEOF {
|
|
return len(data), data, nil
|
|
}
|
|
return 0, nil, nil
|
|
}
|