code: def __init__(self, config: RouteConfig) -> None:
      super().__init__(config)
      if config.model.config is None or not isinstance(
          config.model.config, AI21LabsConfig
      ):
          raise TypeError(
              f"Unexpected config type {config.model.config}"
          )
      self.ai21labs_config: AI21LabsConfig = config.model.config
      self.headers = {
          "Authorization": f"Bearer {self.ai21labs_config.ai21labs_api_key}"
      }
      self.base_url = (
          f"https://api.ai21.com/studio/v1/{self.config.model.name}/"
      )
  
file_path: mlflow/mlflow/gateway/providers/ai21labs.py
imports: import time
  from fastapi import HTTPException
  from fastapi.encoders import jsonable_encoder
  from mlflow.gateway.config import AI21LabsConfig, RouteConfig
  from mlflow.gateway.providers.base import BaseProvider
  from mlflow.gateway.providers.utils import rename_payload_keys, send_request
  from mlflow.gateway.schemas import chat, completions, embeddings
label_desc: Use image node_id 1 to create a new AI21LabsProvider object from inherited base classes: BaseProvider with example: obj = AI21LabsProvider(config)