Made changes for making protect work with different projects

backend/api/main.py

@@ -82,7 +82,7 @@ default_dataset_name = env_variables["GALILEO_DATASET_NAME"]
 
 # Create Galileo platform object
 galileo_platform_config = GalileoPlatformConfig(
-    protect_project_name=env_variables["GALILEO_PROJECT_NAME"],
+    protect_project_name=env_variables["GALILEO_PROTECT_PROJECT_NAME"],
     protect_stage_name=env_variables["GALILEO_PROTECT_STAGE_NAME"],
 )
 galileo_platform = GalileoPlatform(galileo_platform_config)

backend/classes/galileo_platform.py

@@ -1,7 +1,6 @@
-from pydantic import BaseModel
+from pydantic import BaseModel, UUID4
 from dotenv import load_dotenv
 from typing import Any
-from pydantic import UUID4
 
 from galileo import GalileoLogger, GalileoScorers, StageType
 from galileo.log_streams import create_log_stream, get_log_stream
@@ -48,20 +47,18 @@ class GalileoPlatform:
 
     def get_protect_stage_id(self) -> str | UUID4 | None:
         """Get or create a Galileo Protect stage."""
-        try:
-            protect_stage = get_protect_stage(
-                project_name=self.config.protect_project_name,
-                stage_name=self.config.protect_stage_name,
-            )
-            return protect_stage.id
-        except Exception as _:
+
+        protect_stage = get_protect_stage(
+            project_name=self.config.protect_project_name,
+            stage_name=self.config.protect_stage_name,
+        )
+        if not protect_stage:
             protect_stage = create_protect_stage(
                 project_name=self.config.protect_project_name,
                 name=self.config.protect_stage_name,
-                stage_type=StageType.local,
-                description="Deutsche Bank RFP RAG Protect Stage"
+                stage_type=StageType.local
             )
-            return protect_stage.id
+        return protect_stage.id
 
     def run_protect(self, input: str, output: str, logger: GalileoLogger | None = None) -> dict[Any, Any]:
         """Run Galileo Protect on input and output."""

backend/conf/config.yaml

@@ -8,6 +8,7 @@ env_variables:
     - GALILEO_DATASET_NAME
     - GALILEO_LOGSTREAM_NAME
     - GALILEO_PROJECT_NAME
+    - GALILEO_PROTECT_PROJECT_NAME
     - GALILEO_PROTECT_STAGE_NAME
     - OPENAI_MODEL
     - OPENAI_API_KEY