"""
SolAgent-FHE — AI-powered FHE smart contract generator for Fhenix CoFHE.

Usage:
    from solagent_fhe import generate, benchmark, FHE_TASKS

    # Generate a contract
    result = await generate("Create a private voting contract")

    # Run benchmark
    results = await benchmark()
"""

from solagent_fhe.generator import generate_contract as generate, run_benchmark as benchmark, FHE_TASKS
from solagent_fhe.llm import chat, extract_solidity
from solagent_fhe.compiler import run_compile, write_contract, read_contract

__version__ = "0.1.0"
__all__ = [
    "generate",
    "benchmark",
    "FHE_TASKS",
    "chat",
    "extract_solidity",
    "run_compile",
    "write_contract",
    "read_contract",
]
