o ni @slddlmZddlZddlZddlmZdZdZddd ZdddZ dddZ ej ddGdddZ dS)) annotationsN)env2tcpreturnServiceToken | Nonec Cstjtj}|s dS|d}t|dkrtd||\}}}}}|tkr6tdtd|d|d|t vrEtd |d|dzt |t |||t |d WStyg}ztd||d}~ww) zReads the token from environment variables. Returns: The token if the correct environment variable is set, or None. Raises: ValueError: If the environment variable is set but cannot be parsed. N-zInvalid token: zExpected version z , but got z (token=)Unsupported transport: )versionpid transporthostport) osenvirongetrSERVICEsplitlen ValueError_CURRENT_VERSION_SUPPORTED_TRANSPORTS ServiceTokenint)tokenpartsr pid_strrrport_strer!Y/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/wandb/sdk/lib/service_token.pyget_service_token s>   r# parent_pidrrstrrrNonecCs>|tvr td|dtt|||t|ftjtj<dS)aStores a service token in an environment variable. Args: parent_pid: The process ID of the process that started the service. transport: The transport used to communicate with the service. host: The host part of the internet address on which the service is listening (e.g. localhost). port: The port the service is listening on. Raises: ValueError: If given an unsupported transport. r rN) rrjoinrr%rrrr)r$rrrr!r!r"set_service_token6s r(cCstjtjddS)z:Clears the environment variable storing the service token.N)rrpoprrr!r!r!r"clear_service_tokenQsr*T)frozenc@s:eZdZUdZded<ded<ded<ded<ded<d S) rz,An identifier for a running service process.r%r rr rrrN)__name__ __module__ __qualname____doc____annotations__r!r!r!r"rVs  r)rr) r$rrr%rr%rrrr&)rr&) __future__r dataclassesrwandbrrrr#r(r* dataclassrr!r!r!r"s    *