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.
46 lines
1.4 KiB
46 lines
1.4 KiB
package gemini
|
|
|
|
import "context"
|
|
|
|
type (
|
|
// Message 与 sys/doubao 保持一致：role + content，方便模块层 provider 切换
|
|
Message struct {
|
|
Role string `json:"role"`
|
|
Content string `json:"content"`
|
|
// Images 可选图片 url 列表：非空时下载并以 inlineData 形式随该消息发给 Gemini 视觉模型。
|
|
// 为空时按纯文本发送，行为不变。
|
|
Images []string `json:"images,omitempty"`
|
|
}
|
|
|
|
// ChatResponseChoice 与 sys/doubao 保持一致
|
|
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)
|
|
}
|
|
|