simple_a2a 0.1.0 → 0.3.0

data/lib/simple_a2a/server/base.rb

@@ -3,33 +3,35 @@
 module A2A
   module Server
     class Base
-      attr_reader :agent_card, :executor, :storage, :event_router, :push_sender
+      attr_reader :agent_card, :executor, :storage, :push_sender
 
       def initialize(
         agent_card:,
         executor:,
-        storage:      Storage::Memory.new,
-        push_sender:  nil,
-        host:         "localhost",
-        port:         9292
+        storage:           Storage::Memory.new,
+        push_sender:       nil,
+        push_config_store: nil,
+        host:              "localhost",
+        port:              9292
       )
-        @agent_card   = agent_card
-        @executor     = executor
-        @storage      = storage
-        @event_router = EventRouter.new
-        @push_sender  = push_sender
-        @host         = host
-        @port         = port
+        @agent_card        = agent_card
+        @executor          = executor
+        @storage           = storage
+        @push_sender       = push_sender
+        @push_config_store = push_config_store
+        @host              = host
+        @port              = port
       end
 
       def rack_app
         klass = Class.new(App)
         klass.configure(
-          agent_card:   @agent_card,
-          storage:      @storage,
-          executor:     @executor,
-          event_router: @event_router,
-          push_sender:  @push_sender
+          agent_card:         @agent_card,
+          storage:            @storage,
+          executor:           @executor,
+          broadcast_registry: BroadcastRegistry.new,
+          push_sender:        @push_sender,
+          push_config_store:  @push_config_store
         )
         klass.freeze.app
       end

data/lib/simple_a2a/server/broadcast_registry.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module A2A
+  module Server
+    class BroadcastRegistry
+      def initialize
+        @broadcasts = {}
+        @mutex      = Mutex.new
+      end
+
+      def register(task_id, broadcast)
+        @mutex.synchronize { @broadcasts[task_id] = broadcast }
+      end
+
+      def unregister(task_id)
+        @mutex.synchronize { @broadcasts.delete(task_id) }
+      end
+
+      def find(task_id)
+        @mutex.synchronize { @broadcasts[task_id] }
+      end
+    end
+  end
+end

data/lib/simple_a2a/server/multi_agent.rb

@@ -40,7 +40,7 @@ module A2A
           agent_card:   cfg[:agent_card],
           storage:      cfg[:storage]      || Storage::Memory.new,
           executor:     cfg[:executor],
-          event_router: cfg[:event_router] || EventRouter.new,
+          broadcast_registry: BroadcastRegistry.new,
           push_sender:  cfg[:push_sender]
         )
         klass.freeze.app

data/lib/simple_a2a/server/push_config_store.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module A2A
+  module Server
+    class PushConfigStore
+      def initialize
+        @configs = {}
+        @mutex   = Mutex.new
+      end
+
+      def set(task_id, config)
+        @mutex.synchronize { @configs[task_id] = config }
+        config
+      end
+
+      def get(task_id)
+        @mutex.synchronize { @configs[task_id] }
+      end
+
+      def delete(task_id)
+        @mutex.synchronize { @configs.delete(task_id) }
+      end
+
+      def list
+        @mutex.synchronize { @configs.dup }
+      end
+    end
+  end
+end

data/lib/simple_a2a/server/push_sender.rb

@@ -1,5 +1,6 @@
 # frozen_string_literal: true
 
+require "digest"
 require "jwt"
 require "net/http"
 require "uri"

data/lib/simple_a2a/server/task_broadcast.rb

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+require "ractor_queue"
+
+module A2A
+  module Server
+    class TaskBroadcast
+      DONE = :done.freeze
+
+      BroadcastError = Struct.new(:message)
+
+      def initialize
+        @queues = []
+        @mutex  = Mutex.new
+      end
+
+      def subscribe(capacity: 64)
+        RactorQueue.new(capacity: capacity).tap do |q|
+          @mutex.synchronize { @queues << q }
+        end
+      end
+
+      def unsubscribe(queue)
+        @mutex.synchronize { @queues.delete(queue) }
+      end
+
+      # Duck-type compatible with the old EventRouter interface.
+      # task_id is accepted but ignored — the broadcast is already task-scoped.
+      def publish(_task_id, event)
+        snapshot = @mutex.synchronize { @queues.dup }
+        snapshot.each { |q| q.async_push(event) }
+      end
+
+      def error(message)
+        ev = BroadcastError.new(message)
+        snapshot = @mutex.synchronize { @queues.dup }
+        snapshot.each { |q| q.async_push(ev) }
+      end
+
+      def close
+        snapshot = @mutex.synchronize { @queues.dup }
+        snapshot.each { |q| q.async_push(DONE) }
+      end
+    end
+  end
+end

data/lib/simple_a2a/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module A2A
-  VERSION = "0.1.0"
+  VERSION = "0.3.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: simple_a2a
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Dewayne VanHoozer
@@ -122,33 +122,19 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '2.0'
 - !ruby/object:Gem::Dependency
