84 lines
2.6 KiB
Go
84 lines
2.6 KiB
Go
|
package openai
|
||
|
|
|
||
|
|
import (
|
||
|
|
"apigo.cc/ai/llm/llm"
|
||
|
|
"context"
|
||
|
|
"github.com/sashabaranov/go-openai"
|
||
|
|
"strings"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
// func (lm *LLM) FastMakeImage(prompt, size, refImage string) ([]string, llm.Usage, error) {
|
||
|
|
// return lm.MakeImage(ModelDallE3Std, prompt, size, refImage)
|
||
|
|
// }
|
||
|
|
//
|
||
|
|
// func (lm *LLM) BestMakeImage(prompt, size, refImage string) ([]string, llm.Usage, error) {
|
||
|
|
// return lm.MakeImage(ModelDallE3HD, prompt, size, refImage)
|
||
|
|
// }
|
||
|
|
//
|
||
|
|
// func (lm *LLM) MakeImage(model, prompt, size, refImage string) ([]string, llm.Usage, error) {
|
||
|
|
func (lm *LLM) FastMakeImage(prompt string, config llm.GCConfig) ([]string, llm.Usage, error) {
|
||
|
|
config.Model = ModelDallE3Std
|
||
|
|
return lm.MakeImage(prompt, config)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (lm *LLM) BestMakeImage(prompt string, config llm.GCConfig) ([]string, llm.Usage, error) {
|
||
|
|
config.Model = ModelDallE3HD
|
||
|
|
return lm.MakeImage(prompt, config)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (lm *LLM) MakeImage(prompt string, config llm.GCConfig) ([]string, llm.Usage, error) {
|
||
|
|
openaiConf := openai.DefaultConfig(lm.config.ApiKey)
|
||
|
|
if lm.config.Endpoint != "" {
|
||
|
|
openaiConf.BaseURL = lm.config.Endpoint
|
||
|
|
}
|
||
|
|
config.SetDefault(&lm.config.GCConfig)
|
||
|
|
c := openai.NewClientWithConfig(openaiConf)
|
||
|
|
style := openai.CreateImageStyleVivid
|
||
|
|
if (!strings.Contains(prompt, "vivid") || !strings.Contains(prompt, "生动的")) && (strings.Contains(prompt, "natural") || strings.Contains(prompt, "自然的")) {
|
||
|
|
style = openai.CreateImageStyleNatural
|
||
|
|
}
|
||
|
|
quality := openai.CreateImageQualityStandard
|
||
|
|
model := config.GetModel()
|
||
|
|
if strings.HasSuffix(model, "-hd") {
|
||
|
|
quality = openai.CreateImageQualityHD
|
||
|
|
model = model[0 : len(model)-3]
|
||
|
|
}
|
||
|
|
t1 := time.Now().UnixMilli()
|
||
|
|
r, err := c.CreateImage(context.Background(), openai.ImageRequest{
|
||
|
|
Prompt: prompt,
|
||
|
|
Model: model,
|
||
|
|
Quality: quality,
|
||
|
|
Size: config.GetSize(),
|
||
|
|
Style: style,
|
||
|
|
ResponseFormat: openai.CreateImageResponseFormatURL,
|
||
|
|
})
|
||
|
|
t2 := time.Now().UnixMilli() - t1
|
||
|
|
if err == nil {
|
||
|
|
results := make([]string, 0)
|
||
|
|
for _, item := range r.Data {
|
||
|
|
results = append(results, item.URL)
|
||
|
|
}
|
||
|
|
return results, llm.Usage{
|
||
|
|
AskTokens: 0,
|
||
|
|
AnswerTokens: 0,
|
||
|
|
TotalTokens: 0,
|
||
|
|
UsedTime: t2,
|
||
|
|
}, nil
|
||
|
|
} else {
|
||
|
|
return nil, llm.Usage{}, err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (lm *LLM) FastMakeVideo(prompt string, config llm.GCConfig) ([]string, []string, llm.Usage, error) {
|
||
|
|
return lm.MakeVideo(prompt, config)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (lm *LLM) BestMakeVideo(prompt string, config llm.GCConfig) ([]string, []string, llm.Usage, error) {
|
||
|
|
return lm.MakeVideo(prompt, config)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (lm *LLM) MakeVideo(prompt string, config llm.GCConfig) ([]string, []string, llm.Usage, error) {
|
||
|
|
return nil, nil, llm.Usage{}, nil
|
||
|
|
}
|