DBT/clawgo
1
0
Fork 0
mirror of https://github.com/YspCoder/clawgo.git synced 2026-04-29 11:07:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: improve telegram streaming stability and task context with memory/ekg

Browse Source
This commit is contained in:
lpf
2026-03-04 14:05:47 +08:00
parent a7dcafae60
commit b007a403a1
4 changed files with 259 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
pkg/channels/telegram.go
View File

@@ -65,7 +65,7 @@ type telegramRenderedChunk struct {
func (c *TelegramChannel) SupportsAction(action string) bool {
switch strings.ToLower(strings.TrimSpace(action)) {
case "", "send", "edit", "delete", "react", "stream":
case "", "send", "edit", "delete", "react", "stream", "finalize":
return true
default:
return false
@@ -866,6 +866,8 @@ func renderTelegramStreamChunks(content string) []telegramRenderedChunk {
parts = splitTelegramMarkdown(body, telegramStreamSplitMaxRunes)
case telego.ModeHTML:
parts = splitTelegramText(body, telegramSafeHTMLMaxRunes)
case "text":
parts = splitTelegramText(body, telegramStreamSplitMaxRunes)
default:
parts = splitTelegramMarkdown(body, telegramStreamSplitMaxRunes)
mode = "auto_markdown"
@@ -894,6 +896,14 @@ func renderTelegramStreamChunks(content string) []telegramRenderedChunk {
if strings.TrimSpace(payload) != "" {
out = append(out, telegramRenderedChunk{payload: payload, parseMode: telego.ModeMarkdownV2})
}
case "text":
payload := trimmed
if len([]rune(payload)) > telegramStreamSplitMaxRunes {
payload = splitTelegramText(payload, telegramStreamSplitMaxRunes)[0]
}
if strings.TrimSpace(payload) != "" {
out = append(out, telegramRenderedChunk{payload: payload, parseMode: ""})
}
default:
payload := sanitizeTelegramHTML(markdownToTelegramHTML(trimmed))
if len([]rune(payload)) > telegramSafeHTMLMaxRunes {
@@ -914,6 +924,8 @@ func detectTelegramStreamMode(content string) (mode string, body string) {
return telego.ModeHTML, strings.TrimSpace(trimmed[len("[mode:html]"):])
case strings.HasPrefix(strings.ToLower(trimmed), "[mode:markdownv2]"):
return telego.ModeMarkdownV2, strings.TrimSpace(trimmed[len("[mode:markdownv2]"):])
case strings.HasPrefix(strings.ToLower(trimmed), "[mode:text]"):
return "text", strings.TrimSpace(trimmed[len("[mode:text]"):])
default:
return "auto_markdown", content
}
@@ -1068,6 +1080,23 @@ func (c *TelegramChannel) handleAction(ctx context.Context, chatID int64, action
return err
case "stream":
return c.handleStreamAction(ctx, chatID, msg)
case "finalize":
if strings.TrimSpace(msg.Content) != "" {
// Final pass in auto-markdown mode to recover rich formatting after plain streaming.
if err := c.handleStreamAction(ctx, chatID, bus.OutboundMessage{
ChatID: msg.ChatID,
ReplyToID: msg.ReplyToID,
Content: msg.Content,
Action: "stream",
}); err != nil {
return err
}
}
streamKey := telegramStreamKey(chatID, msg.ReplyToID)
c.streamMu.Lock()
delete(c.streamState, streamKey)
c.streamMu.Unlock()
return nil
case "delete":
delCtx, cancel := withTelegramAPITimeout(ctx)
defer cancel()