ag-ui 0.2.0 → 1.0.0

data/lib/ag_ui/server/info.rb

@@ -26,19 +26,19 @@ module AgUi
           end
 
           base = {
-            "version" => VERSION_PARITY,
-            "agents" => { agent_id => agent },
+            "version"                       => VERSION_PARITY,
+            "agents"                        => { agent_id => agent },
             "audioFileTranscriptionEnabled" => false,
-            "mode" => "sse",
-            "threadEndpoints" => {
-              "list" => false,
-              "inspect" => false,
-              "mutations" => false,
+            "mode"                          => "sse",
+            "threadEndpoints"               => {
+              "list"             => false,
+              "inspect"          => false,
+              "mutations"        => false,
               "realtimeMetadata" => false,
             },
-            "a2uiEnabled" => a2ui_enabled,
-            "openGenerativeUIEnabled" => false,
-            "telemetryDisabled" => true,
+            "a2uiEnabled"                   => a2ui_enabled,
+            "openGenerativeUIEnabled"       => false,
+            "telemetryDisabled"             => true,
           }
 
           if a2ui_enabled

data/lib/ag_ui/server/middleware/sse_stream.rb

@@ -67,9 +67,9 @@ module AgUi
                  on_event: nil, on_finish: nil, on_task: nil, &block)
           stream = SSE::Stream.new(
             thread_id: thread_id,
-            run_id: run_id,
-            validate: validate,
-            on_event: on_event,
+            run_id:    run_id,
+            validate:  validate,
+            on_event:  on_event,
           )
 
           @env["ag_ui.stream"] = stream

data/lib/ag_ui/server/sse/event_encoder.rb

@@ -23,20 +23,20 @@ module AgUi
 
         private
 
-        def strip_nils(value)
-          case value
-          when Hash
-            value.each_with_object({}) do |(k, v), out|
-              unless v.nil?
-                out[k] = strip_nils(v)
+          def strip_nils(value)
+            case value
+            when Hash
+              value.each_with_object({}) do |(k, v), out|
+                unless v.nil?
+                  out[k] = strip_nils(v)
+                end
               end
+            when Array
+              value.map { |v| strip_nils(v) }
+            else
+              value
             end
-          when Array
-            value.map { |v| strip_nils(v) }
-          else
-            value
           end
-        end
       end
     end
   end

data/lib/ag_ui/server/sse/stream.rb

@@ -90,7 +90,11 @@ module AgUi
           property_defaults = AgUi::Protocol::JsonSchema.raw_schema
             .dig("definitions", definition_name, "properties")
             .each_with_object({}) do |(camel, prop), defaults|
-              value = prop["default"].nil? ? prop["const"] : prop["default"]
+              if prop["default"].nil?
+                value = prop["const"]
+              else
+                value = prop["default"]
+              end
               unless value.nil?
                 defaults[camel] = value
               end

data/lib/ag_ui/server/triage.rb

@@ -45,7 +45,7 @@ module AgUi
           @app.call(
             env.merge(
               "ag_ui.operation" => "stop",
-              "ag_ui.agent_id" => m[:agent_id],
+              "ag_ui.agent_id"  => m[:agent_id],
               "ag_ui.thread_id" => m[:thread_id],
             ),
           )
@@ -66,8 +66,8 @@ module AgUi
           @app.call(
             env.merge(
               "ag_ui.operation" => "run",
-              "ag_ui.agent_id" => "default",
-              "ag_ui.input" => input,
+              "ag_ui.agent_id"  => "default",
+              "ag_ui.input"     => input,
             ),
           )
         rescue RunInput::InvalidError => e
@@ -80,8 +80,8 @@ module AgUi
           @app.call(
             env.merge(
               "ag_ui.operation" => match[:action],
-              "ag_ui.agent_id" => match[:agent_id],
-              "ag_ui.input" => input,
+              "ag_ui.agent_id"  => match[:agent_id],
+              "ag_ui.input"     => input,
             ),
           )
         rescue RunInput::InvalidError => e
@@ -109,17 +109,20 @@ module AgUi
             "access-control-allow-origin"  => "*",
             "access-control-allow-methods" => "GET, POST, OPTIONS",
             "access-control-allow-headers" => "*",
