Привет! помоги мне интегрировать функцию в бота. Функция:
import requests
import json

class YandexGPT:
    def generate(self, prompt, apikey, sa_id):
        url = 'https://llm.api.cloud.yandex.net/foundationModels/v1/completion'
        headers = {
            'Content-Type': 'application/json',
            'Authorization': f'Api-Key {apikey}'
        }
        data = {
            "modelUri": f"gpt://{sa_id}/yandexgpt/latest",
            "completionOptions": {
                "stream": False,
                "temperature": 0.6,
                "maxTokens": "1000"
            },
            "messages": [
                {
                    "role": "system",
                    "text": "Твоя задача - создать небольшой текст о биографии человека в соответствии с следующими ответами на вопросы (пишется вопрос? ответ). Также напиши короткую эпитафию в соответствии с биографией. Свой ответ пиши только по шаблону:\nБиография:\n\nЭпитафий:\n\nБиография должна быть не сильно длинной, но интересной.  Если биография очевидно неправильная (ответы на вопросы невозможно распознать), то пиши: Извините, я не понял ваш запрос. Попробуйте ответить на вопросы еще раз. "
                },
                {
                    "role": "user",
                    "text": prompt
                }
            ]
        }

        response = requests.post(url, json=data, headers=headers)

        return response.text

self = YandexGPT()

string = YandexGPT.generate(self, "Имя? - Артем\nДата рождения? - 12.03.2005\n Дата смерти? - 12.03.2024\nБыл женат? - Да.\n Кем работал? строителем ", "AQVN1J4sCxYR98rj-tVppyp6gXQthbdmYvmgtO7a", "b1g5og37bgh1ghh2s2qc")
jsons = json.loads(string)
print(jsons['result']['alternatives'][0]['message']['text'])

Существующий код в боте:
async def generate(prompt: str, apikey: str, sa_id: str, user_id : str):
    url = 'https://llm.api.cloud.yandex.net/foundationModels/v1/completion'
    headers = {
        'Content-Type': 'application/json',
        'Authorization': f'Api-Key {apikey}'
    }
    data = {
        "modelUri": f"gpt://{sa_id}/yandexgpt-lite/latest",
        "completionOptions": {
            "stream": False,
            "temperature": 0.6,
            "maxTokens": "1000"
        },
        "messages": [
            {
                "role": "system",
                "text": "Твоя задача - создать небольшой текст о биографии человека в соответствии с следующими ответами на вопросы (пишется вопрос? ответ). Также напиши короткую эпитафию в соответствии с биографией. Свой ответ пиши только по шаблону:\nБиография:\n\nЭпитафий:\n\nБиография должна быть не сильно длинной, но интересной.  Если биография очевидно неправильная (ответы на вопросы невозможно распознать), то пиши: Извините, я не понял ваш запрос. Попробуйте ответить на вопросы еще раз. "
            },
            {
                "role": "user",
                "text": prompt
            }
        ]
    }

    try:
        response = requests.post(url, json=data, headers=headers)
        response_data = response.json()
        jsons = response_data.loads(string)
        answer = jsons['result']['alternatives'][0]['message']['text']
        await bot.send_message(user_id, answer)
    except Exception as e:
        return await bot.send_message(user_id, f"Произошла ошибка: {e}. Пожалуйста, попробуйте еще раз.")