code: def ai21_test_completion():
      litellm.AI21Config(maxTokens=10)
      litellm.set_verbose = True
      try:
          # OVERRIDE WITH DYNAMIC MAX TOKENS
          response_1 = litellm.completion(
              model="j2-mid",
              messages=[
                  {
                      "content": "Hello, how are you? Be as verbose as possible",
                      "role": "user",
                  }
              ],
              max_tokens=100,
          )
          response_1_text = response_1.choices[0].message.content
          print(f"response_1_text: {response_1_text}")
  
          # USE CONFIG TOKENS
          response_2 = litellm.completion(
              model="j2-mid",
              messages=[
                  {
                      "content": "Hello, how are you? Be as verbose as possible",
                      "role": "user",
                  }
              ],
          )
          response_2_text = response_2.choices[0].message.content
          print(f"response_2_text: {response_2_text}")
  
          assert len(response_2_text) < len(response_1_text)
  
          response_3 = litellm.completion(
              model="j2-light",
              messages=[
                  {"content": "Hello, how are you?", "role": "user"}
              ],
              n=2,
          )
          assert len(response_3.choices) > 1
      except Exception as e:
          pytest.fail(f"Error occurred: {e}")
  
file_path: litellm/litellm/tests/test_provider_specific_config.py
imports: import sys, os
  import traceback
  import pytest
  import litellm
  from litellm import completion
  from litellm import RateLimitError
label_desc: Use image node_id 4 for calling a global function with example usage: ai21_test_completion() without return types