ai/zhipu
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

little fix

Browse Source
This commit is contained in:
Star 2024-10-29 12:33:18 +08:00
parent 3e64ec275a
commit dba6f196df
2 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config.go
View File

@ -44,7 +44,7 @@ func (lm *LLM) Support() llm.Support {
return llm.Support{
Ask: true,
AskWithImage: true,
AskWithVideo: true,
AskWithVideo: false,
AskWithCodeInterpreter: true,
AskWithWebSearch: true,
MakeImage: true,

9
gc.go
View File

@ -1,11 +1,12 @@
package zhipu
import (
"apigo.cc/ai/llm/llm"
"context"
"errors"
"github.com/yankeguo/zhipu"
"time"
"apigo.cc/ai/llm/llm"
"github.com/yankeguo/zhipu"
)
func (lm *LLM) FastMakeImage(prompt string, config llm.GCConfig) ([]string, llm.Usage, error) {
@ -25,7 +26,7 @@ func (lm *LLM) MakeImage(prompt string, config llm.GCConfig) ([]string, llm.Usag
}
config.SetDefault(&lm.config.GCConfig)
cc := c.ImageGeneration(config.Model).SetPrompt(prompt)
cc := c.ImageGeneration(config.GetModel()).SetPrompt(prompt)
//cc.SetSize(config.GetSize())
t1 := time.Now().UnixMilli()
@ -60,7 +61,7 @@ func (lm *LLM) MakeVideo(prompt string, config llm.GCConfig) ([]string, []string
}
config.SetDefault(&lm.config.GCConfig)
cc := c.VideoGeneration(config.Model).SetPrompt(prompt)
cc := c.VideoGeneration(config.GetModel()).SetPrompt(prompt)
cc.SetImageURL(config.GetRef())
t1 := time.Now().UnixMilli()