File size: 665 Bytes
c8fb072 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | from src.sdlc.states.states import State
from src.sdlc.prompts.prompts import TESTCASES_GEN_INSTRNS
from src.sdlc import logger
class TestCasesNode:
"""
Node logic implementation.
"""
def __init__(self,model):
self.llm = model
def process(self, state: State):
"""
Processes the input state and generates test cases for the code
"""
design_summary=state.get('design_summary', '')
response=self.llm.invoke(TESTCASES_GEN_INSTRNS.format(design_documents=design_summary))
logger.info("In TEST CASES NODE received response")
return {"test_cases":response.content}
|