yao/agent/robot/executor/standard/host.go
Max ea9e070f29 Enhance robot integration with Telegram and improve event handling
- Add Telegram integration support by introducing a dispatcher for handling Telegram events and messages.
- Implement event notifications for robot configuration changes (creation, update, deletion) to facilitate integration with external services.
- Refactor the robot initialization process to load robots into cache and start the dispatcher, improving the overall system setup.
- Update the delivery event structure to include additional metadata for better context during message handling.
- Enhance logging capabilities for better observability during robot execution and event processing.
2026-03-01 22:03:25 +08:00

62 lines
1.9 KiB
Go

package standard
import (
"encoding/json"
"fmt"
kunlog "github.com/yaoapp/kun/log"
robottypes "github.com/yaoapp/yao/agent/robot/types"
)
// CallHostAgent calls the Host Agent with structured input and parses structured output.
// The Host Agent mediates all human-robot interactions through three scenarios:
// - "assign": new task assignment with multi-round confirmation
// - "guide": guidance during execution
// - "clarify": answering questions from waiting tasks
func (e *Executor) CallHostAgent(ctx *robottypes.Context, robot *robottypes.Robot, input *robottypes.HostInput, chatID string) (*robottypes.HostOutput, error) {
if robot == nil {
return nil, fmt.Errorf("robot cannot be nil")
}
agentID := ""
if robot.Config != nil && robot.Config.Resources != nil {
agentID = robot.Config.Resources.GetPhaseAgent(robottypes.PhaseHost)
}
if agentID == "" {
return nil, fmt.Errorf("no Host Agent configured for robot %s", robot.MemberID)
}
inputJSON, err := json.Marshal(input)
if err != nil {
return nil, fmt.Errorf("failed to marshal host input: %w", err)
}
kunlog.Info("calling Host Agent %s for scenario=%s chatID=%s", agentID, input.Scenario, chatID)
caller := NewConversationCaller(chatID)
result, err := caller.CallWithMessages(ctx, agentID, string(inputJSON))
if err != nil {
return nil, fmt.Errorf("host agent (%s) call failed: %w", agentID, err)
}
data, err := result.GetJSON()
if err != nil {
text := result.GetText()
kunlog.Warn("Host Agent returned non-JSON response, treating as confirm: %s", text)
return &robottypes.HostOutput{
Reply: text,
Action: robottypes.HostActionConfirm,
}, nil
}
output := &robottypes.HostOutput{}
raw, _ := json.Marshal(data)
if err := json.Unmarshal(raw, output); err != nil {
return &robottypes.HostOutput{
Reply: result.GetText(),
Action: robottypes.HostActionConfirm,
}, nil
}
return output, nil
}