diff --git a/src/modules/helper/views/ai-code.vue b/src/modules/helper/views/ai-code.vue
index 5b4056a..6dc5db0 100644
--- a/src/modules/helper/views/ai-code.vue
+++ b/src/modules/helper/views/ai-code.vue
@@ -396,7 +396,8 @@ const code = reactive({
 	async getColumns() {
 		return ai
 			.invokeFlow("comm-column", {
-				name: form.entity
+				name: form.entity,
+				modules: module.dirs
 			})
 			.then((res) => {
 				form.column = res.columns;