Я хочу добавить аргумент maxOutputTokens в свой код, но не знаю, как это сделать.
import requests
API_KEY = 'INPUT_API_KEY'
API_URL = 'https://generativelanguage.googleapis.c ... ateContent'
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)
Подробнее здесь: https://stackoverflow.com/questions/790 ... pi-request
