ljydlv
/
saw-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

添加制作ppt测试数据，不进行数据请求，只读当前数据

This commit is contained in:
junleea 2025-04-04 17:39:02 +08:00
parent b085fd37ba
commit 7620d9e9b0
1 changed files with 66 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

77
service/spark/spark-ppt.go
View File

@ -244,11 +244,13 @@ func SparkDoCreateOutline(userReq *proto.SparkCreateOutlineRequest, base *proto.
var outline proto.SparkCreateOutlineResponse
//获取大纲
if userReq.FileUrl == "" {
outline, err = SparkCreateOutline(&modelParam, userReq) //根据query生成大纲
} else {
outline, err = SparkCreateOutlineByDoc(&modelParam, userReq) //根据文档生成大纲
}
//测试阶段
outline = testPPTOutline()
//if userReq.FileUrl == "" {
// outline, err = SparkCreateOutline(&modelParam, userReq) //根据query生成大纲
//} else {
// outline, err = SparkCreateOutlineByDoc(&modelParam, userReq) //根据文档生成大纲
//}
//进行保存
if err != nil {
log.Println("Spark create outline Error:", err)
@ -299,18 +301,21 @@ func SparkDoCreatePPTByOutline(userReq *proto.SparkCreatePPTByOutlineUserRequest
log.Println("Create message error:", err3, userMsgID)
return proto.SparkCreatePPTResponse{}, err3
}
//测试阶段
resp := testPPTInfo()
resp, err4 := SparkCreatePPTByOutline(&modelParam, userReq) //根据大纲生成ppt
if err4 != nil {
log.Println("Spark create ppt by outline Error:", err4)
return proto.SparkCreatePPTResponse{}, err4
}
//resp, err4 := SparkCreatePPTByOutline(&modelParam, userReq) //根据大纲生成ppt
//
//if err4 != nil {
// log.Println("Spark create ppt by outline Error:", err4)
// return proto.SparkCreatePPTResponse{}, err4
//}
//进行保存
service.CreateAIStreamMsg(base.UserID, base.ModelID, int(base.SessionID), resp, modelParam.Model) //保存消息
//新协程获取ppt制作进度
userReq.Sid = resp.Data.SID
go SparkDoGetPPTInfoBySID(&modelParam, userReq, base)
return resp, err4
return resp, nil
}
// 每3秒查询一次ppt制作情况
@ -402,3 +407,53 @@ func DownloadFile(url, filePath string) error {
_, err = io.Copy(out, resp.Body)
return err
}
func testPPTInfo() proto.SparkCreatePPTResponse {
path := "/home/saw/saw_ppt_info.json"
var pptInfo proto.SparkCreatePPTResponse
//读取文件
file, err := os.Open(path)
if err != nil {
log.Println("Error opening file:", err)
return pptInfo
}
defer file.Close()
//读取文件内容
data, err := io.ReadAll(file)
if err != nil {
log.Println("Error reading file:", err)
return pptInfo
}
err = json.Unmarshal(data, &pptInfo)
if err != nil {
log.Println("Error unmarshalling JSON:", err)
return pptInfo
}
return pptInfo
}
func testPPTOutline() proto.SparkCreateOutlineResponse {
path := "/home/saw/saw_ppt_outline.json"
var outline proto.SparkCreateOutlineResponse
//读取文件
file, err := os.Open(path)
if err != nil {
log.Println("Error opening file:", err)
return outline
}
defer file.Close()
//读取文件内容
data, err := io.ReadAll(file)
if err != nil {
log.Println("Error reading file:", err)
return outline
}
err = json.Unmarshal(data, &outline)
if err != nil {
log.Println("Error unmarshalling JSON:", err)
return outline
}
return outline
}