diff --git a/api/core/tools/provider/builtin/azuredalle/tools/dalle3.py b/api/core/tools/provider/builtin/azuredalle/tools/dalle3.py
index 45cb9b1fe5..2ffdd38b72 100644
--- a/api/core/tools/provider/builtin/azuredalle/tools/dalle3.py
+++ b/api/core/tools/provider/builtin/azuredalle/tools/dalle3.py
@@ -1,3 +1,4 @@
+import random
 from base64 import b64decode
 from typing import Any, Union