feat: support remote storage URI injection for serving runtimes (kserve#4492)

Signed-off-by: de0725 <debosora.52901@gmail.com>

Author: de0725

pkg/constants/constants.go

@@ -72,6 +72,11 @@ var (
 	ModelConfigFileName = "models.json"
 )
 
+// Remote Storage URI
+const (
+	RemoteStorageEnvVarName = "REMOTE_STORAGE_URI"
+)
+
 // Model agent Constants
 const (
 	AgentContainerName        = "agent"

pkg/webhook/admission/pod/storage_initializer_injector.go

@@ -36,6 +36,7 @@ import (
 	"github.com/kserve/kserve/pkg/constants"
 	"github.com/kserve/kserve/pkg/credentials"
 	"github.com/kserve/kserve/pkg/credentials/s3"
+	"github.com/kserve/kserve/pkg/utils"
 )
 
 const (
@@ -54,6 +55,7 @@ const (
 	ModelInitModeEnv                        = "MODEL_INIT_MODE"
 	CpuModelcarDefault                      = "10m"
 	MemoryModelcarDefault                   = "15Mi"
+	RemoteStorageEnvVarName                 = "REMOTE_STORAGE_URI"
 )
 
 type StorageInitializerConfig struct {
@@ -245,6 +247,16 @@ func (mi *StorageInitializerInjector) InjectStorageInitializer(pod *corev1.Pod)
 		}
 	}
 
+	if _, exists := utils.GetEnvVarValue(userContainer.Env, RemoteStorageEnvVarName); !exists {
+		addOrReplaceEnv(userContainer, RemoteStorageEnvVarName, srcURI)
+	}
+
+	if transformerContainer != nil {
+		if _, exists := utils.GetEnvVarValue(transformerContainer.Env, RemoteStorageEnvVarName); !exists {
+			addOrReplaceEnv(transformerContainer, RemoteStorageEnvVarName, srcURI)
+		}
+	}
+
 	// Mount pvc directly if local model label exists
 	if modelName, ok := pod.ObjectMeta.Labels[constants.LocalModelLabel]; ok {
 		subPath, _ := strings.CutPrefix(srcURI, pod.ObjectMeta.Annotations[constants.LocalModelSourceUriAnnotationKey])