pipelines/message.py

from utils.file import File

def send_message(user_query: str, pipeline):
    # Estrutura das mensagens para a chamada da API
    conversation_chat = []

    # Adiciona a mensagem do sistema
    conversation_chat.append({
        "role": "system",
        "content": File("prompts/system.md")
    })

    # Adiciona a mensagem do usuário
    conversation_chat.append({
        "role": "user",
        "content": user_query
    })

    # Chama o pipeline para obter a resposta
    text = pipeline(conversation_chat)

    return text