diff --git a/src/backend/langflow/field_typing/__init__.py b/src/backend/langflow/field_typing/__init__.py index 9a2161d3f..2af9477ea 100644 --- a/src/backend/langflow/field_typing/__init__.py +++ b/src/backend/langflow/field_typing/__init__.py @@ -31,6 +31,7 @@ from .constants import ( AgentExecutor, NestedDict, Data, + BaseLanguageModel, ) __all__ = [ @@ -41,6 +42,7 @@ __all__ = [ "Chain", "BaseChatMemory", "BaseLLM", + "BaseLanguageModel", "BaseLoader", "BaseMemory", "BaseOutputParser", diff --git a/src/backend/langflow/field_typing/constants.py b/src/backend/langflow/field_typing/constants.py index 3ce429548..68179f05a 100644 --- a/src/backend/langflow/field_typing/constants.py +++ b/src/backend/langflow/field_typing/constants.py @@ -1,7 +1,7 @@ from langchain.agents.agent import AgentExecutor from langchain.chains.base import Chain from langchain.document_loaders.base import BaseLoader -from langchain.llms.base import BaseLLM +from langchain.llms.base import BaseLLM, BaseLanguageModel from langchain.memory.chat_memory import BaseChatMemory from langchain.prompts import PromptTemplate from langchain.schema import BaseOutputParser, BaseRetriever, Document @@ -25,6 +25,7 @@ LANGCHAIN_BASE_TYPES = { "AgentExecutor": AgentExecutor, "Tool": Tool, "BaseLLM": BaseLLM, + "BaseLanguageModel": BaseLanguageModel, "PromptTemplate": PromptTemplate, "BaseLoader": BaseLoader, "Document": Document,