小鱼开发
d18e705a99
- provider: 增加 need_contour/contour_color/contour_size/need_crop_background 参数 - service: 默认 scene=human，human/product 场景自动启用白色描边 + 裁剪背景 - adapter: 透传新参数到 provider - API: scene 默认值改为 human - 前端: removeBackground 默认 scene 改为 human
96 lines
3.2 KiB
Python
96 lines
3.2 KiB
Python
"""
|
|
火山引擎 MediaKit Adapter
|
|
==========================
|
|
|
|
实现 PlatformAdapter + SyncCapable。
|
|
直接接入 VolcengineMediakitProvider，提供标准 Protocol 接口。
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from typing import Any
|
|
|
|
from app.ai.adapters.base import AdapterResponse, PlatformAdapter, SyncCapable
|
|
from app.ai.adapters.constants import Method
|
|
from app.ai.providers.volcengine_mediakit_provider import VolcengineMediakitProvider
|
|
from app.core.exceptions import PlatformError, PlatformErrorType
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class VolcengineMediakitAdapter(PlatformAdapter, SyncCapable):
|
|
"""火山引擎 MediaKit 平台标准 Adapter"""
|
|
|
|
platform_id = "volcengine_mediakit"
|
|
|
|
def __init__(self, provider: VolcengineMediakitProvider):
|
|
self.provider = provider
|
|
|
|
# ── PlatformAdapter ──
|
|
|
|
async def health(self) -> AdapterResponse:
|
|
try:
|
|
# 用无效 URL 测试连通性（400 说明网络通且认证通过）
|
|
await self.provider.remove_background(
|
|
image_url="https://example.com/health-check.jpg",
|
|
scene="general",
|
|
)
|
|
return AdapterResponse(success=True)
|
|
except PlatformError as e:
|
|
if e.error_type in (
|
|
PlatformErrorType.AUTH_FAILED,
|
|
PlatformErrorType.BAD_REQUEST,
|
|
):
|
|
return AdapterResponse(success=True)
|
|
return AdapterResponse(
|
|
success=False,
|
|
error_message=str(e),
|
|
retryable=e.retryable,
|
|
)
|
|
except Exception as e:
|
|
return AdapterResponse(
|
|
success=False,
|
|
error_message=str(e),
|
|
retryable=False,
|
|
)
|
|
|
|
async def close(self) -> None:
|
|
await self.provider.close()
|
|
|
|
# ── SyncCapable ──
|
|
|
|
async def call(self, method: str, payload: dict[str, Any]) -> AdapterResponse:
|
|
try:
|
|
if method == Method.REMOVE_BACKGROUND:
|
|
result = await self.provider.remove_background(
|
|
image_url=payload["image_url"],
|
|
scene=payload.get("scene", "general"),
|
|
need_contour=payload.get("need_contour", False),
|
|
contour_color=payload.get("contour_color", "#FFFFFF"),
|
|
contour_size=payload.get("contour_size", 10),
|
|
need_crop_background=payload.get("need_crop_background", False),
|
|
)
|
|
data = result.get("data", {})
|
|
return AdapterResponse(
|
|
success=True,
|
|
data={"image_url": data.get("image_url")},
|
|
)
|
|
|
|
else:
|
|
return AdapterResponse(
|
|
success=False,
|
|
error_message=f"不支持的方法: {method}",
|
|
retryable=False,
|
|
)
|
|
|
|
except PlatformError:
|
|
raise
|
|
except Exception as e:
|
|
raise PlatformError(
|
|
f"MediaKit {method} 调用失败: {e}",
|
|
platform="volcengine_mediakit",
|
|
retryable=False,
|
|
error_type=PlatformErrorType.UNKNOWN,
|
|
) from e
|