a2a 0.1.0.pre → 0.1.0

data/lib/a2a/types/get_task_push_notification_request.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # Request to retrieve the currently configured push notification configuration for a task.
+  class GetTaskPushNotificationRequest < JSONRPCRequest
+    # @return [String] Method name for getting task notification configuration.
+    attribute :method, Types::String.constant('tasks/pushNotification/get')
+
+    # @return [TaskIdParams] Parameters for the get task push notification config method.
+    attribute :params, Types::Constructor(TaskIdParams)
+
+    # @return [String] Method name for getting task notification configuration.
+    def method = attributes[:method]
+  end
+end

data/lib/a2a/types/get_task_push_notification_response.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module A2A
+  # Response to a `tasks/pushNotification/get` request. Contains the TaskPushNotificationConfig or an error.
+  class GetTaskPushNotificationResponse < JSONRPCResponse
+    # @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
+  end
+end

data/lib/a2a/types/get_task_request.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # Request to retrieve the current state of a task.
+  class GetTaskRequest < JSONRPCRequest
+    # @return [String] Method name for getting task status.
+    attribute :method, Types::String.constant('tasks/get')
+
+    # @return [TaskQueryParams] Parameters for the get task method.
+    attribute :params, Types::Constructor(TaskQueryParams)
+
+    # @return [String] Method name for getting task status.
+    def method = attributes[:method]
+  end
+end

data/lib/a2a/types/get_task_response.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module A2A
+  # Response to a `tasks/get` request. Contains the Task object or an error.
+  class GetTaskResponse < JSONRPCResponse
+    # @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
+  end
+end

data/lib/a2a/types/internal_error.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # Error for internal server errors
+  class InternalError < JSONRPCError
+    # @return [Integer] Error code (-32603)
+    attribute :code, Types::Integer.constant(ErrorCodes::INTERNAL_ERROR)
+
+    # @return [String] Error message
+    attribute :message, Types::String.constant('Internal error')
+
+    # @return [Object, nil] Additional error data
+    attribute? :data, Types::Nominal::Any.optional
+  end
+end

data/lib/a2a/types/invalid_params_error.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # Error for invalid parameters
+  class InvalidParamsError < JSONRPCError
+    # @return [Integer] Error code (-32602)
+    attribute :code, Types::Integer.constant(ErrorCodes::INVALID_PARAMS)
+
+    # @return [String] Error message
+    attribute :message, Types::String.constant('Invalid parameters')
+
+    # @return [Object, nil] Additional error data
+    attribute? :data, Types::Nominal::Any.optional
+  end
+end

data/lib/a2a/types/invalid_request_error.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # Error for invalid request structure
+  class InvalidRequestError < JSONRPCError
+    # @return [Integer] Error code (-32600)
+    attribute :code, Types::Integer.constant(ErrorCodes::INVALID_REQUEST)
+
+    # @return [String] Error message
+    attribute :message, Types::String.constant('Request payload validation error')
+
+    # @return [Object, nil] Additional error data
+    attribute? :data, Types::Nominal::Any.optional
+  end
+end

data/lib/a2a/types/json_parse_error.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # Error for invalid JSON payload
+  class JSONParseError < JSONRPCError
+    # @return [Integer] Error code (-32700)
+    attribute :code, Types::Integer.constant(ErrorCodes::PARSE_ERROR)
+
+    # @return [String] Error message
+    attribute :message, Types::String.constant('Invalid JSON payload')
+
+    # @return [Object, nil] Additional error data
+    attribute? :data, Types::Nominal::Any.optional
+  end
+end

data/lib/a2a/types/jsonrpc_error.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # Represents a JSON-RPC error object.
+  class JSONRPCError < ProtocolStruct
+    # @return [Integer] A number indicating the error type that occurred.
+    attribute :code, Types::Integer.constant(ErrorCodes::PARSE_ERROR)
+
+    # @return [String] A string providing a short description of the error.
+    attribute :message, Types::String.constant('Invalid JSON payload')
+
+    # @return [Object, nil] Optional additional data about the error.
+    attribute? :data, Types::Nominal::Any.optional
+  end
+end

data/lib/a2a/types/jsonrpc_message.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module A2A
+  # Represents a base message for JSON-RPC communication.
+  class JSONRPCMessage < ProtocolStruct
+    # @return [String] Specifies the JSON-RPC version. Must be "2.0".
+    attribute? :jsonrpc, Types::String.constant('2.0')
+
+    # @return [Integer, String, nil] Request identifier. Responses must have the same ID as the request they relate to.
+    attribute? :id, (Types::Integer | Types::String).optional
+  end
+end

data/lib/a2a/types/jsonrpc_request.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # Represents a JSON-RPC request object.
+  class JSONRPCRequest < JSONRPCMessage
+    # @return [String] The name of the method to be invoked.
+    attribute :method, Types::String
+
+    # @return [Hash, nil] Parameters for the method. Can be a structured object, an array, or null/omitted.
+    attribute? :params, Types::Hash.optional
+
+    # @return [String] The name of the method to be invoked.
+    def method = attributes[:method]
+  end
+end

data/lib/a2a/types/jsonrpc_response.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module A2A
+  # Represents a JSON-RPC response object.
+  class JSONRPCResponse < JSONRPCMessage
+    # @return [Object, nil] The result of the method invocation. Required on success.
+    attribute? :result, Types::Nominal::Any.optional
+
+    # @return [JSONRPCError, nil] An error object if an error occurred during the request. Required on failure.
+    attribute? :error, Types::Constructor(JSONRPCError).optional
+  end
+end

data/lib/a2a/types/message.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # Represents a message exchanged between a user and an agent.
+  class Message < ProtocolStruct
+    # @return [String] The role of the sender (user or agent).
+    attribute :role, Types::String.enum('user', 'agent')
+
+    # @return [Array<Part>] The content of the message, composed of one or more parts.
+    attribute :parts, Types::Array.of(Types::Constructor(Part))
+
+    # @return [Hash, nil] Optional metadata associated with the message.
+    attribute? :metadata, Types::Hash.optional
+  end
+end

data/lib/a2a/types/method_not_found_error.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # Error for method not found
+  class MethodNotFoundError < JSONRPCError
+    # @return [Integer] Error code (-32601)
+    attribute :code, Types::Integer.constant(ErrorCodes::METHOD_NOT_FOUND)
+
+    # @return [String] Error message
+    attribute :message, Types::String.constant('Method not found')
+
+    # @return [nil] No additional data
+    attribute? :data, Types::Nil.constant(nil)
+  end
+end

data/lib/a2a/types/part.rb

@@ -0,0 +1,6 @@
+# frozen_string_literal: true
+
+module A2A
+  # Represents a single part of a multi-part message. Can be text, file, or data.
+  Part = TextPart | FilePart | DataPart
+end

data/lib/a2a/types/protocol_struct.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+require 'dry-struct'
+
+module A2A
+  # Base class for all A2A struct objects
+  # @!visibility private
+  class ProtocolStruct < Dry::Struct
+    include Extensions::CaseTransformation
+    include Extensions::JSONDeserialization
+  end
+end

data/lib/a2a/types/push_notification_config.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # Information required for setting up push notifications.
+  class PushNotificationConfig < ProtocolStruct
+    # @return [String] The URL endpoint where the agent should send notifications.
+    attribute :url, Types::String
+
+    # @return [String, nil] A token to be included in push notification requests for verification/authentication.
+    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
+  end
+end

data/lib/a2a/types/push_notification_not_supported_error.rb

@@ -0,0 +1,15 @@
+# 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)
+  end
+end

data/lib/a2a/types/send_task_request.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # Request to send a message/initiate a task.
+  class SendTaskRequest < JSONRPCRequest
+    # @return [String] Method name for sending a task message.
+    attribute :method, Types::String.constant('tasks/send')
+
+    # @return [TaskSendParams] Parameters for the send task method.
+    attribute :params, Types::Constructor(TaskSendParams)
+
+    # @return [String] The name of the method to be invoked.
+    def method = attributes[:method]
+  end
+end

