Update crewai/agent.py

crewai/agent.py

@@ -1,12 +1,13 @@
 import uuid
 from typing import Any, List, Optional
 
-from langchain.agents.agent import RunnableAgent
 from langchain.agents.format_scratchpad import format_log_to_str
+from langchain.agents.agent import RunnableAgent
 from langchain.memory import ConversationSummaryMemory
 from langchain.tools.render import render_text_description
 from langchain_core.runnables.config import RunnableConfig
 from langchain_openai import ChatOpenAI
+from langchain_core.language_models import BaseLanguageModel
 from pydantic import (
     UUID4,
     BaseModel,
@@ -30,10 +31,8 @@ from crewai.utilities import I18N, Logger, Prompts, RPMController
 
 class Agent(BaseModel):
     """Represents an agent in a system.
-
     Each agent has a role, a goal, a backstory, and an optional language model (llm).
     The agent can also have memory, can operate in verbose mode, and can delegate tasks to other agents.
-
     Attributes:
             agent_executor: An instance of the CrewAgentExecutor class.
             role: The role of the agent.
@@ -130,12 +129,10 @@ class Agent(BaseModel):
         tools: Optional[List[Any]] = None,
     ) -> str:
         """Execute a task with the agent.
-
         Args:
             task: Task to execute.
             context: Context to execute the task in.
             tools: Tools to use for the task.
-
         Returns:
             Output of the agent
         """
@@ -164,27 +161,24 @@ class Agent(BaseModel):
 
     def set_cache_handler(self, cache_handler: CacheHandler) -> None:
         """Set the cache handler for the agent.
-
         Args:
             cache_handler: An instance of the CacheHandler class.
         """
         self.cache_handler = cache_handler
         self.tools_handler = ToolsHandler(cache=self.cache_handler)
-        self._create_agent_executor()
+        self.__create_agent_executor()
 
     def set_rpm_controller(self, rpm_controller: RPMController) -> None:
         """Set the rpm controller for the agent.
-
         Args:
             rpm_controller: An instance of the RPMController class.
         """
         if not self._rpm_controller:
             self._rpm_controller = rpm_controller
-            self._create_agent_executor()
+            self.__create_agent_executor()
 
-    def _create_agent_executor(self) -> None:
+    def __create_agent_executor(self) -> None:
         """Create an agent executor for the agent.
-
         Returns:
             An instance of the CrewAgentExecutor class.
         """
@@ -240,4 +234,4 @@ class Agent(BaseModel):
 
     @staticmethod
     def __tools_names(tools) -> str:
-        return ", ".join([t.name for t in tools])
+        return ", ".join([t.name for t in tools])