diff --git a/plugins/ai_gen_image/__init__.py b/plugins/ai_gen_image/__init__.py
index 708d018..33849d1 100644
--- a/plugins/ai_gen_image/__init__.py
+++ b/plugins/ai_gen_image/__init__.py
@@ -1,2 +1,9 @@
 # AI绘图插件
-# 基于pollinations.ai API生成图片
\ No newline at end of file
+# 基于pollinations.ai API生成图片
+
+from .main import AIGenImagePlugin
+
+# 提供get_plugin函数，返回插件实例
+def get_plugin():
+    """获取插件实例"""
+    return AIGenImagePlugin()
\ No newline at end of file