From 23763f8795c4fe7b8713656fa724bfe8fb1b81d3 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@langflow.org>
Date: Wed, 19 Jun 2024 20:23:15 -0300
Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20(field=5Ftyping):=20introduce=20Bas?=
 =?UTF-8?q?eChatModel=20to=20field=5Ftyping=20module=20for=20better=20orga?=
 =?UTF-8?q?nization=20and=20structure=20=F0=9F=93=9D=20(constants):=20add?=
 =?UTF-8?q?=20BaseChatModel=20to=20LANGCHAIN=5FBASE=5FTYPES=20dictionary?=
 =?UTF-8?q?=20for=20consistency=20and=20completeness?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

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

diff --git a/src/backend/base/langflow/field_typing/__init__.py b/src/backend/base/langflow/field_typing/__init__.py
index 1038e438e..57eba2f47 100644
--- a/src/backend/base/langflow/field_typing/__init__.py
+++ b/src/backend/base/langflow/field_typing/__init__.py
@@ -7,6 +7,7 @@ from .constants import (
     BaseLLM,
     BaseLoader,
     BaseMemory,
+    BaseChatModel,
     BaseOutputParser,
     BasePromptTemplate,
     BaseRetriever,
@@ -80,4 +81,5 @@ __all__ = [
     "TextSplitter",
     "Tool",
     "VectorStore",
+    "BaseChatModel",
 ]
diff --git a/src/backend/base/langflow/field_typing/constants.py b/src/backend/base/langflow/field_typing/constants.py
index 38846ba77..3e9ca973f 100644
--- a/src/backend/base/langflow/field_typing/constants.py
+++ b/src/backend/base/langflow/field_typing/constants.py
@@ -1,5 +1,5 @@
 from typing import Callable, Dict, Text, Union
-
+from langchain_core.language_models.chat_models import BaseChatModel
 from langchain.agents.agent import AgentExecutor
 from langchain.chains.base import Chain
 from langchain.memory.chat_memory import BaseChatMemory
@@ -48,6 +48,7 @@ LANGCHAIN_BASE_TYPES = {
     "BaseOutputParser": BaseOutputParser,
     "BaseMemory": BaseMemory,
     "BaseChatMemory": BaseChatMemory,
+    "BaseChatModel": BaseChatModel,
 }
 # Langchain base types plus Python base types
 CUSTOM_COMPONENT_SUPPORTED_TYPES = {