diff --git a/dbgpt/app/scene/chat_db/auto_execute/chat.py b/dbgpt/app/scene/chat_db/auto_execute/chat.py
index 30e6fcbaa..6c0cdd187 100644
--- a/dbgpt/app/scene/chat_db/auto_execute/chat.py
+++ b/dbgpt/app/scene/chat_db/auto_execute/chat.py
@@ -1,5 +1,6 @@
 from typing import Dict
 
+from dbgpt.agent.commands.command_mange import ApiCall
 from dbgpt.app.scene import BaseChat, ChatScene
 from dbgpt._private.config import Config
 from dbgpt.util.executor_utils import blocking_func_to_async