|
9 | 9 | from ..meta import AGENT_METHODS |
10 | 10 | from ..router import MessageRouter, Route, _resolve_handler, _warn_legacy_handler |
11 | 11 | from ..schema import ( |
| 12 | + AcceptNesNotification, |
12 | 13 | AuthenticateRequest, |
13 | 14 | CancelNotification, |
| 15 | + CloseNesRequest, |
14 | 16 | CloseSessionRequest, |
| 17 | + DeleteSessionRequest, |
| 18 | + DidChangeDocumentNotification, |
| 19 | + DidCloseDocumentNotification, |
| 20 | + DidFocusDocumentNotification, |
| 21 | + DidOpenDocumentNotification, |
| 22 | + DidSaveDocumentNotification, |
| 23 | + DisableProviderRequest, |
15 | 24 | ForkSessionRequest, |
16 | 25 | InitializeRequest, |
| 26 | + ListProvidersRequest, |
17 | 27 | ListSessionsRequest, |
18 | 28 | LoadSessionRequest, |
| 29 | + LogoutRequest, |
| 30 | + MessageMcpNotification, |
| 31 | + MessageMcpRequest, |
19 | 32 | NewSessionRequest, |
20 | 33 | PromptRequest, |
| 34 | + RejectNesNotification, |
21 | 35 | ResumeSessionRequest, |
| 36 | + SetProviderRequest, |
22 | 37 | SetSessionConfigOptionBooleanRequest, |
23 | 38 | SetSessionConfigOptionSelectRequest, |
24 | 39 | SetSessionModeRequest, |
| 40 | + StartNesRequest, |
| 41 | + SuggestNesRequest, |
25 | 42 | ) |
26 | 43 | from ..utils import model_to_kwargs, normalize_result |
27 | 44 |
|
@@ -102,6 +119,56 @@ def build_agent_router(agent: Agent, use_unstable_protocol: bool = False) -> Mes |
102 | 119 |
|
103 | 120 | router.route_notification(AGENT_METHODS["session_cancel"], CancelNotification, agent, "cancel") |
104 | 121 |
|
| 122 | + router.route_request( |
| 123 | + AGENT_METHODS["session_delete"], |
| 124 | + DeleteSessionRequest, |
| 125 | + agent, |
| 126 | + "delete_session", |
| 127 | + adapt_result=normalize_result, |
| 128 | + ) |
| 129 | + router.route_request(AGENT_METHODS["providers_list"], ListProvidersRequest, agent, "list_providers", unstable=True) |
| 130 | + router.route_request( |
| 131 | + AGENT_METHODS["providers_set"], |
| 132 | + SetProviderRequest, |
| 133 | + agent, |
| 134 | + "set_provider", |
| 135 | + unstable=True, |
| 136 | + adapt_result=normalize_result, |
| 137 | + ) |
| 138 | + router.route_request( |
| 139 | + AGENT_METHODS["providers_disable"], |
| 140 | + DisableProviderRequest, |
| 141 | + agent, |
| 142 | + "disable_provider", |
| 143 | + unstable=True, |
| 144 | + adapt_result=normalize_result, |
| 145 | + ) |
| 146 | + router.route_request(AGENT_METHODS["logout"], LogoutRequest, agent, "logout", adapt_result=normalize_result) |
| 147 | + router.route_request(AGENT_METHODS["mcp_message"], MessageMcpRequest, agent, "mcp_message", unstable=True) |
| 148 | + router.route_notification(AGENT_METHODS["mcp_message"], MessageMcpNotification, agent, "notify_mcp", unstable=True) |
| 149 | + router.route_request(AGENT_METHODS["nes_start"], StartNesRequest, agent, "start_nes", unstable=True) |
| 150 | + router.route_request(AGENT_METHODS["nes_suggest"], SuggestNesRequest, agent, "suggest_nes", unstable=True) |
| 151 | + router.route_request( |
| 152 | + AGENT_METHODS["nes_close"], CloseNesRequest, agent, "close_nes", unstable=True, adapt_result=normalize_result |
| 153 | + ) |
| 154 | + router.route_notification(AGENT_METHODS["nes_accept"], AcceptNesNotification, agent, "accept_nes", unstable=True) |
| 155 | + router.route_notification(AGENT_METHODS["nes_reject"], RejectNesNotification, agent, "reject_nes", unstable=True) |
| 156 | + router.route_notification( |
| 157 | + AGENT_METHODS["document_did_open"], DidOpenDocumentNotification, agent, "did_open", unstable=True |
| 158 | + ) |
| 159 | + router.route_notification( |
| 160 | + AGENT_METHODS["document_did_change"], DidChangeDocumentNotification, agent, "did_change", unstable=True |
| 161 | + ) |
| 162 | + router.route_notification( |
| 163 | + AGENT_METHODS["document_did_close"], DidCloseDocumentNotification, agent, "did_close", unstable=True |
| 164 | + ) |
| 165 | + router.route_notification( |
| 166 | + AGENT_METHODS["document_did_save"], DidSaveDocumentNotification, agent, "did_save", unstable=True |
| 167 | + ) |
| 168 | + router.route_notification( |
| 169 | + AGENT_METHODS["document_did_focus"], DidFocusDocumentNotification, agent, "did_focus", unstable=True |
| 170 | + ) |
| 171 | + |
105 | 172 | @router.handle_extension_request |
106 | 173 | async def _handle_extension_request(name: str, payload: dict[str, Any]) -> Any: |
107 | 174 | ext = getattr(agent, "ext_method", None) |
|
0 commit comments