Refactor recommendation request and mentee query text handling

- Updated the RecommendationRequest model to remove unnecessary fields and added `mentor_domain_ids` and `career_id` for improved data structure.
- Modified the `build_mentee_query_text` function to reflect changes in the mentee data structure, enhancing clarity by including `career_id`, `domain_ids`, and `mentor_domain_ids` while removing references to `desired_skills`, `current_skills`, and `interests`.

models/schemas.py

@@ -51,15 +51,10 @@ class MentorUpsertResponse(BaseModel):
 
 class RecommendationRequest(BaseModel):
     goals: Optional[str] = None
-    desired_skills: Optional[List[Union[SkillDto, Dict[str, Any], str]]] = None
-    current_skills: Optional[List[Union[SkillDto, Dict[str, Any], str]]] = None
-    interests: Optional[List[Union[DomainDto, Dict[str, Any], str]]] = None
-    domains: Optional[List[Union[DomainDto, Dict[str, Any], str]]] = None
-    min_rating: Optional[float] = Field(None, ge=0.0, le=5.0)
     skill_ids: Optional[List[int]] = None
     domain_ids: Optional[List[int]] = None
+    mentor_domain_ids: Optional[List[int]] = None
     career_id: Optional[int] = None
-    status: Optional[str] = "ACTIVATED"
     top_k: Optional[int] = Field(None, ge=1, le=100)
     final_count: Optional[int] = Field(None, ge=1, le=20)
 

test_api.py

@@ -10,16 +10,10 @@ def test_recommend():
     
     recommendation_data = {
         "goals": "Tôi muốn học Python và FastAPI để phát triển web",
-        "desired_skills": [
-            {"id": 1, "name": "Python"},
-            {"id": 2, "name": "FastAPI"}
-        ],
-        "interests": [
-            {"id": 1, "name": "Web Development"}
-        ],
-        "min_rating": 4.0,
-        "skill_ids": [1, 2],
+        "career_id": 1,
         "domain_ids": [1],
+        "mentor_domain_ids": [2, 3],
+        "skill_ids": [1, 2],
         "top_k": 30,
         "final_count": 6
     }

utils/text_builder.py

@@ -108,48 +108,21 @@ def build_mentee_query_text(mentee_data: Dict[str, Any]) -> str:
     if goals:
         parts.append(f"Goals: {goals}")
     
-    desired_skills = mentee_data.get("desired_skills", [])
-    if desired_skills:
-        skill_names = []
-        for skill in desired_skills:
-            if isinstance(skill, dict):
-                skill_names.append(skill.get("name", ""))
-            elif isinstance(skill, str):
-                skill_names.append(skill)
-        if skill_names:
-            parts.append(f"Desired Skills: {', '.join(skill_names)}")
+    career_id = mentee_data.get("career_id")
+    if career_id:
+        parts.append(f"Career ID: {career_id}")
     
-    current_skills = mentee_data.get("current_skills", [])
-    if current_skills:
-        skill_names = []
-        for skill in current_skills:
-            if isinstance(skill, dict):
-                skill_names.append(skill.get("name", ""))
-            elif isinstance(skill, str):
-                skill_names.append(skill)
-        if skill_names:
-            parts.append(f"Current Skills: {', '.join(skill_names)}")
+    domain_ids = mentee_data.get("domain_ids", [])
+    if domain_ids:
+        parts.append(f"Preferred Domains (IDs): {', '.join(map(str, domain_ids))}")
     
-    interests = mentee_data.get("interests", [])
-    domains = mentee_data.get("domains", [])
-    if interests:
-        domain_names = []
-        for domain in interests:
-            if isinstance(domain, dict):
-                domain_names.append(domain.get("name", ""))
-            elif isinstance(domain, str):
-                domain_names.append(domain)
-        if domain_names:
-            parts.append(f"Interests: {', '.join(domain_names)}")
-    elif domains:
-        domain_names = []
-        for domain in domains:
-            if isinstance(domain, dict):
-                domain_names.append(domain.get("name", ""))
-            elif isinstance(domain, str):
-                domain_names.append(domain)
-        if domain_names:
-            parts.append(f"Interests: {', '.join(domain_names)}")
+    mentor_domain_ids = mentee_data.get("mentor_domain_ids", [])
+    if mentor_domain_ids:
+        parts.append(f"Previous Mentor Domains (IDs): {', '.join(map(str, mentor_domain_ids))}")
+    
+    skill_ids = mentee_data.get("skill_ids", [])
+    if skill_ids:
+        parts.append(f"Skills of Previous Mentors (IDs): {', '.join(map(str, skill_ids))}")
     
     return "\n".join(parts)