omni_agent 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 68e52e8d8eb8a4f59c708d32e7464b7526c92f8e5342d6a85fe3c1244bf643e7
-  data.tar.gz: 36d892f5d7593582f31cc15d957deab720c80c8416a48ba32df4253f24d10ac6
+  metadata.gz: 1513c46e356438c23dfd193f25c390df240f2d1d812813114e2e582222de2965
+  data.tar.gz: a9b05b4445128aea795b3f7f5ea80cb6bd668bf2b674c15651011ce6780e0c2e
 SHA512:
-  metadata.gz: afaf69e6f85c7db759d6796477601945577550a80f86e53a694f52d07910ffe081e79aa18db0688a8aaf3b4cb5dae5e141beaefc43e4083c59e46af2f6496df3
-  data.tar.gz: efc7e291b645b15c0fe2d528737597948aaa28b9cc5a2f1bffd153313f071c0bd659d3d5b885932a607cf21f55dd217f907f85e82a667cbc797c8d8b85b9674d
+  metadata.gz: 4ee226683fb316a82a18457b50bbe783712fe4ad9115befa4ab7c9421b31747acf2b3f609360c126a599ac8dded16a9cc4dda670e115f3bed3711147776956a3
+  data.tar.gz: 6ab849bcae377ccc919d6daf13b0556d9d6ed07b1cd16d645fe66fb026df4dd941d4f9a06b396ea044bb4ad0db886457c25ce1ca4a4c4d4de6306dc31edf47f2

data/CHANGELOG.md

@@ -4,25 +4,27 @@ All notable changes to this project will be documented in this file.
 
 ## [Unreleased]
 
-## [0.1.0] - 2026-06-09
+## [0.1.5] - 2026-06-18
 
 ### Added
-- Initial release of OmniAgent as a Rails engine for building application-native AI agents.
-- Agent runtime with provider abstraction, tool-calling loop, callbacks, and prompt rendering.
-- OpenAI provider support via the `openai` gem.
-- Tool DSL with JSON schema generation.
-- Rails generators and tasks for agent scaffolding.
-- RSpec coverage for agent runtime, providers, tools, and integration flows.
+- Added max tool iterations limit on the agent loop, with corresponding error raised when exceeded.
+- Added retry logic to the OpenAI provider, configurable via `Configuration`.
 
-## [0.1.1] - 2026-06-09
-
-### Changed
-- Lowered Rails version needed.
+### Fixed
+- Improved provider resolution error handling for unknown providers.
+- Forced eager load of error aliases to avoid Zeitwerk autoloading issues.
 
-## [0.1.2] - 2026-06-11
+# [0.1.4] - 2026-06-16
 
 ### Fixed
-- Fixed Zeitwerk issue with `OmniAgent::Errors`
+- Change invoke method to an instance method instead of a class method, to actually catch invocations of the `stop_generation!` method.
+
+### Changed
+- Remove agent file from directory to prevent needing to change Zeitwerk initializer as Railtie wasn't allowing to change behaviour without it. 
+
+### Added
+- Add Railtie to project to handle concerns.
+- Added docusaurus, the actual docs will be done later. 
 
 ## [0.1.3] - 2026-06-12
 
@@ -36,14 +38,22 @@ All notable changes to this project will be documented in this file.
 - Added `history` as a context variable which are the messages to be sent to the provider.
 - Added support for nested datatypes in the `array` schema for tools
 
-# [0.1.4] - 2026-06-16
+## [0.1.2] - 2026-06-11
 
 ### Fixed
-- Change invoke method to an instance method instead of a class method, to actually catch invocations of the `stop_generation!` method.
+- Fixed Zeitwerk issue with `OmniAgent::Errors`
+
+## [0.1.1] - 2026-06-09
 
 ### Changed
-- Remove agent file from directory to prevent needing to change Zeitwerk initializer as Railtie wasn't allowing to change behaviour without it. 
+- Lowered Rails version needed.
+
+## [0.1.0] - 2026-06-09
 
 ### Added
-- Add Railtie to project to handle concerns.
-- Added docusaurus, the actual docs will be done later. 
+- Initial release of OmniAgent as a Rails engine for building application-native AI agents.
+- Agent runtime with provider abstraction, tool-calling loop, callbacks, and prompt rendering.
+- OpenAI provider support via the `openai` gem.
+- Tool DSL with JSON schema generation.
+- Rails generators and tasks for agent scaffolding.
+- RSpec coverage for agent runtime, providers, tools, and integration flows.

data/lib/generators/omni_agent/agent/agent_generator.rb

@@ -46,7 +46,7 @@ module OmniAgent
       end
 
       def agent_file_path
-        File.join(destination_root, "#{file_name}.rb")
+        File.join(destination_root, "app", "agents", "#{file_name}.rb")
       end
 
       def prompt_file_path

data/lib/omni_agent/agent.rb

@@ -159,8 +159,17 @@ module OmniAgent
       initial_messages_count = messages.length - 1
 
       filtered_tools = tool_filter(tools: available_tools, agent_tags: self.class.tags)
+      max_iterations = OmniAgent.configuration.max_tool_iterations
+      iterations = 0
 
       loop do
