312 lines
8.2 KiB
Go
312 lines
8.2 KiB
Go
package moonshot
|
||
|
||
import (
|
||
"bytes"
|
||
"encoding/json"
|
||
"fmt"
|
||
"github.com/PuerkitoBio/goquery"
|
||
"github.com/tidwall/gjson"
|
||
"io"
|
||
"log"
|
||
"mime/multipart"
|
||
"net/http"
|
||
"os"
|
||
"strings"
|
||
"week/tools"
|
||
"week/udesk"
|
||
)
|
||
|
||
//请求月之暗面所需要的常量
|
||
|
||
const (
|
||
MoonshotFileApiUrl = "https://api.moonshot.cn/v1/files/"
|
||
moonshotApiKey = "sk-DwKNLIVfGFi5CUocISixedd2kLcNzYVRFShbScKtbxMNOPkQ"
|
||
)
|
||
|
||
// 上传文件
|
||
|
||
func ReqMoonShotFileSendApi() *string {
|
||
filePath := "./周报.xlsx"
|
||
// 打开文件
|
||
file, err := os.Open(filePath)
|
||
if err != nil {
|
||
log.Printf("打开周报excel失败，错误是:%v", err)
|
||
}
|
||
defer file.Close()
|
||
|
||
// 创建一个缓冲区
|
||
body := &bytes.Buffer{}
|
||
writer := multipart.NewWriter(body)
|
||
|
||
// 创建一个文件表单字段
|
||
part, err := writer.CreateFormFile("file", filePath)
|
||
if err != nil {
|
||
log.Printf("写文件缓存失败，错误是%v", err)
|
||
|
||
}
|
||
|
||
// 将文件内容拷贝到表单字段
|
||
_, err = io.Copy(part, file)
|
||
if err != nil {
|
||
log.Printf("拷贝文件到缓存失败，错误是%v", err)
|
||
|
||
}
|
||
|
||
// 添加其他表单字段
|
||
writer.WriteField("purpose", "file-extract")
|
||
|
||
// 完成表单写入
|
||
writer.Close()
|
||
|
||
// 创建 HTTP 请求
|
||
req, err := http.NewRequest("POST", MoonshotFileApiUrl, body)
|
||
if err != nil {
|
||
log.Printf("请求月之暗面api过程，创建http请求失败，错误是：%s", err)
|
||
|
||
}
|
||
|
||
// 设置请求头
|
||
req.Header.Set("Content-Type", writer.FormDataContentType())
|
||
req.Header.Set("Authorization", "Bearer "+moonshotApiKey)
|
||
|
||
// 发送请求
|
||
client := &http.Client{}
|
||
resp, err := client.Do(req)
|
||
if err != nil {
|
||
log.Printf("请求月之暗面api过程，请求失败，错误是：%s", err)
|
||
|
||
}
|
||
defer resp.Body.Close()
|
||
|
||
// 读取响应
|
||
responseBody, err := io.ReadAll(resp.Body)
|
||
if err != nil {
|
||
log.Printf("请求月之暗面api过程，读取响应失败，错误是：%s", err)
|
||
|
||
}
|
||
|
||
// 打印响应内容
|
||
fileId := gjson.Get(string(responseBody), "id").String()
|
||
log.Printf("文件上传成功，你上传的文件id是：%s", fileId)
|
||
return &fileId
|
||
}
|
||
|
||
// 删除文件
|
||
|
||
func ReqMoonShotFileDeleteApi(fileID *string) {
|
||
// 设置 API endpoint
|
||
url := fmt.Sprintf(MoonshotFileApiUrl + *fileID)
|
||
|
||
// 创建 HTTP 请求
|
||
req, err := http.NewRequest("DELETE", url, nil)
|
||
if err != nil {
|
||
log.Printf("调用月之暗面删除文件api过程，创建http请求失败，错误是%v", err)
|
||
return
|
||
}
|
||
|
||
// 设置请求头（如果需要的话）
|
||
req.Header.Set("Authorization", "Bearer "+moonshotApiKey)
|
||
|
||
// 发送请求
|
||
client := &http.Client{}
|
||
resp, err := client.Do(req)
|
||
if err != nil {
|
||
log.Printf("调用月之暗面删除文件api过程，发送http请求失败，错误是%v", err)
|
||
return
|
||
}
|
||
defer resp.Body.Close()
|
||
|
||
// 检查响应状态码
|
||
if resp.StatusCode == http.StatusOK {
|
||
log.Printf("请求月之暗面删除文件api成功~")
|
||
} else {
|
||
log.Println("请求月之暗面删除文件api失败>_<, http状态码是:", resp.StatusCode)
|
||
}
|
||
}
|
||
|
||
// 开启ai对话主函数
|
||
|
||
func AiChat() (string, error) {
|
||
fileId := ReqMoonShotFileSendApi()
|
||
contentStr := GetFileContent(fileId)
|
||
// 设置 API endpoint
|
||
url := "https://api.moonshot.cn/v1/chat/completions"
|
||
|
||
// 构造请求体
|
||
requestBody := map[string]interface{}{
|
||
"model": "moonshot-v1-128k",
|
||
"messages": []map[string]string{
|
||
{"role": "system", "content": contentStr},
|
||
{"role": "user", "content": "帮忙分析下工单中的错别字，告诉我对应的工单id"},
|
||
},
|
||
"temperature": 0.2,
|
||
}
|
||
|
||
// 将请求体转换为 JSON 格式
|
||
jsonBody, err := json.Marshal(requestBody)
|
||
if err != nil {
|
||
log.Printf("开启会话过程，将请求体转换为json格式失败，错误是：%v", err)
|
||
return "", err
|
||
}
|
||
|
||
// 创建 HTTP 请求
|
||
req, err := http.NewRequest("POST", url, bytes.NewBuffer(jsonBody))
|
||
if err != nil {
|
||
log.Printf("开启会话过程，创建请求失败，错误是：%v", err)
|
||
return "", err
|
||
|
||
}
|
||
|
||
// 设置请求头
|
||
req.Header.Set("Content-Type", "application/json")
|
||
req.Header.Set("Authorization", "Bearer "+moonshotApiKey)
|
||
|
||
// 发送请求
|
||
client := &http.Client{}
|
||
resp, err := client.Do(req)
|
||
if err != nil {
|
||
log.Printf("开启会话过程，发送请求失败，错误是：%v:", err)
|
||
return "", err
|
||
|
||
}
|
||
defer resp.Body.Close()
|
||
|
||
// 读取响应
|
||
responseBody := new(bytes.Buffer)
|
||
_, err = responseBody.ReadFrom(resp.Body)
|
||
if err != nil {
|
||
log.Printf("开启会话过程，读取响应失败，错误是：%v:", err)
|
||
return "", err
|
||
}
|
||
|
||
defer ReqMoonShotFileDeleteApi(fileId) //对话后清空文件
|
||
// 打印响应内容
|
||
jsonData := responseBody.String()
|
||
repFromAi := gjson.Get(jsonData, "choices.#.message.content").String()
|
||
repFromAi = strings.Replace(repFromAi, "\\n", "\n", -1)
|
||
repFromAi = strings.Replace(repFromAi, "\\", "", -1)
|
||
repFromAi = strings.Replace(repFromAi, "[", "", -1)
|
||
repFromAi = strings.Replace(repFromAi, "]", "", -1)
|
||
repFromAi = strings.Replace(repFromAi, "\"", "", 1)
|
||
log.Println(repFromAi)
|
||
return repFromAi, nil
|
||
}
|
||
|
||
// 获取文件内容
|
||
|
||
func GetFileContent(fileID *string) string {
|
||
|
||
// 设置 API endpoint
|
||
url := fmt.Sprintf(MoonshotFileApiUrl + *fileID + "/content")
|
||
|
||
// 创建 HTTP 请求
|
||
req, err := http.NewRequest("GET", url, nil)
|
||
if err != nil {
|
||
log.Printf("月之暗面api，获取文件内容过程，创建http请求，错误是：%v:", err)
|
||
|
||
}
|
||
|
||
// 设置请求头
|
||
req.Header.Set("Authorization", "Bearer "+moonshotApiKey)
|
||
|
||
// 发送请求
|
||
client := &http.Client{}
|
||
resp, err := client.Do(req)
|
||
if err != nil {
|
||
log.Printf("月之暗面api，获取文件内容过程，发送请求，错误是：%v", err)
|
||
|
||
}
|
||
defer resp.Body.Close()
|
||
|
||
// 读取响应内容
|
||
content, err := io.ReadAll(resp.Body)
|
||
if err != nil {
|
||
log.Printf("月之暗面api，获取文件内容过程，读取响应失败，错误是：%v", err)
|
||
|
||
}
|
||
contentStr := string(content)
|
||
// 打印文件内容
|
||
fmt.Println(contentStr)
|
||
return contentStr
|
||
}
|
||
|
||
// 根据工单id循环获取工单回复内容
|
||
|
||
func CreateExcel() error {
|
||
totalPages, err := udesk.GetUdeskTicketId()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
id, err := udesk.LoopGetTicketId(totalPages)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
contentMap := make(map[string][]string)
|
||
j := 1
|
||
for _, v := range id {
|
||
|
||
idUrl := fmt.Sprintf("https://servicecenter-alauda.udesk.cn/open_api_v1/tickets/%v/replies?", v)
|
||
repliesUrl := udesk.Geturlstring(idUrl)
|
||
resp, err := http.Get(repliesUrl)
|
||
if err != nil {
|
||
// 处理错误
|
||
log.Printf("获取工单过滤器下的回复信息失败，发送请求过程，错误是：%v\n", err)
|
||
return err
|
||
}
|
||
defer func(Body io.ReadCloser) {
|
||
err := Body.Close()
|
||
if err != nil {
|
||
|
||
}
|
||
}(resp.Body)
|
||
body, err := io.ReadAll(resp.Body)
|
||
if err != nil {
|
||
// 处理错误
|
||
log.Printf("获取工单过滤器下的回复信息失败，读取返回体过程，错误是：%v\n", err)
|
||
return err
|
||
}
|
||
jsonData := string(body)
|
||
repliesContent := gjson.Get(jsonData, "replies.#.content").String()
|
||
repliesContent = strings.Trim(repliesContent, "[]")
|
||
//repliesContent = trimHtml(repliesContent)
|
||
//查找回复内容和配图
|
||
doc, err := goquery.NewDocumentFromReader(strings.NewReader(repliesContent))
|
||
var s = make([]string, 0)
|
||
doc.Find("p,img").Each(func(i int, selection *goquery.Selection) {
|
||
imgSrc, exists := selection.Attr("src")
|
||
if exists {
|
||
s = append(s, imgSrc)
|
||
s = append(s, "\n")
|
||
}
|
||
s = append(s, selection.Text())
|
||
s = append(s, "\n")
|
||
})
|
||
|
||
//跳过没有回复内容的工单,即运维服务,减少给模型的错误条件
|
||
if len(s) == 0 {
|
||
continue
|
||
}
|
||
|
||
repliesContentSlice := tools.ReverseSlice(s)
|
||
|
||
cloudId := udesk.Id2CloudId(v)
|
||
log.Printf("正在处理map中的第%v个数据,键是%v\n", j, v)
|
||
if cloudId == "" {
|
||
contentMap[v] = repliesContentSlice
|
||
} else {
|
||
contentMap[cloudId] = repliesContentSlice
|
||
}
|
||
log.Printf("第%v个数据数据处理完成\n", j)
|
||
j++
|
||
}
|
||
log.Printf("所有回复数据处理完成,map中键的数量：%v\n", len(contentMap))
|
||
//fmt.Printf("map的值:%v\n", contentMap) //回复内容的map，待办：将这个map中的建存储第一列，将这个map中的值作为第二列存储
|
||
// 生成excel表格内容为id --> 回复内容
|
||
// 保存到某地
|
||
tools.CreateNewExcel(contentMap) //创建了excel
|
||
//sk-DwKNLIVfGFi5CUocISixedd2kLcNzYVRFShbScKtbxMNOPkQ
|
||
//卡点，如何根据文件内容获取回复，目前这个结构只能开启单轮对话
|
||
return nil
|
||
|
||
}
|