acra-cli 0.1.1__py3-none-any.whl

acra/__init__.py

@@ -0,0 +1,7 @@
+"""acra package root."""
+
+__version__ = "0.1.0"
+
+from .cli import main
+
+__all__ = ["main", "__version__"]

acra/__main__.py

@@ -0,0 +1,4 @@
+from .cli import app_main
+
+if __name__ == "__main__":
+    app_main()

acra/agents/coder/coder_agent.py

@@ -0,0 +1,235 @@
+from typing import Dict, Optional
+import re
+
+from langchain_core.exceptions import OutputParserException
+from langchain_core.messages import AIMessage
+
+from acra.graph.state import AgentState
+from acra.agents.coder.coder_chain import create_coder_chain
+from acra.schemas.coder_schema import CoderOutput
+
+
+def _build_fallback_response(project_name: str, entry_point: str, generated_files: Dict) -> CoderOutput:
+    """Create a safe fallback coder response when structured output parsing fails."""
+    return CoderOutput(
+        generated_files=generated_files or {},
+        explanation="Generated project files from the latest workflow state.",
+        coding_status="completed",
+        next_agent="executor",
+        project_name=project_name or "current_project",
+        entry_point=entry_point or "app.py",
+    )
+
+
+def _extract_project_name(user_request: str) -> str:
+    """
+    Extract project name from user request.
+    
+    Tries to extract from:
+    1. Quoted names: "project 'name'" or "create 'name'"
+    2. Filenames: "calculator.html" -> "calculator"
+    3. First few words of request
+    """
+    
+    # Try quoted names: 'project_name' or "project_name"
+    quoted_match = re.search(r"['\"]([^'\"]+)['\"]", user_request)
+    if quoted_match:
+        name = quoted_match.group(1)
+        # Remove file extension if present
+        name = name.rsplit('.', 1)[0]
+        # Sanitize name (remove spaces, special chars)
+        name = re.sub(r'[^\w-]', '_', name)
+        if name:
+            return name
+    
+    # Fallback: use first few words (max 2)
+    words = user_request.split()[:2]
+    project_name = '_'.join(words).lower()
+    project_name = re.sub(r'[^\w-]', '_', project_name)
+    
+    return project_name or "generated_project"
+
+
+def coder_agent(state: AgentState) -> Dict:
+    """
+    Coding Agent
+
+    Responsibilities:
+    - Generate source code
+    - Create project files
+    - Fix execution errors
+    - Update generated codebase
+    - Prepare output for execution
+    """
+
+    #extract state
+
+    user_request = state.get("user_request", "")
+    
+    # Extract or preserve project name
+    project_name = state.get("project_name", "")
+    if not project_name:
+        project_name = _extract_project_name(user_request)
+
+    current_step = state.get("current_step", "")
+
+    plan = state.get("plan", [])
+
+    generated_files = state.get(
+        "generated_files",
+        {}
+    )
+
+    interactive = state.get("interactive", False)
+
+    error_message = state.get(
+        "error_message",
+        ""
+    )
+
+    entry_point = state.get("entry_point", "app.py") or "app.py"
+
+    retry_count = state.get(
+        "retry_count",
+        0
+    )
+
+    critic_feedback = state.get(
+        "critic_feedback",
+        ""
+    )
+
+    research_data = state.get(
+        "research_data",
+        []
+    )
+
+
+    #create coder chain
+
+    coder_chain = create_coder_chain()
+
+
+    #invoke coder chain
+
+    used_fallback = False
+    try:
+        response = coder_chain.invoke({
+
+            "user_request": user_request,
+
+            "current_step": current_step,
+
+            "plan": plan,
+
+            "generated_files": generated_files,
+
+            "error_message": error_message,
+
+            "retry_count": retry_count,
+
+            "critic_feedback": critic_feedback,
+
+            "research_data": research_data,
+        })
+
+        #extract structured output
+
+        updated_files = response.generated_files
+        explanation = response.explanation
+        coding_status = response.coding_status
+        entry_point = getattr(response, "entry_point", entry_point) or entry_point
+        next_agent = response.next_agent
+
+    except (OutputParserException, AttributeError, TypeError, ValueError) as exc:
+        response = _build_fallback_response(
+            project_name=project_name,
+            entry_point=entry_point,
+            generated_files=generated_files,
+        )
+        updated_files = response.generated_files
+        explanation = response.explanation
+        coding_status = response.coding_status
+        entry_point = response.entry_point
+        next_agent = response.next_agent
+        used_fallback = True
+        if error_message:
+            explanation = f"{explanation} Fallback due to parser error: {exc}"
+
+    if interactive and getattr(response, "requires_human_approval", False):
+        next_agent = "human"
+
+
+    #coder workflow logic
+
+    # first genration
+    if used_fallback:
+        coder_message = (
+            "⚠️ Coding Agent: "
+            "The model returned an incomplete structured response. "
+            "Using a safe fallback structure to continue the workflow."
+        )
+    elif retry_count == 0 and not error_message:
+
+        coder_message = (
+            "💻 Coding Agent: "
+            "Generating project files and implementation..."
+        )
+
+    # error fioxing mode
+    elif error_message:
+
+        coder_message = (
+            "🛠️ Coding Agent: "
+            "Execution failure detected.\n\n"
+            f"Error:\n{error_message}\n\n"
+            "Attempting autonomous repair..."
+        )
+
+    # critic revision mode
+    elif critic_feedback:
+
+        coder_message = (
+            "🧪 Coding Agent: "
+            "Critic feedback received.\n\n"
+            "Improving implementation quality..."
+        )
+
+    # general update
+    else:
+
+        coder_message = (
+            "⚡ Coding Agent: "
+            "Updating implementation..."
+        )
+
+
+    #return update state
+
+    return {
+
+        # convarsation update
+        "messages": [
+            AIMessage(content=coder_message)
+        ],
+
+        # generate codebase 
+        "generated_files": updated_files,
+        "project_name": project_name,
+        "entry_point": entry_point,
+
+        # coding metadata
+        "coding_status": coding_status,
+        "coding_explanation": explanation,
+
+        # workflow
+        "next_agent": next_agent,
+        "current_agent": "coder",
+
+        # reset execution state before rerun
+        "execution_success": False,
+        "execution_output": "",
+
+        # clear previous errors after regeneration
+        "error_message": "",
+    }

