From 5320da2f06eb3a214d4a7444f276b9f0286716f2 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Thu, 2 Nov 2023 22:19:19 -0300
Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20fix(field=5Ftyping):=20import=20?=
 =?UTF-8?q?BaseLanguageModel=20in=20=5F=5Finit=5F=5F.py=20to=20resolve=20m?=
 =?UTF-8?q?issing=20import=20error=20=F0=9F=94=A7=20fix(constants.py):=20i?=
 =?UTF-8?q?mport=20BaseLanguageModel=20in=20constants.py=20to=20resolve=20?=
 =?UTF-8?q?missing=20import=20error?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/field_typing/__init__.py  | 2 ++
 src/backend/langflow/field_typing/constants.py | 3 ++-
 2 files changed, 4 insertions(+), 1 deletion(-)

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,