mirror of
https://github.com/eryajf/chatgpt-dingtalk.git
synced 2026-04-22 23:47:15 +08:00
二丫讲梵
f7326b6797
* 将ai交互切换为go-openai * add stream * ✨ feat(stream): 优化流式响应机制，实现实时卡片更新 - 将固定1.5秒更新改为基于300ms最小间隔的实时更新策略 - 新增内容缓冲区机制，避免过于频繁的API调用 - 改进流式中断处理，保护已接收的内容不丢失 🔧 chore(llm): 优化HTTP客户端配置 - 增加连接池设置（MaxIdleConns: 100, MaxIdleConnsPerHost: 10） - 设置空闲连接超时时间为90秒 - 添加HTTP/2禁用选项注释，用于解决流式错误问题 📝 docs(stream): 更新流式更新策略文档 - 详细说明实时流式更新机制和缓冲策略 - 新增HTTP/2流式错误的故障排除指南 - 更新配置参数说明和建议范围 🐛 fix(stream): 修复流式中断时的内容丢失问题 - 在流式接收中断时，确保已接收的内容不会丢失 - 改进错误处理逻辑，区分有内容和无内容的情况 * modify ai
73 lines
2.1 KiB
Go
73 lines
2.1 KiB
Go
package process
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/eryajf/chatgpt-dingtalk/pkg/db"
|
|
"github.com/eryajf/chatgpt-dingtalk/pkg/dingbot"
|
|
"github.com/eryajf/chatgpt-dingtalk/pkg/llm"
|
|
"github.com/eryajf/chatgpt-dingtalk/pkg/logger"
|
|
"github.com/eryajf/chatgpt-dingtalk/public"
|
|
)
|
|
|
|
// ImageGenerate openai生成图片
|
|
func ImageGenerate(ctx context.Context, rmsg *dingbot.ReceiveMsg) error {
|
|
if public.Config.AzureOn {
|
|
_, err := rmsg.ReplyToDingtalk(string(dingbot.
|
|
MARKDOWN), "azure 模式下暂不支持图片创作功能")
|
|
if err != nil {
|
|
logger.Warning(fmt.Errorf("send message error: %v", err))
|
|
}
|
|
return err
|
|
}
|
|
qObj := db.Chat{
|
|
Username: rmsg.SenderNick,
|
|
Source: rmsg.GetChatTitle(),
|
|
ChatType: db.Q,
|
|
ParentContent: 0,
|
|
Content: rmsg.Text.Content,
|
|
}
|
|
qid, err := qObj.Add()
|
|
if err != nil {
|
|
logger.Error("往MySQL新增数据失败,错误信息：", err)
|
|
}
|
|
reply, err := llm.ImageQa(ctx, rmsg.Text.Content, rmsg.GetSenderIdentifier())
|
|
if err != nil {
|
|
logger.Info(fmt.Errorf("gpt request error: %v", err))
|
|
_, err = rmsg.ReplyToDingtalk(string(dingbot.TEXT), fmt.Sprintf("请求openai失败了，错误信息：%v", err))
|
|
if err != nil {
|
|
logger.Error(fmt.Errorf("send message error: %v", err))
|
|
return err
|
|
}
|
|
}
|
|
if reply == "" {
|
|
logger.Warning(fmt.Errorf("get gpt result falied: %v", err))
|
|
return nil
|
|
} else {
|
|
reply = strings.TrimSpace(reply)
|
|
reply = strings.Trim(reply, "\n")
|
|
reply = fmt.Sprintf(">点击图片可旋转或放大。\n", reply)
|
|
aObj := db.Chat{
|
|
Username: rmsg.SenderNick,
|
|
Source: rmsg.GetChatTitle(),
|
|
ChatType: db.A,
|
|
ParentContent: qid,
|
|
Content: reply,
|
|
}
|
|
_, err := aObj.Add()
|
|
if err != nil {
|
|
logger.Error("往MySQL新增数据失败,错误信息：", err)
|
|
}
|
|
logger.Info(fmt.Sprintf("🤖 %s得到的答案: %#v", rmsg.SenderNick, reply))
|
|
// 回复@我的用户
|
|
_, err = rmsg.ReplyToDingtalk(string(dingbot.MARKDOWN), reply)
|
|
if err != nil {
|
|
logger.Error(fmt.Errorf("send message error: %v", err))
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|