activeagent 0.2.6.8 → 0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 44f424a63296642766c739571e7833995ef70b05f1d16ed91fab3b4f1657136a
-  data.tar.gz: cf242b606153176e0e7a11efb77306df9bf0f526b601301dc6f8213f1a86ff26
+  metadata.gz: bf6889f20afa4e61e46ab6edbe801186aee0d3fe85284f3ce6379509a6a76bc4
+  data.tar.gz: 3340dcc2d65740cea5a98f4c0640ac806e1686bb48bee2a3260cf3e65b5478b2
 SHA512:
-  metadata.gz: 75265e0675be1e59bfa5721ddf3c87cb9487ce45cbefdcd6457d43b064d70f0ab2c5256a913dc8630cbd05efada6e41af17d05f85347414e2f0568659c9cbca6
-  data.tar.gz: 6c0a82f30aed52a0bb12f8f1d89d1b6310c1400f7456e0351b3d5765d27613546d1fdd87dd9add64944a054cdecdf797ef09a4767eee239a7f405c9b74c439ca
+  metadata.gz: 8c63eaaad2c5049fac65902d5d5676f976580fffd40477287b149d29695c15b75143e1bbf8e60128c04adfc23531770c9ab1c7893a6cf73bd4f258315a93e251
+  data.tar.gz: 0be4d79a4755d23d1c7e2c659871d8e0480c05583e6fe1d6eae48571fd1d2540cc88ea53a827192109bf76476e237e8c09f1bbc64f97bca429416b3c7503154c

data/lib/active_agent/action_prompt/message.rb

@@ -3,14 +3,13 @@ module ActiveAgent
     class Message
       VALID_ROLES = %w[system assistant user tool function].freeze
 
-      attr_accessor :action_id, :content, :role, :name, :action_requested, :requested_actions, :content_type, :charset
+      attr_accessor :action_id, :content, :role, :action_requested, :requested_actions, :content_type, :charset
 
       def initialize(attributes = {})
         @action_id = attributes[:action_id]
         @charset = attributes[:charset] || "UTF-8"
         @content = attributes[:content] || ""
         @content_type = attributes[:content_type] || "text/plain"
-        @name = attributes[:name]
         @role = attributes[:role] || :user
         @requested_actions = attributes[:requested_actions] || []
         @action_requested = @requested_actions.any?
@@ -26,7 +25,6 @@ module ActiveAgent
           charset: charset
         }
 
-        hash[:name] = name if name
         hash[:action_requested] = requested_actions.any?
         hash[:requested_actions] = requested_actions if requested_actions.any?
         hash

data/lib/active_agent/action_prompt/prompt.rb

@@ -3,10 +3,11 @@ require_relative "message"
 module ActiveAgent
   module ActionPrompt
     class Prompt
-      attr_accessor :actions, :body, :content_type, :instructions, :message, :messages, :options, :mime_version, :charset, :context, :parts
+      attr_accessor :actions, :body, :content_type, :context_id, :instructions, :message, :messages, :options, :mime_version, :charset, :context, :parts
 
       def initialize(attributes = {})
         @options = attributes.fetch(:options, {})
+        @agent_class = attributes.fetch(:agent_class, ApplicationAgent)
         @actions = attributes.fetch(:actions, [])
         @action_choice = attributes.fetch(:action_choice, "")
         @instructions = attributes.fetch(:instructions, "")
@@ -18,6 +19,7 @@ module ActiveAgent
         @mime_version = attributes.fetch(:mime_version, "1.0")
         @charset = attributes.fetch(:charset, "UTF-8")
         @context = attributes.fetch(:context, [])
+        @context_id = attributes.fetch(:context_id, nil)
         @headers = attributes.fetch(:headers, {})
         @parts = attributes.fetch(:parts, [])
 

data/lib/active_agent/base.rb

@@ -129,6 +129,7 @@ module ActiveAgent
       def generate_with(provider, **options)
         self.generation_provider = provider
         self.options = (options || {}).merge(options)
