make sure agent key is present

2 years ago · 5c2e7b5b7b
parent 71717fe348
commit 5c2e7b5b7b
1 changed files with 3 additions and 0 deletions
--- a/swarms/agents/utils/AgentManager.py
+++ b/swarms/agents/utils/AgentManager.py
@ -84,7 +84,10 @@ class AgentManager:
                callback_manager=callback_manager,
                verbose=True,
            )
+            if 'agent' not in executor.__dict__:
+                executor.__dict__['agent'] = agent
            self.executors[session] = executor
+
            return executor
        except Exception as e:
            logging.error(f"Error while creating executor: {str(e)}")