fix: gen code generator; some unit tests

test/chatgpt_test.go

@@ -6,46 +6,50 @@ import (
 	"github.com/0xJacky/Nginx-UI/settings"
 	"github.com/pkg/errors"
 	"github.com/sashabaranov/go-openai"
+	"github.com/uozi-tech/cosy/sandbox"
 	"io"
 	"os"
 	"testing"
 )
 
 func TestChatGPT(t *testing.T) {
-	settings.Init("../../app.ini")
-	c := openai.NewClient(settings.OpenAISettings.Token)
+	sandbox.NewInstance("../../app.ini", "sqlite").
+		Run(func(instance *sandbox.Instance) {
+			c := openai.NewClient(settings.OpenAISettings.Token)
 
-	ctx := context.Background()
+			ctx := context.Background()
 
-	req := openai.ChatCompletionRequest{
-		Model: openai.GPT3Dot5Turbo0301,
-		Messages: []openai.ChatCompletionMessage{
-			{
-				Role:    openai.ChatMessageRoleUser,
-				Content: "帮我写一个 nginx 配置文件的示例",
-			},
-		},
-		Stream: true,
-	}
-	stream, err := c.CreateChatCompletionStream(ctx, req)
-	if err != nil {
-		fmt.Printf("CompletionStream error: %v\n", err)
-		return
-	}
-	defer stream.Close()
+			req := openai.ChatCompletionRequest{
+				Model: openai.GPT3Dot5Turbo0301,
+				Messages: []openai.ChatCompletionMessage{
+					{
+						Role:    openai.ChatMessageRoleUser,
+						Content: "帮我写一个 nginx 配置文件的示例",
+					},
+				},
+				Stream: true,
+			}
+			stream, err := c.CreateChatCompletionStream(ctx, req)
+			if err != nil {
+				fmt.Printf("CompletionStream error: %v\n", err)
+				return
+			}
+			defer stream.Close()
 
-	for {
-		response, err := stream.Recv()
-		if errors.Is(err, io.EOF) {
-			return
-		}
+			for {
+				response, err := stream.Recv()
+				if errors.Is(err, io.EOF) {
+					return
+				}
 
-		if err != nil {
-			fmt.Printf("Stream error: %v\n", err)
-			return
-		}
+				if err != nil {
+					fmt.Printf("Stream error: %v\n", err)
+					return
+				}
+
+				fmt.Printf("%v", response.Choices[0].Delta.Content)
+				_ = os.Stdout.Sync()
+			}
+		})
 
-		fmt.Printf("%v", response.Choices[0].Delta.Content)
-		_ = os.Stdout.Sync()
-	}
 }