-          }, []]
+          }, []
+]
         end
 
         def bad_request(details)
           [400, { "content-type" => "application/json" },
-           [JSON.generate({ "error" => "Invalid request body", "details" => details })]]
+           [JSON.generate({ "error" => "Invalid request body", "details" => details })]
+]
         end
 
         def not_found
           [404, { "content-type" => "application/json" },
-           [JSON.generate({ "error" => "Not found" })]]
+           [JSON.generate({ "error" => "Not found" })]
+]
         end
     end
   end

data/lib/ag_ui/server.rb

@@ -45,10 +45,10 @@ module AgUi
 
       @agent_id = agent_id
       @info = Info.payload(
-        agent_id: agent_id,
-        description: description,
+        agent_id:     agent_id,
+        description:  description,
         a2ui_enabled: a2ui_enabled,
-        overrides: info_overrides,
+        overrides:    info_overrides,
       )
       @validate = validate
       @store = store
@@ -116,8 +116,8 @@ module AgUi
         thread_id = input&.thread_id.to_s
         stream = SSE::Stream.new(
           thread_id: thread_id,
-          run_id: input&.run_id.to_s,
-          validate: false,
+          run_id:    input&.run_id.to_s,
+          validate:  false,
         )
 
         if @store
@@ -150,7 +150,11 @@ module AgUi
       # Cancel the thread's in-flight run (the store stops its Async
       # task; the stream closes via the builder's ensure) and ack.
       def respond_stop(env)
-        stopped = @store ? @store.stop(env["ag_ui.thread_id"].to_s) : false
+        if @store
+          stopped = @store.stop(env["ag_ui.thread_id"].to_s)
+        else
+          stopped = false
+        end
         [200, JSON_HEADERS.dup, [JSON.generate({ "stopped" => stopped })]]
       end
 
@@ -166,11 +170,11 @@ module AgUi
 
           @builder.open(
             thread_id: thread_id,
-            run_id: run_id,
-            validate: validate,
-            on_event: ->(payload) { @store.record(thread_id, payload) },
+            run_id:    run_id,
+            validate:  validate,
+            on_event:  ->(payload) { @store.record(thread_id, payload) },
             on_finish: -> { @store.finish_run(thread_id) },
-            on_task: ->(task) { @store.attach_task(thread_id, task) },
+            on_task:   ->(task) { @store.attach_task(thread_id, task) },
             &block
           )
         end

data/lib/ag_ui/terminals/ruby_llm.rb

@@ -20,7 +20,7 @@ module AgUi
     # One assistant turn per call: seeds the chat from env[:messages]
     # (including prior tool-call turns), registers env[:tools] as
     # schema-only CLIENT tools (they halt — never execute server-side),
-    # streams text deltas into env[:events], then appends either the
+    # streams text deltas as AG-UI events through the env, then appends either the
     # assistant text or the assistant tool-call message to env[:messages]
     # for the ToolRouter to route.
     #
@@ -44,7 +44,7 @@ module AgUi
         apply_tool_choice(chat, env)
         seed(chat, env[:messages])
 
-        emitter = TurnEmitter.new(env[:events])
+        emitter = TurnEmitter.new(env)
         chat.complete do |chunk|
           thinking = chunk.thinking
           if thinking&.text
@@ -89,9 +89,13 @@ module AgUi
       # until its first non-empty delta (tool-call-only turns produce no
       # empty bubbles), and reasoning closes as soon as text begins —
       # providers stream thinking strictly before the answer.
+      # Turns a provider's chunk stream into the AG-UI text/reasoning
+      # vocabulary, emitted through the env's hooks: a reasoning phase opens
+      # lazily and closes before any text, and the text phase starts on its
+      # first non-empty delta.
       class TurnEmitter
-        def initialize(events)
-          @events = events
+        def initialize(env)
+          @env = env
           @text_id = SecureRandom.uuid
           @reasoning_id = SecureRandom.uuid
           @text_started = false
@@ -101,10 +105,10 @@ module AgUi
         def thinking_delta(text)
           unless text.empty?
             open_reasoning
