Merge pull request #202 from kimtth/main

fix: correct typo
This commit is contained in:
warmshao
2025-02-09 08:17:38 +08:00
committed by GitHub
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__)
@@ -118,7 +118,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,