import time

from linksoul_agentsdk import (
    AgentAuthCallback,
    AgentParam,
    AgentSdk,
    ExtSkillInvokeCallback,
    ExtSkillQueryCallback,
    ExtSkillRequest,
    IdGenerator,
)

url        = "wss://open.agibot.com/api/V1/open-portal/app/wss/agent-sdk"
app_id     = "您通过灵心开放平台创建的应用获取的appId"
app_key    = "您通过灵心开放平台创建的应用获取的appKey"
app_secret = "您通过灵心开放平台创建的应用获取的appSecret"
agent_id   = "您通过灵心平台创建的二开智能体获取的agentId"

# 第一步：创建 SDK 实例
agent_sdk = AgentSdk.create(url, app_id, app_key, app_secret)


# 第二步：注册验签回调
class MyAuthCallback(AgentAuthCallback):
    def on_auth_state(self, app_id: str, code: int, msg: str) -> None:
        print(f"on_auth_state => app_id: {app_id}, code: {code}, msg: {msg}")


agent_sdk.register_auth(MyAuthCallback())

# 第三步：初始化
agent_sdk.initialize()
time.sleep(2)

# 第四步：注册扩展技能请求实例
request_id = IdGenerator.generate_ext_skill_request_id()
ext_skill_request = ExtSkillRequest(agent_sdk, agent_id, request_id)
agent_sdk.register_ext_skill(ext_skill_request)


# 4.1 查询可用扩展技能清单
class MyQueryCallback(ExtSkillQueryCallback):
    def on_query_result(self, agent_id: str, request_id: str, event_id: str,
                        code: int, msg: str, result) -> None:
        print(f"on_query_result => agent_id={agent_id}, request_id={request_id}, "
              f"event_id={event_id}, code={code}, msg={msg}, result={result}")


query_event_id = ext_skill_request.query(MyQueryCallback(), 2000)
print(f"query event_id: {query_event_id}")


# 4.2 触发某个扩展技能执行
ext_skill_id = "test_ext_skill_id_001"           # 扩展技能 ID
input_param = (                                   # 扩展技能入参
    AgentParam.create()
    .set_string("role", "male")
    .set_integer("threshold", 0)
)


class MyInvokeCallback(ExtSkillInvokeCallback):
    def on_invoke_ack(self, agent_id: str, request_id: str, event_id: str,
                      code: int, msg: str) -> None:
        # 第一段：网关已受理
        print(f"on_invoke_ack => agent_id={agent_id}, request_id={request_id}, "
              f"event_id={event_id}, code={code}, msg={msg}")

    def on_state(self, agent_id: str, request_id: str, event_id: str,
                 state_name, state_value) -> None:
        # 第二段：执行期间状态上报（0 到多次）
        print(f"on_state => agent_id={agent_id}, request_id={request_id}, "
              f"event_id={event_id}, state_name={state_name}, state_value={state_value}")

    def on_invoke_result(self, agent_id: str, request_id: str, event_id: str,
                         code: int, msg: str, param) -> None:
        # 第三段：技能执行输出
        print(f"on_invoke_result => agent_id={agent_id}, request_id={request_id}, "
              f"event_id={event_id}, code={code}, msg={msg}, param={param}")


invoke_event_id = ext_skill_request.invoke(ext_skill_id, input_param, MyInvokeCallback(), 2000)
print(f"invoke event_id: {invoke_event_id}")

time.sleep(5)

# 第五步：反注册释放
agent_sdk.unregister_ext_skill(request_id)
# agent_sdk.release()