data/lib/a2a/types/send_task_response.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module A2A
+  # Response to a `tasks/send` request.
+  # Contains the Task object or an error.
+  class SendTaskResponse < JSONRPCResponse
+    # @return [Task, nil] The resulting task if successful.
+    attribute? :result, Types::Constructor(Task).optional
+
+    # @return [JSONRPCError, nil] Error information if the request failed.
+    attribute? :error, Types::Constructor(JSONRPCError).optional
+  end
+end

data/lib/a2a/types/send_task_streaming_request.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # Request to send a message/initiate a task and subscribe to streaming updates.
+  class SendTaskStreamingRequest < JSONRPCRequest
+    # @return [String] Method name for sending a task message and subscribing to updates.
+    attribute :method, Types::String.constant('tasks/sendSubscribe')
+
+    # @return [TaskSendParams] Parameters for the streaming task send method.
+    attribute :params, Types::Constructor(TaskSendParams)
+
+    # @return [String] Method name for sending a task message and subscribing to updates.
+    def method = attributes[:method]
+  end
+end

data/lib/a2a/types/send_task_streaming_response.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # Response to a streaming task operation, either through `tasks/sendSubscribe` or a subscription.
+  # Contains a TaskStatusUpdateEvent, TaskArtifactUpdateEvent, or an error.
+  class SendTaskStreamingResponse < JSONRPCResponse
+    # @return [TaskStatusUpdateEvent, TaskArtifactUpdateEvent, nil] Event data if successful.
+    attribute? :result do
+      Types.Constructor(TaskStatusUpdateEvent) | Types.Constructor(TaskArtifactUpdateEvent)
+    end.optional
+
+    # @return [JSONRPCError, nil] Error information if the request failed.
+    attribute? :error, Types::Constructor(JSONRPCError).optional
+  end
+end

data/lib/a2a/types/set_task_push_notification_request.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # Request to set or update the push notification config for a task.
+  class SetTaskPushNotificationRequest < JSONRPCRequest
+    # @return [String] Method name for setting a task notifications.
+    attribute :method, Types::String.constant('tasks/pushNotification/set')
+
+    # @return [TaskPushNotificationConfig] Parameters for the set task push notification method.
+    attribute :params, Types::Constructor(TaskPushNotificationConfig)
+
+    # @return [String] Method name for setting a task notifications.
+    def method = attributes[:method]
+  end
+end

data/lib/a2a/types/set_task_push_notification_response.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module A2A
+  # Response to a `tasks/pushNotification/set` request. Contains the TaskPushNotificationConfig or an error.
+  class SetTaskPushNotificationResponse < JSONRPCResponse
+    # @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
+  end
+end

data/lib/a2a/types/task.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module A2A
+  # Represents a task being processed by an agent.
+  class Task < ProtocolStruct
+    # @return [String] Unique identifier for the task.
+    attribute :id, Types::String
+
+    # @return [String, nil] Optional identifier for the session this task belongs to.
+    attribute? :session_id, Types::String.optional
+
+    # @return [TaskStatus] The current status of the task.
+    attribute :status, Types::Constructor(TaskStatus)
+
+    # @return [Array<Artifact>, nil] Optional list of artifacts associated with the task
+    # (e.g., outputs, intermediate files).
+    attribute? :artifacts, Types::Array.of(Types::Constructor(Artifact)).optional
+
+    # @return [Array<Message>, nil] Optional history of messages exchanged during the task.
+    attribute? :history, Types::Array.of(Types::Constructor(Message)).optional
+
+    # @return [Hash, nil] Optional metadata associated with the task.
+    attribute? :metadata, Types::Hash.optional
+  end
+end

data/lib/a2a/types/task_artifact_update_event.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module A2A
+  # Represents an artifact update event for a task, typically used in streaming scenarios.
+  class TaskArtifactUpdateEvent < ProtocolStruct
+    # @return [String] The ID of the task being updated.
+    attribute :id, Types::String
+
+    # @return [Artifact] The new or updated artifact for the task.
+    attribute :artifact, Types::Constructor(Artifact)
+
+    # @return [Boolean, nil] Flag indicating if this is the final update for the task.
+    attribute? :final, Types::Bool.optional
+
+    # @return [Hash, nil] Optional metadata associated with this update event.
+    attribute? :metadata, Types::Hash.optional
+  end
+end

data/lib/a2a/types/task_id_params.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module A2A
+  # Basic parameters used for task ID operations.
+  class TaskIdParams < ProtocolStruct
+    # @return [String] The unique identifier of the task.
+    attribute :id, Types::String
+
+    # @return [Hash, nil] Optional metadata to include with the operation.
+    attribute? :metadata, Types::Hash.optional
+  end
+end

data/lib/a2a/types/task_not_cancelable_error.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # Error for task not cancelable
+  class TaskNotCancelableError < JSONRPCError
+    # @return [Integer] Error code (-32002)
+    attribute :code, Types::Integer.constant(ErrorCodes::TASK_NOT_CANCELABLE)
+
+    # @return [String] Error message
+    attribute :message, Types::String.constant('Task cannot be canceled')
+
+    # @return [nil] No additional data
+    attribute? :data, Types::Nil.constant(nil)
+  end
+end

data/lib/a2a/types/task_not_found_error.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # Error for task not found
+  class TaskNotFoundError < JSONRPCError
+    # @return [Integer] Error code (-32001)
+    attribute :code, Types::Integer.constant(ErrorCodes::TASK_NOT_FOUND)
+
+    # @return [String] Error message
+    attribute :message, Types::String.constant('Task not found')
+
+    # @return [nil] No additional data
+    attribute? :data, Types::Nil.constant(nil)
+  end
+end

data/lib/a2a/types/task_push_notification_config.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module A2A
+  # Represents the push notification information associated with a specific task ID.
+  class TaskPushNotificationConfig < ProtocolStruct
+    # @return [String] The ID of the task the notification config is associated with.
+    attribute :id, Types::String
+
+    # @return [PushNotificationConfig] The push notification configuration details.
+    attribute :push_notification_config, Types::Constructor(PushNotificationConfig)
+  end
+end

data/lib/a2a/types/task_query_params.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+module A2A
+  # Parameters used for querying task-related information by ID.
+  class TaskQueryParams < TaskIdParams
+    # @return [Integer, nil] Optional history length to retrieve for the task.
+    attribute? :history_length, Types::Integer.optional
+  end
+end

data/lib/a2a/types/task_resubscription_request.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # Request to resubscribe to updates for a task after a connection interruption.
+  class TaskResubscriptionRequest < JSONRPCRequest
+    # @return [String] Method name for resubscribing to task updates.
+    attribute :method, Types::String.constant('tasks/resubscribe')
+
+    # @return [TaskQueryParams] Parameters for the task resubscription method.
+    attribute :params, Types::Constructor(TaskQueryParams)
+
+    # @return [String] Method name for resubscribing to task updates.
+    def method = attributes[:method]
+  end
+end

data/lib/a2a/types/task_send_params.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+module A2A
+  # Parameters for the `tasks/send` method.
+  class TaskSendParams < ProtocolStruct
+    # @return [String] Unique identifier for the task being initiated or continued.
+    attribute :id, Types::String
+
+    # @return [String, nil] Optional identifier for the session this task belongs to. If not provided, a new
+    # session might be implicitly created depending on the agent.
+    attribute? :session_id, Types::String.optional
+
+    # @return [Message] The message content to send to the agent for processing.
+    attribute :message, Types::Constructor(Message)
+
+    # @return [PushNotificationConfig, nil] Optional pushNotification information for receiving notifications about
+    # this task. Requires agent capability.
+    attribute? :push_notification, Types::Constructor(PushNotificationConfig).optional
+
+    # @return [Integer, nil] Optional parameter to specify how much message history to include in the response.
+    attribute? :history_length, Types::Integer.optional
+
+    # @return [Hash, nil] Optional metadata associated with sending this message.
+    attribute? :metadata, Types::Hash.optional
+  end
+end

data/lib/a2a/types/task_state.rb

@@ -0,0 +1,6 @@
+# frozen_string_literal: true
+
+module A2A
+  # The current state of the task
+  TaskState = Types::String.enum('submitted', 'working', 'input-required', 'completed', 'canceled', 'failed', 'unknown')
+end

data/lib/a2a/types/task_status.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # Represents the status of a task at a specific point in time.
+  class TaskStatus < ProtocolStruct
+    # @return [String] The current state of the task.
+    attribute :state, TaskState
+
+    # @return [Message, nil] An optional message associated with the current status.
+    attribute? :message, Types::Constructor(Message).optional
+
+    # @return [DateTime, nil] The timestamp when this status was recorded (ISO 8601 format).
+    attribute? :timestamp, Types::JSON::DateTime.optional
+  end
+end

data/lib/a2a/types/task_status_update_event.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module A2A
+  # Represents a status update event for a task, typically used in streaming scenarios.
+  class TaskStatusUpdateEvent < ProtocolStruct
+    # @return [String] The ID of the task being updated.
+    attribute :id, Types::String
+
+    # @return [TaskStatus] The new status of the task.
+    attribute :status, Types::Constructor(TaskStatus)
+
+    # @return [Boolean] Flag indicating if this is the final update for the task.
+    attribute? :final, Types::Bool.default(false)
+
+    # @return [Hash, nil] Optional metadata associated with this update event.
+    attribute? :metadata, Types::Hash.optional
+  end
+end

data/lib/a2a/types/text_part.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # Represents a part of a message containing text content.
+  class TextPart < ProtocolStruct
+    # @return [String] Type identifier for this part.
+    attribute :type, Types::String.constant('text')
+
+    # @return [String] The text content.
+    attribute :text, Types::String
+
+    # @return [Hash, nil] Optional metadata associated with this text part.
+    attribute? :metadata, Types::Hash.optional
+  end
+end

data/lib/a2a/types/unsupported_operation_error.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module A2A
+  # Error for unsupported operations
+  class UnsupportedOperationError < JSONRPCError
+    # @return [Integer] Error code (-32004)
+    attribute :code, Types::Integer.constant(ErrorCodes::UNSUPPORTED_OPERATION)
+
+    # @return [String] Error message
+    attribute :message, Types::String.constant('This operation is not supported')
+
+    # @return [nil] No additional data
+    attribute? :data, Types::Nil.constant(nil)
+  end
+end

data/lib/a2a/types.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+require 'dry-types'
+
+module A2A
+  # @!visibility private
+  module Types
+    include Dry.Types()
+
+    Dry::Types.define_builder(:constant) { |type, value| type.default(value).enum(value) }
+  end
+end

data/lib/a2a/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module A2A
-  VERSION = '0.1.0.pre'
+  VERSION = '0.1.0'
 end

data/lib/a2a.rb

@@ -1,7 +1,26 @@
 # frozen_string_literal: true
 
-require_relative 'a2a/version'
+require 'dry/inflector'
+require 'zeitwerk'
 
 # Encapsulates all the gem's logic
 module A2A
+  # @!visibility private
+  # @return [Dry::Inflector] The inflector instance used for camelizing keys
+  INFLECTOR = Dry::Inflector.new
 end
+
+loader = Zeitwerk::Loader.for_gem
+loader.inflector.inflect(
+  'a2a' => 'A2A',
+  'json_deserialization' => 'JSONDeserialization',
+  'jsonrpc_error' => 'JSONRPCError',
+  'jsonrpc_message' => 'JSONRPCMessage',
+  'jsonrpc_request' => 'JSONRPCRequest',
+  'jsonrpc_response' => 'JSONRPCResponse',
+  'json_parse_error' => 'JSONParseError'
+)
+loader.collapse("#{__dir__}/a2a/types")
+loader.enable_reloading
+loader.setup
+loader.eager_load