agentscope.rpc package
Submodules
- agentscope.rpc.rpc_agent_client module
RpcAgentClient
RpcAgentClient.__init__()
RpcAgentClient.call_agent_func()
RpcAgentClient.is_alive()
RpcAgentClient.stop()
RpcAgentClient.create_agent()
RpcAgentClient.delete_agent()
RpcAgentClient.delete_all_agent()
RpcAgentClient.clone_agent()
RpcAgentClient.update_placeholder()
RpcAgentClient.get_agent_list()
RpcAgentClient.get_server_info()
RpcAgentClient.set_model_configs()
RpcAgentClient.get_agent_memory()
RpcAgentClient.download_file()
ResponseStub
call_in_thread()
- agentscope.rpc.rpc_agent_pb2 module
- agentscope.rpc.rpc_agent_pb2_grpc module
RpcAgentStub
RpcAgentServicer
RpcAgentServicer.is_alive()
RpcAgentServicer.stop()
RpcAgentServicer.create_agent()
RpcAgentServicer.delete_agent()
RpcAgentServicer.delete_all_agents()
RpcAgentServicer.clone_agent()
RpcAgentServicer.get_agent_list()
RpcAgentServicer.get_server_info()
RpcAgentServicer.set_model_configs()
RpcAgentServicer.get_agent_memory()
RpcAgentServicer.call_agent_func()
RpcAgentServicer.update_placeholder()
RpcAgentServicer.download_file()
add_RpcAgentServicer_to_server()
RpcAgent
RpcAgent.is_alive()
RpcAgent.stop()
RpcAgent.create_agent()
RpcAgent.delete_agent()
RpcAgent.delete_all_agents()
RpcAgent.clone_agent()
RpcAgent.get_agent_list()
RpcAgent.get_server_info()
RpcAgent.set_model_configs()
RpcAgent.get_agent_memory()
RpcAgent.call_agent_func()
RpcAgent.update_placeholder()
RpcAgent.download_file()
Module contents
Import all rpc related modules in the package.
- class agentscope.rpc.RpcAgentClient(host: str, port: int, agent_id: str | None = None)[source]
Bases:
object
A client of Rpc agent server
- __init__(host: str, port: int, agent_id: str | None = None) None [source]
Init a rpc agent client
- Parameters:
host (str) – The hostname of the rpc agent server which the
connected. (is)
port (int) – The port of the rpc agent server which the client
connected.
agent_id (str) – The agent id of the agent being called.
None. (Defaults to)
- call_agent_func(func_name: str, value: str | None = None, timeout: int = 300) str [source]
Call the specific function of an agent running on the server.
- Parameters:
func_name (str) – The name of the function being called.
value (str, optional) – The serialized function input value.
None. (Defaults to)
timeout (int, optional) – The timeout for the RPC call in seconds.
300. (Defaults to)
- Returns:
serialized return data.
- Return type:
str
- is_alive() bool [source]
Check if the agent server is alive.
- Returns:
Indicate whether the server is alive.
- Return type:
bool
- create_agent(agent_configs: dict, agent_id: str | None = None) bool [source]
Create a new agent for this client.
- Parameters:
agent_configs (dict) – Init configs of the agent, generated by
_AgentMeta.
agent_id (str) – agent_id of the created agent.
- Returns:
Indicate whether the creation is successful
- Return type:
bool
- delete_agent(agent_id: str | None = None) bool [source]
Delete agents with the specific agent_id.
- Parameters:
agent_id (str) – id of the agent to be deleted.
- Returns:
Indicate whether the deletion is successful
- Return type:
bool
- clone_agent(agent_id: str) str | None [source]
Clone a new agent instance from the origin instance.
- Parameters:
agent_id (str) – The agent_id of the agent to be cloned.
- Returns:
The agent_id of the generated agent.
- Return type:
str
- update_placeholder(task_id: int) str [source]
Update the placeholder value.
- Parameters:
task_id (int) – task_id of the PlaceholderMessage.
- Returns:
Whether the update is successful. str: Serialized message value.
- Return type:
bool
- get_agent_list() Sequence[dict] [source]
Get the summary of all agents on the server as a list.
- Returns:
list of agent summary information.
- Return type:
Sequence[str]
- set_model_configs(model_configs: dict | list[dict]) bool [source]
Set the model configs of the server.
- get_agent_memory(agent_id: str) list[Msg] | Msg [source]
Get the memory usage of the specific agent.
- download_file(path: str) str [source]
Download a file from a remote server to the local machine.
Args: path (str): The path of the file to be downloaded. Note that
it is the path on the remote server.
- Returns:
The path of the downloaded file. Note that it is the path on the local machine.
- Return type:
str
- class agentscope.rpc.ResponseStub[source]
Bases:
object
A stub used to save the response of a rpc call in a sub-thread.
- class agentscope.rpc.RpcMsg
Bases:
Message
,Message
- DESCRIPTOR = <google._upb._message.Descriptor object>
- agentscope.rpc.call_in_thread(client: RpcAgentClient, value: str, func_name: str) ResponseStub [source]
Call rpc function in a sub-thread.
- Parameters:
client (RpcAgentClient) – The rpc client.
value (str) – The value of the request.
func_name (str) – The name of the function being called.
- Returns:
A stub to get the response.
- Return type:
ResponseStub