liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

去掉AIlogo

Browse Source
This commit is contained in:
liuwei
2025-08-27 09:55:54 +08:00
parent b2f94f540d
commit 09c10c0db0
1 changed files with 18 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
plugins/ai_gen_image/main.py
View File

@@ -73,14 +73,16 @@ class AIGenImagePlugin(MessagePluginInterface):
self.enable = self._config.get("AIGenImage", {}).get("enable", True)
# API配置
self.image_api_url = self._config.get("AIGenImage", {}).get("image_api_url", "https://image.pollinations.ai/prompt/{prompt}")
self.image_api_url = self._config.get("AIGenImage", {}).get("image_api_url",
"https://image.pollinations.ai/prompt/{prompt}")
self.default_width = self._config.get("AIGenImage", {}).get("default_width", 1024)
self.default_height = self._config.get("AIGenImage", {}).get("default_height", 1024)
self.default_model = self._config.get("AIGenImage", {}).get("default_model", "turbo")
self.default_timeout = self._config.get("AIGenImage", {}).get("default_timeout", 300)
# 确保临时目录存在
self.temp_dir = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))), 'temp')
self.temp_dir = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))),
'temp')
os.makedirs(self.temp_dir, exist_ok=True)
self.LOG.info(f"[{self.name}] 插件初始化完成，指令：{self._commands}")
@@ -160,7 +162,8 @@ class AIGenImagePlugin(MessagePluginInterface):
"width": self.default_width,
"height": self.default_height,
"model": self.default_model,
"seed": int(time.time()) % 1000000 # 使用时间戳作为随机种子
"seed": int(time.time()) % 1000000, # 使用时间戳作为随机种子
"nologo": "true" # Optional, set to "true" for registered referrers/tokens
}
# 编码提示词