Spaces:
Sleeping
Sleeping
| 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 | |
| } |