🤖 自动格式化代码 [skip ci]
This commit is contained in:
@@ -24,6 +24,7 @@
|
||||
- 拍照Action - 生成自拍照
|
||||
- 展示照片Command - 展示最近生成的照片
|
||||
"""
|
||||
|
||||
from typing import List, Tuple, Type, Optional
|
||||
import random
|
||||
import datetime
|
||||
@@ -67,11 +68,7 @@ class TakePictureAction(BaseAction):
|
||||
|
||||
action_parameters = {}
|
||||
|
||||
action_require = [
|
||||
"当用户想看你的照片时使用",
|
||||
"当用户让你发自拍时使用"
|
||||
"当想随手拍眼前的场景时使用"
|
||||
]
|
||||
action_require = ["当用户想看你的照片时使用", "当用户让你发自拍时使用当想随手拍眼前的场景时使用"]
|
||||
|
||||
associated_types = ["text", "image"]
|
||||
|
||||
@@ -108,16 +105,12 @@ class TakePictureAction(BaseAction):
|
||||
bot_nickname = self.api.get_global_config("bot.nickname", "麦麦")
|
||||
bot_personality = self.api.get_global_config("personality.personality_core", "")
|
||||
|
||||
|
||||
personality_sides = self.api.get_global_config("personality.personality_sides", [])
|
||||
if personality_sides:
|
||||
bot_personality += random.choice(personality_sides)
|
||||
|
||||
# 准备模板变量
|
||||
template_vars = {
|
||||
"name": bot_nickname,
|
||||
"personality": bot_personality
|
||||
}
|
||||
template_vars = {"name": bot_nickname, "personality": bot_personality}
|
||||
|
||||
logger.info(f"{self.log_prefix} 使用的全局配置: name={bot_nickname}, personality={bot_personality}")
|
||||
|
||||
@@ -136,7 +129,7 @@ class TakePictureAction(BaseAction):
|
||||
|
||||
# 从配置获取参数
|
||||
model = self.api.get_config("picture.default_model", "doubao-seedream-3-0-t2i-250415")
|
||||
style = self.api.get_config("picture.default_style", "动漫")
|
||||
self.api.get_config("picture.default_style", "动漫")
|
||||
size = self.api.get_config("picture.default_size", "1024x1024")
|
||||
watermark = self.api.get_config("picture.default_watermark", True)
|
||||
guidance_scale = self.api.get_config("picture.default_guidance_scale", 2.5)
|
||||
@@ -160,7 +153,7 @@ class TakePictureAction(BaseAction):
|
||||
# 缓存失败,清除这个缓存项并继续正常流程
|
||||
del self._request_cache[cache_key]
|
||||
|
||||
await self.send_text(f"正在为你拍照,请稍候...")
|
||||
await self.send_text("正在为你拍照,请稍候...")
|
||||
|
||||
try:
|
||||
seed = random.randint(1, 1000000)
|
||||
@@ -230,7 +223,7 @@ class TakePictureAction(BaseAction):
|
||||
async with file_lock:
|
||||
try:
|
||||
if os.path.exists(log_path):
|
||||
with open(log_path, 'r', encoding='utf-8') as f:
|
||||
with open(log_path, "r", encoding="utf-8") as f:
|
||||
log_data = json.load(f)
|
||||
else:
|
||||
log_data = []
|
||||
@@ -238,18 +231,16 @@ class TakePictureAction(BaseAction):
|
||||
log_data = []
|
||||
|
||||
# 添加新照片
|
||||
log_data.append({
|
||||
"prompt": prompt,
|
||||
"image_url": image_url,
|
||||
"timestamp": datetime.datetime.now().isoformat()
|
||||
})
|
||||
log_data.append(
|
||||
{"prompt": prompt, "image_url": image_url, "timestamp": datetime.datetime.now().isoformat()}
|
||||
)
|
||||
|
||||
# 如果超过最大数量,删除最旧的
|
||||
if len(log_data) > max_photos:
|
||||
log_data = sorted(log_data, key=lambda x: x.get('timestamp', ''), reverse=True)[:max_photos]
|
||||
log_data = sorted(log_data, key=lambda x: x.get("timestamp", ""), reverse=True)[:max_photos]
|
||||
|
||||
try:
|
||||
with open(log_path, 'w', encoding='utf-8') as f:
|
||||
with open(log_path, "w", encoding="utf-8") as f:
|
||||
json.dump(log_data, f, ensure_ascii=False, indent=4)
|
||||
except Exception as e:
|
||||
logger.error(f"{self.log_prefix} 写入照片日志文件失败: {e}", exc_info=True)
|
||||
@@ -325,7 +316,7 @@ class TakePictureAction(BaseAction):
|
||||
with urllib.request.urlopen(image_url) as response:
|
||||
image_data = response.read()
|
||||
|
||||
base64_encoded = base64.b64encode(image_data).decode('utf-8')
|
||||
base64_encoded = base64.b64encode(image_data).decode("utf-8")
|
||||
return True, base64_encoded
|
||||
except Exception as e:
|
||||
logger.error(f"图片下载编码失败: {e}", exc_info=True)
|
||||
@@ -404,7 +395,7 @@ class ShowRecentPicturesCommand(BaseCommand):
|
||||
await self.send_text("最近还没有拍过照片哦,快让我自拍一张吧!")
|
||||
return True, "没有照片日志文件"
|
||||
|
||||
with open(log_path, 'r', encoding='utf-8') as f:
|
||||
with open(log_path, "r", encoding="utf-8") as f:
|
||||
log_data = json.load(f)
|
||||
|
||||
if not log_data:
|
||||
@@ -412,7 +403,7 @@ class ShowRecentPicturesCommand(BaseCommand):
|
||||
return True, "没有照片"
|
||||
|
||||
# 获取最新的5张照片
|
||||
recent_pics = sorted(log_data, key=lambda x: x['timestamp'], reverse=True)[:5]
|
||||
recent_pics = sorted(log_data, key=lambda x: x["timestamp"], reverse=True)[:5]
|
||||
|
||||
# 先发送文本消息
|
||||
await self.send_text("这是我最近拍的几张照片~")
|
||||
@@ -420,19 +411,17 @@ class ShowRecentPicturesCommand(BaseCommand):
|
||||
# 逐个发送图片
|
||||
for pic in recent_pics:
|
||||
# 尝试获取图片URL
|
||||
image_url = pic.get('image_url')
|
||||
image_url = pic.get("image_url")
|
||||
if image_url:
|
||||
try:
|
||||
# 下载图片并转换为Base64
|
||||
with urllib.request.urlopen(image_url) as response:
|
||||
image_data = response.read()
|
||||
base64_encoded = base64.b64encode(image_data).decode('utf-8')
|
||||
base64_encoded = base64.b64encode(image_data).decode("utf-8")
|
||||
|
||||
# 发送图片
|
||||
await self.send_type(
|
||||
message_type="image",
|
||||
content=base64_encoded,
|
||||
display_message="发送最近的照片"
|
||||
message_type="image", content=base64_encoded, display_message="发送最近的照片"
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error(f"{self.log_prefix} 下载或发送照片失败: {e}", exc_info=True)
|
||||
@@ -451,6 +440,7 @@ class ShowRecentPicturesCommand(BaseCommand):
|
||||
@register_plugin
|
||||
class TakePicturePlugin(BasePlugin):
|
||||
"""拍照插件"""
|
||||
|
||||
plugin_name = "take_picture_plugin"
|
||||
plugin_description = "提供生成自拍照和展示最近照片的功能"
|
||||
plugin_version = "1.0.0"
|
||||
@@ -473,7 +463,9 @@ class TakePicturePlugin(BasePlugin):
|
||||
"name": ConfigField(type=str, default="take_picture_plugin", description="插件名称", required=True),
|
||||
"version": ConfigField(type=str, default="1.3.0", description="插件版本号"),
|
||||
"enabled": ConfigField(type=bool, default=False, description="是否启用插件"),
|
||||
"description": ConfigField(type=str, default="提供生成自拍照和展示最近照片的功能", description="插件描述", required=True),
|
||||
"description": ConfigField(
|
||||
type=str, default="提供生成自拍照和展示最近照片的功能", description="插件描述", required=True
|
||||
),
|
||||
},
|
||||
"api": {
|
||||
"base_url": ConfigField(
|
||||
@@ -511,9 +503,7 @@ class TakePicturePlugin(BasePlugin):
|
||||
),
|
||||
"default_seed": ConfigField(type=int, default=42, description="随机种子,用于复现图片"),
|
||||
"prompt_templates": ConfigField(
|
||||
type=list,
|
||||
default=TakePictureAction.DEFAULT_PROMPT_TEMPLATES,
|
||||
description="用于生成自拍照的prompt模板"
|
||||
type=list, default=TakePictureAction.DEFAULT_PROMPT_TEMPLATES, description="用于生成自拍照的prompt模板"
|
||||
),
|
||||
},
|
||||
"storage": {
|
||||
@@ -521,7 +511,7 @@ class TakePicturePlugin(BasePlugin):
|
||||
"log_file": ConfigField(type=str, default="picture_log.json", description="照片日志文件名"),
|
||||
"enable_cache": ConfigField(type=bool, default=True, description="是否启用请求缓存"),
|
||||
"max_cache_size": ConfigField(type=int, default=10, description="最大缓存数量"),
|
||||
}
|
||||
},
|
||||
}
|
||||
|
||||
def get_plugin_components(self) -> List[Tuple[ComponentInfo, Type]]:
|
||||
|
||||
Reference in New Issue
Block a user