fix: randomize tuhui designer alias

This commit is contained in:
2026-03-10 14:35:31 +08:00
parent c399b8cfc1
commit 3f45a4badd

View File

@@ -4,6 +4,7 @@ import json
import logging import logging
import mimetypes import mimetypes
import os import os
import random
import re import re
from pathlib import Path from pathlib import Path
from typing import Dict, List, Optional from typing import Dict, List, Optional
@@ -24,7 +25,6 @@ logger = logging.getLogger("cs_agent")
AUTO_PROCESS_PRICE = int(os.getenv("AUTO_PROCESS_DEFAULT_PRICE", "12")) AUTO_PROCESS_PRICE = int(os.getenv("AUTO_PROCESS_DEFAULT_PRICE", "12"))
AUTO_PROCESS_CATEGORY = os.getenv("AUTO_PROCESS_CATEGORY", "设计素材") AUTO_PROCESS_CATEGORY = os.getenv("AUTO_PROCESS_CATEGORY", "设计素材")
AUTO_PROCESS_DESIGNER_ALIAS = os.getenv("AUTO_PROCESS_DESIGNER_ALIAS", "图绘平台")
AUTO_PROCESS_ROOT = Path( AUTO_PROCESS_ROOT = Path(
os.getenv("AUTO_PROCESS_ROOT", str(Path(__file__).resolve().parents[1] / "runtime" / "auto_processed")) os.getenv("AUTO_PROCESS_ROOT", str(Path(__file__).resolve().parents[1] / "runtime" / "auto_processed"))
) )
@@ -52,6 +52,8 @@ _CONTENT_TYPE_SUFFIX = {
"image/avif": ".avif", "image/avif": ".avif",
"image/gif": ".gif", "image/gif": ".gif",
} }
_DESIGNER_ALIAS_PREFIXES = ("青木", "星野", "白川", "南栀", "言川", "木也", "安可", "拾光", "云岸", "知禾")
_DESIGNER_ALIAS_SUFFIXES = ("设计", "studio", "视觉", "创意", "图像", "工坊", "素材", "像素")
def _safe_name(text: str, fallback: str = "image") -> str: def _safe_name(text: str, fallback: str = "image") -> str:
@@ -95,6 +97,10 @@ def _build_upload_title(intent: str, analysis: Dict, requirement_text: str, idx:
return base return base
def _build_designer_alias() -> str:
return f"{random.choice(_DESIGNER_ALIAS_PREFIXES)}{random.choice(_DESIGNER_ALIAS_SUFFIXES)}"
class AutoImagePipelineService: class AutoImagePipelineService:
def __init__(self): def __init__(self):
self.customer_db = CustomerDatabase() self.customer_db = CustomerDatabase()
@@ -300,7 +306,7 @@ class AutoImagePipelineService:
price=AUTO_PROCESS_PRICE, price=AUTO_PROCESS_PRICE,
category=AUTO_PROCESS_CATEGORY, category=AUTO_PROCESS_CATEGORY,
tags="AI处理,自动转接", tags="AI处理,自动转接",
designer_name=AUTO_PROCESS_DESIGNER_ALIAS, designer_name=_build_designer_alias(),
) )
if not upload_result.success: if not upload_result.success:
if task_id: if task_id: