小鱼开发
1a0679049e
Replace menu list (使用明细 + 设置) with recent transactions table: - Add back recentTx state and loading state - Fetch last 5 transactions in loadData - Display table with type/amount/description/time columns - Add '查看全部' link to usage-detail page - Remove unused icon components (FileTextIcon, SettingsIcon, ChevronRightIcon)
84 lines
2.1 KiB
Python
84 lines
2.1 KiB
Python
"""
|
|
火山引擎 MediaKit Service
|
|
=========================
|
|
|
|
通过 PlatformGateway 调用 MediaKit 第三方 API，自身负责：
|
|
- 参数校验
|
|
- 结果提取与格式化
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from dataclasses import dataclass, field
|
|
from typing import Any
|
|
|
|
from app.ai.adapters.constants import Method
|
|
from app.platform_gateway import PlatformGateway
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@dataclass
|
|
class RemoveBackgroundResult:
|
|
"""抠图结果"""
|
|
|
|
image_url: str = ""
|
|
raw: dict[str, Any] = field(default_factory=dict)
|
|
|
|
|
|
class VolcengineMediakitService:
|
|
"""火山引擎 MediaKit 服务封装"""
|
|
|
|
# 支持的场景
|
|
SUPPORTED_SCENES = {"general", "product"}
|
|
|
|
def __init__(self, gateway: PlatformGateway):
|
|
self.gateway = gateway
|
|
|
|
async def remove_background(
|
|
self,
|
|
image_url: str,
|
|
scene: str = "general",
|
|
) -> RemoveBackgroundResult:
|
|
"""同步抠图
|
|
|
|
Args:
|
|
image_url: 原始图片 URL
|
|
scene: 场景类型，"general"（通用）或 "product"（商品）
|
|
|
|
Returns:
|
|
RemoveBackgroundResult: 包含抠图结果图片 URL
|
|
|
|
Raises:
|
|
ValueError: 参数校验失败
|
|
PlatformError: 平台调用失败
|
|
"""
|
|
if not image_url:
|
|
raise ValueError("image_url 不能为空")
|
|
|
|
if scene not in self.SUPPORTED_SCENES:
|
|
raise ValueError(f"不支持的场景: {scene}，可选: {self.SUPPORTED_SCENES}")
|
|
|
|
payload = {
|
|
"image_url": image_url,
|
|
"scene": scene,
|
|
}
|
|
|
|
response = await self.gateway.call_sync(
|
|
platform="volcengine_mediakit",
|
|
method=Method.REMOVE_BACKGROUND,
|
|
payload=payload,
|
|
)
|
|
|
|
if not response.success:
|
|
raise RuntimeError(
|
|
response.error_message or "抠图失败"
|
|
)
|
|
|
|
result_image_url = (response.data or {}).get("image_url", "")
|
|
return RemoveBackgroundResult(
|
|
image_url=result_image_url,
|
|
raw=response.data or {},
|
|
)
|