+        self.options[:stream] = new.agent_stream if self.options[:stream]
         generation_provider.config.merge!(self.options)
       end
 
@@ -176,13 +177,8 @@ module ActiveAgent
 
       def set_payload_for_prompt(payload, prompt)
         payload[:prompt] = prompt.encoded
-        payload[:agent] = name
+        payload[:agent] = agent_name
         payload[:message_id] = prompt.message_id
-        payload[:subject] = prompt.subject
-        payload[:to] = prompt.to
-        payload[:from] = prompt.from
-        payload[:bcc] = prompt.bcc if prompt.bcc.present?
-        payload[:cc] = prompt.cc if prompt.cc.present?
         payload[:date] = prompt.date
         payload[:perform_generations] = prompt.perform_generations
       end
@@ -200,47 +196,56 @@ module ActiveAgent
       end
     end
 
-    attr_internal :context
+    attr_internal :prompt_context
+
+    def agent_stream
+      proc do |message, delta, stop|
+        run_stream_callbacks(message, delta, stop) do |message, delta, stop|
+          yield message, delta, stop if block_given?
+        end
+      end
+    end
 
     def embed
-      context.options.merge(options)
-      generation_provider.embed(context) if context && generation_provider
+      prompt_context.options.merge(options)
+      generation_provider.embed(prompt_context) if prompt_context && generation_provider
       handle_response(generation_provider.response)
     end
 
     def perform_generation
-      context.options.merge(options)
-      generation_provider.generate(context) if context && generation_provider
+      prompt_context.options.merge(options)
+      generation_provider.generate(prompt_context) if prompt_context && generation_provider
       handle_response(generation_provider.response)
     end
 
     def handle_response(response)
       perform_actions(requested_actions: response.message.requested_actions) if response.message.requested_actions.present?
 
-      update_context(response)
+      update_prompt_context(response)
     end
 
-    def update_context(response)
-      response.prompt
+    def update_prompt_context(response)
+      # response.prompt = prompt_context
+      # response.message = response.messages.last
       response
     end
 
     def perform_actions(requested_actions:)
       requested_actions.each do |action|
         perform_action(action)
-        prompt.messages.last.role = :tool
-        prompt.messages.last.action_id = action.id
       end
     end
 
     def perform_action(action)
       process(action.name, *action.params)
+      prompt_context.messages.last.role = :tool
+      prompt_context.messages.last.action_id = action.id
     end
 
     def initialize
       super
       @_prompt_was_called = false
-      @_context = ActiveAgent::ActionPrompt::Prompt.new(instructions: options[:instructions], options: options)
+      @_prompt_context = ActiveAgent::ActionPrompt::Prompt.new(instructions: options[:instructions], options: options)
     end
 
     def process(method_name, *args) # :nodoc:
@@ -252,7 +257,7 @@ module ActiveAgent
 
       ActiveSupport::Notifications.instrument("process.active_agent", payload) do
         super
-        @_context = ActiveAgent::ActionPrompt::Prompt.new unless @_prompt_was_called
+        @_prompt_context = ActiveAgent::ActionPrompt::Prompt.new unless @_prompt_was_called
       end
     end
     ruby2_keywords(:process)
@@ -282,60 +287,39 @@ module ActiveAgent
 
     def headers(args = nil)
       if args
-        @_context.headers(args)
+        @_prompt_context.headers(args)
       else
-        @_context
-      end
-    end
-
-    # def attachments
-    #   if @_prompt_was_called
-    #     LateAttachmentsProxy.new(@_context.attachments)
-    #   else
-    #     @_context.attachments
-    #   end
-    # end
-
-    class LateAttachmentsProxy < SimpleDelegator
-      def inline
-        self
-      end
-
-      def []=(_name, _content)
-        _raise_error
-      end
-
-      private
-
-      def _raise_error
-        raise "Can't add attachments after `prompt` was called.\n" \
-                              "Make sure to use `attachments[]=` before calling `prompt`."
+        @_prompt_context
       end
     end
 
     def prompt_with(*)
