<SYSTEM_TASK:>
Solve the following problem using Python, implementing the functions described below, one line at a time
<END_TASK>
<USER_TASK:>
Description:
 def set_agent(self, short_name, client_id):
 """Register a client id that handlers commands for a service. Args: short_name (str):
 The name of the service to set an agent for. client_id (str):
 A globally unique id for the client that should receive commands for this service. """


        if short_name not in self.services:
            raise ArgumentError("Unknown service name", short_name=short_name)

        self.agents[short_name] = client_id