From bc7192d664d38ae95f596bf9857103e3f507cdbc Mon Sep 17 00:00:00 2001
From: mudler <mudler@localai.io>
Date: Sat, 20 Apr 2024 10:35:57 +0200
Subject: [PATCH] handle corner case

---
 agent/agent.go | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/agent/agent.go b/agent/agent.go
index 30a7130..46753dd 100644
--- a/agent/agent.go
+++ b/agent/agent.go
@@ -243,8 +243,10 @@ func (a *Agent) consumeJob(job *Job, role string) {
 
 	defer func() {
 		a.Lock()
-		a.actionContext.Cancel()
-		a.actionContext = nil
+		if a.actionContext != nil {
+			a.actionContext.Cancel()
+			a.actionContext = nil
+		}
 		a.Unlock()
 	}()