Merge pull request #51 from PentabyteDevAlign/feat/profile-menu

Feat/profile menu

Author: zrahadiz

AI/src/agents/recommendation_agent/model.py

@@ -5,6 +5,7 @@
 
 class CandidateModel(BaseModel):
     """Model to represent a candidate with relevant metrics."""
+    _id: str
     name: str
     position: str
     skills: List[str]

AI/src/api/roster.py

@@ -289,6 +289,7 @@ def cosine_similarity(vec_a, vec_b):
 
         # 5. Merge all the data
         result = {
+            "_id": str(user_id),
             "name": user.get("name"),
             "position": position_name,
             "skills":  [skill["name"] for skill in skills],
@@ -313,6 +314,7 @@ def cosine_similarity(vec_a, vec_b):
     # group by position
     grouped = defaultdict(list)
     for s in scores:
+        print(s)
         grouped[s["position"]].append(s)
 
     required_map = {p.name: p.numOfRequest for p in required_positions}
@@ -339,6 +341,7 @@ def cosine_similarity(vec_a, vec_b):
             project_description=project_description,
             candidates=candidates
         )
+        print(response)
 
         indexes = response.ordered_indexes
         reasoning = response.reasoning
@@ -354,9 +357,13 @@ def cosine_similarity(vec_a, vec_b):
 
         # Terapkan urutan ke kandidat
         ordered = [top_candidates[position][i] for i in indexes]
+        print("odde")
+        print(ordered)
 
         # Tambahkan rank number
         for idx, c in enumerate(ordered, start=1):
+            print("caca")
+            print(c)
             c["rank"] = idx
             c["reason"] = reasoning
 
@@ -365,5 +372,6 @@ def cosine_similarity(vec_a, vec_b):
 
     logs["total_execution_time"] = time.time() - total_start_time
     print(logs)
-
+    print("ptada")
+    print(top_candidates)
     return {"success": True, "data": top_candidates}

AI/src/models/roster.py

@@ -1,11 +1,13 @@
-from pydantic import BaseModel
+from pydantic import BaseModel, field_validator
 from typing import List, Dict
+from bson import ObjectId
 
 class ProjectEmbeddingsResponse(BaseModel):
     status: str
     project_id: str
 
 class Candidate(BaseModel):
+    _id: str
     name: str
     position: str
     skills: List[str]
@@ -16,6 +18,12 @@ class Candidate(BaseModel):
     rank: int
     reason: str
 
+    @field_validator('_id', mode='before')
+    def convert_objectId(cls, v):
+        if isinstance(v, ObjectId):
+            return str(v)
+        return v
+
 class RosterRecommendationsResponse(BaseModel):
     success: bool
     data: Dict[str, List[Candidate]]

Backend/controllers/hr.controller.js

@@ -201,7 +201,7 @@ const listEmployees = async (req, res) => {
     if (userIds.length > 0) {
       const counts = await ProjectAssignment.aggregate([
         { $match: { userId: { $in: userIds } } },
-        { $group: { _id: '$userId', count: { $sum: 1 } } },
+        { $group: { _id: "$userId", count: { $sum: 1 } } },
       ]).exec();
       counts.forEach((c) => projectCountsMap.set(String(c._id), c.count));
     }
@@ -263,7 +263,9 @@ const getEmployee = async (req, res) => {
     const response = userDto.mapUserToUserResponse(user);
     // attach project count for this employee
     try {
-      const projCount = await ProjectAssignment.countDocuments({ userId: user._id });
+      const projCount = await ProjectAssignment.countDocuments({
+        userId: user._id,
+      });
       response.projectCount = projCount;
     } catch (e) {
       // non-fatal: if counting fails, default to 0
@@ -802,32 +804,33 @@ const parseCv = async (req, res) => {
 const getColleagues = async (req, res) => {
   try {
     const currentUserId = req.user.id || req.user._id;
-    const currentUser = await User.findById(currentUserId).select('role managerId').lean();
+    const currentUser = await User.findById(currentUserId)
+      .select("role managerId")
+      .lean();
 
     if (!currentUser) {
       return res.status(404).json({
         success: false,
         error: "Not Found",
-        message: "Current user not found"
+        message: "Current user not found",
       });
     }
 
     let colleagues = [];
     let directManager = null;
 
-    if (currentUser.role === 'manager') {
+    if (currentUser.role === "manager") {
       // If user is manager, get all direct subordinates (members with managerId = currentUser._id)
       colleagues = await User.find({
         managerId: currentUserId,
         active: true,
-        _id: { $ne: currentUserId }  // Exclude self
+        _id: { $ne: currentUserId }, // Exclude self
       })
         .populate("position", "name")
         .populate("skills", "name")
         .select("_id name email role position skills")
         .sort({ name: 1 })
         .lean();
-
     } else {
       // If user is staff or HR, get teammates (colleagues with same manager) and include direct manager
       if (currentUser.managerId) {
@@ -841,7 +844,7 @@ const getColleagues = async (req, res) => {
         const teammates = await User.find({
           managerId: currentUser.managerId,
           active: true,
-          _id: { $ne: currentUserId }  // Exclude self
+          _id: { $ne: currentUserId }, // Exclude self
         })
           .populate("position", "name")
           .populate("skills", "name")
@@ -854,28 +857,32 @@ const getColleagues = async (req, res) => {
     }
 
     // Format response
-    const formattedColleagues = colleagues.map(colleague => ({
+    const formattedColleagues = colleagues.map((colleague) => ({
       id: colleague._id,
       name: colleague.name,
       email: colleague.email,
       role: colleague.role,
-      position: colleague.position ? {
-        id: colleague.position._id,
-        name: colleague.position.name
-      } : null,
-      skills: colleague.skills ? colleague.skills.map(skill => ({
-        id: skill._id,
-        name: skill.name
-      })) : []
+      position: colleague.position
+        ? {
+            id: colleague.position._id,
+            name: colleague.position.name,
+          }
+        : null,
+      skills: colleague.skills
+        ? colleague.skills.map((skill) => ({
+            id: skill._id,
+            name: skill.name,
+          }))
+        : [],
     }));
 
     const response = {
       success: true,
       data: {
         userRole: currentUser.role,
         colleagues: formattedColleagues,
-        totalColleagues: formattedColleagues.length
-      }
+        totalColleagues: formattedColleagues.length,
+      },
     };
 
     // Include manager info if it exists
@@ -885,10 +892,12 @@ const getColleagues = async (req, res) => {
         name: directManager.name,
         email: directManager.email,
         role: directManager.role,
-        position: directManager.position ? {
-          id: directManager.position._id,
-          name: directManager.position.name
-        } : null
+        position: directManager.position
+          ? {
+              id: directManager.position._id,
+              name: directManager.position.name,
+            }
+          : null,
       };
     }
 
@@ -897,7 +906,7 @@ const getColleagues = async (req, res) => {
     return res.status(500).json({
       success: false,
       error: "Internal Server Error",
-      message: err.message
+      message: err.message,
     });
   }
 };