VincentA2K's picture
Product Recommendation RestAPI
480e694
raw
history blame contribute delete
420 Bytes
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]