From c72ee729bfa7127b34e19a646b626e7a1501836b Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Thu, 22 Feb 2024 15:44:26 -0300
Subject: [PATCH] Add Code class to field_typing module

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

diff --git a/src/backend/langflow/field_typing/__init__.py b/src/backend/langflow/field_typing/__init__.py
index de8f020c6..15ce03693 100644
--- a/src/backend/langflow/field_typing/__init__.py
+++ b/src/backend/langflow/field_typing/__init__.py
@@ -13,6 +13,7 @@ from .constants import (
     Callable,
     Chain,
     ChatPromptTemplate,
+    Code,
     Data,
     Document,
     Embeddings,
@@ -73,4 +74,5 @@ __all__ = [
     "Prompt",
     "RangeSpec",
     "TemplateField",
+    "Code",
 ]
diff --git a/src/backend/langflow/field_typing/constants.py b/src/backend/langflow/field_typing/constants.py
index ed0c0e442..5977cd9f8 100644
--- a/src/backend/langflow/field_typing/constants.py
+++ b/src/backend/langflow/field_typing/constants.py
@@ -34,6 +34,10 @@ class Prompt:
     pass
 
 
+class Code:
+    pass
+
+
 LANGCHAIN_BASE_TYPES = {
     "Chain": Chain,
     "AgentExecutor": AgentExecutor,