diff --git a/config.go b/config.go index 8f30389..9dfcbdd 100644 --- a/config.go +++ b/config.go @@ -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, diff --git a/gc.go b/gc.go index 8a3b920..88947ed 100644 --- a/gc.go +++ b/gc.go @@ -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()