diff --git a/templates/neo4j-semantic-layer/main.py b/templates/neo4j-semantic-layer/main.py
index b5fecae1d6cd2..3a197463ab03b 100644
--- a/templates/neo4j-semantic-layer/main.py
+++ b/templates/neo4j-semantic-layer/main.py
@@ -1,4 +1,3 @@
-from langchain.schema import AIMessage, HumanMessage
 from neo4j_semantic_layer import agent_executor
 
 if __name__ == "__main__":