diff --git a/libs/experimental/langchain_experimental/graph_transformers/llm.py b/libs/experimental/langchain_experimental/graph_transformers/llm.py
index f55cb81feb..5b432a94bf 100644
--- a/libs/experimental/langchain_experimental/graph_transformers/llm.py
+++ b/libs/experimental/langchain_experimental/graph_transformers/llm.py
@@ -332,6 +332,7 @@ def create_simple_model(
             ),
         ),
     }
+
     if node_properties:
         if isinstance(node_properties, list) and "id" in node_properties:
             raise ValueError("The node property 'id' is reserved and cannot be used.")
@@ -347,6 +348,7 @@ def create_simple_model(
                 node_properties_mapped,
                 description="Property key.",
                 input_type="property",
+                llm_type=llm_type,
             )
             value: str = Field(..., description="value")
 
@@ -370,6 +372,7 @@ def create_simple_model(
                 node_labels,
                 description="The type or label of the source node.",
                 input_type="node",
+                llm_type=llm_type,
             ),
         ),
         "target_node_id": (
@@ -385,6 +388,7 @@ def create_simple_model(
                 node_labels,
                 description="The type or label of the target node.",
                 input_type="node",
+                llm_type=llm_type,
             ),
         ),
         "type": (
@@ -393,6 +397,7 @@ def create_simple_model(
                 rel_types,
                 description="The type of the relationship.",
                 input_type="relationship",
+                llm_type=llm_type,
             ),
         ),
     }
@@ -416,6 +421,7 @@ def create_simple_model(
                 relationship_properties_mapped,
                 description="Property key.",
                 input_type="property",
+                llm_type=llm_type,
             )
             value: str = Field(..., description="value")