Source code for kusibot.chatbot.assesment_states.asking_question_state

from kusibot.chatbot.assesment_states.base_state import BaseState
from kusibot.chatbot.assesment_states.waiting_free_state import WaitingFreeTextState



class AskingQuestionState(BaseState):
    """State representing the agent asking a question in an assessment."""



    def generate_response(self, user_input, conversation_id, assessment_id):
        
        # Get the next question to ask
        question_json = self.context._get_question_json(assessment_id)

        # Get the model natural phrase
        user_id = self.context.assess_repo.get_assessment(assessment_id).user_id
        bot_response = self.context._naturalize_question(question_json['question'],
                                                         question_json['id'],
                                                         user_id)
        
        # Changing state of AssesmentAgent to WaitingFreeTextState
        self.context._transition_to_next_state(WaitingFreeTextState())

        # Updating state Assesment in DB
        self.context.assess_repo.update_assessment(
            assessment_id,
            current_state=self.context.STATE_WAITING_FREE_TEXT
        )
        
        return bot_response