330 lines
11 KiB
Python
330 lines
11 KiB
Python
"""
|
||
Preset Service
|
||
预设服务层 - 处理预设的 CRUD 操作
|
||
"""
|
||
import json
|
||
import os
|
||
from pathlib import Path
|
||
from typing import List, Dict, Any, Optional
|
||
from datetime import datetime
|
||
|
||
from core.config import settings
|
||
|
||
|
||
class PresetService:
|
||
"""预设服务类"""
|
||
|
||
@staticmethod
|
||
def _extract_preset_name_from_filename(filename: str) -> str:
|
||
"""
|
||
从文件名提取预设名称，去掉时间戳和文件后缀
|
||
|
||
Args:
|
||
filename: 文件名（不含路径）
|
||
|
||
Returns:
|
||
清理后的预设名称
|
||
|
||
Examples:
|
||
"Default.json" -> "Default"
|
||
"MyPreset_1234567890.json" -> "MyPreset"
|
||
"Test_1714567890123.json" -> "Test"
|
||
"""
|
||
# 去掉 .json 后缀
|
||
name = filename.replace('.json', '')
|
||
|
||
# 去掉末尾的时间戳（下划线+数字组合）
|
||
# 匹配模式：_后面跟着10-13位数字（Unix时间戳）
|
||
import re
|
||
name = re.sub(r'_\d{10,13}$', '', name)
|
||
|
||
return name
|
||
|
||
@staticmethod
|
||
def _get_preset_path(name: str) -> Path:
|
||
"""获取预设文件路径"""
|
||
return settings.PRESET_PATH / f"{name}.json"
|
||
|
||
@staticmethod
|
||
def _load_preset(name: str) -> Optional[Dict[str, Any]]:
|
||
"""加载预设 JSON 文件"""
|
||
path = PresetService._get_preset_path(name)
|
||
if not path.exists():
|
||
return None
|
||
|
||
try:
|
||
with open(path, 'r', encoding='utf-8') as f:
|
||
return json.load(f)
|
||
except Exception as e:
|
||
raise ValueError(f"Failed to load preset '{name}': {str(e)}")
|
||
|
||
@staticmethod
|
||
def _save_preset(name: str, data: Dict[str, Any]):
|
||
"""保存预设到 JSON 文件"""
|
||
path = PresetService._get_preset_path(name)
|
||
try:
|
||
# 确保 prompts 数组和 prompt_order 的顺序一致
|
||
if "prompts" in data and "prompt_order" in data:
|
||
prompts = data["prompts"]
|
||
prompt_order = data.get("prompt_order", [{}])[0].get("order", [])
|
||
|
||
if prompts and prompt_order:
|
||
# 创建 identifier 到 prompt 的映射
|
||
prompt_map = {prompt["identifier"]: prompt for prompt in prompts}
|
||
|
||
# 按照 prompt_order 的顺序重新排列 prompts
|
||
reordered_prompts = []
|
||
for order_item in prompt_order:
|
||
identifier = order_item.get("identifier")
|
||
if identifier and identifier in prompt_map:
|
||
reordered_prompts.append(prompt_map[identifier])
|
||
|
||
# 添加 prompt_order 中不存在的 prompts（如果有）
|
||
existing_identifiers = {item.get("identifier") for item in prompt_order}
|
||
for prompt in prompts:
|
||
if prompt["identifier"] not in existing_identifiers:
|
||
reordered_prompts.append(prompt)
|
||
|
||
data["prompts"] = reordered_prompts
|
||
|
||
with open(path, 'w', encoding='utf-8') as f:
|
||
json.dump(data, f, ensure_ascii=False, indent=2)
|
||
except Exception as e:
|
||
raise ValueError(f"Failed to save preset '{name}': {str(e)}")
|
||
|
||
@staticmethod
|
||
def list_presets() -> List[Dict[str, Any]]:
|
||
"""
|
||
获取所有预设的列表（仅基本信息）
|
||
|
||
Returns:
|
||
预设列表，每个包含 name, description, component_count, temperature 等
|
||
"""
|
||
presets = []
|
||
|
||
for json_file in settings.PRESET_PATH.glob("*.json"):
|
||
try:
|
||
with open(json_file, 'r', encoding='utf-8') as f:
|
||
data = json.load(f)
|
||
|
||
# 计算组件数量 - 支持 SillyTavern 格式 (prompts) 和内部格式 (entries)
|
||
prompts = data.get("prompts", [])
|
||
entries = data.get("entries", [])
|
||
component_count = len(prompts) if prompts else len(entries)
|
||
|
||
# 提取温度参数 - 使用 SillyTavern 标准字段名
|
||
temperature = data.get("temperature", 1.0)
|
||
|
||
# 从文件名提取预设名称（去掉时间戳和后缀）
|
||
preset_name = PresetService._extract_preset_name_from_filename(json_file.name)
|
||
|
||
preset_info = {
|
||
"name": preset_name,
|
||
"description": data.get("description", ""),
|
||
"component_count": component_count,
|
||
"temperature": temperature
|
||
}
|
||
|
||
presets.append(preset_info)
|
||
except Exception as e:
|
||
print(f"Error loading preset {json_file.name}: {e}")
|
||
continue
|
||
|
||
# 按名称排序
|
||
presets.sort(key=lambda x: x.get("name", ""))
|
||
return presets
|
||
|
||
@staticmethod
|
||
def get_preset(name: str) -> Dict[str, Any]:
|
||
"""
|
||
获取指定预设的完整数据
|
||
|
||
Args:
|
||
name: 预设名称
|
||
|
||
Returns:
|
||
预设完整数据
|
||
"""
|
||
data = PresetService._load_preset(name)
|
||
if not data:
|
||
raise FileNotFoundError(f"Preset '{name}' not found")
|
||
|
||
return data
|
||
|
||
@staticmethod
|
||
def create_preset(name: str, preset_data: Dict[str, Any]) -> Dict[str, Any]:
|
||
"""
|
||
创建新预设
|
||
|
||
Args:
|
||
name: 预设名称
|
||
preset_data: 预设数据
|
||
|
||
Returns:
|
||
创建的预设数据
|
||
"""
|
||
# 检查是否已存在
|
||
if PresetService._get_preset_path(name).exists():
|
||
raise ValueError(f"Preset '{name}' already exists")
|
||
|
||
# 确保有必要的字段
|
||
if "name" not in preset_data:
|
||
preset_data["name"] = name
|
||
|
||
# 添加时间戳
|
||
now = int(datetime.now().timestamp())
|
||
if "createdAt" not in preset_data:
|
||
preset_data["createdAt"] = now
|
||
if "updatedAt" not in preset_data:
|
||
preset_data["updatedAt"] = now
|
||
|
||
PresetService._save_preset(name, preset_data)
|
||
return preset_data
|
||
|
||
@staticmethod
|
||
def update_preset(name: str, update_data: Dict[str, Any]) -> Dict[str, Any]:
|
||
"""
|
||
更新预设
|
||
|
||
Args:
|
||
name: 预设名称
|
||
update_data: 要更新的数据
|
||
|
||
Returns:
|
||
更新后的预设数据
|
||
"""
|
||
data = PresetService._load_preset(name)
|
||
if not data:
|
||
raise FileNotFoundError(f"Preset '{name}' not found")
|
||
|
||
# 更新字段
|
||
for key, value in update_data.items():
|
||
if key not in ["name", "createdAt"]: # 不允许修改名称和创建时间
|
||
data[key] = value
|
||
|
||
# 更新时间戳
|
||
data["updatedAt"] = int(datetime.now().timestamp())
|
||
|
||
PresetService._save_preset(name, data)
|
||
return data
|
||
|
||
@staticmethod
|
||
def delete_preset(name: str) -> bool:
|
||
"""
|
||
删除预设
|
||
|
||
Args:
|
||
name: 预设名称
|
||
|
||
Returns:
|
||
是否删除成功
|
||
"""
|
||
path = PresetService._get_preset_path(name)
|
||
if not path.exists():
|
||
raise FileNotFoundError(f"Preset '{name}' not found")
|
||
|
||
path.unlink()
|
||
return True
|
||
|
||
@staticmethod
|
||
def rename_preset(old_name: str, new_name: str) -> Dict[str, Any]:
|
||
"""
|
||
重命名预设（同时修改文件名和内部 name 字段）
|
||
|
||
Args:
|
||
old_name: 原预设名称
|
||
new_name: 新预设名称
|
||
|
||
Returns:
|
||
更新后的预设数据
|
||
"""
|
||
# 检查原预设是否存在
|
||
old_path = PresetService._get_preset_path(old_name)
|
||
if not old_path.exists():
|
||
raise FileNotFoundError(f"Preset '{old_name}' not found")
|
||
|
||
# 检查新名称是否已存在
|
||
new_path = PresetService._get_preset_path(new_name)
|
||
if new_path.exists() and old_name != new_name:
|
||
raise ValueError(f"Preset '{new_name}' already exists")
|
||
|
||
# 加载原预设数据
|
||
data = PresetService._load_preset(old_name)
|
||
if not data:
|
||
raise FileNotFoundError(f"Preset '{old_name}' not found")
|
||
|
||
# 更新内部的 name 字段
|
||
data["name"] = new_name
|
||
|
||
# 更新时间戳
|
||
data["updatedAt"] = int(datetime.now().timestamp())
|
||
|
||
# 保存到新文件
|
||
PresetService._save_preset(new_name, data)
|
||
|
||
# 删除旧文件（如果名称不同）
|
||
if old_name != new_name:
|
||
old_path.unlink()
|
||
|
||
return data
|
||
|
||
@staticmethod
|
||
def reorder_components(name: str, component_order: List[str]) -> Dict[str, Any]:
|
||
"""
|
||
重新排序预设组件 - 支持 SillyTavern 标准格式
|
||
|
||
Args:
|
||
name: 预设名称
|
||
component_order: 组件 identifier 列表，按新顺序排列
|
||
|
||
Returns:
|
||
更新后的预设数据
|
||
"""
|
||
data = PresetService._load_preset(name)
|
||
if not data:
|
||
raise FileNotFoundError(f"Preset '{name}' not found")
|
||
|
||
# 支持 SillyTavern 格式的 prompts
|
||
if "prompts" in data and isinstance(data["prompts"], list):
|
||
# 创建 identifier 到 prompt 的映射
|
||
prompt_map = {prompt["identifier"]: prompt for prompt in data["prompts"]}
|
||
|
||
# 按新顺序重新排列
|
||
reordered_prompts = []
|
||
for identifier in component_order:
|
||
if identifier in prompt_map:
|
||
reordered_prompts.append(prompt_map[identifier])
|
||
|
||
data["prompts"] = reordered_prompts
|
||
|
||
# 更新 prompt_order
|
||
if "prompt_order" in data and isinstance(data["prompt_order"], list) and len(data["prompt_order"]) > 0:
|
||
data["prompt_order"][0]["order"] = [
|
||
{"identifier": identifier, "enabled": True}
|
||
for identifier in component_order
|
||
if identifier in prompt_map
|
||
]
|
||
|
||
# 也支持内部格式的 entries（向后兼容）
|
||
elif "entries" in data and isinstance(data["entries"], list):
|
||
# 创建 identifier 到 entry 的映射
|
||
entry_map = {entry["identifier"]: entry for entry in data["entries"]}
|
||
|
||
# 按新顺序重新排列
|
||
reordered_entries = []
|
||
for identifier in component_order:
|
||
if identifier in entry_map:
|
||
reordered_entries.append(entry_map[identifier])
|
||
|
||
# 更新 order 字段
|
||
for index, entry in enumerate(reordered_entries):
|
||
entry["order"] = index
|
||
|
||
data["entries"] = reordered_entries
|
||
|
||
# 更新时间戳
|
||
data["updatedAt"] = int(datetime.now().timestamp())
|
||
|
||
PresetService._save_preset(name, data)
|
||
return data
|