Update kubernetes.use_secret_as_volume to accept secret name as a par…

…ameter

Signed-off-by: ddalvi <ddalvi@redhat.com>

kubernetes_platform/python/kfp/kubernetes/secret.py

@@ -18,7 +18,8 @@
 from kfp.dsl import PipelineTask
 from kfp.kubernetes import common
 from kfp.kubernetes import kubernetes_executor_config_pb2 as pb
-
+from typing import Union
+from kfp.dsl.pipeline_channel import PipelineParameterChannel
 
 def use_secret_as_env(
     task: PipelineTask,
@@ -59,7 +60,7 @@ def use_secret_as_env(
 
 def use_secret_as_volume(
     task: PipelineTask,
-    secret_name: str,
+    secret_name: Union[str, PipelineParameterChannel],
     mount_path: str,
     optional: bool = False,
 ) -> PipelineTask:
@@ -75,7 +76,9 @@ def use_secret_as_volume(
     Returns:
         Task object with updated secret configuration.
     """
-
+    # Extract the actual string value if secret_name is a PipelineParameterChannel
+    if isinstance(secret_name, PipelineParameterChannel):
+        secret_name = secret_name.name
     msg = common.get_existing_kubernetes_config_as_message(task)
 
     secret_as_vol = pb.SecretAsVolume(