activeagent 0.0.1.alpha → 0.1.1

data/lib/active_agent/test_case.rb

@@ -0,0 +1,125 @@
+# # frozen_string_literal: true
+
+# require_relative "test_helper"
+# require "active_support/test_case"
+# require "rails-dom-testing"
+
+# module ActiveAgent
+#   class NonInferrableAgentError < ::StandardError
+#     def initialize(name)
+#       super("Unable to determine the agent to test from #{name}. " \
+#         "You'll need to specify it using tests YourAgent in your " \
+#         "test case definition")
+#     end
+#   end
+
+#   class TestCase < ActiveSupport::TestCase
+#     module ClearTestDeliveries
+#       extend ActiveSupport::Concern
+
+#       included do
+#         setup :clear_test_generations
+#         teardown :clear_test_generations
+#       end
+
+#       private
+
+#       def clear_test_generations
+#         if ActiveAgent::Base.generation_method == :test
+#           ActiveAgent::Base.generations.clear
+#         end
+#       end
+#     end
+
+#     module Behavior
+#       extend ActiveSupport::Concern
+
+#       include ActiveSupport::Testing::ConstantLookup
+#       include TestHelper
+#       include Rails::Dom::Testing::Assertions::SelectorAssertions
+#       include Rails::Dom::Testing::Assertions::DomAssertions
+
+#       included do
+#         class_attribute :_agent_class
+#         setup :initialize_test_generations
+#         setup :set_expected_prompt
+#         teardown :restore_test_generations
+#         ActiveSupport.run_load_hooks(:active_agent_test_case, self)
+#       end
+
+#       module ClassMethods
+#         def tests(agent)
+#           case agent
+#           when String, Symbol
+#             self._agent_class = agent.to_s.camelize.constantize
+#           when Module
+#             self._agent_class = agent
+#           else
+#             raise NonInferrableAgentError.new(agent)
+#           end
+#         end
+
+#         def agent_class
+#           if agent = _agent_class
+#             agent
+#           else
+#             tests determine_default_agent(name)
+#           end
+#         end
+
+#         def determine_default_agent(name)
+#           agent = determine_constant_from_test_name(name) do |constant|
+#             Class === constant && constant < ActiveAgent::Base
+#           end
+#           raise NonInferrableAgentError.new(name) if agent.nil?
+#           agent
+#         end
+#       end
+
+#       # Reads the fixture file for the given agent.
+#       #
+#       # This is useful when testing agents by being able to write the body of
+#       # an promt inside a fixture. See the testing guide for a concrete example:
+#       # https://guides.rubyonrails.org/testing.html#revenge-of-the-fixtures
+#       def read_fixture(action)
+#         IO.readlines(File.join(Rails.root, "test", "fixtures", self.class.agent_class.name.underscore, action))
+#       end
+
+#       private
+
+#       def initialize_test_generations
+#         set_generation_method :test
+#         @old_perform_generations = ActiveAgent::Base.perform_generations
+#         ActiveAgent::Base.perform_generations = true
+#         ActiveAgent::Base.generations.clear
+#       end
+
+#       def restore_test_generations
+#         restore_generation_method
+#         ActiveAgent::Base.perform_generations = @old_perform_generations
+#       end
+
+#       def set_generation_method(method)
+#         @old_generation_method = ActiveAgent::Base.generation_method
+#         ActiveAgent::Base.generation_method = method
+#       end
+
+#       def restore_generation_method
+#         ActiveAgent::Base.generations.clear
+#         ActiveAgent::Base.generation_method = @old_generation_method
+#       end
+
+#       def set_expected_prompt
+#         @expected = ActiveAgent::ActionPrompt::Prompt.new
+#         @expected.content_type ["text", "plain", {"charset" => charset}]
+#         @expected.mime_version = "1.0"
+#       end
+
+#       def charset
+#         "UTF-8"
+#       end
+#     end
+
+#     include Behavior
+#   end
+# end

data/lib/active_agent/version.rb

@@ -0,0 +1,3 @@
+module ActiveAgent
+  VERSION = "0.1.1"
+end

data/lib/active_agent.rb

