添加制作ppt测试数据,不进行数据请求,只读当前数据
This commit is contained in:
parent
b085fd37ba
commit
7620d9e9b0
|
|
@ -244,11 +244,13 @@ func SparkDoCreateOutline(userReq *proto.SparkCreateOutlineRequest, base *proto.
|
||||||
|
|
||||||
var outline proto.SparkCreateOutlineResponse
|
var outline proto.SparkCreateOutlineResponse
|
||||||
//获取大纲
|
//获取大纲
|
||||||
if userReq.FileUrl == "" {
|
//测试阶段
|
||||||
outline, err = SparkCreateOutline(&modelParam, userReq) //根据query生成大纲
|
outline = testPPTOutline()
|
||||||
} else {
|
//if userReq.FileUrl == "" {
|
||||||
outline, err = SparkCreateOutlineByDoc(&modelParam, userReq) //根据文档生成大纲
|
// outline, err = SparkCreateOutline(&modelParam, userReq) //根据query生成大纲
|
||||||
}
|
//} else {
|
||||||
|
// outline, err = SparkCreateOutlineByDoc(&modelParam, userReq) //根据文档生成大纲
|
||||||
|
//}
|
||||||
//进行保存
|
//进行保存
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Println("Spark create outline Error:", err)
|
log.Println("Spark create outline Error:", err)
|
||||||
|
|
@ -299,18 +301,21 @@ func SparkDoCreatePPTByOutline(userReq *proto.SparkCreatePPTByOutlineUserRequest
|
||||||
log.Println("Create message error:", err3, userMsgID)
|
log.Println("Create message error:", err3, userMsgID)
|
||||||
return proto.SparkCreatePPTResponse{}, err3
|
return proto.SparkCreatePPTResponse{}, err3
|
||||||
}
|
}
|
||||||
|
//测试阶段
|
||||||
|
resp := testPPTInfo()
|
||||||
|
|
||||||
resp, err4 := SparkCreatePPTByOutline(&modelParam, userReq) //根据大纲生成ppt
|
//resp, err4 := SparkCreatePPTByOutline(&modelParam, userReq) //根据大纲生成ppt
|
||||||
if err4 != nil {
|
//
|
||||||
log.Println("Spark create ppt by outline Error:", err4)
|
//if err4 != nil {
|
||||||
return proto.SparkCreatePPTResponse{}, err4
|
// 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) //保存消息
|
service.CreateAIStreamMsg(base.UserID, base.ModelID, int(base.SessionID), resp, modelParam.Model) //保存消息
|
||||||
//新协程获取ppt制作进度
|
//新协程获取ppt制作进度
|
||||||
userReq.Sid = resp.Data.SID
|
userReq.Sid = resp.Data.SID
|
||||||
go SparkDoGetPPTInfoBySID(&modelParam, userReq, base)
|
go SparkDoGetPPTInfoBySID(&modelParam, userReq, base)
|
||||||
return resp, err4
|
return resp, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// 每3秒查询一次ppt制作情况
|
// 每3秒查询一次ppt制作情况
|
||||||
|
|
@ -402,3 +407,53 @@ func DownloadFile(url, filePath string) error {
|
||||||
_, err = io.Copy(out, resp.Body)
|
_, err = io.Copy(out, resp.Body)
|
||||||
return err
|
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
|
||||||
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue