a2a 0.1.0 → 0.2.0

data/lib/a2a/types/get_task_push_notification_response.rb

@@ -1,12 +1,12 @@
 # frozen_string_literal: true
 
 module A2A
-  # Response to a `tasks/pushNotification/get` request. Contains the TaskPushNotificationConfig or an error.
-  class GetTaskPushNotificationResponse < JSONRPCResponse
+  # Response to a `tasks/pushNotificationConfig/get` request. Contains the TaskPushNotificationConfig or an error.
+  class GetTaskPushNotificationResponse < JSONRPC::Response
     # @return [TaskPushNotificationConfig, nil] The push notification config if successful.
     attribute? :result, Types::Constructor(TaskPushNotificationConfig).optional
 
-    # @return [JSONRPCError, nil] Error information if the request failed.
-    attribute? :error, Types::Constructor(JSONRPCError).optional
+    # @return [JSONRPC::Error, nil] Error information if the request failed.
+    attribute? :error, Types::Constructor(JSONRPC::Error).optional
   end
 end

data/lib/a2a/types/get_task_request.rb

@@ -2,9 +2,12 @@
 
 module A2A
   # Request to retrieve the current state of a task.
-  class GetTaskRequest < JSONRPCRequest
+  class GetTaskRequest < JSONRPC::Request
     # @return [String] Method name for getting task status.
-    attribute :method, Types::String.constant('tasks/get')
+    METHOD = 'tasks/get'
+
+    # @return [String] Method name for getting task status.
+    attribute :method, Types::String.constant(METHOD)
 
     # @return [TaskQueryParams] Parameters for the get task method.
     attribute :params, Types::Constructor(TaskQueryParams)

data/lib/a2a/types/get_task_response.rb

@@ -2,11 +2,11 @@
 
 module A2A
   # Response to a `tasks/get` request. Contains the Task object or an error.
-  class GetTaskResponse < JSONRPCResponse
+  class GetTaskResponse < JSONRPC::Response
     # @return [Task, nil] The task information if successful.
     attribute? :result, Types::Constructor(Task).optional
 
-    # @return [JSONRPCError, nil] Error information if the request failed.
-    attribute? :error, Types::Constructor(JSONRPCError).optional
+    # @return [JSONRPC::Error, nil] Error information if the request failed.
+    attribute? :error, Types::Constructor(JSONRPC::Error).optional
   end
 end

data/lib/a2a/types/http_auth_security_scheme.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module A2A
+  # Defines a security scheme using HTTP authentication.
+  class HTTPAuthSecurityScheme < SecuritySchemeBase
+    # @return [String] The type of the security scheme. Must be 'http'.
+    attribute :type, Types::String.constant('http')
+
+    # @return [String] The name of the HTTP Authentication scheme to be used in the Authorization header,
+    #   as defined in RFC7235 (e.g., "Bearer").
+    #   This value should be registered in the IANA Authentication Scheme registry.
+    attribute :scheme, Types::String
+
+    # @return [String, nil] A hint to the client to identify how the bearer token is formatted (e.g., "JWT").
+    #   This is primarily for documentation purposes.
+    attribute? :bearer_format, Types::String.optional
+  end
+end

data/lib/a2a/types/implicit_oauth_flow.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+module A2A
+  # Defines configuration details for the OAuth 2.0 Implicit flow.
+  class ImplicitOAuthFlow < ProtocolStruct
+    # @return [URI] The authorization URL to be used for this flow. This MUST be a URL.
+    attribute :authorization_url, Types::URI
+
+    # @return [URI, nil] The URL to be used for obtaining refresh tokens. This MUST be a URL.
+    attribute? :refresh_url, Types::URI.optional
+
+    # @return [Hash] The available scopes for the OAuth2 security scheme. A map between the scope
+    #   name and a short description for it.
+    attribute :scopes, Types::Hash.map(Types::String, Types::String)
+  end
+end

data/lib/a2a/types/invalid_agent_response_error.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+module A2A
+  # Error for invalid agent responses (-32006)
+  #
+  # Raised when agent generates an invalid response for the requested method.
+  #
+  # @api public
+  #
+  # @example Create an invalid agent response error
+  #   error = JSONRPC::InvalidAgentResponseError.new
+  #
+  class InvalidAgentResponseError < JSONRPC::Error
+    # Creates a new Invalid Agent Response Error with code -32006
+    #
+    # @api public
+    #
+    # @example Create an invalid agent response error with default message
+    #   error = JSONRPC::InvalidAgentResponseError.new
+    #
+    # @example Create an invalid agent response error with exception details
+    #   error = JSONRPC::InvalidAgentResponseError.new(data: { response_type: 'malformed' }, request_id: 1)
+    #
+    # @param message [String] short description of the error
+    # @param data [Hash, Array, String, Number, Boolean, nil] additional error information
+    # @param request_id [String, Integer, nil] the request identifier
+    #
+    def initialize(
+      message = 'Invalid agent response type.',
+      data: nil,
+      request_id: nil
+    )
+      super(
+        message,
+        code: ErrorCodes::INVALID_AGENT_RESPONSE,
+        data:,
+        request_id:
+      )
+    end
+  end
+end

data/lib/a2a/types/list_task_push_notification_config_params.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+module A2A
+  # Defines parameters for listing all push notification configurations associated with a task.
+  class ListTaskPushNotificationConfigParams < TaskIdParams
+    # No additional attributes beyond those inherited from TaskIdParams
+  end
+end

data/lib/a2a/types/list_task_push_notification_config_request.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module A2A
+  # Represents a JSON-RPC request for the `tasks/pushNotificationConfig/list` method.
+  class ListTaskPushNotificationConfigRequest < JSONRPC::Request
+    # @return [String] Method name for listing push notification configurations.
+    METHOD = 'tasks/pushNotificationConfig/list'
+
+    # @return [String] Method name for listing push notification configurations.
+    attribute :method, Types::String.constant(METHOD)
+
+    # @return [ListTaskPushNotificationConfigParams] Parameters for listing push notification configurations.
+    attribute :params, Types::Constructor(ListTaskPushNotificationConfigParams)
+
+    # @return [String] Method name for listing push notification configurations.
+    def method = attributes[:method]
+  end
+end

data/lib/a2a/types/list_task_push_notification_config_response.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module A2A
+  # Response to a `tasks/pushNotificationConfig/list` request.
+  # Returns an array of all push notification configurations for the specified task.
+  class ListTaskPushNotificationConfigResponse < JSONRPC::Response
+    # @return [Array<TaskPushNotificationConfig>, nil] Array of push notification configurations if successful.
+    attribute? :result, Types::Array.of(Types::Constructor(TaskPushNotificationConfig)).optional
+
+    # @return [JSONRPC::Error, nil] Error information if the request failed.
+    attribute? :error, Types::Constructor(JSONRPC::Error).optional
+  end
+end

data/lib/a2a/types/message.rb

@@ -1,15 +1,35 @@
 # frozen_string_literal: true
 
 module A2A
-  # Represents a message exchanged between a user and an agent.
+  # Represents a single message in the conversation between a user and an agent.
   class Message < ProtocolStruct
-    # @return [String] The role of the sender (user or agent).
+    # @return [String] Identifies the sender of the message. `user` for the client, `agent` for the service.
     attribute :role, Types::String.enum('user', 'agent')
 
-    # @return [Array<Part>] The content of the message, composed of one or more parts.
+    # @return [Array<Part>] An array of content parts that form the message body. A message can be
+    #   composed of multiple parts of different types (e.g., text and files).
     attribute :parts, Types::Array.of(Types::Constructor(Part))
 
-    # @return [Hash, nil] Optional metadata associated with the message.
+    # @return [Hash, nil] Optional metadata for extensions. The key is an extension-specific identifier.
     attribute? :metadata, Types::Hash.optional
+
+    # @return [Array<String>, nil] The URIs of extensions that are relevant to this message.
+    attribute? :extensions, Types::Array.of(Types::String).optional
+
+    # @return [Array<String>, nil] A list of other task IDs that this message references for additional context.
+    attribute? :reference_task_ids, Types::Array.of(Types::String).optional
+
+    # @return [String] A unique identifier for the message, typically a UUID, generated by the sender.
+    attribute :message_id, Types::String
+
+    # @return [String, nil] The ID of the task this message is part of. Can be omitted for the first message of a
+    #   new task.
+    attribute? :task_id, Types::String.optional
+
+    # @return [String, nil] The context ID for this message, used to group related interactions.
+    attribute? :context_id, Types::String.optional
+
+    # @return [String] The type of this object, used as a discriminator. Always 'message' for a Message.
+    attribute :kind, Types::String.constant('message')
   end
 end

data/lib/a2a/types/message_send_configuration.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+module A2A
+  # Defines configuration options for a `message/send` or `message/stream` request.
+  class MessageSendConfiguration < ProtocolStruct
+    # @return [Array<String>, nil] A list of output MIME types the client is prepared to accept in the response.
+    attribute? :accepted_output_modes, Types::Array.of(Types::String).optional
+
+    # @return [Integer, nil] The number of most recent messages from the task's history to retrieve in the response.
+    attribute? :history_length, Types::Integer.optional
+
+    # @return [PushNotificationConfig, nil] Configuration for the agent to send push notifications for updates after
+    #   the initial response.
+    attribute? :push_notification_config, Types::Constructor(PushNotificationConfig).optional
+
+    # @return [Boolean] If true, the client will wait for the task to complete. The server may reject this if
+    #   the task is long-running.
+    attribute? :blocking, Types::Bool.default(false)
+  end
+end

data/lib/a2a/types/message_send_params.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+module A2A
+  # Defines the parameters for a request to send a message to an agent. This can be used
+  # to create a new task, continue an existing one, or restart a task.
+  class MessageSendParams < ProtocolStruct
+    # @return [Message] The message object being sent to the agent.
+    attribute :message, Types::Constructor(Message)
+
+    # @return [MessageSendConfiguration, nil] Optional configuration for the send request.
+    attribute? :configuration, Types::Constructor(MessageSendConfiguration).optional
+
+    # @return [Hash, nil] Optional metadata for extensions.
+    attribute? :metadata, Types::Hash.optional
+  end
+end

data/lib/a2a/types/mutual_tls_security_scheme.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+module A2A
+  # Defines a security scheme using mTLS authentication.
+  class MutualTLSSecurityScheme < SecuritySchemeBase
+    # @return [String] The type of the security scheme. Must be 'mutualTLS'.
+    attribute :type, Types::String.constant('mutualTLS')
+  end
+end

data/lib/a2a/types/oauth2_security_scheme.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+module A2A
+  # Defines a security scheme using OAuth 2.0.
+  class OAuth2SecurityScheme < SecuritySchemeBase
+    # @return [String] The type of the security scheme. Must be 'oauth2'.
+    attribute :type, Types::String.constant('oauth2')
+
+    # @return [OAuthFlows] An object containing configuration information for the supported OAuth 2.0 flows.
+    attribute :flows, Types::Constructor(OAuthFlows)
+
+    # @return [URI, nil] URL to the oauth2 authorization server metadata
+    #   RFC8414. TLS is required.
+    attribute? :oauth2_metadata_url, Types::URI.optional
+  end
+end

data/lib/a2a/types/oauth_flows.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+module A2A
+  # Defines the configuration for the supported OAuth 2.0 flows.
+  class OAuthFlows < ProtocolStruct
+    # @return [AuthorizationCodeOAuthFlow, nil] Configuration for the OAuth Authorization Code flow.
+    #   Previously called accessCode in OpenAPI 2.0.
+    attribute? :authorization_code, Types::Constructor(AuthorizationCodeOAuthFlow).optional
+
+    # @return [ClientCredentialsOAuthFlow, nil] Configuration for the OAuth Client Credentials flow.
+    #   Previously called application in OpenAPI 2.0.
+    attribute? :client_credentials, Types::Constructor(ClientCredentialsOAuthFlow).optional
+
+    # @return [ImplicitOAuthFlow, nil] Configuration for the OAuth Implicit flow.
+    attribute? :implicit, Types::Constructor(ImplicitOAuthFlow).optional
+
+    # @return [PasswordOAuthFlow, nil] Configuration for the OAuth Resource Owner Password flow.
+    attribute? :password, Types::Constructor(PasswordOAuthFlow).optional
+  end
+end

