ea9e070f29
- 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.
70 lines
1.9 KiB
Go
70 lines
1.9 KiB
Go
package robot
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/yaoapp/yao/agent/robot/cache"
|
|
"github.com/yaoapp/yao/agent/robot/dedup"
|
|
"github.com/yaoapp/yao/agent/robot/events/integrations"
|
|
"github.com/yaoapp/yao/agent/robot/events/integrations/telegram"
|
|
"github.com/yaoapp/yao/agent/robot/executor"
|
|
"github.com/yaoapp/yao/agent/robot/logger"
|
|
"github.com/yaoapp/yao/agent/robot/manager"
|
|
"github.com/yaoapp/yao/agent/robot/plan"
|
|
"github.com/yaoapp/yao/agent/robot/pool"
|
|
"github.com/yaoapp/yao/agent/robot/store"
|
|
robottypes "github.com/yaoapp/yao/agent/robot/types"
|
|
)
|
|
|
|
var (
|
|
log = logger.New("robot")
|
|
|
|
globalManager *manager.Manager
|
|
globalCache *cache.Cache
|
|
globalPool *pool.Pool
|
|
globalDedup *dedup.Dedup
|
|
globalStore *store.Store
|
|
globalExecutor executor.Executor
|
|
globalPlan *plan.Plan
|
|
globalDispatcher *integrations.Dispatcher
|
|
)
|
|
|
|
// Init initializes the robot agent system
|
|
func Init() error {
|
|
globalCache = cache.New()
|
|
globalDedup = dedup.New()
|
|
globalStore = store.New()
|
|
globalPool = pool.New()
|
|
globalExecutor = executor.New()
|
|
globalManager = manager.New()
|
|
globalPlan = plan.New()
|
|
|
|
// Load robots into cache from database before starting dispatcher
|
|
rCtx := robottypes.NewContext(context.Background(), nil)
|
|
if err := globalCache.Load(rCtx); err != nil {
|
|
log.Warn("robot.Init: cache load failed (will rely on config events): %v", err)
|
|
}
|
|
|
|
adapters := map[string]integrations.Adapter{
|
|
"telegram": telegram.NewAdapter(),
|
|
}
|
|
globalDispatcher = integrations.NewDispatcher(globalCache, adapters)
|
|
if err := globalDispatcher.Start(context.Background()); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Shutdown gracefully shuts down the robot agent system
|
|
func Shutdown() error {
|
|
if globalDispatcher != nil {
|
|
globalDispatcher.Stop()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Manager returns the global manager instance
|
|
func Manager() *manager.Manager {
|
|
return globalManager
|
|
}
|