@cascadeflow.agent()

def agent(
    budget: Optional[float] = None,
    compliance: Optional[str] = None,
    kpi_weights: Optional[dict[str, float]] = None,
    kpi_targets: Optional[dict[str, float]] = None,
    max_tool_calls: Optional[int] = None,
)

@cascadeflow.agent(budget=0.20)
async def my_agent(query: str):
    return await llm.complete(query)

@cascadeflow.agent(budget=0.50, compliance="gdpr")
async def eu_agent(query: str):
    return await llm.complete(query)

@cascadeflow.agent(
    budget=1.00,
    kpi_weights={"quality": 0.8, "cost": 0.2},
    kpi_targets={"quality": 0.9},
)
async def premium_agent(query: str):
    return await llm.complete(query)

@cascadeflow.agent(budget=0.10, kpi_weights={"cost": 0.9, "quality": 0.1})
async def triage_agent(query: str):
    return await llm.complete(query)

@cascadeflow.agent(budget=2.00, kpi_weights={"quality": 0.9, "cost": 0.1})
async def analysis_agent(query: str):
    return await llm.complete(query)