a.rb

my_container = Dry::Container.new
my_container.register(:'one.data_store', -> { A.new })
my_container.register(:'two.data_store', -> { A.new })


Import = Dry::AutoInject(my_container)

class OP

  attr_reader :context

  def initialize(**args)
    @context = args.fetch(:context)
  end

end

class B < OP
  include Import.kwargs[blah: 'one.data_store', bleh: 'two.data_store']

  def initialize(**args)
    super
  end

  def call
    p context.x
    bleh.call
  end
end

B.new(context: OpenStruct.new(x: :y)).call