-            @events << {
-              type: :reasoning_message_content,
-              data: { message_id: @reasoning_id, delta: text },
-            }
+            @env.emit(
+              :reasoning_message_content,
+              { message_id: @reasoning_id, delta: text },
+            )
           end
         end
 
@@ -112,20 +116,17 @@ module AgUi
           unless text.empty?
             close_reasoning
             unless @text_started
-              @events << { type: :text_message_start, data: { message_id: @text_id } }
+              @env.emit(:text_message_start, { message_id: @text_id })
               @text_started = true
             end
-            @events << {
-              type: :text_message_content,
-              data: { message_id: @text_id, delta: text },
-            }
+            @env.emit(:text_message_content, { message_id: @text_id, delta: text })
           end
         end
 
         def finish
           close_reasoning
           if @text_started
-            @events << { type: :text_message_end, data: { message_id: @text_id } }
+            @env.emit(:text_message_end, { message_id: @text_id })
           end
         end
 
@@ -133,16 +134,16 @@ module AgUi
 
           def open_reasoning
             unless @reasoning_open
-              @events << { type: :reasoning_start, data: { message_id: @reasoning_id } }
-              @events << { type: :reasoning_message_start, data: { message_id: @reasoning_id } }
+              @env.emit(:reasoning_start, { message_id: @reasoning_id })
+              @env.emit(:reasoning_message_start, { message_id: @reasoning_id })
               @reasoning_open = true
             end
           end
 
           def close_reasoning
             if @reasoning_open
-              @events << { type: :reasoning_message_end, data: { message_id: @reasoning_id } }
-              @events << { type: :reasoning_end, data: { message_id: @reasoning_id } }
+              @env.emit(:reasoning_message_end, { message_id: @reasoning_id })
+              @env.emit(:reasoning_end, { message_id: @reasoning_id })
               @reasoning_open = false
             end
           end
@@ -168,8 +169,16 @@ module AgUi
         # force copilotkitSuggest.
         def apply_tool_choice(chat, env)
           props = env[:forwarded_props]
-          choice = props.is_a?(Hash) ? props["toolChoice"] : nil
-          name = choice.is_a?(Hash) ? choice.dig("function", "name") : nil
+          if props.is_a?(Hash)
+            choice = props["toolChoice"]
+          else
+            choice = nil
+          end
+          if choice.is_a?(Hash)
+            name = choice.dig("function", "name")
+          else
+            name = nil
+          end
           if name
             chat.with_tools(choice: name)
           end
@@ -179,9 +188,9 @@ module AgUi
           (tools || []).each do |tool|
             chat.with_tool(
               ClientTool.new(
-                name: dig_tool(tool, "name"),
+                name:        dig_tool(tool, "name"),
                 description: dig_tool(tool, "description"),
-                parameters: dig_tool(tool, "parameters"),
+                parameters:  dig_tool(tool, "parameters"),
               ),
             )
           end
@@ -212,8 +221,8 @@ module AgUi
               seed_assistant(chat, message)
             when :tool
               chat.add_message(
-                role: :tool,
-                content: message.content.to_s,
+                role:         :tool,
+                content:      message.content.to_s,
                 tool_call_id: message.tool_call_id,
               )
             end
