bugfix:can not stop typing

ivhu · ivhu · commit a0115a0137cc · 2024-11-24T20:10:10.000+08:00
diff --git a/ai/gemini/gemini.go b/ai/gemini/gemini.go
@@ -104,6 +104,7 @@ func (g *gemini) Chat(chatId string, msg string) (string, error) {
 			result := fmt.Sprint(resp.Candidates[0].Content.Parts[0])
 			if err := g.db.Add(models.NewChat(chatId, false, result)); err != nil {
 				log.Error().Err(err).Msg("failed to add chat record")
+				return "", err
 			}
 			return result, nil
 		}
diff --git a/handler/gemini_handler.go b/handler/gemini_handler.go
@@ -163,6 +163,7 @@ func handlePrivateChat(b *gotgbot.Bot, ctx *ext.Context, ai ai.AiInterface) erro
 	if err != nil {
 		log.Error().Err(err).Msg("gemini chat error")
 		ctx.EffectiveMessage.Reply(b, "gemini chat error", nil)
+		a <- struct{}{}
 		return err
 	}
 	log.Debug().Msgf("%s say: %s", sender, input)

Original file line number	Diff line number	Diff line change
`@@ -104,6 +104,7 @@ func (g *gemini) Chat(chatId string, msg string) (string, error) {`
`104`	`104`	`result := fmt.Sprint(resp.Candidates[0].Content.Parts[0])`
`105`	`105`	`if err := g.db.Add(models.NewChat(chatId, false, result)); err != nil {`
`106`	`106`	`log.Error().Err(err).Msg("failed to add chat record")`
	`107`	`+ return "", err`
`107`	`108`	`}`
`108`	`109`	`return result, nil`
`109`	`110`	`}`
Original file line number	Diff line number	Diff line change
`@@ -163,6 +163,7 @@ func handlePrivateChat(b gotgbot.Bot, ctx ext.Context, ai ai.AiInterface) erro`
`163`	`163`	`if err != nil {`
`164`	`164`	`log.Error().Err(err).Msg("gemini chat error")`
`165`	`165`	`ctx.EffectiveMessage.Reply(b, "gemini chat error", nil)`
	`166`	`+ a <- struct{}{}`
`166`	`167`	`return err`
`167`	`168`	`}`
`168`	`169`	`log.Debug().Msgf("%s say: %s", sender, input)`