data/lib/a2a/types/openid_connect_security_scheme.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module A2A
+  # Defines a security scheme using OpenID Connect.
+  class OpenIdConnectSecurityScheme < SecuritySchemeBase
+    # @return [String] The type of the security scheme. Must be 'openIdConnect'.
+    attribute :type, Types::String.constant('openIdConnect')
+
+    # @return [URI] The OpenID Connect Discovery URL for the OIDC provider's metadata.
+    attribute :open_id_connect_url, Types::URI
+  end
+end

data/lib/a2a/types/part.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 module A2A
-  # Represents a single part of a multi-part message. Can be text, file, or data.
+  # A discriminated union representing a part of a message or artifact, which can
+  # be text, a file, or structured data.
   Part = TextPart | FilePart | DataPart
 end

data/lib/a2a/types/part_base.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+module A2A
+  # Defines base properties common to all message or artifact parts.
+  class PartBase < ProtocolStruct
+    # @return [Hash<String, Object>] Optional metadata associated with this part.
+    attribute? :metadata, Types::Hash.map(Types::String, Types::Any).optional
+  end
+end

data/lib/a2a/types/password_oauth_flow.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+module A2A
+  # Defines configuration details for the OAuth 2.0 Resource Owner Password flow.
+  class PasswordOAuthFlow < ProtocolStruct
+    # @return [URI] The token URL to be used for this flow. This MUST be a URL.
+    attribute :token_url, Types::URI
+
+    # @return [URI, nil] The URL to be used for obtaining refresh tokens. This MUST be a URL.
+    attribute? :refresh_url, Types::URI.optional
+
+    # @return [Hash] The available scopes for the OAuth2 security scheme. A map between the scope
+    #   name and a short description for it.
+    attribute :scopes, Types::Hash.map(Types::String, Types::String)
+  end
+end

data/lib/a2a/types/push_notification_authentication_info.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module A2A
+  # Defines authentication details for a push notification endpoint.
+  class PushNotificationAuthenticationInfo < ProtocolStruct
+    # @return [Array<String>] A list of supported authentication schemes (e.g., 'Basic', 'Bearer').
+    attribute :schemes, Types::Array.of(Types::String)
+
+    # @return [String, nil] Optional credentials required by the push notification endpoint.
+    attribute? :credentials, Types::String.optional
+  end
+end

data/lib/a2a/types/push_notification_config.rb

@@ -1,15 +1,20 @@
 # frozen_string_literal: true
 
 module A2A
-  # Information required for setting up push notifications.
+  # Defines the configuration for setting up push notifications for task updates.
   class PushNotificationConfig < ProtocolStruct
-    # @return [String] The URL endpoint where the agent should send notifications.
-    attribute :url, Types::String
+    # @return [String, nil] A unique identifier (e.g. UUID) for the push notification configuration, set by the client
+    #   to support multiple notification callbacks.
+    attribute? :id, Types::String.optional
 
-    # @return [String, nil] A token to be included in push notification requests for verification/authentication.
+    # @return [URI] The callback URL where the agent should send push notifications.
+    attribute :url, Types::URI
+
+    # @return [String, nil] A unique token for this task or session to validate incoming push notifications.
     attribute? :token, Types::String.optional
 
-    # @return [AuthenticationInfo, nil] Optional authentication details needed by the agent to call the notification URL
-    attribute? :authentication, Types::Constructor(AuthenticationInfo).optional
+    # @return [PushNotificationAuthenticationInfo, nil] Optional authentication details for the agent to use when
+    #   calling the notification URL.
+    attribute? :authentication, Types::Constructor(PushNotificationAuthenticationInfo).optional
   end
 end

data/lib/a2a/types/push_notification_not_supported_error.rb