-      context.update_context(*)
+      prompt_context.update_prompt_context(*)
     end
 
     def prompt(headers = {}, &block)
-      return context if @_prompt_was_called && headers.blank? && !block
+      return prompt_context if @_prompt_was_called && headers.blank? && !block
 
       content_type = headers[:content_type]
 
       headers = apply_defaults(headers)
 
-      context.charset = charset = headers[:charset]
+      prompt_context.context_id = headers[:context_id]
+
+      prompt_context.options = options.merge(headers[:options] || {})
+
+      prompt_context.charset = charset = headers[:charset]
 
       responses = collect_responses(headers, &block)
 
       @_prompt_was_called = true
 
-      create_parts_from_responses(context, responses)
+      create_parts_from_responses(prompt_context, responses)
 
-      context.content_type = set_content_type(context, content_type, headers[:content_type])
-      context.charset = charset
-      context.actions = headers[:actions] || action_schemas
-      context
+      prompt_context.content_type = set_content_type(prompt_context, content_type, headers[:content_type])
+      prompt_context.charset = charset
+      prompt_context.actions = headers[:actions] || action_schemas
+      prompt_context
     end
 
     def action_schemas
@@ -352,7 +336,7 @@ module ActiveAgent
       if user_content_type.present?
         user_content_type
       else
-        context.content_type || class_default
+        prompt_context.content_type || class_default
       end
     end
 
@@ -383,10 +367,10 @@ module ActiveAgent
       end
     end
 
-    def assign_headers_to_context(context, headers)
+    def assign_headers_to_prompt_context(prompt_context, headers)
       assignable = headers.except(:parts_order, :content_type, :body, :template_name,
         :template_path, :delivery_method, :delivery_method_options)
-      assignable.each { |k, v| context[k] = v }
+      assignable.each { |k, v| prompt_context[k] = v }
     end
 
     def collect_responses(headers, &)
@@ -437,24 +421,24 @@ module ActiveAgent
       end
     end
 
-    def create_parts_from_responses(context, responses)
+    def create_parts_from_responses(prompt_context, responses)
       if responses.size > 1
         # prompt_container = ActiveAgent::ActionPrompt::Prompt.new
         # prompt_container.content_type = "multipart/alternative"
-        responses.each { |r| insert_part(context, r, context.charset) }
-        # context.add_part(prompt_container)
+        responses.each { |r| insert_part(prompt_context, r, prompt_context.charset) }
+        # prompt_context.add_part(prompt_container)
       else
-        responses.each { |r| insert_part(context, r, context.charset) }
+        responses.each { |r| insert_part(prompt_context, r, prompt_context.charset) }
       end
     end
 
-    def insert_part(context, response, charset)
+    def insert_part(prompt_context, response, charset)
       message = ActiveAgent::ActionPrompt::Message.new(
         content: response[:body],
         content_type: response[:content_type],
         charset: charset
       )
-      context.add_part(message)
+      prompt_context.add_part(message)
     end
 
     # This and #instrument_name is for caching instrument

data/lib/active_agent/generation.rb

@@ -11,14 +11,14 @@ module ActiveAgent
     ruby2_keywords(:initialize)
 
     def __getobj__
-      @prompt_context ||= processed_agent.context
+      @prompt_context ||= processed_agent.prompt_context
     end
 
     def __setobj__(prompt_context)
       @prompt_context = prompt_context
     end
 
-    def context
+    def prompt_context
       __getobj__
     end
 

data/lib/active_agent/generation_provider/anthropic_provider.rb

