diff --git a/.gitattributes b/.gitattributes index a6344aac8c09253b3b630fb776ae94478aa0275b..40bf842adace62cc95e02d19435e1d65c6453fb0 100644 --- a/.gitattributes +++ b/.gitattributes @@ -1,35 +1,36 @@ -*.7z filter=lfs diff=lfs merge=lfs -text -*.arrow filter=lfs diff=lfs merge=lfs -text -*.bin filter=lfs diff=lfs merge=lfs -text -*.bz2 filter=lfs diff=lfs merge=lfs -text -*.ckpt filter=lfs diff=lfs merge=lfs -text -*.ftz filter=lfs diff=lfs merge=lfs -text -*.gz filter=lfs diff=lfs merge=lfs -text -*.h5 filter=lfs diff=lfs merge=lfs -text -*.joblib filter=lfs diff=lfs merge=lfs -text -*.lfs.* filter=lfs diff=lfs merge=lfs -text -*.mlmodel filter=lfs diff=lfs merge=lfs -text -*.model filter=lfs diff=lfs merge=lfs -text -*.msgpack filter=lfs diff=lfs merge=lfs -text -*.npy filter=lfs diff=lfs merge=lfs -text -*.npz filter=lfs diff=lfs merge=lfs -text -*.onnx filter=lfs diff=lfs merge=lfs -text -*.ot filter=lfs diff=lfs merge=lfs -text -*.parquet filter=lfs diff=lfs merge=lfs -text -*.pb filter=lfs diff=lfs merge=lfs -text -*.pickle filter=lfs diff=lfs merge=lfs -text -*.pkl filter=lfs diff=lfs merge=lfs -text -*.pt filter=lfs diff=lfs merge=lfs -text -*.pth filter=lfs diff=lfs merge=lfs -text -*.rar filter=lfs diff=lfs merge=lfs -text -*.safetensors filter=lfs diff=lfs merge=lfs -text -saved_model/**/* filter=lfs diff=lfs merge=lfs -text -*.tar.* filter=lfs diff=lfs merge=lfs -text -*.tar filter=lfs diff=lfs merge=lfs -text -*.tflite filter=lfs diff=lfs merge=lfs -text -*.tgz filter=lfs diff=lfs merge=lfs -text -*.wasm filter=lfs diff=lfs merge=lfs -text -*.xz filter=lfs diff=lfs merge=lfs -text -*.zip filter=lfs diff=lfs merge=lfs -text -*.zst filter=lfs diff=lfs merge=lfs -text -*tfevents* filter=lfs diff=lfs merge=lfs -text +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text +chroma_db/chroma.sqlite3 filter=lfs diff=lfs merge=lfs -text diff --git a/Dockerfile b/Dockerfile index 5f51ead59f36f13043e036290df9440e25fe8cbe..67841b1d027dea81e3e169e0b3efc59b728be7ac 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,20 +1,22 @@ -FROM python:3.13.5-slim - -WORKDIR /app - -RUN apt-get update && apt-get install -y \ - build-essential \ - curl \ - git \ - && rm -rf /var/lib/apt/lists/* - -COPY requirements.txt ./ -COPY src/ ./src/ - -RUN pip3 install -r requirements.txt - -EXPOSE 8501 - -HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health - -ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"] \ No newline at end of file +# AI Virtual Pod - Streamlit application +FROM python:3.11-slim + +WORKDIR /app + +# Install dependencies first for better layer caching +COPY requirements.txt . +RUN pip install --no-cache-dir -r requirements.txt + +# Copy application code +COPY app.py . +COPY agents/ agents/ +COPY core/ core/ + +# ChromaDB may write to disk; optional data dir +ENV CHROMA_PERSIST_DIR=/app/chroma_db + +# Streamlit runs on 8501 by default +EXPOSE 8501 + +# Run the app (bind to 0.0.0.0 so it's reachable from outside the container) +CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0"] diff --git a/agents/__pycache__/business_analyst.cpython-310.pyc b/agents/__pycache__/business_analyst.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..9c6100f61078c7b11a2dbd64177628570b3978c2 Binary files /dev/null and b/agents/__pycache__/business_analyst.cpython-310.pyc differ diff --git a/agents/__pycache__/business_analyst.cpython-311.pyc b/agents/__pycache__/business_analyst.cpython-311.pyc new file mode 100644 index 0000000000000000000000000000000000000000..176e42b167963677ed660317630e08abc6fb1833 Binary files /dev/null and b/agents/__pycache__/business_analyst.cpython-311.pyc differ diff --git a/agents/__pycache__/business_analyst.cpython-313.pyc b/agents/__pycache__/business_analyst.cpython-313.pyc new file mode 100644 index 0000000000000000000000000000000000000000..6b556d5573e8b884b0b728a5b1cebec284ea4e3c Binary files /dev/null and b/agents/__pycache__/business_analyst.cpython-313.pyc differ diff --git a/agents/__pycache__/designer.cpython-310.pyc b/agents/__pycache__/designer.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..0956f10f0daf5ab3eefb1d9b6e7687164cdfe7bf Binary files /dev/null and b/agents/__pycache__/designer.cpython-310.pyc differ diff --git a/agents/__pycache__/designer.cpython-311.pyc b/agents/__pycache__/designer.cpython-311.pyc new file mode 100644 index 0000000000000000000000000000000000000000..77ad7bcbdaa7b9259cfe96694d70c7481cebfaef Binary files /dev/null and b/agents/__pycache__/designer.cpython-311.pyc differ diff --git a/agents/__pycache__/designer.cpython-313.pyc b/agents/__pycache__/designer.cpython-313.pyc new file mode 100644 index 0000000000000000000000000000000000000000..5dd07a7637eb0fc40b97cec27e396e35add14422 Binary files /dev/null and b/agents/__pycache__/designer.cpython-313.pyc differ diff --git a/agents/__pycache__/developer.cpython-310.pyc b/agents/__pycache__/developer.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..873ae4b59efbdcd065f0382d2b9d385aec5ac993 Binary files /dev/null and b/agents/__pycache__/developer.cpython-310.pyc differ diff --git a/agents/__pycache__/developer.cpython-311.pyc b/agents/__pycache__/developer.cpython-311.pyc new file mode 100644 index 0000000000000000000000000000000000000000..8a57b9f06b666275c44077f334d30f02820ff1c1 Binary files /dev/null and b/agents/__pycache__/developer.cpython-311.pyc differ diff --git a/agents/__pycache__/developer.cpython-313.pyc b/agents/__pycache__/developer.cpython-313.pyc new file mode 100644 index 0000000000000000000000000000000000000000..a6e0e19c1b807bdd7a69eb5e8b92306e69c66af2 Binary files /dev/null and b/agents/__pycache__/developer.cpython-313.pyc differ diff --git a/agents/__pycache__/project_manager.cpython-310.pyc b/agents/__pycache__/project_manager.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..d4a1b193a72a65fc50c4a719b2ac9216aa73e5da Binary files /dev/null and b/agents/__pycache__/project_manager.cpython-310.pyc differ diff --git a/agents/__pycache__/project_manager.cpython-311.pyc b/agents/__pycache__/project_manager.cpython-311.pyc new file mode 100644 index 0000000000000000000000000000000000000000..0e5ea2525aa485dbd350aa088238e7b5a3fe6a9d Binary files /dev/null and b/agents/__pycache__/project_manager.cpython-311.pyc differ diff --git a/agents/__pycache__/project_manager.cpython-313.pyc b/agents/__pycache__/project_manager.cpython-313.pyc new file mode 100644 index 0000000000000000000000000000000000000000..712e2e8c3c0f080d57d0f85748a8a2b03a2f3f2f Binary files /dev/null and b/agents/__pycache__/project_manager.cpython-313.pyc differ diff --git a/agents/__pycache__/tester.cpython-310.pyc b/agents/__pycache__/tester.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..5a9b3f14803ed55929403fd76a54847c9f0c0d75 Binary files /dev/null and b/agents/__pycache__/tester.cpython-310.pyc differ diff --git a/agents/__pycache__/tester.cpython-311.pyc b/agents/__pycache__/tester.cpython-311.pyc new file mode 100644 index 0000000000000000000000000000000000000000..639b61a1e8936436ec72e0995edf9a06a498c072 Binary files /dev/null and b/agents/__pycache__/tester.cpython-311.pyc differ diff --git a/agents/__pycache__/tester.cpython-313.pyc b/agents/__pycache__/tester.cpython-313.pyc new file mode 100644 index 0000000000000000000000000000000000000000..3acd63ff5b9770e1c05b46fcad8770254b3fc590 Binary files /dev/null and b/agents/__pycache__/tester.cpython-313.pyc differ diff --git a/agents/business_analyst.py b/agents/business_analyst.py new file mode 100644 index 0000000000000000000000000000000000000000..bf499a980a87d7a5d70f92fd3c20073559c46828 --- /dev/null +++ b/agents/business_analyst.py @@ -0,0 +1,46 @@ +from core.base_agent import BaseAgent +from core.database import db +from typing import Dict, Any + +class BusinessAnalystAgent(BaseAgent): + def __init__(self): + super().__init__("Business Analyst") + self.create_chain(""" + You are a Business Analyst. Your task is to create detailed user stories from the given business requirements. + + Business Requirements: + {input} + + Create user stories following this format: + As a [type of user] + I want [some goal] + So that [some reason] + + Acceptance Criteria: + 1. [First criterion] + 2. [Second criterion] + ... + + Please provide clear, testable user stories with specific acceptance criteria. + """) + + async def create_user_stories(self, requirements: str) -> Dict[str, Any]: + """Create user stories from business requirements""" + result = await self.process({"input": requirements}) + + # Store the user stories in the database + db.store_artifact( + "user_stories", + result, + { + "type": "user_story", + "source": "business_analyst", + "status": "created" + } + ) + + return { + "status": "success", + "user_stories": result, + "message": "User stories created successfully" + } \ No newline at end of file diff --git a/agents/designer.py b/agents/designer.py new file mode 100644 index 0000000000000000000000000000000000000000..747f44c042be9cd89179c55222d329df7314b293 --- /dev/null +++ b/agents/designer.py @@ -0,0 +1,44 @@ +from core.base_agent import BaseAgent +from core.database import db +from typing import Dict, Any + +class DesignerAgent(BaseAgent): + def __init__(self): + super().__init__("Designer") + self.create_chain(""" + You are a Software Designer. Your task is to create detailed design specifications from the given user stories. + + User Stories: + {input} + + Create a comprehensive design document that includes: + 1. System Architecture + 2. Component Design + 3. Database Schema + 4. API Endpoints + 5. Data Flow Diagrams + 6. Security Considerations + + Please provide a detailed, implementable design that follows best practices and design patterns. + """) + + async def create_design(self, user_stories: str) -> Dict[str, Any]: + """Create design specifications from user stories""" + result = await self.process({"input": user_stories}) + + # Store the design in the database + db.store_artifact( + "designs", + result, + { + "type": "design", + "source": "designer", + "status": "created" + } + ) + + return { + "status": "success", + "design": result, + "message": "Design specifications created successfully" + } \ No newline at end of file diff --git a/agents/developer.py b/agents/developer.py new file mode 100644 index 0000000000000000000000000000000000000000..016a110d4b1effd572a099988bb22c2f450ae227 --- /dev/null +++ b/agents/developer.py @@ -0,0 +1,111 @@ +from core.base_agent import BaseAgent +from core.database import db +from typing import Dict, Any + +class DeveloperAgent(BaseAgent): + def __init__(self): + super().__init__("Developer") + self.create_chain(""" + You are an expert Software Developer. Your task is to generate production-ready, well-structured code based on the given design specifications. + + Design Specifications: + {input} + + Generate code that follows these strict guidelines: + + 1. Code Structure: + - Use proper project structure with separate modules + - Follow SOLID principles + - Implement proper error handling + - Use type hints + - Include comprehensive docstrings + - Follow PEP 8 style guide + + 2. Implementation Details: + - Use modern Python features (Python 3.9+) + - Implement proper logging + - Add input validation + - Include unit tests + - Use dependency injection where appropriate + - Implement proper configuration management + + 3. Security: + - Sanitize all inputs + - Implement proper authentication/authorization + - Use secure coding practices + - Handle sensitive data properly + + 4. Performance: + - Optimize database queries + - Implement caching where appropriate + - Use async/await for I/O operations + - Implement proper resource management + + 5. Documentation: + - Include detailed module docstrings + - Document all public methods + - Include usage examples + - Document configuration options + + The code should be: + - Production-ready + - Well-tested + - Scalable + - Maintainable + - Secure + - Performant + + Provide the complete implementation with all necessary files and dependencies. + """) + + async def generate_code(self, design: str) -> Dict[str, Any]: + """Generate code from design specifications""" + result = await self.process({"input": design}) + + # Store the code in the database + db.store_artifact( + "code", + result, + { + "type": "code", + "source": "developer", + "status": "created", + "version": "1.0.0" + } + ) + + return { + "status": "success", + "code": result, + "message": "Code generated successfully", + "metadata": { + "language": "Python", + "framework": "Standard Library", + "dependencies": self._extract_dependencies(result) + } + } + + def _extract_dependencies(self, code: str) -> list: + """Extract dependencies from the generated code""" + dependencies = set() + # Add common dependencies + dependencies.update([ + "python-dotenv", + "pydantic", + "fastapi", + "uvicorn", + "pytest", + "pytest-asyncio" + ]) + + # Add dependencies based on code content + if "import sqlalchemy" in code: + dependencies.add("sqlalchemy") + if "import aiohttp" in code: + dependencies.add("aiohttp") + if "import redis" in code: + dependencies.add("redis") + if "import jwt" in code: + dependencies.add("PyJWT") + + return sorted(list(dependencies)) \ No newline at end of file diff --git a/agents/project_manager.py b/agents/project_manager.py new file mode 100644 index 0000000000000000000000000000000000000000..d264bdca2424918c02eb8b1fb378cf138a706912 --- /dev/null +++ b/agents/project_manager.py @@ -0,0 +1,123 @@ +from core.base_agent import BaseAgent +from core.database import db +from core.workflow import DevelopmentWorkflow +from typing import Dict, Any + +class ProjectManagerAgent(BaseAgent): + def __init__(self): + super().__init__("Project Manager") + self.workflow = DevelopmentWorkflow() + + self.create_chain(""" + You are a Project Manager. Your task is to oversee the development process and provide status updates. + + Current Status: + {input} + + Please provide a comprehensive status report including: + 1. Project progress + 2. Any blockers or issues + 3. Next steps + 4. Quality metrics + 5. Recommendations + + Please provide clear, actionable insights. + """) + + async def start_project(self, requirements: str) -> Dict[str, Any]: + """Start a new project with the given requirements""" + # Run the complete workflow + result = await self.workflow.run(requirements) + + # Store all artifacts in the database + self._store_artifacts(result) + + return { + "status": "success", + "user_stories": result["user_stories"], + "design": result["design"], + "code": result["code"], + "test_results": result["test_results"], + "messages": result["messages"], + "message": "Project completed successfully" + } + + def _store_artifacts(self, result: Dict[str, Any]): + """Store all project artifacts in the database""" + # Store user stories + db.store_artifact( + "user_stories", + result["user_stories"], + { + "type": "user_story", + "source": "business_analyst", + "status": "created" + } + ) + + # Store design + db.store_artifact( + "designs", + result["design"], + { + "type": "design", + "source": "designer", + "status": "created" + } + ) + + # Store code + db.store_artifact( + "code", + result["code"], + { + "type": "code", + "source": "developer", + "status": "created", + "version": "1.0.0" + } + ) + + # Store test results + db.store_artifact( + "test_results", + result["test_results"], + { + "type": "test_result", + "source": "tester", + "status": "executed" + } + ) + + async def get_status(self) -> Dict[str, Any]: + """Get the current project status""" + # Query all artifacts from the database + user_stories = db.query_artifacts("user_stories", "status:created") + designs = db.query_artifacts("designs", "status:created") + code = db.query_artifacts("code", "status:created") + test_results = db.query_artifacts("test_results", "status:executed") + + status = { + "user_stories": len(user_stories["ids"]), + "designs": len(designs["ids"]), + "code": len(code["ids"]), + "test_results": len(test_results["ids"]) + } + + return { + "status": "success", + "data": status, + "message": "Status retrieved successfully" + } + + async def check_quality(self) -> Dict[str, Any]: + """Check the quality of project artifacts""" + result = await self.process({ + "input": "Checking quality of all project artifacts" + }) + + return { + "status": "success", + "quality_report": result, + "message": "Quality check completed successfully" + } \ No newline at end of file diff --git a/agents/tester.py b/agents/tester.py new file mode 100644 index 0000000000000000000000000000000000000000..fd144c97aa2846f084ade85f7b52ed34627f0aac --- /dev/null +++ b/agents/tester.py @@ -0,0 +1,66 @@ +from core.base_agent import BaseAgent +from core.database import db +from typing import Dict, Any + +class TesterAgent(BaseAgent): + def __init__(self): + super().__init__("Tester") + self.create_chain(""" + You are a Software Tester. Your task is to create comprehensive test cases and execute them on the given code. + + Code to Test: + {input} + + Create test cases that: + 1. Cover all user stories and acceptance criteria + 2. Include unit tests, integration tests, and system tests + 3. Test edge cases and error conditions + 4. Follow testing best practices + 5. Include test data and expected results + + Please provide detailed test cases with clear steps and expected outcomes. + """) + + async def create_test_cases(self, code: str) -> Dict[str, Any]: + """Create test cases for the given code""" + result = await self.process({"input": code}) + + # Store the test cases in the database + db.store_artifact( + "test_cases", + result, + { + "type": "test_case", + "source": "tester", + "status": "created" + } + ) + + return { + "status": "success", + "test_cases": result, + "message": "Test cases created successfully" + } + + async def execute_tests(self, code: str, test_cases: str) -> Dict[str, Any]: + """Execute test cases on the given code""" + result = await self.process({ + "input": f"Code:\n{code}\n\nTest Cases:\n{test_cases}" + }) + + # Store the test results in the database + db.store_artifact( + "test_results", + result, + { + "type": "test_result", + "source": "tester", + "status": "executed" + } + ) + + return { + "status": "success", + "test_results": result, + "message": "Tests executed successfully" + } \ No newline at end of file diff --git a/api/__pycache__/main.cpython-310.pyc b/api/__pycache__/main.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..99669fb6710cb7be0312806defa8e48bf524ba5a Binary files /dev/null and b/api/__pycache__/main.cpython-310.pyc differ diff --git a/app.py b/app.py new file mode 100644 index 0000000000000000000000000000000000000000..bd58fbc86846c3dd8f7949d963e1d596ff4ff5b1 --- /dev/null +++ b/app.py @@ -0,0 +1,167 @@ +import streamlit as st +import asyncio +from agents.project_manager import ProjectManagerAgent +import json +from typing import Dict, Any + +# Initialize the project manager +project_manager = ProjectManagerAgent() + +# Set page config +st.set_page_config( + page_title="AI Development Pod", + page_icon="🤖", + layout="wide" +) + +# Custom CSS +st.markdown(""" + +""", unsafe_allow_html=True) + +# Title and description +st.title("🤖 AI Development Pod") +st.markdown(""" + This is an AI-powered virtual development team that can handle your software project from requirements to testing. + Simply enter your project requirements and let the AI team do the rest! +""") + +# Sidebar for navigation +st.sidebar.title("Navigation") +page = st.sidebar.radio("Go to", ["Start Project", "Project Status", "Quality Check"]) + +def display_progress(messages: list): + """Display progress based on workflow messages""" + steps = { + "User stories created successfully": 20, + "Design created successfully": 40, + "Code generated successfully": 60, + "Test cases created successfully": 80, + "Tests executed successfully": 100 + } + + current_progress = 0 + for message in messages: + if message in steps: + current_progress = max(current_progress, steps[message]) + + st.markdown(f""" +
+
+
+

{current_progress}% Complete

+ """, unsafe_allow_html=True) + + for message in messages: + st.info(message) + +async def run_project(requirements: str) -> Dict[str, Any]: + """Run the project workflow""" + return await project_manager.start_project(requirements) + +async def get_project_status() -> Dict[str, Any]: + """Get the project status""" + return await project_manager.get_status() + +async def run_quality_check() -> Dict[str, Any]: + """Run the quality check""" + return await project_manager.check_quality() + +if page == "Start Project": + st.header("Start New Project") + + # Project requirements input + requirements = st.text_area( + "Enter your project requirements:", + placeholder="Describe your project requirements in detail...", + height=200 + ) + + if st.button("Start Project"): + if requirements: + with st.spinner("Processing your project..."): + # Run the async function + result = asyncio.run(run_project(requirements)) + + # Display progress + display_progress(result["messages"]) + + # Display results in tabs + tab1, tab2, tab3, tab4 = st.tabs(["User Stories", "Design", "Code", "Test Results"]) + + with tab1: + st.subheader("User Stories") + st.markdown(result["user_stories"]) + + with tab2: + st.subheader("Design") + st.markdown(result["design"]) + + with tab3: + st.subheader("Generated Code") + st.code(result["code"], language="python") + + with tab4: + st.subheader("Test Results") + st.markdown(result["test_results"]) + else: + st.error("Please enter project requirements") + +elif page == "Project Status": + st.header("Project Status") + + if st.button("Check Status"): + with st.spinner("Getting project status..."): + status = asyncio.run(get_project_status()) + + # Display status metrics + col1, col2, col3, col4 = st.columns(4) + + with col1: + st.metric("User Stories", status["data"]["user_stories"]) + with col2: + st.metric("Designs", status["data"]["designs"]) + with col3: + st.metric("Code Files", status["data"]["code"]) + with col4: + st.metric("Test Results", status["data"]["test_results"]) + +elif page == "Quality Check": + st.header("Quality Check") + + if st.button("Run Quality Check"): + with st.spinner("Running quality check..."): + quality_report = asyncio.run(run_quality_check()) + + st.subheader("Quality Report") + st.markdown(quality_report["quality_report"]) + +# Footer +st.markdown("---") +st.markdown(""" +
+

AI Development Pod - Powered by Groq's Gemma2-9b-it and LangGraph

+
+""", unsafe_allow_html=True) \ No newline at end of file diff --git a/chroma_db/0895c0fd-aea2-4db2-9ea7-55fff1798415/data_level0.bin b/chroma_db/0895c0fd-aea2-4db2-9ea7-55fff1798415/data_level0.bin new file mode 100644 index 0000000000000000000000000000000000000000..0fee824df425bbd6a2ddf45894dfa0ad3633141e --- /dev/null +++ b/chroma_db/0895c0fd-aea2-4db2-9ea7-55fff1798415/data_level0.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b8146ecc3e4c3a36ea9b3edc3778630c452f483990ec942d38e8006f4661e430 +size 16760000 diff --git a/chroma_db/0895c0fd-aea2-4db2-9ea7-55fff1798415/header.bin b/chroma_db/0895c0fd-aea2-4db2-9ea7-55fff1798415/header.bin new file mode 100644 index 0000000000000000000000000000000000000000..51ca55818d180b505757e2a099351f24a94d6fbe --- /dev/null +++ b/chroma_db/0895c0fd-aea2-4db2-9ea7-55fff1798415/header.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:18f1e924efbb5e1af5201e3fbab86a97f5c195c311abe651eeec525884e5e449 +size 100 diff --git a/chroma_db/0895c0fd-aea2-4db2-9ea7-55fff1798415/length.bin b/chroma_db/0895c0fd-aea2-4db2-9ea7-55fff1798415/length.bin new file mode 100644 index 0000000000000000000000000000000000000000..c7c63e13ec48a5d18544a5a6ca045ab5daf10bc4 --- /dev/null +++ b/chroma_db/0895c0fd-aea2-4db2-9ea7-55fff1798415/length.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a6bbf80e711a9a4d340bd13f1144da633010eb9d4088769ff99715e83ff21386 +size 40000 diff --git a/chroma_db/0895c0fd-aea2-4db2-9ea7-55fff1798415/link_lists.bin b/chroma_db/0895c0fd-aea2-4db2-9ea7-55fff1798415/link_lists.bin new file mode 100644 index 0000000000000000000000000000000000000000..fc8e42b32efb9a9bf3ae0234a18a948e499490f8 --- /dev/null +++ b/chroma_db/0895c0fd-aea2-4db2-9ea7-55fff1798415/link_lists.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855 +size 0 diff --git a/chroma_db/14bf569c-0412-405e-93d7-9bb5e922d766/data_level0.bin b/chroma_db/14bf569c-0412-405e-93d7-9bb5e922d766/data_level0.bin new file mode 100644 index 0000000000000000000000000000000000000000..0fee824df425bbd6a2ddf45894dfa0ad3633141e --- /dev/null +++ b/chroma_db/14bf569c-0412-405e-93d7-9bb5e922d766/data_level0.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b8146ecc3e4c3a36ea9b3edc3778630c452f483990ec942d38e8006f4661e430 +size 16760000 diff --git a/chroma_db/14bf569c-0412-405e-93d7-9bb5e922d766/header.bin b/chroma_db/14bf569c-0412-405e-93d7-9bb5e922d766/header.bin new file mode 100644 index 0000000000000000000000000000000000000000..51ca55818d180b505757e2a099351f24a94d6fbe --- /dev/null +++ b/chroma_db/14bf569c-0412-405e-93d7-9bb5e922d766/header.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:18f1e924efbb5e1af5201e3fbab86a97f5c195c311abe651eeec525884e5e449 +size 100 diff --git a/chroma_db/14bf569c-0412-405e-93d7-9bb5e922d766/length.bin b/chroma_db/14bf569c-0412-405e-93d7-9bb5e922d766/length.bin new file mode 100644 index 0000000000000000000000000000000000000000..0759830aaa2a38a9f8cb2b910a34075fdd4088f7 --- /dev/null +++ b/chroma_db/14bf569c-0412-405e-93d7-9bb5e922d766/length.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:19cd81fc650082cb4a5ef6a84ce1d62be39af6da611b317c12e9ebdd8c3fd603 +size 40000 diff --git a/chroma_db/14bf569c-0412-405e-93d7-9bb5e922d766/link_lists.bin b/chroma_db/14bf569c-0412-405e-93d7-9bb5e922d766/link_lists.bin new file mode 100644 index 0000000000000000000000000000000000000000..fc8e42b32efb9a9bf3ae0234a18a948e499490f8 --- /dev/null +++ b/chroma_db/14bf569c-0412-405e-93d7-9bb5e922d766/link_lists.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855 +size 0 diff --git a/chroma_db/7bd06384-5d74-44db-8b6f-08b881d638bb/data_level0.bin b/chroma_db/7bd06384-5d74-44db-8b6f-08b881d638bb/data_level0.bin new file mode 100644 index 0000000000000000000000000000000000000000..0fee824df425bbd6a2ddf45894dfa0ad3633141e --- /dev/null +++ b/chroma_db/7bd06384-5d74-44db-8b6f-08b881d638bb/data_level0.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b8146ecc3e4c3a36ea9b3edc3778630c452f483990ec942d38e8006f4661e430 +size 16760000 diff --git a/chroma_db/7bd06384-5d74-44db-8b6f-08b881d638bb/header.bin b/chroma_db/7bd06384-5d74-44db-8b6f-08b881d638bb/header.bin new file mode 100644 index 0000000000000000000000000000000000000000..51ca55818d180b505757e2a099351f24a94d6fbe --- /dev/null +++ b/chroma_db/7bd06384-5d74-44db-8b6f-08b881d638bb/header.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:18f1e924efbb5e1af5201e3fbab86a97f5c195c311abe651eeec525884e5e449 +size 100 diff --git a/chroma_db/7bd06384-5d74-44db-8b6f-08b881d638bb/length.bin b/chroma_db/7bd06384-5d74-44db-8b6f-08b881d638bb/length.bin new file mode 100644 index 0000000000000000000000000000000000000000..68d73c6f21b941c1595336c58ad34416609496a2 --- /dev/null +++ b/chroma_db/7bd06384-5d74-44db-8b6f-08b881d638bb/length.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:75312a06f8f6d7e8829ef723c70b4ce7aa48d66f2b4b8ba91ed0d6e82889c107 +size 40000 diff --git a/chroma_db/7bd06384-5d74-44db-8b6f-08b881d638bb/link_lists.bin b/chroma_db/7bd06384-5d74-44db-8b6f-08b881d638bb/link_lists.bin new file mode 100644 index 0000000000000000000000000000000000000000..fc8e42b32efb9a9bf3ae0234a18a948e499490f8 --- /dev/null +++ b/chroma_db/7bd06384-5d74-44db-8b6f-08b881d638bb/link_lists.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855 +size 0 diff --git a/chroma_db/a2f05e8f-89be-4d76-83a9-7a93d98352c3/data_level0.bin b/chroma_db/a2f05e8f-89be-4d76-83a9-7a93d98352c3/data_level0.bin new file mode 100644 index 0000000000000000000000000000000000000000..0fee824df425bbd6a2ddf45894dfa0ad3633141e --- /dev/null +++ b/chroma_db/a2f05e8f-89be-4d76-83a9-7a93d98352c3/data_level0.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b8146ecc3e4c3a36ea9b3edc3778630c452f483990ec942d38e8006f4661e430 +size 16760000 diff --git a/chroma_db/a2f05e8f-89be-4d76-83a9-7a93d98352c3/header.bin b/chroma_db/a2f05e8f-89be-4d76-83a9-7a93d98352c3/header.bin new file mode 100644 index 0000000000000000000000000000000000000000..51ca55818d180b505757e2a099351f24a94d6fbe --- /dev/null +++ b/chroma_db/a2f05e8f-89be-4d76-83a9-7a93d98352c3/header.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:18f1e924efbb5e1af5201e3fbab86a97f5c195c311abe651eeec525884e5e449 +size 100 diff --git a/chroma_db/a2f05e8f-89be-4d76-83a9-7a93d98352c3/length.bin b/chroma_db/a2f05e8f-89be-4d76-83a9-7a93d98352c3/length.bin new file mode 100644 index 0000000000000000000000000000000000000000..e6cfd56206c93a7875848cdf1ffda685ceb13983 --- /dev/null +++ b/chroma_db/a2f05e8f-89be-4d76-83a9-7a93d98352c3/length.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fdcfb6f1f26aa69981243c6b7f28102259bfd257a0d4f46ca0ad930da0e247e9 +size 40000 diff --git a/chroma_db/a2f05e8f-89be-4d76-83a9-7a93d98352c3/link_lists.bin b/chroma_db/a2f05e8f-89be-4d76-83a9-7a93d98352c3/link_lists.bin new file mode 100644 index 0000000000000000000000000000000000000000..fc8e42b32efb9a9bf3ae0234a18a948e499490f8 --- /dev/null +++ b/chroma_db/a2f05e8f-89be-4d76-83a9-7a93d98352c3/link_lists.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855 +size 0 diff --git a/chroma_db/chroma.sqlite3 b/chroma_db/chroma.sqlite3 new file mode 100644 index 0000000000000000000000000000000000000000..8aefc9eade84e4b8c0a8334e14e1016c61a8336b --- /dev/null +++ b/chroma_db/chroma.sqlite3 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:05db23910207a92cde2e7f64cb2269ce331113916513eee4eefd22e989a51345 +size 1581056 diff --git a/chroma_db/d08fb65a-aa5f-4895-9fa1-1c9781b3421e/data_level0.bin b/chroma_db/d08fb65a-aa5f-4895-9fa1-1c9781b3421e/data_level0.bin new file mode 100644 index 0000000000000000000000000000000000000000..0fee824df425bbd6a2ddf45894dfa0ad3633141e --- /dev/null +++ b/chroma_db/d08fb65a-aa5f-4895-9fa1-1c9781b3421e/data_level0.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b8146ecc3e4c3a36ea9b3edc3778630c452f483990ec942d38e8006f4661e430 +size 16760000 diff --git a/chroma_db/d08fb65a-aa5f-4895-9fa1-1c9781b3421e/header.bin b/chroma_db/d08fb65a-aa5f-4895-9fa1-1c9781b3421e/header.bin new file mode 100644 index 0000000000000000000000000000000000000000..51ca55818d180b505757e2a099351f24a94d6fbe --- /dev/null +++ b/chroma_db/d08fb65a-aa5f-4895-9fa1-1c9781b3421e/header.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:18f1e924efbb5e1af5201e3fbab86a97f5c195c311abe651eeec525884e5e449 +size 100 diff --git a/chroma_db/d08fb65a-aa5f-4895-9fa1-1c9781b3421e/length.bin b/chroma_db/d08fb65a-aa5f-4895-9fa1-1c9781b3421e/length.bin new file mode 100644 index 0000000000000000000000000000000000000000..5ef8d6296ca5788253bcad0e57ef6a6d86ea7a77 --- /dev/null +++ b/chroma_db/d08fb65a-aa5f-4895-9fa1-1c9781b3421e/length.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:378f45a28e48d022520cb763bf98854799329f959f17c44a79ba7198a8f6d1bb +size 40000 diff --git a/chroma_db/d08fb65a-aa5f-4895-9fa1-1c9781b3421e/link_lists.bin b/chroma_db/d08fb65a-aa5f-4895-9fa1-1c9781b3421e/link_lists.bin new file mode 100644 index 0000000000000000000000000000000000000000..fc8e42b32efb9a9bf3ae0234a18a948e499490f8 --- /dev/null +++ b/chroma_db/d08fb65a-aa5f-4895-9fa1-1c9781b3421e/link_lists.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855 +size 0 diff --git a/core/__pycache__/base_agent.cpython-310.pyc b/core/__pycache__/base_agent.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..ae4afb72aedb90a96f029d43ed00f5c31a1f8683 Binary files /dev/null and b/core/__pycache__/base_agent.cpython-310.pyc differ diff --git a/core/__pycache__/base_agent.cpython-311.pyc b/core/__pycache__/base_agent.cpython-311.pyc new file mode 100644 index 0000000000000000000000000000000000000000..93e025eabb1bcaec446410e6b271f65eb5a3de62 Binary files /dev/null and b/core/__pycache__/base_agent.cpython-311.pyc differ diff --git a/core/__pycache__/base_agent.cpython-313.pyc b/core/__pycache__/base_agent.cpython-313.pyc new file mode 100644 index 0000000000000000000000000000000000000000..79497d17b29f53e383f92bc2fb2ff191be4e0134 Binary files /dev/null and b/core/__pycache__/base_agent.cpython-313.pyc differ diff --git a/core/__pycache__/config.cpython-310.pyc b/core/__pycache__/config.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..2c466a16fcf25d5abeb6b2384534bba38d6df79c Binary files /dev/null and b/core/__pycache__/config.cpython-310.pyc differ diff --git a/core/__pycache__/config.cpython-311.pyc b/core/__pycache__/config.cpython-311.pyc new file mode 100644 index 0000000000000000000000000000000000000000..bc3f33f477acf86161c79e21a70c90c6f8206c85 Binary files /dev/null and b/core/__pycache__/config.cpython-311.pyc differ diff --git a/core/__pycache__/config.cpython-313.pyc b/core/__pycache__/config.cpython-313.pyc new file mode 100644 index 0000000000000000000000000000000000000000..955147fa5e0d7199010018136866ad2f9f5a91e4 Binary files /dev/null and b/core/__pycache__/config.cpython-313.pyc differ diff --git a/core/__pycache__/database.cpython-310.pyc b/core/__pycache__/database.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..22ed612284c03efef216ede0a16abf29e7f4c741 Binary files /dev/null and b/core/__pycache__/database.cpython-310.pyc differ diff --git a/core/__pycache__/database.cpython-311.pyc b/core/__pycache__/database.cpython-311.pyc new file mode 100644 index 0000000000000000000000000000000000000000..76b8c702e4549f76c8807b00f68afe0ff094bfcc Binary files /dev/null and b/core/__pycache__/database.cpython-311.pyc differ diff --git a/core/__pycache__/database.cpython-313.pyc b/core/__pycache__/database.cpython-313.pyc new file mode 100644 index 0000000000000000000000000000000000000000..ff9f2e7c72b64a1004f43da85fe27405616f50a8 Binary files /dev/null and b/core/__pycache__/database.cpython-313.pyc differ diff --git a/core/__pycache__/workflow.cpython-310.pyc b/core/__pycache__/workflow.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..ea5c981f7f01f5cbf1a0c3ca1cd8a116a91bac61 Binary files /dev/null and b/core/__pycache__/workflow.cpython-310.pyc differ diff --git a/core/__pycache__/workflow.cpython-311.pyc b/core/__pycache__/workflow.cpython-311.pyc new file mode 100644 index 0000000000000000000000000000000000000000..4021a99c77ad446db4ebe838655f4f5f61b3a3de Binary files /dev/null and b/core/__pycache__/workflow.cpython-311.pyc differ diff --git a/core/__pycache__/workflow.cpython-313.pyc b/core/__pycache__/workflow.cpython-313.pyc new file mode 100644 index 0000000000000000000000000000000000000000..24e9b61e2c38bab3457dcb2ab6acc4ab11c49c4e Binary files /dev/null and b/core/__pycache__/workflow.cpython-313.pyc differ diff --git a/core/base_agent.py b/core/base_agent.py new file mode 100644 index 0000000000000000000000000000000000000000..d4dab27d44b1a86ec0c8ebfd5ba2ffd3dcf3d9ef --- /dev/null +++ b/core/base_agent.py @@ -0,0 +1,36 @@ +from typing import Dict, Any, List +from langchain_groq import ChatGroq +from langchain_core.prompts import ChatPromptTemplate +from langchain_core.output_parsers import StrOutputParser +from langchain_core.runnables import RunnablePassthrough +from core.config import settings + +class BaseAgent: + def __init__(self, role: str): + self.role = role + self.llm = ChatGroq( + api_key=settings.GROQ_API_KEY, + model_name=settings.MODEL_NAME, + temperature=settings.TEMPERATURE + ) + self.chain = None + + def create_chain(self, template: str): + """Create a LangChain chain with the given template""" + prompt = ChatPromptTemplate.from_template(template) + self.chain = ( + {"input": RunnablePassthrough()} + | prompt + | self.llm + | StrOutputParser() + ) + + async def process(self, input_data: Dict[str, Any]) -> str: + """Process input data and return the result""" + if not self.chain: + raise ValueError("Chain not initialized. Call create_chain first.") + return await self.chain.ainvoke(input_data) + + def get_role(self) -> str: + """Get the agent's role""" + return self.role \ No newline at end of file diff --git a/core/config.py b/core/config.py new file mode 100644 index 0000000000000000000000000000000000000000..1df76f4d2e040fe122fb966d6117d0d210fc4d06 --- /dev/null +++ b/core/config.py @@ -0,0 +1,18 @@ +from pydantic_settings import BaseSettings +from typing import Optional +import os +from dotenv import load_dotenv + +load_dotenv() + +class Settings(BaseSettings): + GROQ_API_KEY: str = os.getenv("GROQ_API_KEY", "") + CHROMA_DB_PATH: str = "chroma_db" + MODEL_NAME: str = "llama-3.3-70b-versatile" + TEMPERATURE: float = 0.7 + MAX_TOKENS: int = 2048 + + class Config: + env_file = ".env" + +settings = Settings() \ No newline at end of file diff --git a/core/database.py b/core/database.py new file mode 100644 index 0000000000000000000000000000000000000000..c8abd10675ce77da5e02312d2b7bc49e7d5f9422 --- /dev/null +++ b/core/database.py @@ -0,0 +1,38 @@ +import chromadb +from chromadb.config import Settings +from core.config import settings +import os + +class Database: + def __init__(self): + self.client = chromadb.PersistentClient( + path=settings.CHROMA_DB_PATH, + settings=Settings(allow_reset=True) + ) + + def get_collection(self, name: str): + """Get or create a collection""" + try: + return self.client.get_collection(name) + except: + return self.client.create_collection(name) + + def store_artifact(self, collection_name: str, document: str, metadata: dict): + """Store a project artifact in the database""" + collection = self.get_collection(collection_name) + collection.add( + documents=[document], + metadatas=[metadata], + ids=[f"doc_{len(collection.get()['ids'])}"] + ) + + def query_artifacts(self, collection_name: str, query: str, n_results: int = 5): + """Query artifacts in a collection""" + collection = self.get_collection(collection_name) + results = collection.query( + query_texts=[query], + n_results=n_results + ) + return results + +db = Database() \ No newline at end of file diff --git a/core/workflow.py b/core/workflow.py new file mode 100644 index 0000000000000000000000000000000000000000..940ab371d89bf06d081447b7eb82e32eb0ed5208 --- /dev/null +++ b/core/workflow.py @@ -0,0 +1,127 @@ +from typing import Dict, Any, TypedDict, Sequence +from langgraph.graph import StateGraph +from agents.business_analyst import BusinessAnalystAgent +from agents.designer import DesignerAgent +from agents.developer import DeveloperAgent +from agents.tester import TesterAgent +from core.database import db + + +class AgentState(TypedDict): + requirements: str + user_stories: str + design: str + code: str + test_cases: str + test_results: str + current_step: str + status: str + messages: Sequence[str] + + +class DevelopmentWorkflow: + def __init__(self): + self.business_analyst = BusinessAnalystAgent() + self.designer = DesignerAgent() + self.developer = DeveloperAgent() + self.tester = TesterAgent() + self.workflow = self._create_workflow() + + def _create_workflow(self): + workflow = StateGraph(AgentState) + + workflow.add_node("create_user_stories", self._create_user_stories) + workflow.add_node("create_design", self._create_design) + workflow.add_node("generate_code", self._generate_code) + workflow.add_node("create_tests", self._create_tests) + workflow.add_node("execute_tests", self._execute_tests) + + workflow.add_edge("create_user_stories", "create_design") + workflow.add_edge("create_design", "generate_code") + workflow.add_edge("generate_code", "create_tests") + workflow.add_edge("create_tests", "execute_tests") + + workflow.set_entry_point("create_user_stories") + workflow.set_finish_point("execute_tests") + + return workflow.compile() + + async def _create_user_stories(self, state: AgentState) -> AgentState: + """Create user stories from requirements""" + result = await self.business_analyst.create_user_stories(state["requirements"]) + return { + **state, + "user_stories": result["user_stories"], + "current_step": "create_user_stories", + "status": "in_progress", + "messages": [*state["messages"], "User stories created successfully"] + } + + async def _create_design(self, state: AgentState) -> AgentState: + """Create design from user stories""" + result = await self.designer.create_design(state["user_stories"]) + return { + **state, + "design": result["design"], + "current_step": "create_design", + "status": "in_progress", + "messages": [*state["messages"], "Design created successfully"] + } + + async def _generate_code(self, state: AgentState) -> AgentState: + """Generate code from design""" + result = await self.developer.generate_code(state["design"]) + return { + **state, + "code": result["code"], + "current_step": "generate_code", + "status": "in_progress", + "messages": [*state["messages"], "Code generated successfully"] + } + + async def _create_tests(self, state: AgentState) -> AgentState: + """Create test cases from code""" + result = await self.tester.create_test_cases(state["code"]) + return { + **state, + "test_cases": result["test_cases"], + "current_step": "create_tests", + "status": "in_progress", + "messages": [*state["messages"], "Test cases created successfully"] + } + + async def _execute_tests(self, state: AgentState) -> AgentState: + """Execute tests on the code""" + result = await self.tester.execute_tests(state["code"], state["test_cases"]) + return { + **state, + "test_results": result["test_results"], + "current_step": "execute_tests", + "status": "completed", + "messages": [*state["messages"], "Tests executed successfully"] + } + + async def run(self, requirements: str) -> Dict[str, Any]: + """Run the complete workflow""" + initial_state = AgentState( + requirements=requirements, + user_stories="", + design="", + code="", + test_cases="", + test_results="", + current_step="start", + status="initialized", + messages=[] + ) + + final_state = await self.workflow.ainvoke(initial_state) + + return { + "status": "success", + "user_stories": final_state["user_stories"], + "design": final_state["design"], + "code": final_state["code"], + "test_results": final_state["test_results"], + "messages": final_state["messages"] + } \ No newline at end of file diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000000000000000000000000000000000000..0ad0a3078ee7f24e37b4c3f7e1c8ad1c36a72b82 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,12 @@ +langgraph +langchain +langchain-core +langchain-groq +chromadb +pydantic +python-dotenv +streamlit +pytest +pytest-asyncio +typing-extensions +pydantic-settings \ No newline at end of file diff --git a/tests/__pycache__/test_agents.cpython-310-pytest-8.3.5.pyc b/tests/__pycache__/test_agents.cpython-310-pytest-8.3.5.pyc new file mode 100644 index 0000000000000000000000000000000000000000..19992762cf364d87bbcdae57cc41d76ac82ee978 Binary files /dev/null and b/tests/__pycache__/test_agents.cpython-310-pytest-8.3.5.pyc differ diff --git a/tests/test_agents.py b/tests/test_agents.py new file mode 100644 index 0000000000000000000000000000000000000000..4987dc1555b72eea4e76978005045ff2874f2d99 --- /dev/null +++ b/tests/test_agents.py @@ -0,0 +1,46 @@ +import pytest +from agents.business_analyst import BusinessAnalystAgent +from agents.designer import DesignerAgent +from agents.developer import DeveloperAgent +from agents.tester import TesterAgent +from agents.project_manager import ProjectManagerAgent + +@pytest.mark.asyncio +async def test_business_analyst(): + agent = BusinessAnalystAgent() + requirements = "Create a simple todo list application" + result = await agent.create_user_stories(requirements) + assert result["status"] == "success" + assert "user_stories" in result + +@pytest.mark.asyncio +async def test_designer(): + agent = DesignerAgent() + user_stories = "As a user, I want to add tasks to my todo list" + result = await agent.create_design(user_stories) + assert result["status"] == "success" + assert "design" in result + +@pytest.mark.asyncio +async def test_developer(): + agent = DeveloperAgent() + design = "Create a REST API for todo list management" + result = await agent.generate_code(design) + assert result["status"] == "success" + assert "code" in result + +@pytest.mark.asyncio +async def test_tester(): + agent = TesterAgent() + code = "def add_task(task): return True" + result = await agent.create_test_cases(code) + assert result["status"] == "success" + assert "test_cases" in result + +@pytest.mark.asyncio +async def test_project_manager(): + agent = ProjectManagerAgent() + requirements = "Create a simple todo list application" + result = await agent.start_project(requirements) + assert result["status"] == "success" + assert all(key in result for key in ["user_stories", "design", "code", "test_results"]) \ No newline at end of file