@@ -1,25 +1,47 @@
-# lib/active_agent.rb
 require "yaml"
+require "abstract_controller"
 require "active_agent/version"
 require "active_agent/deprecator"
-require "active_agent/action_prompt/prompt_helper"
+require "active_agent/railtie" if defined?(Rails)
+
 require "active_support"
+require "active_support/rails"
+require "active_support/core_ext/class"
+require "active_support/core_ext/module/attr_internal"
+require "active_support/core_ext/string/inflections"
+require "active_support/lazy_load_hooks"
 
 module ActiveAgent
   extend ActiveSupport::Autoload
 
+  eager_autoload do
+    autoload :Collector
+  end
+
   autoload :Base
   autoload :Callbacks
-  autoload :ActionPrompt
-  autoload :Parameterized
+  autoload :InlinePreviewInterceptor
+  autoload :PromptHelper
   autoload :Generation
+  autoload :GenerationMethods
   autoload :GenerationProvider
-  autoload :GenerationJob
   autoload :QueuedGeneration
+  autoload :Parameterized
+  autoload :Preview
+  autoload :Previews, "active_agent/preview"
+  autoload :GenerationJob
 
   class << self
     attr_accessor :config
 
+    def eager_load!
+      super
+
+      Base.descendants.each do |agent|
+        agent.eager_load! unless agent.abstract?
+      end
+    end
+
     def configure
       yield self
     end

data/lib/generators/active_agent/USAGE

@@ -0,0 +1,18 @@
+Description:
+    Generates a new agent and its views. Passes the agent name, either
+    CamelCased or under_scored, and an optional list of prompts as arguments.
+
+    This generates a agent class in app/agents and invokes your template
+    engine and test framework generators.
+
+Examples:
+    `bin/rails generate agent inventory search`
+
+    creates a sign up mailer class, views, and test:
+        Agent:     app/agents/inventory_agent.rb
+        Views:      app/views/inventory_agent/search.text.erb [...]
+        Test:       test/agents/inventory_agent_test.rb
+
+    `bin/rails generate agent inventory search update report`
+
+    creates an inventory agent with search, update, and report actions.

data/lib/generators/active_agent/agent_generator.rb

@@ -0,0 +1,63 @@
+# frozen_string_literal: true
+
+module ActiveAgent
+  module Generators
+    class AgentGenerator < ::Rails::Generators::NamedBase
+      source_root File.expand_path("templates", __dir__)
+
+      argument :actions, type: :array, default: [], banner: "method method"
+
+      check_class_collision suffix: "Agent"
+
+      def create_agent_file
+        template "agent.rb", File.join("app/agents", class_path, "#{file_name}_agent.rb")
+
+        in_root do
+          if behavior == :invoke && !File.exist?(application_agent_file_name)
+            template "application_agent.rb", application_agent_file_name
+          end
+        end
+      end
+
+      def create_test_file
+        if test_framework == :rspec
+          template "agent_spec.rb", File.join("spec/agents", class_path, "#{file_name}_agent_spec.rb")
+        else
+          template "agent_test.rb", File.join("test/agents", class_path, "#{file_name}_agent_test.rb")
+        end
+      end
+
+      def create_view_files
+        actions.each do |action|
+          @action = action
+          @schema_path = File.join("app/views", class_path, file_name, "#{action}.json.jbuilder")
+          @view_path = File.join("app/views", class_path, file_name, "#{action}.html.#{template_engine}")
+          template "action.json.jbuilder", @schema_path
+          template "action.html.#{template_engine}", @view_path
+        end
+      end
+      
+
+      private
+        def test_framework
+          ::Rails.application.config.generators.options[:rails][:test_framework]
+        end
+        
+        def template_engine
+          ::Rails.application.config.generators.options[:rails][:template_engine]
+        end
+        
+        def file_name # :doc:
+          @_file_name ||= super + "_agent"
+        end
+
+        def application_agent_file_name
+          @_application_agent_file_name ||= if mountable_engine?
+            "app/agents/#{namespaced_path}/application_agent.rb"
+          else
+            "app/agents/application_agent.rb"
+          end
+        end
+    end
+  end
+end

