From dba6f196df75fa7becea75175fe100e8e644ebad Mon Sep 17 00:00:00 2001
From: Star <>
Date: Tue, 29 Oct 2024 12:33:18 +0800
Subject: [PATCH] little fix

---
 config.go | 2 +-
 gc.go     | 9 +++++----
 2 files changed, 6 insertions(+), 5 deletions(-)

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()