|
12 | 12 | from utils.agent_card import get_agent_card
|
13 | 13 |
|
14 | 14 |
|
15 |
| -def agent_list_page(app_state: AppState): |
16 |
| - """Agents List Page""" |
| 15 | +def agent_list_page(app_state: AppState) -> None: |
| 16 | + """Agents List Page.""" |
17 | 17 | state = me.state(AgentState)
|
18 | 18 | with page_scaffold(): # pylint: disable=not-context-manager
|
19 | 19 | with page_frame():
|
@@ -65,41 +65,41 @@ def agent_list_page(app_state: AppState):
|
65 | 65 | me.button('Cancel', on_click=cancel_agent_dialog)
|
66 | 66 |
|
67 | 67 |
|
68 |
| -def set_agent_address(e: me.InputBlurEvent): |
| 68 | +def set_agent_address(e: me.InputBlurEvent) -> None: |
69 | 69 | state = me.state(AgentState)
|
70 | 70 | state.agent_address = e.value
|
71 | 71 |
|
72 | 72 |
|
73 |
| -def load_agent_info(e: me.ClickEvent): |
| 73 | +async def load_agent_info(e: me.ClickEvent) -> None: |
74 | 74 | state = me.state(AgentState)
|
75 | 75 | try:
|
76 | 76 | state.error = None
|
77 |
| - agent_card_response = get_agent_card(state.agent_address) |
| 77 | + agent_card_response = await get_agent_card(state.agent_address) |
78 | 78 | state.agent_name = agent_card_response.name
|
79 | 79 | state.agent_description = agent_card_response.description
|
80 | 80 | state.agent_framework_type = (
|
81 | 81 | agent_card_response.provider.organization
|
82 | 82 | if agent_card_response.provider
|
83 | 83 | else ''
|
84 | 84 | )
|
85 |
| - state.input_modes = agent_card_response.defaultInputModes |
86 |
| - state.output_modes = agent_card_response.defaultOutputModes |
| 85 | + state.input_modes = agent_card_response.default_input_modes |
| 86 | + state.output_modes = agent_card_response.default_output_modes |
87 | 87 | state.stream_supported = agent_card_response.capabilities.streaming
|
88 | 88 | state.push_notifications_supported = (
|
89 |
| - agent_card_response.capabilities.pushNotifications |
| 89 | + agent_card_response.capabilities.push_notifications |
90 | 90 | )
|
91 | 91 | except Exception as e:
|
92 | 92 | print(e)
|
93 | 93 | state.agent_name = None
|
94 | 94 | state.error = f'Cannot connect to agent as {state.agent_address}'
|
95 | 95 |
|
96 | 96 |
|
97 |
| -def cancel_agent_dialog(e: me.ClickEvent): |
| 97 | +def cancel_agent_dialog(e: me.ClickEvent) -> None: |
98 | 98 | state = me.state(AgentState)
|
99 | 99 | state.agent_dialog_open = False
|
100 | 100 |
|
101 | 101 |
|
102 |
| -async def save_agent(e: me.ClickEvent): |
| 102 | +async def save_agent(e: me.ClickEvent) -> None: |
103 | 103 | state = me.state(AgentState)
|
104 | 104 | await AddRemoteAgent(state.agent_address)
|
105 | 105 | state.agent_address = ''
|
|
0 commit comments