Spaces:
No application file
No application file
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] | |