diff --git a/src/component/openai_node/__init__.py b/src/component/openai_node/__init__.py
index 6453fbf..40c265a 100644
--- a/src/component/openai_node/__init__.py
+++ b/src/component/openai_node/__init__.py
@@ -1,3 +1,4 @@
 from .open_ai_node import *
 from .openai_tts_node import *
-from .whisper_node import *
\ No newline at end of file
+from .whisper_node import *
+from .dall_e_compute_node import *
\ No newline at end of file
diff --git a/src/aios/environment/dall_e_compute_node.py b/src/component/openai_node/dall_e_compute_node.py
similarity index 95%
rename from src/aios/environment/dall_e_compute_node.py
rename to src/component/openai_node/dall_e_compute_node.py
index 8e63e03..a1ffaa6 100644
--- a/src/aios/environment/dall_e_compute_node.py
+++ b/src/component/openai_node/dall_e_compute_node.py
@@ -9,9 +9,7 @@ import base64
 
 from PIL import Image
 
-from .compute_task import ComputeTask, ComputeTaskResult, ComputeTaskState, ComputeTaskType, ComputeTaskResultCode
-from .compute_node import ComputeNode
-from .storage import AIStorage, UserConfig
+from aios import ComputeTask, ComputeTaskResult, ComputeTaskState, ComputeTaskType, ComputeTaskResultCode,ComputeNode, AIStorage, UserConfig
 
 logger = logging.getLogger(__name__)
 
diff --git a/test/test_workspace.py b/test/test_workspace.py
new file mode 100644
index 0000000..62929d7
--- /dev/null
+++ b/test/test_workspace.py
@@ -0,0 +1,28 @@
+import sys
+import os
+import logging
+import asyncio
+import time
+import unittest
+
+dir_path = os.path.dirname(os.path.realpath(__file__))
+
+sys.path.append("{}/../src/".format(dir_path))
+
+from aios_kernel import WorkspaceEnvironment
+
+async def test_workspace():
+    test_env = WorkspaceEnvironment("test")
+    test_env._add_document_dir(f"{dir_path}/../rootfs/test_doc")
+    test_env._start_scan_document()
+    catalogs = await test_env.get_knowledege_catalog()
+    print(catalogs)
+    asyncio.sleep(60*60)
+
+
+if __name__ == "__main__":
+    asyncio.run(test_workspace())
+    print("OK!")
+    time.sleep(60*60)
+
+