diff --git a/src/backend/base/langflow/base/agents/agent.py b/src/backend/base/langflow/base/agents/agent.py index 029ab4913..d0ce82d76 100644 --- a/src/backend/base/langflow/base/agents/agent.py +++ b/src/backend/base/langflow/base/agents/agent.py @@ -1,6 +1,8 @@ from typing import List, Union from langchain.agents import AgentExecutor, BaseMultiActionAgent, BaseSingleActionAgent +from langchain_core.runnables import Runnable + from langflow.field_typing import BaseMemory, Text, Tool from langflow.interface.custom.custom_component import CustomComponent @@ -38,7 +40,7 @@ class LCAgentComponent(CustomComponent): async def run_agent( self, - agent: Union[BaseSingleActionAgent, BaseMultiActionAgent, AgentExecutor], + agent: Union[Runnable, BaseSingleActionAgent, BaseMultiActionAgent, AgentExecutor], inputs: str, input_variables: list[str], tools: List[Tool], diff --git a/src/backend/base/langflow/components/experimental/__init__.py b/src/backend/base/langflow/components/experimental/__init__.py index d23a73106..2d59a063d 100644 --- a/src/backend/base/langflow/components/experimental/__init__.py +++ b/src/backend/base/langflow/components/experimental/__init__.py @@ -1,6 +1,6 @@ from .ClearMessageHistory import ClearMessageHistoryComponent from .ExtractDataFromRecord import ExtractKeyFromRecordComponent -from .Listen import GetNotifiedComponent +from .Listen import ListenComponent from .ListFlows import ListFlowsComponent from .MergeRecords import MergeRecordsComponent from .Notify import NotifyComponent @@ -11,7 +11,7 @@ from .SQLExecutor import SQLExecutorComponent __all__ = [ "ClearMessageHistoryComponent", "ExtractKeyFromRecordComponent", - "GetNotifiedComponent", + "ListenComponent", "ListFlowsComponent", "MergeRecordsComponent", "MessageHistoryComponent", diff --git a/src/backend/base/langflow/services/database/models/flow/model.py b/src/backend/base/langflow/services/database/models/flow/model.py index 95311909c..cbfdec563 100644 --- a/src/backend/base/langflow/services/database/models/flow/model.py +++ b/src/backend/base/langflow/services/database/models/flow/model.py @@ -1,4 +1,4 @@ -# Path: src/backend/langflow/database/models/flow.py +# Path: src/backend/langflow/services/database/models/flow/model.py from datetime import datetime from typing import TYPE_CHECKING, Dict, Optional