ai-agents 0.10.0 → 0.11.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6d4dc10b4aceae77705002488794ecdd15531f604641adaa03445d72374220f0
-  data.tar.gz: b056fcf690121a790808e6aa1229dc1a16b6e19b609fbaa09a5e32613f5bc58f
+  metadata.gz: '071869beb00d53446a27f489c3817c58b0f8d163f334f0030c95b789a7c4895b'
+  data.tar.gz: 874227bd4dd05dd2947269bec7da2d593777a8600540ef9ec5a8fb50c346a884
 SHA512:
-  metadata.gz: b80b45ccaa92140f6155dceeb2227235150f913c90385197878074af9df0c22b913e611c46424f9efa9b3f8f128b8921795c3136685a889a758c4b8dd81f4dbd
-  data.tar.gz: 63662db9f1dcbed10439dfa64746a868a0ee2cc64266c48e55f613296e90dca0d1661466af01825bbec3015c928209f20f000bc9aeb882f5ff510be7ad7110f7
+  metadata.gz: 7e3814e0e76359be595534eb92ce11d655e19e3997f78aea006be89721aae771c6a1f97ad277ed80581957cdea5a0919506518dfd8e8de1911a03a3641668a8b
+  data.tar.gz: 351b99abda0942df5253e1735faca522b5eee988fc248ec16564dea700b2bcfcf31a8ec4ba8f9db73f3f57103e5f8b37a0997c6e71f4b1a1872ca796046cad2a

data/CHANGELOG.md

