from mem0.configs.prompts import FACT_RETRIEVAL_PROMPT


def get_fact_retrieval_messages(message):
    return FACT_RETRIEVAL_PROMPT, f"Input: {message}"

def parse_messages(messages):
        response = ""
        for msg in messages:
            if msg["role"] == "system":
                response += f"system: {msg['content']}\n"
            if msg["role"] == "user":
                response += f"user: {msg['content']}\n"
            if msg["role"] == "assistant":
                response += f"assistant: {msg['content']}\n"
        return response