protect fix

backend/classes/galileo_platform.py

@@ -1,4 +1,4 @@
-from pydantic import BaseModel, UUID4
+from pydantic import BaseModel
 from dotenv import load_dotenv
 from typing import Any
 
@@ -45,7 +45,7 @@ class GalileoPlatform:
             print(f"Failed to create logger: {e}")
             return None
 
-    def get_protect_stage_id(self) -> str | UUID4 | None:
+    def get_protect_stage_id(self) -> str | None:
         """Get or create a Galileo Protect stage."""
         if not get_project(name=self.config.protect_project_name):
             _ = create_project(name=self.config.protect_project_name)
@@ -61,6 +61,7 @@ class GalileoPlatform:
                 name=self.config.protect_stage_name,
                 stage_type=StageType.local
             )
+            print(f"Protect stage {self.config.protect_stage_name} created in project {self.config.protect_project_name}")
             protect_stage = get_protect_stage(
                 project_name=self.config.protect_project_name,
                 stage_name=self.config.protect_stage_name,