fix: correctly stop agents
Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
@@ -14,7 +14,7 @@ type ActionContext struct {
|
||||
}
|
||||
|
||||
func (ac *ActionContext) Cancel() {
|
||||
if ac.cancelFunc == nil {
|
||||
if ac.cancelFunc != nil {
|
||||
ac.cancelFunc()
|
||||
}
|
||||
}
|
||||
|
||||
@@ -198,6 +198,7 @@ var ErrContextCanceled = fmt.Errorf("context canceled")
|
||||
func (a *Agent) Stop() {
|
||||
a.Lock()
|
||||
defer a.Unlock()
|
||||
xlog.Debug("Stopping agent", "agent", a.Character.Name)
|
||||
a.context.Cancel()
|
||||
}
|
||||
|
||||
|
||||
@@ -369,7 +369,6 @@ func (a *AgentPool) startAgentWithConfig(name string, config *AgentConfig) error
|
||||
go func() {
|
||||
if err := agent.Run(); err != nil {
|
||||
xlog.Error("Agent stopped", "error", err.Error(), "name", name)
|
||||
panic(err)
|
||||
}
|
||||
}()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user