def parse_message()

in orchestration/orchestrator.py [0:0]


    def parse_message(message_str: str) -> dict:
        pattern = r"(\w+)=((?:'(?:\\'|[^'])*'|\"(?:\\\"|[^\"])*\"|\[[^\]]+\]|\([^\)]+\)|[^\s]+))"
        pairs = re.findall(pattern, message_str)
        result = {}
        for key, value in pairs:
            if (value.startswith("'") and value.endswith("'")) or (value.startswith("\"") and value.endswith("\"")):
                value = value[1:-1]
            result[key] = value
        if 'models_usage' in result and result['models_usage'].startswith('RequestUsage'):
            start_index = message_str.find('models_usage=')
            if start_index != -1:
                usage_str = message_str[start_index:].split(' ', 1)[1]
                usage_match = re.match(r"RequestUsage\((.*?)\)", usage_str)
                if usage_match:
                    result['models_usage'] = usage_match.group(1)
        return result