chore: import upstream snapshot with attribution

2026-07-13 12:23:54 +08:00
commit ecb5ae4e59
153 changed files with 21551 additions and 0 deletions
@@ -0,0 +1,62 @@
+import tempfile
+import unittest
+from pathlib import Path
+
+from interfaces import Camera, ShotBriefDescription, ShotDescription
+from pipelines.script2video_pipeline import Script2VideoPipeline, _group_shots_into_cameras
+
+
+class FlakyCameraImageGenerator:
+    def __init__(self):
+        self.calls = 0
+
+    async def construct_camera_tree(self, cameras, shot_descs):
+        self.calls += 1
+        if self.calls == 1:
+            return ["not-a-camera"]
+        return cameras
+
+
+class Script2VideoPipelineGuardTests(unittest.IsolatedAsyncioTestCase):
+    def test_group_shots_into_cameras_does_not_use_camera_idx_as_list_index(self):
+        shots = [
+            ShotDescription(idx=0, is_last=False, cam_idx=2, visual_desc="a", variation_type="small", variation_reason="same", ff_desc="a", ff_vis_char_idxs=[], lf_desc="a", lf_vis_char_idxs=[], motion_desc="a", audio_desc="none"),
+            ShotDescription(idx=1, is_last=True, cam_idx=5, visual_desc="b", variation_type="small", variation_reason="same", ff_desc="b", ff_vis_char_idxs=[], lf_desc="b", lf_vis_char_idxs=[], motion_desc="b", audio_desc="none"),
+            ShotDescription(idx=2, is_last=True, cam_idx=2, visual_desc="c", variation_type="small", variation_reason="same", ff_desc="c", ff_vis_char_idxs=[], lf_desc="c", lf_vis_char_idxs=[], motion_desc="c", audio_desc="none"),
+        ]
+        cameras = _group_shots_into_cameras(shots)
+        self.assertEqual([camera.idx for camera in cameras], [2, 5])
+        self.assertEqual(cameras[0].active_shot_idxs, [0, 2])
+        self.assertEqual(cameras[1].active_shot_idxs, [1])
+
+    async def test_plan_text_artifacts_retries_bad_camera_tree_schema(self):
+        with tempfile.TemporaryDirectory() as tmp:
+            pipeline = Script2VideoPipeline(chat_model=object(), image_generator=object(), video_generator=object(), working_dir=tmp)
+            pipeline.camera_image_generator = FlakyCameraImageGenerator()
+
+            async def design_storyboard(script, characters, user_requirement, quiet=False):
+                return [{"idx": 0, "is_last": True, "cam_idx": 3, "visual_desc": "wide shot", "audio_desc": "waves"}]
+
+            async def decompose_visual_descriptions(shot_brief_descriptions, characters, quiet=False):
+                return [{"idx": 0, "is_last": True, "cam_idx": 3, "visual_desc": "wide shot", "variation_type": "small", "variation_reason": "simple", "ff_desc": "start", "ff_vis_char_idxs": [], "lf_desc": "end", "lf_vis_char_idxs": [], "motion_desc": "walk", "audio_desc": "waves"}]
+
+            pipeline.design_storyboard = design_storyboard
+            pipeline.decompose_visual_descriptions = decompose_visual_descriptions
+            events = []
+            result = await pipeline.plan_text_artifacts(
+                "script",
+                "req",
+                "style",
+                characters=[{"idx": 0, "identifier_in_scene": "Man", "is_visible": True, "static_features": "adult", "dynamic_features": "coat"}],
+                progress=lambda stage, message, metadata=None: events.append(stage),
+                quiet=True,
+            )
+
+            self.assertEqual(pipeline.camera_image_generator.calls, 2)
+            self.assertIn("construct_camera_tree_retry", events)
+            self.assertEqual(result["camera_tree"][0].idx, 3)
+            self.assertTrue((Path(tmp) / "camera_tree.json").exists())
+
+
+if __name__ == "__main__":
+    unittest.main()