Add handoff context frames

2026-09-10 11:44:18 +00:00
parent ea6fae7e56
commit 9923a3e417
3 changed files with 71 additions and 2 deletions
@@ -112,8 +112,17 @@ class DumasH3LongVideosUpstreamWrapperTests(unittest.TestCase):
        self.assertIn("first_frame", schema["optional"])
        self.assertIn("ref_image_1", schema["optional"])
        self.assertIn("latent_upscale", schema["optional"])
+        self.assertIn("handoff_frames", schema["optional"])
+        self.assertEqual(schema["optional"]["handoff_frames"][1]["default"], 1)
        self.assertEqual(node_cls.RETURN_NAMES[0:4], ("images", "audio", "info", "script"))

+    def test_handoff_context_claim_names_reference_range(self):
+        upstream = importlib.import_module("dumas_h3_longvideos_upstream")
+
+        self.assertIn("<Picture 2> through <Picture 22>", upstream.handoff_context_claim(2, 22))
+        self.assertIn("no new subjects", upstream.handoff_context_claim(2, 22))
+        self.assertIn("<Picture 5>", upstream.handoff_context_claim(5, 5))
+

 class _NullContext:
    def __enter__(self):