File size: 420 Bytes
480e694
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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]