add extract_length function

handler.py

@@ -339,6 +339,30 @@ class EndpointHandler():
                 return {"start_date": None, "end_date": None}
         
         return {"start_date": None, "end_date": None}
+    
+    def extract_length(self, start_date, end_date):
+        """
+        Args:
+            start_date (datetime): The earlier date.
+            end_date (datetime): The later date.
+
+        Returns:
+            int: Number of full months between the dates.
+        """
+        try:
+            if start_date > end_date:
+            # raise ValueError("start_date must be before end_date")
+                return 0
+        except:
+            return 0
+
+        # Calculate difference in years and months
+        year_diff = end_date.year - start_date.year
+        month_diff = end_date.month - start_date.month
+
+        # Total months difference
+        total_months = year_diff * 12 + month_diff
+        return total_months
 
     def label_resume(self, text):
         results = self.extract_resume_roles(text)