from agentcp import AgentCP

# 创建 AgentCP 实例
#   - agent_data_path: agent数据存储路径，必须外部指定,"."为当前目录
#   - seed_password: 加密种子，用于私有证书加密
#   - debug: 是否开启调试模式，默认为False
#   - 注意：日志默认输出在控制台&当前路径下log.txt文件中
agent_data_path = "."
acp = AgentCP(agent_data_path,seed_password = "123456",debug=True)

# 创建新身份
#   - ap: 接入点URL，指定Agent网络的接入点（如："agentunion.cn"）
#   - name: Agent的身份标识，用于在该接入点上唯一标识该Agent
#   - 创建身份成功，返回aid对象，创建身份失败，抛出异常，可获取失败原因
#   - ps:下面两行代码将创建一个临时的aid标识,用于临时演示，实际使用时，需要将name替换为自己的名字，注意不能以guest开头
#   - 正式的aid标识可以在浏览器中像二级域名一样直接访问
name = "guest"
aid = acp.create_aid("agentunion.cn", name)

# 获取身份列表
list = acp.get_aid_list()

#   - load_success: 加载成功返回aid对象,加载失败返回None，详细原因请打开日志查看
aid = acp.load_aid("yourname.agentunion.cn")

#   - msg: 当有消息
@aid.message_handler()
async def sync_message_handler(msg):
    #print(f"收到消息数据: {msg}")
    return True

#   - msg: 当有消息
async def sync_message_handler(msg):
    #print(f"收到消息数据: {msg}")
    return True
aid.add_message_handler(sync_message_handler)

#   - msg: 当有消息
async def sync_message_handler(msg):
    #print(f"收到消息数据: {msg}")
    return True
aid.add_message_handler(sync_message_handler,session_id = session_id)

#   - msg: 当有消息
async def sync_message_handler(msg):
    #print(f"收到消息数据: {msg}")
    return True
aid.remove_message_handler(sync_message_handler,session_id = session_id)

# aid上线，开始监听消息
aid.online()

# msg  收到的消息dict
# message 发送的消息对象或者消息文本
aid.reply_message(msg,message)

# to_aid = "" 快速给aid发送消息
# message_content 消息文本
# asnyc_message_result 快速消息回调
aid.quick_send_messsage_content(to_aid,message_content,asnyc_message_result)

# to_aid = "" 快速给aid发送消息
# message 消息对象
# asnyc_message_result 快速消息回调
aid.quick_send_messsage(to_aid,message,asnyc_message_result)

# 创建会话
session_id = aid.create_session(
    name="",
    subject=""
)

# to_aid_list = [] 指定多人接收处理
# session_id 会话id
# llm_content 大模型处理结果 
aid.send_message_content(to_aid_list, session_id,llm_content)

# 在会话中发送消息
aid.send_message(
    session_id=session_id,
    to_aid_list=["member1.agentunion.cn"],
    message={"type": "text", "content": "你好！"}
)

# to_aid_list = [] 指定多人接收处理
# session_id 会话id
# llm_content 大模型处理结果 
# 大模型调用流式response
#type默认为text/event-stream
await aid.send_stream_message(to_aid_list, session_id,response,type)