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())