From a53576c98bcffffca5474451a8ca5de4faf92943 Mon Sep 17 00:00:00 2001
From: Pedro gengo <pedro.gengo.lourenco@gmail.com>
Date: Mon, 24 Mar 2025 21:33:24 -0300
Subject: [PATCH] Exposing new nodes

---
 py/luma_nodes.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/py/luma_nodes.py b/py/luma_nodes.py
index f41633f..98bbef6 100644
--- a/py/luma_nodes.py
+++ b/py/luma_nodes.py
@@ -10,7 +10,8 @@ from .lumaai_api_node import (
     CharacterReference,
     ImageGeneration,
     ModifyImage,
-    Ray2Text2Video,
+    AddAudio2Video,
+    UpscaleGeneration,
 )
 from .imgbb_node import ImgBBUpload
 
@@ -27,7 +28,8 @@ NODE_CLASS_MAPPINGS = {
     "CharacterReference": CharacterReference,
     "LumaImageGeneration": ImageGeneration,
     "LumaModifyImage": ModifyImage,
-    "Ray2Text2Video": Ray2Text2Video,
+    "LumaAddAudio2Video": AddAudio2Video,
+    "LumaUpscaleGeneration": UpscaleGeneration,
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
@@ -43,5 +45,6 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "LumaCharacterReference": "Character Reference",
     "LumaImageGeneration": "Image Generation",
     "LumaModifyImage": "Modify Image",
-    "Ray2Text2Video": "Ray2 Text to Video",
+    "LumaAddAudio2Video": "Add Audio to Video",
+    "LumaUpscaleGeneration": "Upscale Generation",
 }