fix: type alias (#695)

vesoft-inc · Dec 5, 2023 · b0d0fe2 · b0d0fe2
1 parent fa73dd3
commit b0d0fe2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git 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
@@ -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