From c4167a551791cc34e3f53832ce09467544fc250d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Moura?= Date: Sun, 22 Sep 2024 11:20:54 -0300 Subject: [PATCH] respecting OPENAI_MODEL_NAME --- src/crewai/agent.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/crewai/agent.py b/src/crewai/agent.py index 0fceb6a41..ec9fbeda2 100644 --- a/src/crewai/agent.py +++ b/src/crewai/agent.py @@ -119,7 +119,8 @@ class Agent(BaseAgent): def post_init_setup(self): self.agent_ops_agent_name = self.role self.llm = ( - getattr(self.llm, "model_name", None) + os.environ.get("OPENAI_MODEL_NAME") + or getattr(self.llm, "model_name", None) or getattr(self.llm, "deployment_name", None) or self.llm )