diff --git a/api/core/llm_generator/output_parser/structured_output.py b/api/core/llm_generator/output_parser/structured_output.py
index e9c7bb5595..d26c8b3cc3 100644
--- a/api/core/llm_generator/output_parser/structured_output.py
+++ b/api/core/llm_generator/output_parser/structured_output.py
@@ -24,7 +24,6 @@ from core.model_runtime.entities.message_entities import (
     PromptMessage,
     PromptMessageTool,
     SystemPromptMessage,
-    TextPromptMessageContent,
 )
 from core.model_runtime.entities.model_entities import AIModelEntity, ModelFeature, ParameterRule