mirror of
https://github.com/eryajf/chatgpt-dingtalk.git
synced 2025-10-06 00:26:51 +08:00
perf: 当使用prompt但内容为空时,直接返回prompt的内容 (#138)
This commit is contained in:
69
main.go
69
main.go
@@ -19,18 +19,50 @@ func main() {
|
|||||||
Start()
|
Start()
|
||||||
}
|
}
|
||||||
|
|
||||||
var Welcome string = `Commands:
|
var Welcome string = `# 发送信息
|
||||||
=================================
|
|
||||||
🙋 单聊 👉 单独聊天
|
若您想给机器人发送信息,请选择:
|
||||||
📣 串聊 👉 带上下文聊天
|
|
||||||
🔃 重置 👉 重置带上下文聊天
|
1. 在本机器人所在群里@机器人;
|
||||||
💵 余额 👉 查询剩余额度
|
2. 点击机器人的头像后,再点击"发消息"。
|
||||||
🚀 帮助 👉 显示帮助信息
|
|
||||||
🌈 模板 👉 内置的prompt
|
机器人收到您的信息后,默认会交给chatgpt进行处理。除非,您发送的内容是7个**系统指令**之一。
|
||||||
🎨 图片 👉 根据prompt生成图片
|
|
||||||
=================================
|
-----
|
||||||
🚜 例:@我发送 空 或 帮助 将返回此帮助信息
|
|
||||||
💪 Power By https://github.com/eryajf/chatgpt-dingtalk
|
# 系统指令
|
||||||
|
|
||||||
|
系统指令是一些特殊的词语,当您向机器人发送这些词语时,会触发对应的功能:
|
||||||
|
|
||||||
|
**单聊**:每条消息都是单独的对话,不包含上下文
|
||||||
|
|
||||||
|
**串聊**:对话会携带上下文,除非您主动重置对话或对话长度超过限制
|
||||||
|
|
||||||
|
**重置**:重置上下文
|
||||||
|
|
||||||
|
**余额**:查询机器人所用OpenAI账号的余额
|
||||||
|
|
||||||
|
**模板**:查询机器人内置的快捷模板
|
||||||
|
|
||||||
|
**图片**:查看如何根据提示词生成图片
|
||||||
|
|
||||||
|
**帮助**:重新获取帮助信息
|
||||||
|
|
||||||
|
-----
|
||||||
|
|
||||||
|
# 友情提示
|
||||||
|
|
||||||
|
使用"串聊模式"会显著加快机器人所用账号的余额消耗速度。
|
||||||
|
|
||||||
|
因此,若无保留上下文的需求,建议使用"单聊模式"。
|
||||||
|
|
||||||
|
即使有保留上下文的需求,也应适时使用"重置"指令来重置上下文。
|
||||||
|
|
||||||
|
-----
|
||||||
|
|
||||||
|
# 项目地址
|
||||||
|
|
||||||
|
本项目已在GitHub开源,[查看源代码](https://github.com/eryajf/chatgpt-dingtalk)。
|
||||||
`
|
`
|
||||||
|
|
||||||
func Start() {
|
func Start() {
|
||||||
@@ -51,7 +83,7 @@ func Start() {
|
|||||||
// TODO: 校验请求
|
// TODO: 校验请求
|
||||||
if len(msgObj.Text.Content) == 1 || strings.TrimSpace(msgObj.Text.Content) == "帮助" {
|
if len(msgObj.Text.Content) == 1 || strings.TrimSpace(msgObj.Text.Content) == "帮助" {
|
||||||
// 欢迎信息
|
// 欢迎信息
|
||||||
_, err := msgObj.ReplyToDingtalk(string(dingbot.TEXT), Welcome)
|
_, err := msgObj.ReplyToDingtalk(string(dingbot.MARKDOWN), Welcome)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Warning(fmt.Errorf("send message error: %v", err))
|
logger.Warning(fmt.Errorf("send message error: %v", err))
|
||||||
return ship.ErrBadRequest.New(fmt.Errorf("send message error: %v", err))
|
return ship.ErrBadRequest.New(fmt.Errorf("send message error: %v", err))
|
||||||
@@ -62,7 +94,16 @@ func Start() {
|
|||||||
case strings.HasPrefix(strings.TrimSpace(msgObj.Text.Content), "#图片"):
|
case strings.HasPrefix(strings.TrimSpace(msgObj.Text.Content), "#图片"):
|
||||||
return process.ImageGenerate(&msgObj)
|
return process.ImageGenerate(&msgObj)
|
||||||
default:
|
default:
|
||||||
msgObj.Text.Content = process.GeneratePrompt(strings.TrimSpace(msgObj.Text.Content))
|
msgObj.Text.Content, err = process.GeneratePrompt(strings.TrimSpace(msgObj.Text.Content))
|
||||||
|
// err不为空:提示词之后没有文本 -> 直接返回提示词所代表的内容
|
||||||
|
if err != nil {
|
||||||
|
_, err = msgObj.ReplyToDingtalk(string(dingbot.TEXT), msgObj.Text.Content)
|
||||||
|
if err != nil {
|
||||||
|
logger.Warning(fmt.Errorf("send message error: %v", err))
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
logger.Info(fmt.Sprintf("after generate prompt: %#v", msgObj.Text.Content))
|
logger.Info(fmt.Sprintf("after generate prompt: %#v", msgObj.Text.Content))
|
||||||
return process.ProcessRequest(&msgObj)
|
return process.ProcessRequest(&msgObj)
|
||||||
}
|
}
|
||||||
|
@@ -40,7 +40,7 @@ func ProcessRequest(rmsg *dingbot.ReceiveMsg) error {
|
|||||||
for _, v := range *public.Prompt {
|
for _, v := range *public.Prompt {
|
||||||
title = title + v.Title + " | "
|
title = title + v.Title + " | "
|
||||||
}
|
}
|
||||||
_, err := rmsg.ReplyToDingtalk(string(dingbot.TEXT), fmt.Sprintf("%s 您好,当前程序内置集成了这些prompt:\n====================================\n| %s \n====================================\n你可以选择某个prompt开头,然后进行对话。\n以周报为例,可发送 #周报 我本周用Go写了一个钉钉集成ChatGPT的聊天应用", rmsg.SenderNick, title))
|
_, err := rmsg.ReplyToDingtalk(string(dingbot.MARKDOWN), fmt.Sprintf("%s 您好,当前程序内置集成了这些提示词:\n\n-----\n\n| %s \n\n-----\n\n您可以选择某个提示词作为对话内容的开头。\n\n以周报为例,可发送\"#周报 我本周用Go写了一个钉钉集成ChatGPT的聊天应用\",可将工作内容填充为一篇完整的周报。\n\n-----\n\n若您不清楚某个提示词的所代表的含义,您可以直接发送提示词,例如直接发送\"#周报\"", rmsg.SenderNick, title))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Warning(fmt.Errorf("send message error: %v", err))
|
logger.Warning(fmt.Errorf("send message error: %v", err))
|
||||||
}
|
}
|
||||||
|
@@ -1,16 +1,23 @@
|
|||||||
package process
|
package process
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/eryajf/chatgpt-dingtalk/public"
|
"github.com/eryajf/chatgpt-dingtalk/public"
|
||||||
)
|
)
|
||||||
|
|
||||||
// GeneratePrompt 生成当次请求的 Prompt
|
// GeneratePrompt 生成当次请求的 Prompt
|
||||||
func GeneratePrompt(msg string) (rst string) {
|
func GeneratePrompt(msg string) (rst string, err error) {
|
||||||
for _, prompt := range *public.Prompt {
|
for _, prompt := range *public.Prompt {
|
||||||
if strings.HasPrefix(msg, prompt.Title) {
|
if strings.HasPrefix(msg, prompt.Title) {
|
||||||
|
if strings.TrimSpace(msg) == prompt.Title {
|
||||||
|
rst = fmt.Sprintf("%s:\n%s", prompt.Title, prompt.Content)
|
||||||
|
err = errors.New("消息内容为空") // 当提示词之后没有文本,抛出异常,以便直接返回Prompt所代表的内容
|
||||||
|
} else {
|
||||||
rst = prompt.Content + strings.Replace(msg, prompt.Title, "", -1)
|
rst = prompt.Content + strings.Replace(msg, prompt.Title, "", -1)
|
||||||
|
}
|
||||||
return
|
return
|
||||||
} else {
|
} else {
|
||||||
rst = msg
|
rst = msg
|
||||||
|
Reference in New Issue
Block a user