Fix agent notification

Author: benbrandt

docs/protocol/draft/schema.mdx

@@ -167,9 +167,9 @@ This capability is not part of the spec yet, and may be removed or changed at an
 
 Cancels an ongoing request.
 
-This is a notification sent by the agent to cancel any ongoing request.
+This is a notification sent by the client to cancel any ongoing request.
 
-Upon receiving this notification, the Client:
+Upon receiving this notification, the Agent:
 
 1. MUST cancel the corresponding request activity and all nested activities
 2. MAY send any pending notifications.

src/bin/generate.rs

@@ -822,10 +822,7 @@ and control access to resources."
                 "session/prompt" => self.agent_methods.get("PromptRequest").unwrap(),
                 "session/cancel" => self.agent_methods.get("CancelNotification").unwrap(),
                 "session/set_model" => self.agent_methods.get("SetSessionModelRequest").unwrap(),
-                "request/cancel" => self
-                    .client_methods
-                    .get("CancelRequestNotification")
-                    .unwrap(),
+                "request/cancel" => self.agent_methods.get("CancelRequestNotification").unwrap(),
                 _ => panic!("Introduced a method? Add it here :)"),
             }
         }