code: def get_provider(provider: Provider) -> Type[BaseProvider]:
      from mlflow.gateway.providers.ai21labs import AI21LabsProvider
      from mlflow.gateway.providers.anthropic import AnthropicProvider
      from mlflow.gateway.providers.bedrock import AWSBedrockProvider
      from mlflow.gateway.providers.cohere import CohereProvider
      from mlflow.gateway.providers.huggingface import (
          HFTextGenerationInferenceServerProvider,
      )
      from mlflow.gateway.providers.mlflow import (
          MlflowModelServingProvider,
      )
      from mlflow.gateway.providers.mosaicml import MosaicMLProvider
      from mlflow.gateway.providers.openai import OpenAIProvider
      from mlflow.gateway.providers.palm import PaLMProvider
  
      provider_to_class = {
          Provider.OPENAI: OpenAIProvider,
          Provider.ANTHROPIC: AnthropicProvider,
          Provider.COHERE: CohereProvider,
          Provider.AI21LABS: AI21LabsProvider,
          Provider.MOSAICML: MosaicMLProvider,
          Provider.PALM: PaLMProvider,
          Provider.MLFLOW_MODEL_SERVING: MlflowModelServingProvider,
          Provider.HUGGINGFACE_TEXT_GENERATION_INFERENCE: HFTextGenerationInferenceServerProvider,
          Provider.BEDROCK: AWSBedrockProvider,
      }
      if prov := provider_to_class.get(provider):
          return prov
  
      raise MlflowException.invalid_parameter_value(
          f"Provider {provider} not found"
      )
  
imports: from typing import Type
  from mlflow.exceptions import MlflowException
  from mlflow.gateway.config import Provider
  from mlflow.gateway.providers.base import BaseProvider
label_desc: Use image node_id 1 for calling a global function with example usage: get_provider(provider) and returns: prov