from typing import Dict

from app.controllers.abc import ITrainingController
from app.dtos.training import FetchTipsDTO
from app.services.abc import ITrainingService


class TrainingController(ITrainingController):

    def __init__(self, training_service: ITrainingService):
        self._service = training_service

    async def fetch_tips(self, data: FetchTipsDTO):
        return await self._service.fetch_tips(data.context, data.question, data.answer, data.correct_answer)

    async def get_training_content(self, data: Dict):
        return await self._service.get_training_content(data)