code: def ai21_completion_response():
      return {
          "id": 1234,
          "prompt": {
              "text": "This is a test",
              "tokens": [
                  {
                      "generatedToken": {
                          "token": "▁This▁is▁a",
                          "logprob": -7.127955436706543,
                          "raw_logprob": -7.127955436706543,
                      },
                      "topTokens": None,
                      "textRange": {"start": 0, "end": 9},
                  },
                  {
                      "generatedToken": {
                          "token": "▁test",
                          "logprob": -4.926638126373291,
                          "raw_logprob": -4.926638126373291,
                      },
                      "topTokens": None,
                      "textRange": {"start": 9, "end": 14},
                  },
              ],
          },
          "completions": [
              {
                  "data": {
                      "text": "\nIt looks like",
                      "tokens": [
                          {
                              "generatedToken": {
                                  "token": "<|newline|>",
                                  "logprob": -0.021781044080853462,
                                  "raw_logprob": -0.021781044080853462,
                              },
                              "topTokens": None,
                              "textRange": {"start": 0, "end": 1},
                          },
                          {
                              "generatedToken": {
                                  "token": "▁It▁looks▁like",
                                  "logprob": -3.2340049743652344,
                                  "raw_logprob": -3.2340049743652344,
                              },
                              "topTokens": None,
                              "textRange": {"start": 1, "end": 14},
                          },
                          {
                              "generatedToken": {
                                  "token": "<|endoftext|>",
                                  "logprob": -0.01683046855032444,
                                  "raw_logprob": -0.01683046855032444,
                              },
                              "topTokens": None,
                              "textRange": {"start": 14, "end": 14},
                          },
                      ],
                  },
                  "finishReason": {"reason": "endoftext"},
              }
          ],
      }
  
file_path: mlflow/tests/gateway/providers/test_bedrock.py
imports: from unittest import mock
  import pytest
  from fastapi.encoders import jsonable_encoder
  from mlflow.gateway.config import AWSBaseConfig, AWSBedrockConfig, AWSIdAndKey, AWSRole, RouteConfig
  from mlflow.gateway.providers.bedrock import AWSBedrockModelProvider, AWSBedrockProvider
  from mlflow.gateway.schemas import completions
  from tests.gateway.providers.test_anthropic import completions_response
  from tests.gateway.providers.test_anthropic import parsed_completions_response
  from tests.gateway.providers.test_cohere import completions_response
label_desc: Use image node_id 1 for calling a global function with example usage: ai21_completion_response() and returns: dict