acra/agents/coder/coder_chain.py

@@ -0,0 +1,204 @@
+from langchain_core.prompts import ChatPromptTemplate
+from acra.agents.llm import llm
+from acra.schemas.coder_schema import CoderOutput
+
+
+# create coder chain
+
+def create_coder_chain():
+    """
+    Creates the Coding Agent chain.
+
+    Responsibilities:
+    - Generate source code
+    - Create project files
+    - Fix runtime errors
+    - Improve existing code
+    - Respond to critic feedback
+    """
+
+    model = llm()
+
+    coder_prompt = ChatPromptTemplate.from_messages([
+
+        (
+            "system",
+
+            """
+You are the Coding Agent inside an autonomous AI platform called AgentForge.
+
+You are an expert AI software engineer.
+
+Your responsibilities:
+- Generate production-quality code
+- Create complete project files
+- Fix runtime errors
+- Improve architecture
+- Modify existing implementations
+- Respond to critic feedback
+- Ensure clean and executable code
+
+==================================================
+YOUR CAPABILITIES
+==================================================
+
+You can:
+- generate Python applications
+- create APIs
+- write backend systems
+- implement ML pipelines
+- create tests
+- generate configuration files
+- fix broken code
+- improve code quality
+
+==================================================
+IMPORTANT RULES
+==================================================
+
+1. ALWAYS generate executable code
+2. NEVER generate incomplete placeholders
+3. NEVER leave TODO comments
+4. ALWAYS return complete file contents
+5. ALWAYS preserve working code unless fixing issues
+6. Keep architecture modular and clean
+7. Prefer readability and maintainability
+8. Handle runtime errors carefully
+9. Generate minimal but functional implementations
+10. Return ONLY structured output
+11. YOU MUST INCLUDE EVERY REQUIRED FIELD IN THE RESPONSE
+12. DO NOT OMIT explanation, coding_status, next_agent, entry_point, or generated_files
+13. If a field is not applicable, still provide a safe default value rather than omitting it
+14. generated_files must be a dictionary mapping filenames to full file contents
+15. If you are unsure, return a conservative working implementation instead of an empty response
+
+==================================================
+CURRENT WORKFLOW STATE
+==================================================
+
+USER REQUEST:
+{user_request}
+
+CURRENT TASK:
+{current_step}
+
+FULL EXECUTION PLAN:
+{plan}
+
+PREVIOUSLY GENERATED FILES:
+{generated_files}
+
+EXECUTION ERROR:
+{error_message}
+
+RETRY COUNT:
+{retry_count}
+
+CRITIC FEEDBACK:
+{critic_feedback}
+
+RESEARCH DATA:
+{research_data}
+
+==================================================
+YOUR TASK
+==================================================
+
+Analyze the workflow state carefully.
+
+Then:
+
+1. Generate or update project files
+2. Fix runtime issues if errors exist
+3. Improve implementation if critic feedback exists
+4. Maintain consistency across all files
+5. Prepare project for execution
+
+==================================================
+OUTPUT REQUIREMENTS
+==================================================
+
+You MUST return ALL of the following fields and they must be present in every response:
+
+1. generated_files
+   - dictionary of filename → file content
+   - must not be empty unless the task is impossible
+
+2. explanation
+   - short explanation of changes
+   - never omit this field
+
+3. coding_status
+   - one of: generating, updating, fixing, improving, completed, failed
+   - never omit this field
+
+4. next_agent
+   - one of: executor, critic, human
+   - never omit this field
+
+5. entry_point
+   - the file that should be executed to start the project, for example "app.py" or "main.py"
+   - never omit this field
+
+6. project_name
+   - a valid project name string
+   - never omit this field
+
+Before responding, verify that every required field exists and is non-empty.
+If any field is missing in your draft, fill it with a safe default before finalizing.
+
+==================================================
+CODING STATUS OPTIONS
+==================================================
+
+- generating
+- updating
+- fixing
+- improving
+- completed
+
+==================================================
+NEXT AGENT OPTIONS
+==================================================
+
+- executor
+- critic
+
+==================================================
+IMPORTANT
+==================================================
+
+- If execution errors exist:
+  focus on fixing the errors.
+
+- If critic feedback exists:
+  improve code quality and architecture.
+
+- If no files exist:
+  generate the initial implementation.
+
+- Always return FULL updated file contents.
+"""
+        ),
+
+        (
+            "human",
+
+            """
+Generate or update the project implementation based on the workflow state.
+"""
+        )
+
+    ])
+
+
+    # structured output 
+
+    structured_llm = model.with_structured_output(
+        CoderOutput
+    )
+
+
+    #return chain
+
+    return coder_prompt | structured_llm