-  name: simple_flow
+  name: ractor_queue
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.4'
+        version: '0.2'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.4'
-- !ruby/object:Gem::Dependency
-  name: typed_bus
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '0.0'
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '0.0'
+        version: '0.2'
 - !ruby/object:Gem::Dependency
   name: rake
   requirement: !ruby/object:Gem::Requirement
@@ -248,7 +234,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 description: Client and server for the A2A protocol — async-first, Rack-compatible,
-  built on Falcon and TypedBus.
+  built on Falcon.
 email:
 - dvanhoozer@gmail.com
 executables: []
@@ -260,6 +246,7 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
+- compare_agent2agent.md
 - docs/api/client/index.md
 - docs/api/index.md
 - docs/api/models/index.md
@@ -268,9 +255,18 @@ files:
 - docs/architecture/index.md
 - docs/architecture/protocol.md
 - docs/assets/css/custom.css
+- docs/assets/images/simple_a2a.jpg
+- docs/examples/agent-chaining.md
+- docs/examples/auth-headers.md
 - docs/examples/basic-usage.md
+- docs/examples/cancellation.md
 - docs/examples/index.md
+- docs/examples/interrupted-states.md
 - docs/examples/llm-research.md
+- docs/examples/multipart.md
+- docs/examples/push-notifications.md
+- docs/examples/resubscribe.md
+- docs/examples/sqlite-storage.md
 - docs/examples/streaming.md
 - docs/getting-started/installation.md
 - docs/getting-started/quick-start.md
@@ -286,6 +282,26 @@ files:
 - examples/03_llm_research/run
 - examples/03_llm_research/server.rb
 - examples/03_llm_research/web_client.rb
+- examples/04_resubscribe/client.rb
+- examples/04_resubscribe/server.rb
+- examples/05_cancellation/client.rb
+- examples/05_cancellation/server.rb
+- examples/06_push_notifications/client.rb
+- examples/06_push_notifications/server.rb
+- examples/07_agent_chaining/client.rb
+- examples/07_agent_chaining/server.rb
+- examples/08_interrupted_states/client.rb
+- examples/08_interrupted_states/server.rb
+- examples/09_multipart/client.rb
+- examples/09_multipart/server.rb
+- examples/10_auth_headers/client.rb
+- examples/10_auth_headers/server.rb
+- examples/11_sqlite_storage/Brewfile
+- examples/11_sqlite_storage/Gemfile
+- examples/11_sqlite_storage/client.rb
+- examples/11_sqlite_storage/run
+- examples/11_sqlite_storage/server.rb
+- examples/README.md
 - examples/common_config.rb
 - examples/run
 - lib/simple_a2a.rb
@@ -315,12 +331,14 @@ files:
 - lib/simple_a2a/server/agent_executor.rb
 - lib/simple_a2a/server/app.rb
 - lib/simple_a2a/server/base.rb
+- lib/simple_a2a/server/broadcast_registry.rb
 - lib/simple_a2a/server/context.rb
-- lib/simple_a2a/server/event_router.rb
 - lib/simple_a2a/server/falcon_runner.rb
 - lib/simple_a2a/server/multi_agent.rb
+- lib/simple_a2a/server/push_config_store.rb
 - lib/simple_a2a/server/push_sender.rb
 - lib/simple_a2a/server/resume_context.rb
+- lib/simple_a2a/server/task_broadcast.rb
 - lib/simple_a2a/storage/base.rb
 - lib/simple_a2a/storage/memory.rb
 - lib/simple_a2a/version.rb

data/lib/simple_a2a/server/event_router.rb

@@ -1,50 +0,0 @@
-# frozen_string_literal: true
-
-require "typed_bus"
-
-module A2A
-  module Server
-    class EventRouter
-      def initialize
-        @bus = TypedBus::MessageBus.new
-      end
-
-      def open(task_id)
-        return if @bus.channel?(task_id.to_sym)
-        @bus.add_channel(task_id.to_sym, type: nil, timeout: nil)
-      end
-
-      def close(task_id)
-        @bus.remove_channel(task_id.to_sym)
-      end
-
-      def publish(task_id, event)
-        sym = task_id.to_sym
-        open(task_id) unless @bus.channel?(sym)
-        @bus.publish(sym, event)
-      rescue ArgumentError
-        nil
-      end
-
-      def subscribe(task_id, &block)
-        sym = task_id.to_sym
-        open(task_id) unless @bus.channel?(sym)
-        @bus.subscribe(sym) do |delivery|
-          block.call(delivery.message)
-          delivery.ack!
-        end
-      end
-
-      def unsubscribe(task_id, id_or_block)
-        return unless @bus.channel?(task_id.to_sym)
-        @bus.unsubscribe(task_id.to_sym, id_or_block)
-      rescue ArgumentError
-        nil
-      end
-
-      def channel?(task_id)
-        @bus.channel?(task_id.to_sym)
-      end
-    end
-  end
-end