+        iterations += 1
+        if iterations > max_iterations
+          raise OmniAgent::MaxToolIterationsError,
+                "Exceeded max_tool_iterations (#{max_iterations}) without a final response. " \
+                "Increase OmniAgent.configuration.max_tool_iterations if more tool calls are expected."
+        end
+
         response = provider.chat(messages: messages, tools: filtered_tools, **@chat_options)
 
         if response.content && !response.tool_calls?
@@ -309,7 +318,15 @@ module OmniAgent
     end
 
     def resolve_provider(name, model)
-      OmniAgent::Providers.registry[name.to_sym].new(model: model)
+      provider_class = OmniAgent::Providers.registry[name.to_sym]
+
+      unless provider_class
+        known = OmniAgent::Providers.registry.keys.join(", ")
+        raise OmniAgent::UnknownProviderError,
+              "Unknown provider #{name.inspect}. Known providers: #{known}"
+      end
+
+      provider_class.new(model: model)
     end
 
     def run_before_generation_callbacks(input:, context:, messages:)

data/lib/omni_agent/configuration.rb

@@ -1,10 +1,13 @@
 module OmniAgent
   class Configuration
-    attr_accessor :default_provider, :default_model
+    attr_accessor :default_provider, :default_model, :max_retries, :retry_base_delay, :max_tool_iterations
 
     def initialize
       @default_provider = :openai
       @default_model = "gpt-4o-mini"
+      @max_retries = 3
+      @retry_base_delay = 0.5
+      @max_tool_iterations = 10
     end
   end
 end

data/lib/omni_agent/errors.rb

@@ -2,8 +2,12 @@ module OmniAgent
   module Errors
     class Error < StandardError; end
     class MissingDependencyError < Error; end
+    class UnknownProviderError < Error; end
+    class MaxToolIterationsError < Error; end
   end
 
   Error = Errors::Error
   MissingDependencyError = Errors::MissingDependencyError
+  UnknownProviderError = Errors::UnknownProviderError
+  MaxToolIterationsError = Errors::MaxToolIterationsError
 end

data/lib/omni_agent/providers/base.rb

@@ -49,6 +49,29 @@ module OmniAgent
         raise NotImplementedError, "Providers must define a default model"
       end
 
+      # Override in subclasses to flag exceptions worth retrying (rate limits, timeouts, 5xx).
+      def retryable_error?(_error)
+        false
+      end
+
+      def with_retries
+        max_retries = OmniAgent.configuration.max_retries
+        base_delay = OmniAgent.configuration.retry_base_delay
+        attempt = 0
+
+        begin
+          yield
+        rescue => e
+          raise unless retryable_error?(e)
+
+          attempt += 1
+          raise if attempt > max_retries
+
+          sleep(base_delay * (2**(attempt - 1)))
+          retry
+        end
+      end
+
       def validate_message_payload!(message, index, role_name)
         has_content = message.key?(:content) || message.key?("content")
         content_value = message[:content] || message["content"]

data/lib/omni_agent/providers/openai.rb

@@ -44,7 +44,7 @@ module OmniAgent
         payload[:tools] = openai_tools if openai_tools.any?
         payload.merge!(options) if options.any?
 
-        response = client.chat.completions.create(**payload)
+        response = with_retries { client.chat.completions.create(**payload) }
 
         parse_response(response, payload)
       rescue => e
@@ -53,6 +53,13 @@ module OmniAgent
 
       protected
 
+      def retryable_error?(error)
+        defined?(::OpenAI::Errors) &&
+          (error.is_a?(::OpenAI::Errors::RateLimitError) ||
+           error.is_a?(::OpenAI::Errors::InternalServerError) ||
+           error.is_a?(::OpenAI::Errors::APIConnectionError))
+      end
+
       def client
         @client ||= ::OpenAI::Client.new(api_key: @api_key)
       end

data/lib/omni_agent/version.rb

@@ -1,3 +1,3 @@
 module OmniAgent
-  VERSION = "0.1.4"
+  VERSION = "0.1.5"
 end

data/lib/omni_agent.rb

@@ -7,6 +7,12 @@ loader.inflector.inflect("openai" => "OpenAI")
 loader.ignore(File.expand_path("generators", __dir__))
 loader.setup
 
+# errors.rb defines top-level aliases (OmniAgent::MissingDependencyError, etc.)
+# that Zeitwerk cannot autoload, since they don't map to a file path of their
+# own. Force it to load eagerly so the aliases exist regardless of which
+# constant is referenced first.
+require "omni_agent/errors"
+
 
 module OmniAgent
   class << self

metadata

@@ -1,14 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: omni_agent
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.5
 platform: ruby
 authors:
 - ACR1209
-autorequire: 
 bindir: bin
 cert_chain: []
-date: 2026-06-17 00:00:00.000000000 Z
+date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -82,7 +81,6 @@ metadata:
   homepage_uri: https://github.com/ACR1209/omni_agent
   source_code_uri: https://github.com/ACR1209/omni_agent
   changelog_uri: https://github.com/ACR1209/omni_agent/blob/main/CHANGELOG.md
-post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
@@ -97,8 +95,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.11
-signing_key: 
+rubygems_version: 3.6.9
 specification_version: 4
 summary: Rails engine for building AI agents with tools.
 test_files: []