@@ -1,15 +1,42 @@
 # frozen_string_literal: true
 
 module A2A
-  # Error for push notification not supported
-  class PushNotificationNotSupportedError < JSONRPCError
-    # @return [Integer] Error code (-32003)
-    attribute :code, Types::Integer.constant(ErrorCodes::PUSH_NOTIFICATION_NOT_SUPPORTED)
-
-    # @return [String] Error message
-    attribute :message, Types::String.constant('Push Notification is not supported')
-
-    # @return [nil] No additional data
-    attribute? :data, Types::Nil.constant(nil)
+  # Error for unsupported push notifications (-32003)
+  #
+  # Raised when client attempts to use push notification features but the server agent
+  # does not support them (i.e., AgentCard.capabilities.pushNotifications is false).
+  #
+  # @api public
+  #
+  # @example Create a push notification not supported error
+  #   error = JSONRPC::PushNotificationNotSupportedError.new
+  #
+  class PushNotificationNotSupportedError < JSONRPC::Error
+    # Creates a new Push Notification Not Supported Error with code -32003
+    #
+    # @api public
+    #
+    # @example Create a push notification not supported error with default message
+    #   error = JSONRPC::PushNotificationNotSupportedError.new
+    #
+    # @example Create a push notification not supported error with exception details
+    #   error = JSONRPC::PushNotificationNotSupportedError.new(data: { capability: false }, request_id: 1)
+    #
+    # @param message [String] short description of the error
+    # @param data [Hash, Array, String, Number, Boolean, nil] additional error information
+    # @param request_id [String, Integer, nil] the request identifier
+    #
+    def initialize(
+      message = 'Push Notification is not supported.',
+      data: nil,
+      request_id: nil
+    )
+      super(
+        message,
+        code: ErrorCodes::PUSH_NOTIFICATION_NOT_SUPPORTED,
+        data:,
+        request_id:
+      )
+    end
   end
 end

data/lib/a2a/types/request.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # A discriminated union representing all possible JSON-RPC 2.0 requests supported by the A2A specification.
+  Request = SendMessageRequest |
+            SendStreamingMessageRequest |
+            GetTaskRequest |
+            CancelTaskRequest |
+            SetTaskPushNotificationConfigRequest |
+            GetTaskPushNotificationConfigRequest |
+            TaskResubscriptionRequest |
+            ListTaskPushNotificationConfigRequest |
+            DeleteTaskPushNotificationConfigRequest |
+            GetAuthenticatedExtendedCardRequest
+end

data/lib/a2a/types/security_scheme.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module A2A
+  # A discriminated union representing a security scheme that can be used to secure an agent's endpoints.
+  # This is a discriminated union type based on the OpenAPI 3.0 Security Scheme Object.
+  SecurityScheme = APIKeySecurityScheme |
+                   HTTPAuthSecurityScheme |
+                   OAuth2SecurityScheme |
+                   OpenIdConnectSecurityScheme |
+                   MutualTLSSecurityScheme
+end

data/lib/a2a/types/security_scheme_base.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+module A2A
+  # Defines base properties shared by all security scheme objects.
+  class SecuritySchemeBase < ProtocolStruct
+    # @return [String, nil] An optional description for the security scheme.
+    attribute? :description, Types::String.optional
+  end
+end

data/lib/a2a/types/send_message_request.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module A2A
+  # Represents a JSON-RPC request for the `message/send` method.
+  class SendMessageRequest < JSONRPC::Request
+    # @return [String] Method name for sending a message.
+    METHOD = 'message/send'
+
+    # @return [String] Method name for sending a message.
+    attribute :method, Types::String.constant(METHOD)
+
+    # @return [MessageSendParams] Parameters for sending a message.
+    attribute :params, Types::Constructor(MessageSendParams)
+
+    # @return [String] Method name for sending a message.
+    def method = attributes[:method]
+  end
+end

