Pulastya0 commited on
Commit
57aaf9e
·
1 Parent(s): 12d7bc6

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +5 -1
main.py CHANGED
@@ -39,13 +39,17 @@ class InternshipData(BaseModel):
39
  createdAt: str
40
  stipend: int = None
41
 
 
 
 
 
42
  # --- NEW: More descriptive response models ---
43
  class InternshipRecommendation(InternshipData):
44
  internship_id: str
45
  score: float
46
 
47
  class RecommendationResponse(BaseModel):
48
- recommendations: List[InternshipRecommendation]
49
 
50
  class StatusResponse(BaseModel):
51
  status: str
 
39
  createdAt: str
40
  stipend: int = None
41
 
42
+ class SimpleRecommendation(BaseModel):
43
+ internship_id: str
44
+ score: float
45
+
46
  # --- NEW: More descriptive response models ---
47
  class InternshipRecommendation(InternshipData):
48
  internship_id: str
49
  score: float
50
 
51
  class RecommendationResponse(BaseModel):
52
+ recommendations: List[SimpleRecommendation]
53
 
54
  class StatusResponse(BaseModel):
55
  status: str