Plans and quotas

if context.user.plan == "cycls_pass":
    yield "Premium features are enabled."
else:
    yield "Upgrade for full access."
    yield {"type": "ui", "action": "open_plan_modal"}

from datetime import datetime, timezone
import cycls

LIMITS = {"u:free_user": 14}   # messages a month per plan; a plan not listed has no limit


def next_month(now):
    return f"{now.year + 1}-01-01" if now.month == 12 else f"{now.year}-{now.month + 1:02d}-01"


@cycls.agent(image=image, web=web, volumes={"/workspace": cycls.Volume("my-agent")})
async def my_agent(context):
    user = context.user
    exempt = not context.prod                      # local runs are never blocked

    if user.plan == "o:free_org" and not exempt:
        cycls.log("cap_hit", user=user, chat_id=context.chat_id, kind="org_free")
        yield {"type": "callout", "callout": "This workspace needs a paid plan.", "style": "error"}
        yield {"type": "ui", "action": "open_plan_modal"}
        return

    db = cycls.DB(context.workspace)
    now = datetime.now(timezone.utc)
    month = now.strftime("%Y-%m")
    entry = await db.get(f"usage/{month}", {"count": 0})
    limit = None if exempt else LIMITS.get(user.plan)
    usage = lambda n: {"type": "usage", "used": n, "limit": limit, "per": "month", "resets": next_month(now)}

    if limit and entry["count"] >= limit:
        cycls.log("cap_hit", user=user, chat_id=context.chat_id,
                  kind="user_free_monthly", count=entry["count"])
        yield usage(entry["count"])
        yield {"type": "ui", "action": "open_plan_modal", "reason": "limit"}
        return

    entry["count"] += 1
    await db.put(f"usage/{month}", entry)
    if limit:
        yield usage(entry["count"])

    async for ev in llm.run(context=context):
        yield ev

web = cycls.Web().plans(
    pay=("apple_pay", "google_pay", "visa", "mastercard"),
    currency=("SAR", 3.75),
    words={
        "pro": {"ar": {"name": "برو", "description": "لمن يستخدم الذكاء الاصطناعي كل يوم."}},
        "memory": {"ar": {"name": "الذاكرة"}},
    },
)

llm = cycls.LLM().price(input=3, output=15, cache_read=0.30, cache_write=6)

iap = cycls.AppleIAP(
    bundle_id="com.example.app",
    products={"com.example.app.pro.monthly": "u:pro"},
    namespace="00000000-0000-0000-0000-000000000000",
)

web = cycls.Web().auth(cycls.Clerk()).iap(iap)

web = cycls.Web().affiliate(os.environ["REWARDFUL_KEY"])