omni_agent 0.1.3 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '0094c9758c242364af55eeffd4e87143f78b1db9ffabb54c06aaa23234030382'
-  data.tar.gz: 866bc34809e4c1ecbf21679a2d4dab00a0e1383ae3ac3bb8f6f94348d7ae7b6b
+  metadata.gz: 1513c46e356438c23dfd193f25c390df240f2d1d812813114e2e582222de2965
+  data.tar.gz: a9b05b4445128aea795b3f7f5ea80cb6bd668bf2b674c15651011ce6780e0c2e
 SHA512:
-  metadata.gz: bf56557c378dce28538199dc6cf9795923acd7db6ea291dae542845f9f66d923a40ef78e0c1dfdff443e675425c431a58d0cb0d825edc69a171d6b473a0201a1
-  data.tar.gz: a4ee13697bbff34e8a3e0e1f196b282bc05d35ba680b8f19c2860c44527d3b61ed7210ed5a6d756dfcf7acea6bcb4219696d04528b7c3d57a479963f390fc1e4
+  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
 
@@ -35,3 +37,23 @@ All notable changes to this project will be documented in this file.
 - Added `generated_messages` to the agent response, it contains all the messages generated by the agent and the user request.
 - 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.2] - 2026-06-11
+
+### Fixed
+- Fixed Zeitwerk issue with `OmniAgent::Errors`
+
+## [0.1.1] - 2026-06-09
+
+### Changed
+- Lowered Rails version needed.
+
+## [0.1.0] - 2026-06-09
+
+### 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.

data/README.md

@@ -56,10 +56,11 @@ OPENAI_ACCESS_TOKEN=your_api_key_here
 4. Implement your agent prompt and optional tools under:
 
 ```text
-app/agents/research_agent/
+app/agents/
 	research_agent.rb
-	prompt.md.erb
-	tools/
+	research_agent/
+		prompt.md.erb
+		tools/
 ```
 
 ## Agent Example

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

@@ -46,7 +46,7 @@ module OmniAgent
       end
 
       def agent_file_path
-        File.join(agent_directory, "#{file_name}.rb")
+        File.join(destination_root, "app", "agents", "#{file_name}.rb")
       end
 
       def prompt_file_path
@@ -89,7 +89,7 @@ module OmniAgent
 
       def requested_tool_names
         names = Array(options[:with_tools]).map(&:to_s).map(&:strip).reject(&:empty?)
-        names.empty? ? ["ExampleTool"] : names
+        names.empty? ? [ "ExampleTool" ] : names
       end
 
       def generated_model_expression

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?
@@ -189,7 +198,7 @@ module OmniAgent
             tool_instance = tool_class.new
 
             begin
-              result = tool_class.invoke(tool_args)
+              result = tool_instance.invoke(tool_args)
 
               messages << {
                 role: "tool",
@@ -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/railtie.rb

@@ -0,0 +1,31 @@
+# lib/omni_agent/railtie.rb
+module OmniAgent
+  class Railtie < ::Rails::Railtie
+    initializer "omni_agent.remove_agents_from_zeitwerk", before: :setup_main_autoloader do
+      agents_path = Rails.root.join("app", "agents")
+      if agents_path.exist?
+        Rails.autoloaders.main.ignore(agents_path)
+      end
+    end
+
+    config.to_prepare do
+      agents_dir = Rails.root.join("app", "agents")
+      next unless agents_dir.exist?
+
+      Dir.glob(agents_dir.join("concerns/**/*.rb")).each { |f| require_dependency f }
+
+      Dir.glob(agents_dir.join("*")).each do |file|
+        next unless File.directory?(file)
+        next if File.basename(file) == "concerns"
+
+        main_agent_file = File.join(file, "#{File.basename(file)}.rb")
+        require_dependency main_agent_file if File.file?(main_agent_file)
+
+        Dir.glob(File.join(file, "**/*.rb")).each do |sub_file|
+          next if sub_file == main_agent_file
+          require_dependency sub_file
+        end
+      end
+    end
+  end
+end

data/lib/omni_agent/tool.rb

@@ -40,14 +40,10 @@ module OmniAgent
         }
       end
 
-      def invoke(arguments_hash)
+      def parse_arguments(arguments_hash)
         kwargs = arguments_hash.transform_keys(&:to_sym)
-
         valid_keys = (@properties || {}).keys.map(&:to_sym)
-
-        filtered_kwargs = kwargs.slice(*valid_keys)
-
-        new.execute(**filtered_kwargs)
+        kwargs.slice(*valid_keys)
       end
 
       def stops_generation(value = true)
@@ -73,6 +69,11 @@ module OmniAgent
       end
     end
 
+    def invoke(arguments_hash)
+      filtered_kwargs = self.class.parse_arguments(arguments_hash)
+      execute(**filtered_kwargs)
+    end
+
     def stop_generation!
       @stops_generation_instance = true
     end

data/lib/omni_agent/version.rb

@@ -1,3 +1,3 @@
 module OmniAgent
-  VERSION = "0.1.3"
+  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.3
+  version: 0.1.5
 platform: ruby
 authors:
 - ACR1209
-autorequire: 
 bindir: bin
 cert_chain: []
-date: 2026-06-13 00:00:00.000000000 Z
+date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -70,6 +69,7 @@ files:
 - lib/omni_agent/providers/mock.rb
 - lib/omni_agent/providers/openai.rb
 - lib/omni_agent/providers/response.rb
+- lib/omni_agent/railtie.rb
 - lib/omni_agent/tasks/omni_agent_tasks.rake
 - lib/omni_agent/tool.rb
 - lib/omni_agent/tool/schema_builder.rb
@@ -81,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
@@ -96,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: []