Система промежуточного ПО (Middleware)#

# Обработка сообщений перед отправкой в GPT
def custom_message_middleware(notification, message_content, messages, session_data):
    # Добавление пользовательского контекста в разговор
    if notification.get_message_type() == "textMessage" and notification.chat.endswith("@c.us"):
        # Добавление контекста к сообщению
        enhanced_content = f"[Сообщение пользователя] {message_content}"

        return {
            "message_content": enhanced_content,
            "messages": messages
        }

    return {
        "message_content": message_content,
        "messages": messages
    }


# Добавление промежуточного ПО
bot.add_message_middleware(custom_message_middleware)

# Обработка ответов GPT перед отправкой пользователю
def custom_response_middleware(response, messages, session_data):
    # Форматирование или изменение ответа
    formatted_response = response.replace("GPT", "Ассистент").strip()

    # Вы также можете изменить сообщения, которые будут сохранены в истории
    return {
        "response": formatted_response,
        "messages": messages
    }


# Добавление промежуточного ПО
bot.add_response_middleware(custom_response_middleware)