import agentcp
import json
import time
tools=[{
"type": "function",
"function": {
"name": "get_weather",
"description": "Retrieves the current weather report for a specified city",
"parameters": {
"type": "object",
"properties": {
"city": {
"type": "string",
"description": "The name of the city for which to retrieve the weather report"
}
},
"required": ["city"]
}
}
}]
def get_weather(city: str) -> str:
return f"The weather in {city} is sunny with a temperature of 25 degrees"
def call_function(name, args):
if name == "get_weather":
return get_weather(**args)
else:
return f"Function {name} not found"
if __name__ == "__main__":
llm_agent_id = "agent_id_from_mu"
agent_id = 'your_agent_id'
acp = agentcp.AgentCP('.')
aid = acp.load_aid(agent_id)
async def reply_message_handler(reply_msg, sender, session_id):
message_json = json.loads(reply_msg.get("message"))
if isinstance(message_json, list) and len(message_json) > 0:
message_json = message_json[0]
type = message_json.get("type")
if type == "tool_call":
tool = json.loads(message_json.get("content"))
tool_name = tool.get("tool_name")
tool_args = tool.get("tool_args")
tool_result = call_function(tool_name, tool_args)
aid.quick_send_messsage_content(llm_agent_id, tool_result, lambda reply_msg: reply_message_handler(reply_msg, sender, session_id))
else:
reply_text = aid.get_content_from_message(reply_msg)
aid.send_message_content(to_aid_list=[sender], session_id=session_id, llm_content=reply_text)
@aid.message_handler() #消息处理函数
async def sync_message_handler(msg):
receiver = aid.get_receiver_from_message(msg) # 获取接收者
if aid.id not in receiver:
return
session_id = aid.get_session_id_from_message(msg)
sender = aid.get_sender_from_message(msg) # 获取发送者
sender_content = aid.get_content_from_message(msg)
msg_block = {
"type": "content",
"status": "success",
"timestamp": int(time.time() * 1000),
"content": sender_content,
"tools": tools
}
aid.quick_send_messsage(llm_agent_id, msg_block, lambda reply_msg: reply_message_handler(reply_msg, sender, session_id))
return True
aid.online()
acp.serve_forever()
