Merge branch 'zustand/io/migration' of personal:logspace-ai/langflow into zustand/io/migration

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",
 ]

src/backend/langflow/field_typing/constants.py

@@ -34,6 +34,10 @@ class Prompt:
     pass
 
 
+class Code:
+    pass
+
+
 LANGCHAIN_BASE_TYPES = {
     "Chain": Chain,
     "AgentExecutor": AgentExecutor,

src/backend/langflow/interface/custom/utils.py

@@ -121,7 +121,13 @@ def get_field_properties(extra_field):
 def process_type(field_type: str):
     if field_type.startswith("list") or field_type.startswith("List"):
         return extract_inner_type(field_type)
-    return "prompt" if field_type == "Prompt" else field_type
+
+    # field_type is a string can be Prompt or Code too
+    # so we just need to lower if it is the case
+    lowercase_type = field_type.lower()
+    if lowercase_type in ["prompt", "code"]:
+        return lowercase_type
+    return field_type
 
 
 def add_new_custom_field(

src/backend/langflow/utils/types.py

@@ -1,2 +0,0 @@
-class Prompt:
-    pass