diff --git a/server/api/studio/pkg/llm/schedule.go b/server/api/studio/pkg/llm/schedule.go
index b9a4c497..bc362623 100644
--- a/server/api/studio/pkg/llm/schedule.go
+++ b/server/api/studio/pkg/llm/schedule.go
@@ -6,6 +6,9 @@ import (
 	db "github.com/vesoft-inc/nebula-studio/server/api/studio/internal/model"
 )
 
+type LLMJob = db.LLMJob
+type LLMConfig = db.LLMConfig
+
 func InitSchedule() {
 	// TODO
 	for {
diff --git a/server/api/studio/pkg/llm/transformer/openai.go b/server/api/studio/pkg/llm/transformer/openai.go
index a03be1ed..af03866b 100644
--- a/server/api/studio/pkg/llm/transformer/openai.go
+++ b/server/api/studio/pkg/llm/transformer/openai.go
@@ -57,6 +57,7 @@ func (o *OpenAI) HandleResponse(resp *http.Response, callback func(str string))
 		if err := scanner.Err(); err != nil {
 			fmt.Println("reading standard input:", err)
 		}
+		callback("[DONE]")
 		return nil, nil
 	}
 	// Read the response data