andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-09 08:08:32 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

More carefully check atters on llm

Browse Source
This commit is contained in:
Brandon Hancock
2024-08-23 12:27:57 -04:00
parent 2cc6cab6a9
commit 55158438ff
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/crewai/agent.py
View File

@@ -118,7 +118,9 @@ class Agent(BaseAgent):
self.agent_ops_agent_name = self.role self.agent_ops_agent_name = self.role
# Different llms store the model name in different attributes # 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: if model_name:
self._setup_llm_callbacks(model_name) self._setup_llm_callbacks(model_name)