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