from loggers.logger import logger
from template.response_prompt import GENERATE_CHAIN

import sys
sys.dont_write_bytecode = True

def generate(state):
    """
    Generate Answer
    Args:
        state (dict): The current graph state
    Return:
        state (dict): New key added to state
    """
    
    logger.info("STEP: GENERATING FINAL RESPONSE")

    question = state['question']

    try:
        context = state['context']
    except:
        context = " "
    
    generation = GENERATE_CHAIN.invoke(
        {
            "context": context,
            "question": question
        }
    )
    return {
        "generation": generation
    }