from pydantic import BaseModel
from typing import List, Optional

class RecommendationRequest(BaseModel):
    query: str
    top_n: Optional[int] = 5

class ProductRecommendation(BaseModel):
    product_id: str
    product_name: str
    description: str
    brand: str
    price: float
    categories: List[str]
    score: float

class RecommendationResponse(BaseModel):
    recommendations: List[ProductRecommendation]