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

Fix deployment name issue to support Azure

Browse Source
This commit is contained in:
Brandon Hancock
2024-08-23 11:38:29 -04:00
parent 8119edb495
commit 2cc6cab6a9
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/crewai/agent.py
View File

@@ -117,16 +117,19 @@ class Agent(BaseAgent):
def post_init_setup(self): def post_init_setup(self):
self.agent_ops_agent_name = self.role self.agent_ops_agent_name = self.role
if hasattr(self.llm, "model_name"): # Different llms store the model name in different attributes
self._setup_llm_callbacks() model_name = self.llm.model_name or self.llm.deployment_name
if model_name:
self._setup_llm_callbacks(model_name)
if not self.agent_executor: if not self.agent_executor:
self._setup_agent_executor() self._setup_agent_executor()
return self return self
def _setup_llm_callbacks(self): def _setup_llm_callbacks(self, model_name: str):
token_handler = TokenCalcHandler(self.llm.model_name, self._token_process) token_handler = TokenCalcHandler(model_name, self._token_process)
if not isinstance(self.llm.callbacks, list): if not isinstance(self.llm.callbacks, list):
self.llm.callbacks = [] self.llm.callbacks = []