From d0ad0edc06844fe2d95b4d698d45f5d4be639531 Mon Sep 17 00:00:00 2001
From: junleea <354425203@qq.com>
Date: Mon, 7 Apr 2025 16:37:27 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0windows=E7=B3=BB=E7=BB=9F?=
 =?UTF-8?q?=E6=94=AF=E6=8C=81=EF=BC=8C=E4=BF=AE=E6=94=B9=E9=83=A8=E5=88=86?=
 =?UTF-8?q?=E9=85=8D=E7=BD=AE=E5=8F=8Aspark=E9=83=A8=E5=88=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.go                      | 13 ++++++++++---
 service/spark.go             |  2 +-
 service/spark/spark-image.go |  3 +--
 3 files changed, 12 insertions(+), 6 deletions(-)

diff --git a/main.go b/main.go
index 0f3ec43..3612b2d 100644
--- a/main.go
+++ b/main.go
@@ -58,9 +58,16 @@ func init() {
 	os.MkdirAll(proto.CID_BASE_DIR, os.ModePerm)
 	os.MkdirAll(proto.CID_BASE_DIR+"script", os.ModePerm)
 	os.MkdirAll(proto.CID_BASE_DIR+"workspace", os.ModePerm)
-	//读取配置文件
-	//文件地址/home/saw-ai/saw-ai.conf
-	configPath := "/home/saw/saw-ai-go/saw-ai.conf"
+	//系统是linux、macos还是windows
+	var configPath string
+	if os.Getenv("OS") == "Windows_NT" {
+		configPath = "E:/Code/saw-ai/saw-ai.conf"
+	} else if os.Getenv("OS") == "linux" {
+		//文件地址/home/saw-ai/saw-ai.conf
+		configPath = "/home/saw/saw-ai-go/saw-ai.conf"
+	} else {
+		configPath = "/home/saw/saw-ai-go/saw-ai.conf"
+	}
 	//读取配置文件
 	err := proto.ReadConfig(configPath)
 	if err != nil {
diff --git a/service/spark.go b/service/spark.go
index fa5fd5c..c4e3f4f 100644
--- a/service/spark.go
+++ b/service/spark.go
@@ -306,7 +306,7 @@ func genSparkParams(question, appid string, domain string, sessionID int, system
 			Chat: ChatParameter{
 				Domain:      domain,
 				Temperature: 0.8,
-				MaxTokens:   2048,
+				MaxTokens:   4096,
 			},
 		},
 		Payload: SparkRequestPayload{
diff --git a/service/spark/spark-image.go b/service/spark/spark-image.go
index 0883be2..16aa1a6 100644
--- a/service/spark/spark-image.go
+++ b/service/spark/spark-image.go
@@ -26,9 +26,8 @@ var (
 func sparkImage() {
 
 	d := websocket.Dialer{
-		HandshakeTimeout: 5 * time.Second,
+		HandshakeTimeout: 20 * time.Second,
 	}
-	//鎻℃墜骞跺缓绔媤ebsocket 杩炴帴
 	conn, resp, err := d.Dial(assembleAuthUrl(hostUrl, apiKey, apiSecret), nil)
 	if err != nil {
 		panic(readResp(resp) + err.Error())