acra/agents/critic/critic_agent.py

@@ -0,0 +1,199 @@
+from typing import Dict
+
+from langchain_core.messages import AIMessage
+
+from acra.graph.state import AgentState
+
+from acra.agents.critic.critic_chain import (
+    create_critic_chain
+)
+
+
+def critic_agent(state: AgentState) -> Dict:
+    """
+    Critic Agent
+
+    Responsibilities:
+    - Evaluate generated code
+    - Review execution results
+    - Detect vulnerabilities
+    - Analyze architecture quality
+    - Decide whether code is acceptable
+    """
+
+    #extract state
+
+    user_request = state.get(
+        "user_request",
+        ""
+    )
+
+    current_step = state.get(
+        "current_step",
+        ""
+    )
+
+    generated_files = state.get(
+        "generated_files",
+        {}
+    )
+
+    interactive = state.get("interactive", False)
+
+    execution_success = state.get(
+        "execution_success",
+        False
+    )
+
+    execution_logs = state.get(
+        "execution_logs",
+        ""
+    )
+
+    execution_output = state.get(
+        "execution_output",
+        ""
+    )
+
+    error_message = state.get(
+        "error_message",
+        ""
+    )
+
+    retry_count = state.get(
+        "retry_count",
+        0
+    )
+
+
+    #create chain
+
+    critic_chain = create_critic_chain()
+
+
+    #invoke critic agent chain
+
+    response = critic_chain.invoke({
+
+        "user_request": user_request,
+
+        "current_step": current_step,
+
+        "generated_files": generated_files,
+
+        "execution_success": execution_success,
+
+        "execution_logs": execution_logs,
+
+        "execution_output": execution_output,
+
+        "error_message": error_message,
+
+        "retry_count": retry_count,
+    })
+
+
+    #extract structured response
+
+    review_status = response.review_status
+
+    quality_score = response.quality_score
+
+    summary = response.summary
+
+    feedback = response.feedback
+
+    next_agent = response.next_agent
+
+    if interactive and getattr(response, "requires_human_approval", False):
+        next_agent = "human"
+
+    issues = response.issues or []
+
+    security_issues = (
+        response.security_issues or []
+    )
+
+    improvement_suggestions = (
+        response.improvement_suggestions or []
+    )
+
+
+    #review success
+
+    if review_status == "approved":
+
+        critic_message = (
+            "Critic Agent: "
+            "Implementation approved.\n\n"
+            f"Quality Score: {quality_score}/10\n\n"
+            f"Summary:\n{summary}"
+        )
+
+    #review failed
+
+    elif review_status in [
+        "needs_improvement",
+        "failed"
+    ]:
+
+        critic_message = (
+            "Critic Agent: "
+            "Issues detected in implementation.\n\n"
+            f"Quality Score: {quality_score}/10\n\n"
+            f"Feedback:\n{feedback}"
+        )
+
+    #security failure
+
+    elif review_status == "unsafe":
+
+        critic_message = (
+            "Critic Agent: "
+            "Security risks detected.\n\n"
+            f"Security Issues:\n"
+            f"{security_issues}"
+        )
+
+    #feedback
+
+    else:
+
+        critic_message = (
+            "Critic Agent: "
+            "Unexpected review state encountered."
+        )
+
+
+    #return update state
+
+    return {
+
+        "messages": [
+            AIMessage(content=critic_message)
+        ],
+
+        # Review Results
+        "review_status": review_status,
+
+        "quality_score": quality_score,
+
+        "critic_feedback": feedback,
+
+        "critic_summary": summary,
+
+        "review_issues": issues,
+
+        "security_issues": security_issues,
+
+        "improvement_suggestions": (
+            improvement_suggestions
+        ),
+
+        # Workflow
+        "next_agent": next_agent,
+        "current_agent": "critic",
+
+        # Reset retry count on approval so future workflow phases start fresh
+        "retry_count": 0 if review_status == "approved" else state.get("retry_count", 0),
+    }