cf5a84aac1
Merge upstream/main into bugfix/fix-duplicate-telegram-messages. Conflict resolutions: - pkg/agent/loop.go: Adopt upstream's processSystemMessage which removes runAgentLoop call entirely (subagents now communicate via message tool directly). Keep PR's HasSentInRound() check in Run() for normal message processing path. - pkg/tools/message.go: Merge both changes - keep sentInRound tracking from PR and adopt upstream's *ToolResult return type with Silent: true.
104 lines
2.5 KiB
Go
104 lines
2.5 KiB
Go
package tools
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
)
|
|
|
|
type SendCallback func(channel, chatID, content string) error
|
|
|
|
type MessageTool struct {
|
|
sendCallback SendCallback
|
|
defaultChannel string
|
|
defaultChatID string
|
|
sentInRound bool // Tracks whether a message was sent in the current processing round
|
|
}
|
|
|
|
func NewMessageTool() *MessageTool {
|
|
return &MessageTool{}
|
|
}
|
|
|
|
func (t *MessageTool) Name() string {
|
|
return "message"
|
|
}
|
|
|
|
func (t *MessageTool) Description() string {
|
|
return "Send a message to user on a chat channel. Use this when you want to communicate something."
|
|
}
|
|
|
|
func (t *MessageTool) Parameters() map[string]interface{} {
|
|
return map[string]interface{}{
|
|
"type": "object",
|
|
"properties": map[string]interface{}{
|
|
"content": map[string]interface{}{
|
|
"type": "string",
|
|
"description": "The message content to send",
|
|
},
|
|
"channel": map[string]interface{}{
|
|
"type": "string",
|
|
"description": "Optional: target channel (telegram, whatsapp, etc.)",
|
|
},
|
|
"chat_id": map[string]interface{}{
|
|
"type": "string",
|
|
"description": "Optional: target chat/user ID",
|
|
},
|
|
},
|
|
"required": []string{"content"},
|
|
}
|
|
}
|
|
|
|
func (t *MessageTool) SetContext(channel, chatID string) {
|
|
t.defaultChannel = channel
|
|
t.defaultChatID = chatID
|
|
t.sentInRound = false // Reset send tracking for new processing round
|
|
}
|
|
|
|
// HasSentInRound returns true if the message tool sent a message during the current round.
|
|
func (t *MessageTool) HasSentInRound() bool {
|
|
return t.sentInRound
|
|
}
|
|
|
|
func (t *MessageTool) SetSendCallback(callback SendCallback) {
|
|
t.sendCallback = callback
|
|
}
|
|
|
|
func (t *MessageTool) Execute(ctx context.Context, args map[string]interface{}) *ToolResult {
|
|
content, ok := args["content"].(string)
|
|
if !ok {
|
|
return &ToolResult{ForLLM: "content is required", IsError: true}
|
|
}
|
|
|
|
channel, _ := args["channel"].(string)
|
|
chatID, _ := args["chat_id"].(string)
|
|
|
|
if channel == "" {
|
|
channel = t.defaultChannel
|
|
}
|
|
if chatID == "" {
|
|
chatID = t.defaultChatID
|
|
}
|
|
|
|
if channel == "" || chatID == "" {
|
|
return &ToolResult{ForLLM: "No target channel/chat specified", IsError: true}
|
|
}
|
|
|
|
if t.sendCallback == nil {
|
|
return &ToolResult{ForLLM: "Message sending not configured", IsError: true}
|
|
}
|
|
|
|
if err := t.sendCallback(channel, chatID, content); err != nil {
|
|
return &ToolResult{
|
|
ForLLM: fmt.Sprintf("sending message: %v", err),
|
|
IsError: true,
|
|
Err: err,
|
|
}
|
|
}
|
|
|
|
t.sentInRound = true
|
|
// Silent: user already received the message directly
|
|
return &ToolResult{
|
|
ForLLM: fmt.Sprintf("Message sent to %s:%s", channel, chatID),
|
|
Silent: true,
|
|
}
|
|
}
|