SolAgent-FHE

AI-powered FHE (Fully Homomorphic Encryption) smart contract generator for Fhenix CoFHE.

Generate privacy-preserving Solidity contracts from natural language descriptions. The AI understands FHE patterns, encrypted types, access control, and compiles contracts automatically using Hardhat.

Installation

pip install solagent-fhe

Requirements: Node.js 18+ and the cofhe-hardhat-starter project for compilation.

Quick Start

CLI

# Generate a contract from description
solagent-fhe "Create a private voting contract with encrypted vote counts"

# Use a predefined task
solagent-fhe --task-key voting

# Run benchmark (all 9 contract types)
solagent-fhe --benchmark

# List available tasks
solagent-fhe --list-tasks

# Use Groq instead of DeepSeek
solagent-fhe --provider groq "Create a sealed-bid auction"

Python API

import asyncio
from solagent_fhe import generate, benchmark, FHE_TASKS

async def main():
    # Generate a single contract
    result = await generate(
        task="Create a private token with encrypted balances",
        contract_name="MyToken",
        provider="deepseek",  # or "groq"
    )

    if result['compiled']:
        print(f"Contract generated: {result['name']}.sol")
        print(f"Refine rounds: {result['rounds']}")
        print(result['code'])

    # Run full benchmark
    results = await benchmark(provider="deepseek")

asyncio.run(main())

Environment Variables

*At least one LLM provider key required.

Supported Contract Types

How It Works

Description ──→ [Code Agent] ──→ Solidity ──→ [Hardhat Compile]
                  (DeepSeek)        │              │
                                    │         ✓ Success → Done
                                    │              │
                                    │         ✗ Errors
                                    │              │
                                    └──→ [Refine Agent] ──→ Fix ──→ Recompile
                                          (Groq/DeepSeek)     (up to 10 rounds)

Benchmark: 5/5 contracts compile successfully (100%)

FHE Patterns

The generator understands all Fhenix CoFHE patterns:

• Encrypted types: euint8–euint256, ebool, eaddress
• Operations: FHE.add, FHE.sub, FHE.select, FHE.gte, etc.
• Access control: FHE.allowThis(), FHE.allow(ct, addr)
• Decryption: Async FHE.decrypt() + getDecryptResultSafe()
• Random: FHE.randomEuint32() for on-chain randomness
• Input: InEuint64 for client-encrypted values

License

MIT