diff --git a/src/backend/base/langflow/components/agents/OpenAIToolsAgent.py b/src/backend/base/langflow/components/agents/OpenAIToolsAgent.py
index 175f572af72..661d0d8021b 100644
--- a/src/backend/base/langflow/components/agents/OpenAIToolsAgent.py
+++ b/src/backend/base/langflow/components/agents/OpenAIToolsAgent.py
@@ -6,7 +6,7 @@
 from langflow.base.agents.agent import LCToolsAgentComponent
 from langflow.field_typing import Data
 from langflow.inputs import MultilineInput
-from langflow.inputs.inputs import HandleInput
+from langflow.inputs.inputs import HandleInput, DataInput
 
 
 class OpenAIToolsAgentComponent(LCToolsAgentComponent):