From 9fb84b90139d142848e225fbbca2fa6b11ba7cf5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=A5=9E=E4=BB=99=E9=83=BD=E6=B2=A1=E7=94=A8?=
 <615206459@qq.com>
Date: Fri, 28 Jun 2024 15:16:54 +0800
Subject: [PATCH] 1

---
 src/modules/helper/views/ai-code.vue | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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;