src/models/models.py

"""
Pydantic models for Trademark Valuation API
"""
from pydantic import BaseModel
from typing import Optional


class TrademarkRequest(BaseModel):
    """Request model for trademark valuation by serial number"""
    serial_number: str


class ValuationResponse(BaseModel):
    """Response model with simplified valuation results"""
    final_value: str
    confidence: int
    status: str
    market_dominance: str