kkyy
/
tgbot_misaka_5882f7
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Compare commits

base: kkyy:7e9d4e5052ca5917c20e3b29064b254d1575edb3
Branches Tags
kkyy:main
..
compare: kkyy:dcb251d2ad07d317d13d1201997df7a376e6b481
Branches Tags
kkyy:main

No commits in common. "7e9d4e5052ca5917c20e3b29064b254d1575edb3" and "dcb251d2ad07d317d13d1201997df7a376e6b481" have entirely different histories.
7e9d4e5052 ... dcb251d2ad

1 changed files with 7 additions and 25 deletions
Show Stats Expand all files Collapse all files

30
assistant.go
View File

@ -31,12 +31,6 @@ func matchAssistantConversation(botUsr *tele.User, msg *tele.Message) []*tele.Me
}
for _, msg := range thread {
for _, ent := range msg.Entities {
if ent.Type == tele.EntityCommand {
return nil
}
}
from := msg.Sender
chat := msg.Chat
if from == nil || chat == nil {
@ -90,23 +84,13 @@ func matchAssistantConversation(botUsr *tele.User, msg *tele.Message) []*tele.Me
type assistantStreamedResponseCb func(text string, finished bool) (*tele.Message, error)
func assistantStreamedResponse(request openai.ChatRequest, cb assistantStreamedResponseCb) error {
logger.Debugw("Openai chat request", "req", request)
logger.Debugf("Openai chat request: %#+v", request)
ai := openai.NewClient(config.OpenAIApiKey)
var (
resp *openai.ChatResponseStream
err error
)
tries := 2
for {
if tries--; tries < 0 {
resp, err := ai.ChatCompletionStream(request)
if err != nil {
return err
}
if resp, err = ai.ChatCompletionStream(request); err == nil {
break
}
logger.Warnw("assistant: failed to get response", "error", err)
}
nErrs := 0
go func() {
@ -204,7 +188,7 @@ func handleAssistantConversation(c tele.Context, thread []*tele.Message) error {
role := openai.ChatRoleUser
from := thread[l].Sender
if from != nil && from.ID == me.ID {
role = openai.ChatRoleAssistant
role = openai.ChatRoleSystem
}
convMsgs = append(convMsgs, openai.ChatMessage{
@ -237,12 +221,11 @@ func handleAssistantConversation(c tele.Context, thread []*tele.Message) error {
var err error
if replyMsg == nil {
<-typingNotifyCh
replyMsg, err = c.Bot().Reply(lastMsg, text, tele.Silent)
replyMsg, err = c.Bot().Reply(c.Message(), text, tele.Silent)
} else {
replyMsg, err = c.Bot().Edit(replyMsg, text)
}
if finished && err == nil {
replyMsg.ReplyTo = lastMsg // nasty bug
if err := cacheMessage(replyMsg); err != nil {
logger.Warnw("failed to cache message", "error", err)
}
@ -252,9 +235,8 @@ func handleAssistantConversation(c tele.Context, thread []*tele.Message) error {
if reqErr != nil {
logger.Errorw("assistant: failed to complete conversation", "error", reqErr)
return c.Reply("Sorry, there's a technical issue. 😵💫 Please try again later.", tele.Silent)
}
return nil
return reqErr
}
func assistantRemoveMention(msg, name string) string {