diff --git a/apps/api/app/api/routes/viral_video.py b/apps/api/app/api/routes/viral_video.py index aa5e09201..3ddcb8e90 100644 --- a/apps/api/app/api/routes/viral_video.py +++ b/apps/api/app/api/routes/viral_video.py @@ -49,8 +49,8 @@ from packages.adapters.sqlalchemy_impl.viral_video_repository import ( SQLAlchemyViralVideoJobRepository, SQLAlchemyViralVideoStyleTemplateRepository, ) -from packages.domain.viral_video import ViralVideoStatus from packages.domain.points_rules import list_viral_video_models +from packages.domain.viral_video import ViralVideoStatus from packages.shared.dashscope_client import get_dashscope_client logger = logging.getLogger(__name__) diff --git a/apps/worker/worker_app/tasks/viral_video.py b/apps/worker/worker_app/tasks/viral_video.py index 0540b6164..c581edba0 100644 --- a/apps/worker/worker_app/tasks/viral_video.py +++ b/apps/worker/worker_app/tasks/viral_video.py @@ -1126,8 +1126,8 @@ def _step_render(job: ViralVideoJob, copy_result: dict, tts_audio_url: str | Non 返回 (本地视频路径, usage dict|None)。失败抛异常。 """ - from packages.shared.ai_service import call_video_generation from packages.domain.points_rules import get_viral_video_model_config + from packages.shared.ai_service import call_video_generation prompt = _assemble_seedance_prompt(copy_result, job) dur = max(5, min(30, int(getattr(job, "duration", 15) or 15))) diff --git a/tests/unit/test_dashscope_client.py b/tests/unit/test_dashscope_client.py index 500f1a8ed..82c751a4f 100644 --- a/tests/unit/test_dashscope_client.py +++ b/tests/unit/test_dashscope_client.py @@ -2,7 +2,7 @@ from __future__ import annotations -from unittest.mock import MagicMock, patch, mock_open +from unittest.mock import MagicMock, mock_open, patch import pytest