subtract.py•1.08 kB
from fastapi import APIRouter
from pydantic import BaseModel
router = APIRouter()
class SubtractRequest(BaseModel):
a: float
b: float
class Config:
json_schema_extra = {
"example": {
"a": 20.0,
"b": 5.5
}
}
class SubtractResponse(BaseModel):
result: float
operation: str
inputs: dict
class Config:
json_schema_extra = {
"example": {
"result": 14.5,
"operation": "subtraction",
"inputs": {"a": 20.0, "b": 5.5}
}
}
@router.post("/subtract", response_model=SubtractResponse)
def subtract_numbers(request: SubtractRequest):
"""
Subtract b from a.
Args:
request: SubtractRequest containing two numbers 'a' and 'b'
Returns:
SubtractResponse with the difference (a - b)
"""
result = request.a - request.b
return SubtractResponse(
result=result,
operation="subtraction",
inputs={"a": request.a, "b": request.b}
)