fix: correct typo

massage -> message
This commit is contained in:
kimtth
2025-01-30 14:40:03 +09:00
parent dc41476f37
commit fbced39ad5
2 changed files with 3 additions and 3 deletions

View File

@@ -34,7 +34,7 @@ from langchain_core.messages import (
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 @@ class CustomAgent(Agent):
# 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(),

View File

@@ -24,7 +24,7 @@ from .custom_prompts import CustomAgentMessagePrompt
logger = logging.getLogger(__name__)
class CustomMassageManager(MessageManager):
class CustomMessageManager(MessageManager):
def __init__(
self,
llm: BaseChatModel,