Expose individual prompt reference descriptions

This commit is contained in:
2026-09-10 07:59:11 +00:00
parent e973a0d785
commit eac4e73b47
3 changed files with 30 additions and 5 deletions
+10 -1
View File
@@ -510,6 +510,11 @@ class DumasImageNodeTests(unittest.TestCase):
self.assertIn("<Picture 1> Dave", result[24])
self.assertIn("<Picture 2> Coffee Shop", result[24])
self.assertIn("Location context for <Picture 2> Coffee Shop", result[24])
self.assertIn("<Picture 1> Dave", result[25])
self.assertNotIn("<Picture 2> Coffee Shop", result[25])
self.assertIn("<Picture 2> Coffee Shop", result[26])
self.assertIn("Location context for <Picture 2> Coffee Shop", result[26])
self.assertEqual(result[27], "")
def test_h3_prompt_curator_renumbers_explicit_reference_tags(self):
node = self.image_nodes.DumasH3PromptCuratorNode()
@@ -647,11 +652,15 @@ class DumasImageNodeTests(unittest.TestCase):
def test_h3_prompt_curator_uses_documented_reference_limits(self):
node = self.image_nodes.DumasH3PromptCuratorNode()
self.assertEqual(len(node.RETURN_TYPES), 25)
self.assertEqual(len(node.RETURN_TYPES), 34)
self.assertEqual(node.RETURN_NAMES[1:10], tuple(f"ref_image_{i}" for i in range(1, 10)))
self.assertEqual(node.RETURN_NAMES[12:15], ("anchor", "sounds", "bgm"))
self.assertEqual(node.RETURN_NAMES[15:24], tuple(f"original_ref_{i}" for i in range(1, 10)))
self.assertEqual(node.RETURN_NAMES[24], "reference_description")
self.assertEqual(
node.RETURN_NAMES[25:34],
tuple(f"original_ref_description_{i}" for i in range(1, 10)),
)
def test_normalize_reference_upgrades_generic_summary_with_socket_picture_id(self):
image = FakeTensorBatch()