From c6ad9a56fa9065c28d1a8bc1355a36ce86004e6b Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@langflow.org>
Date: Sun, 23 Jun 2024 18:14:56 -0300
Subject: [PATCH] refactor: Update build_config parameter type in
 AgentComponent

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

diff --git a/src/backend/base/langflow/field_typing/__init__.py b/src/backend/base/langflow/field_typing/__init__.py
index 20915fa5c..bd25f6cf4 100644
--- a/src/backend/base/langflow/field_typing/__init__.py
+++ b/src/backend/base/langflow/field_typing/__init__.py
@@ -3,11 +3,11 @@ from typing import Any
 from .constants import (
     AgentExecutor,
     BaseChatMemory,
+    BaseChatModel,
     BaseLanguageModel,
     BaseLLM,
     BaseLoader,
     BaseMemory,
-    BaseChatModel,
     BaseOutputParser,
     BasePromptTemplate,
     BaseRetriever,
@@ -21,10 +21,11 @@ from .constants import (
     NestedDict,
     Object,
     PromptTemplate,
+    Retriever,
+    Text,
     TextSplitter,
     Tool,
     VectorStore,
-    Retriever,
 )
 from .range_spec import RangeSpec
 
@@ -82,4 +83,5 @@ __all__ = [
     "VectorStore",
     "BaseChatModel",
     "Retriever",
+    "Text",
 ]
diff --git a/src/backend/base/langflow/field_typing/constants.py b/src/backend/base/langflow/field_typing/constants.py
index 8f14d7bde..a857de59e 100644
--- a/src/backend/base/langflow/field_typing/constants.py
+++ b/src/backend/base/langflow/field_typing/constants.py
@@ -1,4 +1,4 @@
-from typing import Callable, Dict, TypeAlias, TypeVar, Union
+from typing import Callable, Dict, Text, TypeAlias, TypeVar, Union
 
 from langchain.agents.agent import AgentExecutor
 from langchain.chains.base import Chain
@@ -65,6 +65,7 @@ CUSTOM_COMPONENT_SUPPORTED_TYPES = {
     "NestedDict": NestedDict,
     "Data": Data,
     "Message": Message,
+    "Text": Text,
     "Object": Object,
     "Callable": Callable,
     "LanguageModel": LanguageModel,