context.py

contexts = {}

def update_context(user_id, message, role):
    if user_id not in contexts:
        contexts[user_id] = []

    if message.author.id == user_id:
        role = "user"
    else:
        role = "assistant"

    new_context_entry = {"role": role, "content": message.content}
    contexts[user_id].append(new_context_entry)

    if len(contexts[user_id]) > 5:
        contexts[user_id] = contexts[user_id][-5:]

    return contexts[user_id]

def format_prompt(context):
    prompt = "Привет! Меня зовут Хелпер, и я ваш дружелюбный дроид-путеводитель по миру Звездных войн. Задавайте свои вопросы на русском языке, и я постараюсь помочь вам, используя свои знания из этой захватывающей вселенной. Я буду стараться отвечать на все ваши запросы максимально точно и полезно."
    messages = [{"role": "system", "content": prompt}]
    messages.extend(context)

    return messages

def filter_response(response):
    unwanted_phrases = ["unwanted_phrase1", "unwanted_phrase2"]
    for phrase in unwanted_phrases:
        if phrase in response:
            return "I'm sorry, I can't provide an answer to that question."
    return response

def post_process(response):
    # Add your post-processing logic here
    return response