From 55158438ff2780beeb06d3e3ecdd9306916621cf Mon Sep 17 00:00:00 2001 From: Brandon Hancock Date: Fri, 23 Aug 2024 12:27:57 -0400 Subject: [PATCH] More carefully check atters on llm --- src/crewai/agent.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/crewai/agent.py b/src/crewai/agent.py index 1618675ee..0711c741d 100644 --- a/src/crewai/agent.py +++ b/src/crewai/agent.py @@ -118,7 +118,9 @@ class Agent(BaseAgent): self.agent_ops_agent_name = self.role # Different llms store the model name in different attributes - model_name = self.llm.model_name or self.llm.deployment_name + model_name = getattr(self.llm, "model_name", None) or getattr( + self.llm, "deployment_name", None + ) if model_name: self._setup_llm_callbacks(model_name)