33 lines
823 B
Go
33 lines
823 B
Go
|
package nlp
|
||
|
|
|
||
|
|
import (
|
||
|
|
"catface/app/global/variable"
|
||
|
|
"catface/app/service/nlp/glm"
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
func GenerateTitle(content string) string {
|
||
|
|
message := variable.PromptsYml.GetString("Prompt.Title") + content
|
||
|
|
title, _ := glm.Chat(message)
|
||
|
|
return title
|
||
|
|
}
|
||
|
|
|
||
|
|
// ChatKnoledgeRAG 使用 RAG 模型进行知识问答
|
||
|
|
func ChatKnoledgeRAG(doc, query string) (string, error) {
|
||
|
|
// 读取配置文件中的 KnoledgeRAG 模板
|
||
|
|
promptTemplate := variable.PromptsYml.GetString("Prompt.KnoledgeRAG")
|
||
|
|
|
||
|
|
// 替换模板中的占位符
|
||
|
|
message := strings.Replace(promptTemplate, "{question}", query, -1)
|
||
|
|
message = strings.Replace(message, "{context}", doc, -1)
|
||
|
|
|
||
|
|
// 调用聊天接口
|
||
|
|
response, err := glm.Chat(message)
|
||
|
|
if err != nil {
|
||
|
|
return "", fmt.Errorf("调用聊天接口失败: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return response, nil
|
||
|
|
}
|