from duck_core.tasks.state import TaskState


class ContextBuilder:
    def build_basic_messages(self, task: TaskState) -> list[dict[str, str]]:
        return [
            {
                "role": "user",
                "content": task.user_message,
            }
        ]