@@ -283,8 +292,8 @@ module AgUi
 
           if assistant&.tool_call?
             env[:messages] << Brute::Message.new(
-              role: :assistant,
-              content: assistant.content.to_s,
+              role:       :assistant,
+              content:    assistant.content.to_s,
               tool_calls: assistant.tool_calls.each_value.map do |tc|
                 { id: tc.id, name: tc.name, arguments: tc.arguments }
               end,
@@ -366,21 +375,38 @@ describe "AgUi::Terminals::RubyLLM" do
     end
   end
 
-  build_env = -> do
-    { messages: Brute.log, events: [], metadata: {}, current_iteration: 1 }
+  # Brute 6 replaced the env[:events] sink with the hooks registry, so a test
+  # that wants to see what the terminal streamed subscribes instead of reading
+  # an array off the env. `events` is where the turn's deltas land.
+  TURN_EVENTS = %i[
+    text_message_start text_message_content text_message_end
+    reasoning_start reasoning_message_start reasoning_message_content
+    reasoning_message_end reasoning_end
+  ].freeze
+
+  build_env = ->(into = []) do
+    hooks = Brute::Hooks::Registry.new
+    TURN_EVENTS.each do |event|
+      hooks.on(event) { |_env, data, _trace| into << { type: event, data: data } }
+    end
+    Brute::Hooks::Trace.new(
+      { messages: Brute.log, metadata: {}, current_iteration: 1 },
+      hooks: hooks,
+    )
   end
 
-  it "streams deltas into env[:events] and appends the assistant message" do
+  it "streams deltas as AG-UI events and appends the assistant message" do
     fake = fake_chat_class.new(chunks: ["Hel", "", "lo"], final: "Hello")
     terminal = AgUi::Terminals::RubyLLM.new(chat_factory: ->(**) { fake })
 
-    env = build_env.()
+    events = []
+    env = build_env.(events)
     env[:messages].user("hi")
     terminal.call(env)
 
-    types = env[:events].map { |e| e[:type] }
+    types = events.map { |e| e[:type] }
     types.should == [:text_message_start, :text_message_content, :text_message_content, :text_message_end]
-    env[:events][1][:data][:delta].should == "Hel"
+    events[1][:data][:delta].should == "Hel"
 
     env[:messages].last.role.should == :assistant
     env[:messages].last.content.should == "Hello"
@@ -410,12 +436,13 @@ describe "AgUi::Terminals::RubyLLM" do
     fake = fake_chat_class.new(final: nil, tool_calls: tool_calls)
     terminal = AgUi::Terminals::RubyLLM.new(chat_factory: ->(**) { fake })
 
-    env = build_env.()
+    events = []
+    env = build_env.(events)
     env[:messages].user("go to data")
     terminal.call(env)
 
     # No text events for a tool-only turn (lazy start).
-    env[:events].should == []
+    events.should == []
 
     last = env[:messages].last
     last.tool_call?.should.be.true
@@ -466,21 +493,22 @@ describe "AgUi::Terminals::RubyLLM" do
     )
     terminal = AgUi::Terminals::RubyLLM.new(thinking: { budget: 1024 }, chat_factory: ->(**) { fake })
 
-    env = build_env.()
+    events = []
+    env = build_env.(events)
     env[:messages].user("why?")
     terminal.call(env)
 
     fake.thinking_config.should == { budget: 1024 }
