File size: 745 Bytes
c8fb072 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | from src.sdlc.states.states import State
from src.sdlc.prompts.prompts import SUMMARIZE_DESIGN_DOCS
from src.sdlc import logger
class DesignSummarizeNode:
"""
Node logic implementation.
"""
def __init__(self,model):
self.llm = model
def process(self, state: State):
"""
Processes the input state and generates design document based on user stories.
"""
#Summarize the design documents for the code generation
summarized_docs=self.llm.invoke(SUMMARIZE_DESIGN_DOCS.format(design_documents=state["design_documents"]))
logger.info(f"In DESIGN SUMMARIZN NODE,response received : {summarized_docs.content}")
return {"design_summary": summarized_docs.content}
|