[Python] How to add maxOutputTokens argument to Gemini API request

I want to add maxOutputTokens argument in my code but I don't know how.

import requests


API_KEY = 'INPUT_API_KEY'
API_URL = 'https://generativelanguage.googleapis.com/v1/models/gemini-pro:generateContent'


def call_gemini_api(prompt):
headers = {
'Content-Type': 'application/json',
'x-goog-api-key': API_KEY,
}
data = {
'contents': [
{
'role': 'model',
'parts': [{'text': 'You are AI'}]
},
{
'role': 'user',
'parts': [{'text': prompt}]
}
]
}
response = requests.post(API_URL, headers=headers, json=data)
return response.json()

prompt = "Hello"
response = call_gemini_api(prompt)
print(response)

Continue reading...