fix: correct typo
massage -> message
This commit is contained in:
@@ -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(),
|
||||
|
||||
@@ -24,7 +24,7 @@ from .custom_prompts import CustomAgentMessagePrompt
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class CustomMassageManager(MessageManager):
|
||||
class CustomMessageManager(MessageManager):
|
||||
def __init__(
|
||||
self,
|
||||
llm: BaseChatModel,
|
||||
Reference in New Issue
Block a user