data/lib/generators/active_agent/templates/action.json.jbuilder.tt

@@ -0,0 +1,14 @@
+json.type :function
+json.function do
+  json.name action_name
+  json.description "TODO: Write a description for this action"
+  json.parameters do
+    json.type :object
+    json.properties do
+      json.param_name do
+        json.type :string
+        json.description "The param_description"
+      end
+    end
+  end
+end

data/lib/generators/active_agent/templates/agent.rb.tt

@@ -0,0 +1,14 @@
+<% module_namespacing do -%>
+class <%= class_name %>Agent < ApplicationAgent
+<% actions.each_with_index do |action, index| -%>
+<% if index != 0 -%>
+
+<% end -%>
+  def <%= action %>
+    @message = "Cats go.."
+
+    prompt message: @message
+  end
+<% end -%>
+end
+<% end -%>

data/lib/generators/active_agent/templates/application_agent.rb.tt

@@ -0,0 +1,4 @@
+<% module_namespacing do -%>
+class ApplicationAgent < ActiveAgent::Base
+end
+<% end %>

metadata

@@ -1,15 +1,99 @@
 --- !ruby/object:Gem::Specification
 name: activeagent
 version: !ruby/object:Gem::Version
-  version: 0.0.1.alpha
+  version: 0.1.1
 platform: ruby
 authors:
 - Justin Bowen
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-10-17 00:00:00.000000000 Z
-dependencies: []
+date: 2024-11-14 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: actionpack
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '7.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '7.2'
+- !ruby/object:Gem::Dependency
+  name: actionview
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '7.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '7.2'
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '7.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '7.2'
+- !ruby/object:Gem::Dependency
+  name: activemodel
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '7.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '7.2'
+- !ruby/object:Gem::Dependency
+  name: activejob
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '7.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '7.2'
+- !ruby/object:Gem::Dependency
+  name: rails
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '7.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '7.2'
 description: A simple way to perform long running LLM background jobs and streaming
   responses
 email: jusbowen@gmail.com
@@ -17,7 +101,48 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- README.md
+- Rakefile
 - lib/active_agent.rb
+- lib/active_agent/README.md
+- lib/active_agent/action_prompt.rb
+- lib/active_agent/action_prompt/README.md
+- lib/active_agent/action_prompt/base.rb
+- lib/active_agent/action_prompt/collector.rb
+- lib/active_agent/action_prompt/message.rb
+- lib/active_agent/action_prompt/prompt.rb
+- lib/active_agent/base.rb
+- lib/active_agent/callbacks.rb
+- lib/active_agent/deprecator.rb
+- lib/active_agent/engine.rb
+- lib/active_agent/generation.rb
+- lib/active_agent/generation_job.rb
+- lib/active_agent/generation_methods.rb
+- lib/active_agent/generation_provider.rb
+- lib/active_agent/generation_provider/README.md
+- lib/active_agent/generation_provider/base.rb
+- lib/active_agent/generation_provider/open_ai_provider.rb
+- lib/active_agent/generation_provider/response.rb
+- lib/active_agent/inline_preview_interceptor.rb
+- lib/active_agent/log_subscriber.rb
+- lib/active_agent/operation.rb
+- lib/active_agent/parameterized.rb
+- lib/active_agent/preview.rb
+- lib/active_agent/prompt_helper.rb
+- lib/active_agent/queued_generation.rb
+- lib/active_agent/railtie.rb
+- lib/active_agent/rescuable.rb
+- lib/active_agent/service.rb
+- lib/active_agent/test_case.rb
+- lib/active_agent/version.rb
+- lib/generators/active_agent/USAGE
+- lib/generators/active_agent/agent_generator.rb
+- lib/generators/active_agent/templates/action.html.erb.tt
+- lib/generators/active_agent/templates/action.json.jbuilder.tt
+- lib/generators/active_agent/templates/agent.rb.tt
+- lib/generators/active_agent/templates/agent_spec.rb.tt
+- lib/generators/active_agent/templates/agent_test.rb.tt
+- lib/generators/active_agent/templates/application_agent.rb.tt
 homepage: https://rubygems.org/gems/activeagent
 licenses:
 - MIT