@@ -7,6 +7,13 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [0.11.0] - 2026-05-27
+
+### Added
+- Support RubyLLM provider overrides and custom model/deployment IDs on agents, including Azure configuration passthrough (#66)
+
+### Fixed
+- Preserve per-message assistant agent attribution when restoring and snapshotting multi-agent conversation history, preventing earlier assistant messages from being re-labeled as the final active agent after handoffs (#68)
 
 ## [0.10.0] - 2026-04-20
 

data/README.md

@@ -193,22 +193,40 @@ ruby examples/isp-support/interactive.rb
 Agents.configure do |config|
   # Provider API keys
   config.openai_api_key = ENV['OPENAI_API_KEY']
+  config.azure_api_base = ENV['AZURE_API_BASE']
+  config.azure_api_key = ENV['AZURE_API_KEY']
   config.anthropic_api_key = ENV['ANTHROPIC_API_KEY']
   config.gemini_api_key = ENV['GEMINI_API_KEY']
 
   # Defaults
-  config.default_provider = :openai
   config.default_model = 'gpt-4o'
 
   # Performance
   config.request_timeout = 120
-  config.max_turns = 10
 
   # Debugging
   config.debug = true
 end
 ```
 
+### Azure and Custom Deployments
+
+```ruby
+Agents.configure do |config|
+  config.azure_api_base = ENV["AZURE_API_BASE"]
+  config.azure_api_key = ENV["AZURE_API_KEY"]
+end
+
+agent = Agents::Agent.new(
+  name: "Support",
+  model: "my-azure-deployment",
+  provider: :azure,
+  assume_model_exists: true
+)
+```
+
+`provider` is optional for known, unambiguous registry models. Set it for custom deployment names and for model IDs that can exist under multiple providers, such as Azure and OpenAI deployments.
+
 ## 🔍 Observability
 
 Optional OpenTelemetry instrumentation for tracing agent execution, compatible with

data/docs/guides/rails-integration.md

@@ -24,12 +24,25 @@ Configure your LLM providers in an initializer:
 # config/initializers/ai_agents.rb
 Agents.configure do |config|
   config.openai_api_key = Rails.application.credentials.openai_api_key
+  config.azure_api_base = Rails.application.credentials.azure_api_base
+  config.azure_api_key = Rails.application.credentials.azure_api_key
   config.anthropic_api_key = Rails.application.credentials.anthropic_api_key
   config.default_model = 'gpt-4o-mini'
   config.debug = Rails.env.development?
 end
 ```
 
+For Azure custom deployment names, pass the provider at the agent level:
+
+```ruby
+support = Agents::Agent.new(
+  name: "Support",
+  model: "my-azure-deployment",
+  provider: :azure,
+  assume_model_exists: true
+)
+```
+
 ## ActiveRecord Integration
 
 ### Conversation Persistence

data/docs/index.md

@@ -61,6 +61,8 @@ require 'agents'
 # Configure your API keys
 Agents.configure do |config|
   config.openai_api_key = ENV['OPENAI_API_KEY']
+  # config.azure_api_base = ENV['AZURE_API_BASE']
+  # config.azure_api_key = ENV['AZURE_API_KEY']
   # config.anthropic_api_key = ENV['ANTHROPIC_API_KEY']
   # config.gemini_api_key = ENV['GEMINI_API_KEY']
   config.default_model = 'gpt-4o-mini'
@@ -87,6 +89,24 @@ result = runner.run("I need help with a technical issue")
 puts result.output
 ```
 
+### Azure and Custom Deployments
+
+```ruby
+Agents.configure do |config|
+  config.azure_api_base = ENV["AZURE_API_BASE"]
+  config.azure_api_key = ENV["AZURE_API_KEY"]
+end
+
+agent = Agents::Agent.new(
+  name: "Support",
+  model: "my-azure-deployment",
+  provider: :azure,
+  assume_model_exists: true
+)
+```
+
+`provider` is optional for known, unambiguous registry models. Set it for custom deployment names and duplicate model IDs, such as Azure and OpenAI deployments.
+
 ## Next Steps
 
 - [Learn about Agents](concepts/agents.html)

data/lib/agents/agent.rb

@@ -50,24 +50,29 @@ require_relative "helpers/hash_normalizer"
 #   )
 module Agents
   class Agent
-    attr_reader :name, :instructions, :model, :tools, :handoff_agents, :temperature, :response_schema, :headers, :params
+    attr_reader :name, :instructions, :model, :provider, :assume_model_exists, :tools, :handoff_agents, :temperature,
+                :response_schema, :headers, :params
 
     # Initialize a new Agent instance
     #
     # @param name [String] The name of the agent
     # @param instructions [String, Proc, nil] Static string or dynamic Proc that returns instructions
     # @param model [String] The LLM model to use (default: "gpt-4.1-mini")
+    # @param provider [Symbol, String, nil] Optional RubyLLM provider override
+    # @param assume_model_exists [Boolean] Whether RubyLLM should skip registry validation for custom model IDs
     # @param tools [Array<Agents::Tool>] Array of tool instances the agent can use
     # @param handoff_agents [Array<Agents::Agent>] Array of agents this agent can hand off to
     # @param temperature [Float] Controls randomness in responses (0.0 = deterministic, 1.0 = very random, default: 0.7)
     # @param response_schema [Hash, nil] JSON schema for structured output responses
     # @param headers [Hash, nil] Default HTTP headers applied to LLM requests
     # @param params [Hash, nil] Default provider-specific parameters applied to LLM requests (e.g., service_tier)
-    def initialize(name:, instructions: nil, model: "gpt-4.1-mini", tools: [], handoff_agents: [], temperature: 0.7,
-                   response_schema: nil, headers: nil, params: nil)
+    def initialize(name:, instructions: nil, model: "gpt-4.1-mini", provider: nil, assume_model_exists: false,
+                   tools: [], handoff_agents: [], temperature: 0.7, response_schema: nil, headers: nil, params: nil)
       @name = name
       @instructions = instructions
       @model = model
+      @provider = provider&.to_sym
+      @assume_model_exists = assume_model_exists
       @tools = tools.dup
       @handoff_agents = []
       @temperature = temperature
@@ -155,6 +160,8 @@ module Agents
     # @option changes [String] :name New agent name
     # @option changes [String, Proc] :instructions New instructions
     # @option changes [String] :model New model identifier
+    # @option changes [Symbol, String, nil] :provider New provider override
+    # @option changes [Boolean] :assume_model_exists Whether to skip model registry validation
     # @option changes [Array<Agents::Tool>] :tools New tools array (replaces all tools)
     # @option changes [Array<Agents::Agent>] :handoff_agents New handoff agents
     # @option changes [Float] :temperature Temperature for LLM responses (0.0-1.0)
@@ -165,6 +172,8 @@ module Agents
         name: changes.fetch(:name, @name),
         instructions: changes.fetch(:instructions, @instructions),
         model: changes.fetch(:model, @model),
+        provider: changes.fetch(:provider, @provider),
+        assume_model_exists: changes.fetch(:assume_model_exists, @assume_model_exists),
         tools: changes.fetch(:tools, @tools.dup),
         handoff_agents: changes.fetch(:handoff_agents, @handoff_agents),
         temperature: changes.fetch(:temperature, @temperature),

data/lib/agents/helpers/message_extractor.rb

@@ -18,8 +18,30 @@
 module Agents
   module Helpers
     module MessageExtractor
+      # RubyLLM::Message has no metadata/extension API, so agent ownership is stored
+      # as an SDK-namespaced ivar on the message object until it is extracted.
+      #
+      # Caveat: RubyLLM::Message#instance_variables is overridden to hide :@raw but does
+      # not hide this ivar, so it will appear in instance_variables listings. This is
+      # harmless for our own code path (we read it via instance_variable_get and
+      # serialize through extract_messages, not Marshal), but external introspection
+      # of a message's ivars will see :@agents_authoring_agent.
+      AUTHORING_AGENT_IVAR = :@agents_authoring_agent
+
       module_function
 
+      def assign_agent_name(message, agent_name)
+        return unless message && agent_name
+
+        message.instance_variable_set(AUTHORING_AGENT_IVAR, agent_name)
+      end
+
+      def attributed_agent_name_for(message)
+        return unless message&.instance_variable_defined?(AUTHORING_AGENT_IVAR)
+
+        message.instance_variable_get(AUTHORING_AGENT_IVAR)
+      end
+
       # Check if content is considered empty (handles both String and Hash content)
       #
       # @param content [String, Hash, nil] The content to check
@@ -65,7 +87,8 @@ module Agents
 
         return message unless msg.role == :assistant
 
-        message[:agent_name] = current_agent.name if current_agent
+        attributed_agent_name = attributed_agent_name_for(msg) || current_agent&.name
+        message[:agent_name] = attributed_agent_name if attributed_agent_name
 
         if tool_calls_present
           # RubyLLM stores tool_calls as Hash with call_id => ToolCall object

data/lib/agents/runner.rb

@@ -103,7 +103,11 @@ module Agents
       agent_params = Helpers::HashNormalizer.normalize(current_agent.params, label: "params")
 
       # Create chat and restore conversation history
-      chat = RubyLLM::Chat.new(model: current_agent.model)
+      chat = RubyLLM::Chat.new(
+        model: current_agent.model,
+        provider: current_agent.provider,
+        assume_model_exists: current_agent.assume_model_exists
+      )
       current_headers = Helpers::HashNormalizer.merge(agent_headers, runtime_headers)
       current_params = Helpers::HashNormalizer.merge(agent_params, runtime_params)
       apply_headers(chat, current_headers)
@@ -118,6 +122,7 @@ module Agents
         raise MaxTurnsExceeded, "Exceeded maximum turns: #{max_turns}" if current_turn > max_turns
 
         # Get response from LLM (RubyLLM handles tool execution with halting based handoff detection)
+        message_count_before_response = chat_message_count(chat)
         response = if current_turn == 1
                      # Emit agent thinking event for initial message
                      context_wrapper.callback_manager.emit_agent_thinking(current_agent.name, input, context_wrapper)
@@ -130,6 +135,7 @@ module Agents
                                                                           context_wrapper)
                      chat.complete
                    end
+        assign_agent_name_to_new_assistant_messages(chat, current_agent, message_count_before_response)
         track_usage(response, context_wrapper)
 
         # Emit LLM call complete event with model and response for instrumentation
@@ -266,6 +272,7 @@ module Agents
         next unless message_params # Skip invalid messages
 
         message = RubyLLM::Message.new(**message_params)
+        assign_restored_agent_name(message, msg)
         chat.add_message(message)
 
         if message.role == :assistant && message_params[:tool_calls]
@@ -377,6 +384,33 @@ module Agents
       context_wrapper.context.delete(:pending_handoff)
     end
 
+    def assign_agent_name_to_new_assistant_messages(chat, current_agent, start_index)
+      # Runtime chats are RubyLLM::Chat instances and expose messages. Keep this
+      # no-op guard for chat-like doubles/adapters that do not expose history.
+      return unless chat.respond_to?(:messages)
+
+      chat.messages[start_index..]&.each do |message|
+        next unless message.role == :assistant
+
+        Helpers::MessageExtractor.assign_agent_name(message, current_agent.name)
+      end
+    end
+
+    def chat_message_count(chat)
+      # Runtime chats are RubyLLM::Chat instances and expose messages. Keep this
+      # fallback for chat-like doubles/adapters where attribution is irrelevant.
+      return 0 unless chat.respond_to?(:messages)
+
+      chat.messages.length
+    end
+
+    def assign_restored_agent_name(message, msg)
+      return unless message.role == :assistant
+
+      restored_agent_name = msg[:agent_name] || msg["agent_name"]
+      Helpers::MessageExtractor.assign_agent_name(message, restored_agent_name)
+    end
+
     # Configures a RubyLLM chat instance with agent-specific settings.
     # Uses RubyLLM's replace option to swap agent context while preserving conversation history during handoffs.
     #
@@ -393,7 +427,13 @@ module Agents
       all_tools = build_agent_tools(agent, context_wrapper)
 
       # Switch model if different (important for handoffs between agents using different models)
-      chat.with_model(agent.model) if replace
+      if replace
+        chat.with_model(
+          agent.model,
+          provider: agent.provider,
+          assume_exists: agent.assume_model_exists
+        )
+      end
 
       # Configure chat with instructions, temperature, tools, and schema
       chat.with_instructions(system_prompt, replace: replace) if system_prompt

data/lib/agents/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Agents
-  VERSION = "0.10.0"
+  VERSION = "0.11.0"
 end

data/lib/agents.rb

@@ -56,6 +56,9 @@ module Agents
 
       # Other providers
       apply_if_present(config, :anthropic_api_key)
+      apply_if_present(config, :azure_api_base)
+      apply_if_present(config, :azure_api_key)
+      apply_if_present(config, :azure_ai_auth_token)
       apply_if_present(config, :gemini_api_key)
       apply_if_present(config, :deepseek_api_key)
       apply_if_present(config, :openrouter_api_key)
@@ -83,8 +86,9 @@ module Agents
   class Configuration
     # Provider API keys and configuration
     attr_accessor :openai_api_key, :openai_api_base, :openai_organization_id, :openai_project_id
-    attr_accessor :anthropic_api_key, :gemini_api_key, :deepseek_api_key, :openrouter_api_key, :ollama_api_base,
-                  :bedrock_api_key, :bedrock_secret_key, :bedrock_region, :bedrock_session_token
+    attr_accessor :anthropic_api_key, :azure_api_base, :azure_api_key, :azure_ai_auth_token, :gemini_api_key,
+                  :deepseek_api_key, :openrouter_api_key, :ollama_api_base, :bedrock_api_key, :bedrock_secret_key,
+                  :bedrock_region, :bedrock_session_token
 
     # General configuration
     attr_accessor :request_timeout, :default_model, :debug
@@ -100,7 +104,7 @@ module Agents
     def configured?
       @openai_api_key || @anthropic_api_key || @gemini_api_key ||
         @deepseek_api_key || @openrouter_api_key || @ollama_api_base ||
-        @bedrock_api_key
+        @bedrock_api_key || (@azure_api_base && (@azure_api_key || @azure_ai_auth_token))
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ai-agents
 version: !ruby/object:Gem::Version
-  version: 0.10.0
+  version: 0.11.0
 platform: ruby
 authors:
 - Shivam Mishra