omni_agent 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '0094c9758c242364af55eeffd4e87143f78b1db9ffabb54c06aaa23234030382'
-  data.tar.gz: 866bc34809e4c1ecbf21679a2d4dab00a0e1383ae3ac3bb8f6f94348d7ae7b6b
+  metadata.gz: 68e52e8d8eb8a4f59c708d32e7464b7526c92f8e5342d6a85fe3c1244bf643e7
+  data.tar.gz: 36d892f5d7593582f31cc15d957deab720c80c8416a48ba32df4253f24d10ac6
 SHA512:
-  metadata.gz: bf56557c378dce28538199dc6cf9795923acd7db6ea291dae542845f9f66d923a40ef78e0c1dfdff443e675425c431a58d0cb0d825edc69a171d6b473a0201a1
-  data.tar.gz: a4ee13697bbff34e8a3e0e1f196b282bc05d35ba680b8f19c2860c44527d3b61ed7210ed5a6d756dfcf7acea6bcb4219696d04528b7c3d57a479963f390fc1e4
+  metadata.gz: afaf69e6f85c7db759d6796477601945577550a80f86e53a694f52d07910ffe081e79aa18db0688a8aaf3b4cb5dae5e141beaefc43e4083c59e46af2f6496df3
+  data.tar.gz: efc7e291b645b15c0fe2d528737597948aaa28b9cc5a2f1bffd153313f071c0bd659d3d5b885932a607cf21f55dd217f907f85e82a667cbc797c8d8b85b9674d

data/CHANGELOG.md

@@ -35,3 +35,15 @@ 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.4] - 2026-06-16
+
+### Fixed
+- 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. 

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, "#{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

@@ -189,7 +189,7 @@ module OmniAgent
             tool_instance = tool_class.new
 
             begin
-              result = tool_class.invoke(tool_args)
+              result = tool_instance.invoke(tool_args)
 
               messages << {
                 role: "tool",

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.4"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: omni_agent
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - ACR1209
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2026-06-13 00:00:00.000000000 Z
+date: 2026-06-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -70,6 +70,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