data/lib/a2a/types/send_message_response.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module A2A
+  # Response to a `message/send` request.
+  # Returns either a direct Message reply or the initial Task object.
+  class SendMessageResponse < JSONRPC::Response
+    # @return [Message, Task, nil] The result, which can be a direct reply Message or the initial Task object.
+    attribute? :result, Types::Any.optional
+
+    # @return [JSONRPC::Error, nil] Error information if the request failed.
+    attribute? :error, Types::Constructor(JSONRPC::Error).optional
+  end
+end

data/lib/a2a/types/send_streaming_message_request.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module A2A
+  # Represents a JSON-RPC request for the `message/stream` method.
+  class SendStreamingMessageRequest < JSONRPC::Request
+    # @return [String] Method name for streaming a message.
+    METHOD = 'message/stream'
+
+    # @return [String] Method name for streaming a message.
+    attribute :method, Types::String.constant(METHOD)
+
+    # @return [MessageSendParams] Parameters for streaming a message.
+    attribute :params, Types::Constructor(MessageSendParams)
+
+    # @return [String] Method name for streaming a message.
+    def method = attributes[:method]
+  end
+end

data/lib/a2a/types/send_streaming_message_response.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # Response to a `message/stream` request.
+  # The server may send multiple response objects for a single request.
+  # Returns a Message, Task, or streaming update events.
+  class SendStreamingMessageResponse < JSONRPC::Response
+    # @return [Message, Task, TaskStatusUpdateEvent, TaskArtifactUpdateEvent, nil]
+    #   The result, which can be a Message, Task, or streaming update event.
+    attribute? :result, Types::Any.optional
+
+    # @return [JSONRPC::Error, nil] Error information if the request failed.
+    attribute? :error, Types::Constructor(JSONRPC::Error).optional
+  end
+end

data/lib/a2a/types/set_task_push_notification_config_request.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module A2A
+  # Represents a JSON-RPC request for the `tasks/pushNotificationConfig/set` method.
+  class SetTaskPushNotificationConfigRequest < JSONRPC::Request
+    # @return [String] Method name for setting task push notification configuration.
+    METHOD = 'tasks/pushNotificationConfig/set'
+
+    # @return [String] Method name for setting task push notification configuration.
+    attribute :method, Types::String.constant(METHOD)
+
+    # @return [TaskPushNotificationConfig] Parameters for setting the push notification configuration.
+    attribute :params, Types::Constructor(TaskPushNotificationConfig)
+
+    # @return [String] Method name for setting task push notification configuration.
+    def method = attributes[:method]
+  end
+end

data/lib/a2a/types/set_task_push_notification_request.rb

@@ -2,9 +2,12 @@
 
 module A2A
   # Request to set or update the push notification config for a task.
-  class SetTaskPushNotificationRequest < JSONRPCRequest
+  class SetTaskPushNotificationRequest < JSONRPC::Request
     # @return [String] Method name for setting a task notifications.
-    attribute :method, Types::String.constant('tasks/pushNotification/set')
+    METHOD = 'tasks/pushNotificationConfig/set'
+
+    # @return [String] Method name for setting a task notifications.
+    attribute :method, Types::String.constant(METHOD)
 
     # @return [TaskPushNotificationConfig] Parameters for the set task push notification method.
     attribute :params, Types::Constructor(TaskPushNotificationConfig)

data/lib/a2a/types/set_task_push_notification_response.rb

@@ -1,12 +1,12 @@
 # frozen_string_literal: true
 
 module A2A
-  # Response to a `tasks/pushNotification/set` request. Contains the TaskPushNotificationConfig or an error.
-  class SetTaskPushNotificationResponse < JSONRPCResponse
+  # Response to a `tasks/pushNotificationConfig/set` request. Contains the TaskPushNotificationConfig or an error.
+  class SetTaskPushNotificationResponse < JSONRPC::Response
     # @return [TaskPushNotificationConfig, nil] The push notification config if successful.
     attribute? :result, Types::Constructor(TaskPushNotificationConfig).optional
 
-    # @return [JSONRPCError, nil] Error information if the request failed.
-    attribute? :error, Types::Constructor(JSONRPCError).optional
+    # @return [JSONRPC::Error, nil] Error information if the request failed.
+    attribute? :error, Types::Constructor(JSONRPC::Error).optional
   end
 end