| from src.dev_pilot.state.sdlc_state import SDLCState, UserStoryList |
| from langchain_core.messages import SystemMessage |
|
|
| class ProjectRequirementNode: |
| """ |
| Graph Node for the project requirements |
| |
| """ |
| |
| def __init__(self, model): |
| self.llm = model |
| |
| def initialize_project(self, state: SDLCState): |
| """ |
| Performs the project initilazation |
| """ |
| return state |
| |
| def get_user_requirements(self, state: SDLCState): |
| """ |
| Gets the requirements from the user |
| """ |
| pass |
| |
| def generate_user_stories(self, state: SDLCState): |
| """ |
| Auto-generate highly detailed and accurate user stories for each requirement. |
| """ |
| project_name = state["project_name"] |
| requirements = state["requirements"] |
| feedback_reason = state.get("user_stories_feedback", None) |
|
|
| prompt = f""" |
| You are a senior software analyst specializing in Agile SDLC and user story generation. |
| Your task is to generate **a separate and detailed user story for EACH requirement** from the project details below. |
| |
| --- |
| **Project Name:** "{project_name}" |
| |
| **Requirements:** "{requirements} |
| |
| --- |
| **Instructions for User Story Generation:** |
| - Create **one user story per requirement**. |
| - Assign a **unique identifier** (e.g., US-001, US-002, etc.). |
| - Provide a **clear and concise title** summarizing the user story. |
| - Write a **detailed description** using the "As a [user role], I want [goal] so that [benefit]" format. |
| - Assign a **priority level** (1 = Critical, 2 = High, 3 = Medium, 4 = Low). |
| - Define **acceptance criteria** with bullet points to ensure testability. |
| - Use **domain-specific terminology** for clarity. |
| |
| {f"Additionally, consider the following feedback while refining the user stories: {feedback_reason}" if feedback_reason else ""} |
| |
| --- |
| **Expected Output Format (for each user story):** |
| - Unique Identifier: US-XXX |
| - Title: [User Story Title] |
| - Description: |
| - As a [user role], I want [feature] so that [benefit]. |
| - Priority: [1-4] |
| - Acceptance Criteria: |
| - [Criteria 1] |
| - [Criteria 2] |
| - [Criteria 3] |
| |
| Ensure that the user stories are **specific, testable, and aligned with Agile principles**. |
| """ |
|
|
| llm_with_structured = self.llm.with_structured_output(UserStoryList) |
| response = llm_with_structured.invoke(prompt) |
| state["user_stories"] = response |
| return state |
| |
| def review_user_stories(self, state: SDLCState): |
| return state |
| |
| def revise_user_stories(self, state: SDLCState): |
| pass |
| |
| def review_user_stories_router(self, state: SDLCState): |
| return state.get("user_stories_review_status", "approved") |
