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() {
|
func (ac *ActionContext) Cancel() {
|
||||||
if ac.cancelFunc == nil {
|
if ac.cancelFunc != nil {
|
||||||
ac.cancelFunc()
|
ac.cancelFunc()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -198,6 +198,7 @@ var ErrContextCanceled = fmt.Errorf("context canceled")
|
|||||||
func (a *Agent) Stop() {
|
func (a *Agent) Stop() {
|
||||||
a.Lock()
|
a.Lock()
|
||||||
defer a.Unlock()
|
defer a.Unlock()
|
||||||
|
xlog.Debug("Stopping agent", "agent", a.Character.Name)
|
||||||
a.context.Cancel()
|
a.context.Cancel()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -369,7 +369,6 @@ func (a *AgentPool) startAgentWithConfig(name string, config *AgentConfig) error
|
|||||||
go func() {
|
go func() {
|
||||||
if err := agent.Run(); err != nil {
|
if err := agent.Run(); err != nil {
|
||||||
xlog.Error("Agent stopped", "error", err.Error(), "name", name)
|
xlog.Error("Agent stopped", "error", err.Error(), "name", name)
|
||||||
panic(err)
|
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user