-    env[:events].map { |e| e[:type] }.should == %i[
+    events.map { |e| e[:type] }.should == %i[
       reasoning_start reasoning_message_start
       reasoning_message_content reasoning_message_content
       reasoning_message_end reasoning_end
       text_message_start text_message_content text_message_end
     ]
 
-    reasoning_ids = env[:events].first(6).map { |e| e[:data][:message_id] }.uniq
+    reasoning_ids = events.first(6).map { |e| e[:data][:message_id] }.uniq
     reasoning_ids.length.should == 1
-    env[:events].last[:data][:message_id].should.not == reasoning_ids.first
+    events.last[:data][:message_id].should.not == reasoning_ids.first
   end
 
   it "seeds multimodal user content as ruby_llm Content with attachments" do
@@ -536,7 +564,8 @@ describe "AgUi::Terminals::RubyLLM" do
                    .run(terminal)
       env["ag_ui.stream"].open(thread_id: input.thread_id, run_id: input.run_id) do |stream|
         stream.run_started
-        agent.start(AgUi::Messages.to_brute(input.messages), events: AgUi::EventBridge.new(stream))
+        AgUi::EventBridge.new(stream).subscribe(agent)
+        agent.start(AgUi::Messages.to_brute(input.messages))
         stream.run_finished
       rescue => e
         stream.run_error(message: e.message, code: e.class.name)

data/lib/ag_ui/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module AgUi
-  VERSION = "0.2.0"
+  VERSION = "1.0.0"
 end

data/lib/ag_ui.rb

@@ -25,6 +25,7 @@ require "ag_ui/event_bridge"
 require "ag_ui/middleware/system_prompt"
 require "ag_ui/middleware/forwarded_props"
 require "ag_ui/middleware/tool_router"
+require "ag_ui/middleware/state"
 require "ag_ui/a2ui/catalog"
 require "ag_ui/a2ui/validate"
 require "ag_ui/a2ui/recovery"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ag-ui
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 1.0.0
 platform: ruby
 authors:
 - AgUi Contributors
@@ -66,25 +66,47 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '2.5'
 - !ruby/object:Gem::Dependency
-  name: brute
+  name: hana
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.0'
-    - - ">="
+        version: '1.3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: 3.0.1
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: brute
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.0'
-    - - ">="
+        version: '6.0'
+- !ruby/object:Gem::Dependency
+  name: gem_kit
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: 3.0.1
+        version: '0.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.2'
 - !ruby/object:Gem::Dependency
   name: ruby_llm
   requirement: !ruby/object:Gem::Requirement
@@ -183,6 +205,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '2.1'
+- !ruby/object:Gem::Dependency
+  name: gem_kit-release
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.3'
 description: Ruby server implementation of the AG-UI protocol (SSE event streaming,
   run loop, client-tool round-trips, A2UI) for driving CopilotKit frontends from Rack/Falcon.
 executables: []
@@ -191,7 +227,6 @@ extra_rdoc_files: []
 files:
 - LICENSE
 - data/ag_ui.json
-- data/generate-ag-ui-schema.py
 - lib/ag_ui.rb
 - lib/ag_ui/a2ui/catalog.rb
 - lib/ag_ui/a2ui/recovery.rb
@@ -200,6 +235,7 @@ files:
 - lib/ag_ui/messages.rb
 - lib/ag_ui/middleware/a2ui.rb
 - lib/ag_ui/middleware/forwarded_props.rb
+- lib/ag_ui/middleware/state.rb
 - lib/ag_ui/middleware/system_prompt.rb
 - lib/ag_ui/middleware/tool_router.rb
 - lib/ag_ui/protocol/json_schema.rb

data/data/generate-ag-ui-schema.py

@@ -1,50 +0,0 @@
-# Generates data/ag_ui.json from the reference Python SDK's pydantic models
-# (clone github.com/ag-ui-protocol/ag-ui and point AG_UI_SDK at sdks/python).
-#
-# Run: uv run --with pydantic data/generate-ag-ui-schema.py [/path/to/ag-ui/sdks/python]
-#
-# Output shape mirrors a2a's data/a2a.json: a single top-level "definitions"
-# map keyed by model class name, refs rewritten to #/definitions/<Model>.
-# Wire keys are camelCase (by_alias), matching @ag-ui/core exactly.
-
-import inspect
-import json
-import os
-import sys
-
-sdk = sys.argv[1] if len(sys.argv) > 1 else os.environ.get("AG_UI_SDK", "/tmp/ag-ui/sdks/python")
-sys.path.insert(0, sdk)
-
-from pydantic import BaseModel  # noqa: E402
-from pydantic.json_schema import models_json_schema  # noqa: E402
-
-from ag_ui.core import capabilities, events, types  # noqa: E402
-
-models = []
-for mod in (types, events, capabilities):
-    for _name, obj in sorted(vars(mod).items()):
-        if (
-            inspect.isclass(obj)
-            and issubclass(obj, BaseModel)
-            and obj.__module__ == mod.__name__
-            and obj.__name__ != "ConfiguredBaseModel"
-        ):
-            models.append(obj)
-
-_, top = models_json_schema(
-    [(m, "validation") for m in models],
-    by_alias=True,
-    ref_template="#/definitions/{model}",
-)
-
-out = {
-    "$comment": f"Generated by data/generate-ag-ui-schema.py from ag-ui sdks/python ({len(models)} models). Do not edit.",
-    "definitions": top["$defs"],
-}
-
-dest = os.path.join(os.path.dirname(os.path.abspath(__file__)), "ag_ui.json")
-with open(dest, "w") as f:
-    json.dump(out, f, indent=2, sort_keys=True)
-    f.write("\n")
-
-print(f"wrote {dest}: {len(out['definitions'])} definitions")