chongming-plugins/example/router/divide.py

import time

from chongming_worker import rpc_method

from ..models.pydantic import (
    DivideGetResponse,
    DivideGetRequest,
)

@rpc_method(
    subject="example.divide",
    method="GET",
    summary="Divide two numbers",
    response_model=DivideGetResponse,
)
async def divide(divide_get_request: DivideGetRequest) -> DivideGetResponse:
    """除法运算：a / b（b 不能为 0）"""
    if divide_get_request.b == 0:
        return DivideGetResponse(
            error="除数不能为零",
            operation="divide",
            result=0.0,
            timestamp=time.time(),
        )
    return DivideGetResponse(
        result=divide_get_request.a / divide_get_request.b,
        operation="divide",
        timestamp=time.time(),
    )