From b0d0fe28b1d91fdebb5169acc934937b6b692bda Mon Sep 17 00:00:00 2001
From: Miao Zhuang <1060950782@163.com>
Date: Tue, 5 Dec 2023 14:27:19 +0800
Subject: [PATCH] fix: type alias (#695)

---
 server/api/studio/pkg/llm/schedule.go           | 3 +++
 server/api/studio/pkg/llm/transformer/openai.go | 1 +
 2 files changed, 4 insertions(+)

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