@@ -0,0 +1,142 @@
+# lib/active_agent/generation_provider/anthropic_provider.rb
+
+require "anthropic"
+require "active_agent/action_prompt/action"
+require_relative "base"
+require_relative "response"
+
+module ActiveAgent
+  module GenerationProvider
+    class AnthropicProvider < Base
+      def initialize(config)
+        super
+        @api_key = config["api_key"]
+        @model_name = config["model"] || "claude-3-5-sonnet-20240620"
+        @client = Anthropic::Client.new(access_token: @api_key)
+      end
+
+      def generate(prompt)
+        @prompt = prompt
+
+        chat_prompt(parameters: prompt_parameters)
+      rescue => e
+        raise GenerationProviderError, e.message
+      end
+
+      def chat_prompt(parameters: prompt_parameters)
+        parameters[:stream] = provider_stream if prompt.options[:stream] || config["stream"]
+
+        chat_response(@client.messages(parameters))
+      end
+
+      private
+
+      def provider_stream
+        agent_stream = prompt.options[:stream]
+        message = ActiveAgent::ActionPrompt::Message.new(content: "", role: :assistant)
+        @response = ActiveAgent::GenerationProvider::Response.new(prompt: prompt, message:)
+
+        proc do |chunk|
+          if new_content = chunk.dig(:delta, :text)
+            message.content += new_content
+            agent_stream.call(message) if agent_stream.respond_to?(:call)
+          end
+        end
+      end
+
+      def prompt_parameters(model: @prompt.options[:model] || @model_name, messages: @prompt.messages, temperature: @config["temperature"] || 0.7, tools: @prompt.actions)
+        params = {
+          model: model,
+          messages: provider_messages(messages),
+          temperature: temperature,
+          max_tokens: 4096
+        }
+
+        if tools&.present?
+          params[:tools] = format_tools(tools)
+        end
+
+        params
+      end
+
+      def format_tools(tools)
+        tools.map do |tool|
+          {
+            name: tool[:name] || tool[:function][:name],
+            description: tool[:description],
+            input_schema: tool[:parameters]
+          }
+        end
+      end
+
+      def provider_messages(messages)
+        messages.map do |message|
+          provider_message = {
+            role: convert_role(message.role),
+            content: []
+          }
+
+          provider_message[:content] << if message.content_type == "image_url"
+            {
+              type: "image",
+              source: {
+                type: "url",
+                url: message.content
+              }
+            }
+          else
+            {
+              type: "text",
+              text: message.content
+            }
+          end
+
+          provider_message
+        end
+      end
+
+      def convert_role(role)
+        case role.to_s
+        when "system" then "system"
+        when "user" then "user"
+        when "assistant" then "assistant"
+        when "tool", "function" then "assistant"
+        else "user"
+        end
+      end
+
+      def chat_response(response)
+        return @response if prompt.options[:stream]
+
+        content = response.content.first[:text]
+
+        message = ActiveAgent::ActionPrompt::Message.new(
+          content: content,
+          role: "assistant",
+          action_requested: response.stop_reason == "tool_use",
+          requested_actions: handle_actions(response.tool_use)
+        )
+
+        update_context(prompt: prompt, message: message, response: response)
+
+        @response = ActiveAgent::GenerationProvider::Response.new(
+          prompt: prompt,
+          message: message,
+          raw_response: response
+        )
+      end
+
+      def handle_actions(tool_uses)
+        return unless tool_uses&.present?
+
+        tool_uses.map do |tool_use|
+          ActiveAgent::ActionPrompt::Action.new(
+            id: tool_use[:id],
+            name: tool_use[:name],
+            params: tool_use[:input]
+          )
+        end
+      end
+    end
+  end
+end

data/lib/active_agent/generation_provider/open_ai_provider.rb

@@ -64,9 +64,16 @@ module ActiveAgent
         @response = ActiveAgent::GenerationProvider::Response.new(prompt: prompt, message:)
 
         proc do |chunk, bytesize|
-          if new_content = chunk.dig("choices", 0, "delta", "content")
+          if (new_content = chunk.dig("choices", 0, "delta", "content"))
             message.content += new_content
-            agent_stream.call(message) if agent_stream.respond_to?(:call)
+
+            agent_stream.call(message, new_content, false) do |message, new_content|
+              yield message, new_content if block_given?
+            end
+          end
+
+          agent_stream.call(message, nil, true) do |message|
+            yield message, nil if block_given?
           end
         end
       end
@@ -98,6 +105,7 @@ module ActiveAgent
       end
 
       def chat_response(response)
