diff --git a/src/agent/custom_agent.py b/src/agent/custom_agent.py index 10be78d4..98d38677 100644 --- a/src/agent/custom_agent.py +++ b/src/agent/custom_agent.py @@ -34,7 +34,7 @@ from json_repair import repair_json from src.utils.agent_state import AgentState -from .custom_massage_manager import CustomMassageManager +from .custom_message_manager import CustomMessageManager from .custom_views import CustomAgentOutput, CustomAgentStepInfo logger = logging.getLogger(__name__) @@ -116,7 +116,7 @@ def __init__( # agent_state for Stop self.agent_state = agent_state self.agent_prompt_class = agent_prompt_class - self.message_manager = CustomMassageManager( + self.message_manager = CustomMessageManager( llm=self.llm, task=self.task, action_descriptions=self.controller.registry.get_prompt_description(), diff --git a/src/agent/custom_massage_manager.py b/src/agent/custom_message_manager.py similarity index 99% rename from src/agent/custom_massage_manager.py rename to src/agent/custom_message_manager.py index f39c9998..4cc42e21 100644 --- a/src/agent/custom_massage_manager.py +++ b/src/agent/custom_message_manager.py @@ -24,7 +24,7 @@ logger = logging.getLogger(__name__) -class CustomMassageManager(MessageManager): +class CustomMessageManager(MessageManager): def __init__( self, llm: BaseChatModel,