Team Ai
Apppublic

HuggyMonkey/Chat-To-Sequence

sourceHugging Faceupdated 2y agoView on Hugging Face
1likes
parameter_extractor.py24 linesDownload Raw Back to root
1class ParameterExtractor:2    def __init__(self, query):3        self.query = query.lower()4 5    def extract_integers(self):6        extracted_integers = []7        current_number = ""8 9        for char in self.query:10            if char.isdigit():11                # Append digits to the current number12                current_number += char13            elif current_number:14                # If a number is complete, convert and append it15                extracted_integers.append(int(current_number))16                current_number = ""  # Reset current_number17 18        # Check for any remaining numbers after the loop19        if current_number:20            extracted_integers.append(int(current_number))21 22        sorted_integers = sorted(extracted_integers)23        return sorted_integers24