+        binding.irb
         return @response if prompt.options[:stream]
 
         message_json = response.dig("choices", 0, "message")

data/lib/active_agent/version.rb

@@ -1,3 +1,3 @@
 module ActiveAgent
-  VERSION = "0.2.6.8"
+  VERSION = "0.3"
 end

data/lib/generators/active_agent/agent_generator.rb

@@ -30,13 +30,25 @@ module ActiveAgent
       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
+          
+          # Use configured template engines or fall back to defaults
+          json_template_engine = json_template_engine_for_views || "jbuilder"
+          html_template_engine = html_template_engine_for_views || "erb"
+          
+          @schema_path = File.join("app/views", class_path, file_name, "#{action}.json.#{json_template_engine}")
+          @view_path = File.join("app/views", class_path, file_name, "#{action}.html.#{html_template_engine}")
+          
+          template "action.json.#{json_template_engine}", @schema_path
+          template "action.html.#{html_template_engine}", @view_path
         end
       end
 
+      def create_layout_files
+        # Create the application agent layouts
+        template "layout.text.erb", "app/views/layouts/agent.text.erb"
+        template "layout.html.erb", "app/views/layouts/agent.html.erb"
+      end
+
       private
 
       def test_framework
@@ -47,6 +59,17 @@ module ActiveAgent
         ::Rails.application.config.generators.options[:rails][:template_engine]
       end
 
+      def json_template_engine_for_views
+        # Check if there's a specific JSON template engine configured
+        json_engine = ::Rails.application.config.generators.options[:rails][:json_template_engine]
+        json_engine || "jbuilder" # Default to jbuilder if not specified
+      end
+      
+      def html_template_engine_for_views
+        # Use the configured template engine or default to erb
+        template_engine || "erb"
+      end
+
       def file_name # :doc:
         @_file_name ||= super + "_agent"
       end

data/lib/generators/active_agent/install_generator.rb

@@ -12,6 +12,10 @@ module ActiveAgent
       def create_application_agent
         template "application_agent.rb", "app/agents/application_agent.rb"
       end
+
+      def create_agent_layout_template
+        template "agent.text.erb", "app/views/layouts/agent.text.erb"
+      end
     end
   end
 end

data/lib/generators/active_agent/templates/agent.text.erb

@@ -0,0 +1 @@
+<%= yield %>

data/lib/generators/active_agent/templates/layout.html.erb

@@ -0,0 +1 @@
+<%= yield %>

data/lib/generators/active_agent/templates/layout.text.erb

@@ -0,0 +1 @@
+<%= yield %>

data/lib/tasks/activeagent_tasks.rake

@@ -0,0 +1,4 @@
+# desc "Explaining what the task does"
+# task :activeagent do
+#   # Task goes here
+# end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: activeagent
 version: !ruby/object:Gem::Version
-  version: 0.2.6.8
+  version: '0.3'
 platform: ruby
 authors:
 - Justin Bowen
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-03-17 00:00:00.000000000 Z
+date: 2025-03-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: actionpack
@@ -156,6 +156,7 @@ files:
 - lib/active_agent/generation_methods.rb
 - lib/active_agent/generation_provider.rb
 - lib/active_agent/generation_provider/README.md
+- lib/active_agent/generation_provider/anthropic_provider.rb
 - lib/active_agent/generation_provider/base.rb
 - lib/active_agent/generation_provider/open_ai_provider.rb
 - lib/active_agent/generation_provider/response.rb
@@ -179,9 +180,13 @@ files:
 - lib/generators/active_agent/templates/action.json.jbuilder.tt
 - lib/generators/active_agent/templates/active_agent.yml
 - lib/generators/active_agent/templates/agent.rb.tt
+- lib/generators/active_agent/templates/agent.text.erb
 - 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
+- lib/generators/active_agent/templates/layout.html.erb
+- lib/generators/active_agent/templates/layout.text.erb
+- lib/tasks/activeagent_tasks.rake
 homepage: https://activeagents.ai
 licenses:
 - MIT