wake-ai

🥈Silver

Wake AI is an orchestration framework for structured multi-step workflows with validation. It wraps terminal-based AI agents like Claude Code to build reliable smart contract security analysis and other workflows. It benefits operations teams by making agentic execution more predictable and reliable.

2020Updated 2mo ago

Intermediate30min to implementautomation

Saves ~240 min per use

Quick InstallView Source

git clone https://github.com/wakehacker/wake-ai.git

Works with:

Claude

Overview

About This Skill

Wake AI orchestrates complex AI workflows by breaking tasks into discrete steps with validation between each, ensuring predictable and reproducible results. It wraps terminal-based AI agents like Claude Code to handle smart contract security audits, vulnerability detection, and other structured processes. Each workflow step has clear inputs, outputs, and success criteria, with validation ensuring quality before proceeding to the next stage. The framework uses isolated working directories for context sharing between agents and steps, eliminating state-passing complexity. Operations teams and security auditors benefit from cost-controlled, transparent execution with automatic retries and structured JSON outputs for integration with other tools.

How to Use

Install with `pip install wake-ai`. Run pre-built workflows like `wake-ai audit` for comprehensive security analysis or `wake-ai reentrancy` for vulnerability detection. Define custom workflows by extending the AIWorkflow class with steps, validators, and prompts. Preserve working directories with the `--no-cleanup` flag to review outputs after completion.

Use Cases

Smart contract security audits with multi-step analysis

Reentrancy vulnerability detection in Solidity code

Structured code analysis workflows with validation checkpoints

Multi-agent collaboration on complex vulnerability detection

Setup & Installation

Quick Install

No install command available. Check the GitHub repository for manual installation instructions.

Alternative Install (Git Clone)

git clone https://github.com/wakehacker/wake-ai

Requirements

Claude Code or compatible AI agent
Works with: Claude

Quick Start Guide

Install the Skill

Copy the install command above and run it in your terminal.

Open Your AI Agent

Launch Claude Code, Cursor, or your preferred AI coding agent.

Try It Out

Use the prompt template or examples below to test the skill.

Customize

Adapt the skill to your specific use case and workflow.

Usage Examples

Prompt Template

Analyze the following smart contract code for security vulnerabilities using Wake AI. The contract is written in Solidity and is part of a decentralized finance (DeFi) platform. Here is the code: [SMART_CONTRACT_CODE]. Please provide a detailed analysis, including potential risks and recommendations for mitigation.

Example Output

# Smart Contract Security Analysis

## Overview
The provided Solidity smart contract is part of a DeFi lending platform. The contract manages user deposits, withdrawals, and interest calculations. Below is a detailed security analysis.

## Potential Vulnerabilities
- **Reentrancy Risk**: The contract's withdrawal function does not use the Checks-Effects-Interactions pattern, which could allow for reentrancy attacks.
- **Integer Overflow/Underflow**: The contract does not use SafeMath or similar libraries to prevent integer overflows or underflows in arithmetic operations.
- **Access Control Issues**: The contract lacks proper access control mechanisms, allowing unauthorized users to call critical functions.

## Recommendations
- **Implement Checks-Effects-Interactions**: Restructure the withdrawal function to follow the Checks-Effects-Interactions pattern to prevent reentrancy.
- **Use SafeMath**: Integrate SafeMath or similar libraries to handle arithmetic operations safely.
- **Enhance Access Control**: Implement proper access control mechanisms, such as role-based access control, to restrict access to sensitive functions.

## Conclusion
Addressing the identified vulnerabilities will significantly enhance the security of the smart contract and protect users' funds.