From fbced39ad57da631868a2079a74e745ce7e7d199 Mon Sep 17 00:00:00 2001 From: kimtth Date: Thu, 30 Jan 2025 14:40:03 +0900 Subject: [PATCH] fix: correct typo massage -> message --- src/agent/custom_agent.py | 4 ++-- .../{custom_massage_manager.py => custom_message_manager.py} | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) rename src/agent/{custom_massage_manager.py => custom_message_manager.py} (99%) 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,