Skip to content

fix: simplify API version detection and streamline parameter handling in ttv.py #6818

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
wxg0103 merged 1 commit into from
Aug 31, 2026
Merged

fix: simplify API version detection and streamline parameter handling in ttv.py #6818

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 3 additions & 11 deletions apps/models_provider/impl/minimax_model_provider/model/ttv.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -28,8 +28,6 @@ def __init__(self, **kwargs):
self.params = kwargs.get("params", {}) or {}
self.max_retries = kwargs.get("max_retries", 3)
self.retry_delay = kwargs.get("retry_delay", 10)
# 显式参数可覆盖自动探测（params.api_version: 'v1' / 'v2'）
self.api_version = self.params.get("api_version", "auto")

@staticmethod
def is_cache_model():
Expand Down Expand Up @@ -57,13 +55,7 @@ def check_auth(self):
# ---------- API 版本探测 / URL 构建 ----------

def _detect_api_version(self) -> str:
"""探测当前使用 V1 还是 V2 (MiniMax-H3)。"""
if self.api_version in ("v1", "v2"):
return self.api_version
# 模型名包含 H3 -> V2
if self.model_name and "H3" in self.model_name.upper():
return "v2"
# api_base 路径包含 /v2 -> V2
"""根据 api_base 路径判断 V1 还是 V2 (MiniMax-H3)。"""
base_path = self.api_base.split("://", 1)[-1] if "://" in self.api_base else self.api_base
if "/v2" in base_path:
return "v2"
Expand Down Expand Up @@ -228,8 +220,8 @@ def _generate_video_v1(self, prompt, first_frame_url=None, last_frame_url=None,
else:
maxkb_logger.info("使用文生视频模式")

# 合并额外参数（duration, resolution 等），跳过版本探测专用字段
payload.update({k: v for k, v in self.params.items() if k != "api_version"})
# 合并额外参数（duration, resolution 等）
payload.update(self.params)

# --- 步骤 1: 提交任务 ---
maxkb_logger.info(f"提交视频生成任务，模型: {self.model_name}")
Expand Down
Loading