deepagents_rails 0.1.0

data/lib/generators/deepagents/model/templates/conversation.rb

@@ -0,0 +1,73 @@
+# frozen_string_literal: true
+
+module Deepagents
+  class <%= class_name %>Conversation < ApplicationRecord
+    self.table_name = 'deepagents_<%= file_name %>_conversations'
+    
+    has_many :<%= file_name %>_messages, class_name: 'Deepagents::<%= class_name %>Message', foreign_key: 'conversation_id', dependent: :destroy
+    has_many :<%= file_name %>_files, class_name: 'Deepagents::<%= class_name %>File', foreign_key: 'conversation_id', dependent: :destroy
+    
+    # Alias for easier access
+    alias_method :messages, :<%= file_name %>_messages
+    alias_method :files, :<%= file_name %>_files
+    
+    # Scopes
+    scope :recent, -> { order(created_at: :desc) }
+    
+    # Convert to format expected by DeepAgents
+    def to_agent_format
+      {
+        messages: messages.order(:created_at).map(&:to_agent_format),
+        files: files_hash
+      }
+    end
+    
+    # Run the agent with the current conversation context
+    def run_agent(input, agent_name = nil)
+      # Create a new message for the user input
+      user_message = messages.create!(
+        role: 'user',
+        content: input
+      )
+      
+      # Get the service
+      service = DeepagentsRails.service(agent_name)
+      
+      # Run the agent with the conversation context
+      result = service.run(input, to_agent_format)
+      
+      # Create a new message for the agent response
+      assistant_message = messages.create!(
+        role: 'assistant',
+        content: result[:response]
+      )
+      
+      # Save any files generated by the agent
+      if result[:files].present?
+        result[:files].each do |filename, content|
+          files.create!(
+            filename: filename,
+            content: content
+          )
+        end
+      end
+      
+      # Return the result
+      {
+        message: assistant_message,
+        files: result[:files]
+      }
+    end
+    
+    private
+    
+    # Convert files to hash format expected by DeepAgents
+    def files_hash
+      hash = {}
+      files.each do |file|
+        hash[file.filename] = file.content
+      end
+      hash
+    end
+  end
+end

data/lib/generators/deepagents/model/templates/create_conversations_migration.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+class CreateDeepagents<%= class_name.pluralize %>Conversations < ActiveRecord::Migration[6.0]
+  def change
+    create_table :deepagents_<%= file_name %>_conversations do |t|
+      t.string :title
+      t.references :user, polymorphic: true, index: { name: 'index_<%= file_name %>_conversations_on_user' }
+      t.string :agent_name
+      t.jsonb :metadata, default: {}
+      
+      t.timestamps
+    end
+    
+    add_index :deepagents_<%= file_name %>_conversations, :created_at
+  end
+end

data/lib/generators/deepagents/model/templates/create_files_migration.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+class CreateDeepagents<%= class_name.pluralize %>Files < ActiveRecord::Migration[6.0]
+  def change
+    create_table :deepagents_<%= file_name %>_files do |t|
+      t.references :conversation, null: false, foreign_key: { to_table: :deepagents_<%= file_name %>_conversations }, index: { name: 'index_<%= file_name %>_files_on_conversation_id' }
+      t.string :filename, null: false
+      t.text :content, null: false
+      t.jsonb :metadata, default: {}
+      
+      t.timestamps
+    end
+    
+    add_index :deepagents_<%= file_name %>_files, :created_at
+    add_index :deepagents_<%= file_name %>_files, :filename
+  end
+end

data/lib/generators/deepagents/model/templates/create_messages_migration.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+class CreateDeepagents<%= class_name.pluralize %>Messages < ActiveRecord::Migration[6.0]
+  def change
+    create_table :deepagents_<%= file_name %>_messages do |t|
+      t.references :conversation, null: false, foreign_key: { to_table: :deepagents_<%= file_name %>_conversations }, index: { name: 'index_<%= file_name %>_messages_on_conversation_id' }
+      t.string :role, null: false
+      t.text :content, null: false
+      t.jsonb :metadata, default: {}
+      
+      t.timestamps
+    end
+    
+    add_index :deepagents_<%= file_name %>_messages, :created_at
+    add_index :deepagents_<%= file_name %>_messages, :role
+  end
+end

data/lib/generators/deepagents/model/templates/file.rb

@@ -0,0 +1,96 @@
+# frozen_string_literal: true
+
+module Deepagents
+  class <%= class_name %>File < ApplicationRecord
+    self.table_name = 'deepagents_<%= file_name %>_files'
+    
+    belongs_to :<%= file_name %>_conversation, class_name: 'Deepagents::<%= class_name %>Conversation', foreign_key: 'conversation_id'
+    
+    # Alias for easier access
+    alias_method :conversation, :<%= file_name %>_conversation
+    
+    # Validations
+    validates :filename, presence: true
+    validates :content, presence: true
+    
+    # Get the file extension
+    def extension
+      File.extname(filename).delete('.')
+    end
+    
+    # Check if the file is an image
+    def image?
+      %w[jpg jpeg png gif svg webp].include?(extension.downcase)
+    end
+    
+    # Check if the file is a text file
+    def text?
+      %w[txt md markdown rb py js html css json yaml yml xml].include?(extension.downcase)
+    end
+    
+    # Generate a URL for the file (if using Active Storage)
+    def url
+      if defined?(ActiveStorage) && attachment.present?
+        Rails.application.routes.url_helpers.rails_blob_path(attachment, only_path: true)
+      else
+        "#"
+      end
+    end
+    
+    # Attach the file content to Active Storage (if available)
+    def attach_content
+      return unless defined?(ActiveStorage)
+      
+      # Create a tempfile with the content
+      tempfile = Tempfile.new([filename, ".#{extension}"])
+      tempfile.binmode
+      tempfile.write(content)
+      tempfile.rewind
+      
+      # Attach the tempfile
+      attachment.attach(
+        io: tempfile,
+        filename: filename,
+        content_type: content_type
+      )
+      
+      # Close and delete the tempfile
+      tempfile.close
+      tempfile.unlink
+    end
+    
+    private
+    
+    # Determine the content type based on the extension
+    def content_type
+      case extension.downcase
+      when 'jpg', 'jpeg'
+        'image/jpeg'
+      when 'png'
+        'image/png'
+      when 'gif'
+        'image/gif'
+      when 'svg'
+        'image/svg+xml'
+      when 'txt'
+        'text/plain'
+      when 'md', 'markdown'
+        'text/markdown'
+      when 'html'
+        'text/html'
+      when 'css'
+        'text/css'
+      when 'js'
+        'application/javascript'
+      when 'json'
+        'application/json'
+      when 'yaml', 'yml'
+        'application/yaml'
+      when 'xml'
+        'application/xml'
+      else
+        'application/octet-stream'
+      end
+    end
+  end
+end

data/lib/generators/deepagents/model/templates/message.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+module Deepagents
+  class <%= class_name %>Message < ApplicationRecord
+    self.table_name = 'deepagents_<%= file_name %>_messages'
+    
+    belongs_to :<%= file_name %>_conversation, class_name: 'Deepagents::<%= class_name %>Conversation', foreign_key: 'conversation_id'
+    
+    # Alias for easier access
+    alias_method :conversation, :<%= file_name %>_conversation
+    
+    # Validations
+    validates :role, presence: true, inclusion: { in: %w[user assistant system] }
+    validates :content, presence: true
+    
+    # Scopes
+    scope :user, -> { where(role: 'user') }
+    scope :assistant, -> { where(role: 'assistant') }
+    scope :system, -> { where(role: 'system') }
+    scope :chronological, -> { order(created_at: :asc) }
+    
+    # Convert to format expected by DeepAgents
+    def to_agent_format
+      {
+        role: role,
+        content: content
+      }
+    end
+  end
+end

data/lib/generators/deepagents/tool/templates/tool.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+module Deepagents
+  module Tools
+    class <%= class_name %>Tool
+      # Returns a DeepAgents::Tool instance
+      def self.build
+        DeepAgents::Tool.new(
+          "<%= file_name %>",
+          "<%= tool_description %>"
+        ) do |<%= parameters_with_defaults %>|
+          # Implement your tool's functionality here
+          # This block will be called when the agent uses this tool
+          
+          begin
+            # Example implementation - replace with your actual logic
+            "Result of <%= file_name %> operation with parameters: #{[<%= parameters_as_args %>].join(', ')}"
+          rescue => e
+            "Error in <%= file_name %> tool: #{e.message}"
+          end
+        end
+      end
+      
+      # Optional: Add helper methods for your tool below
+    end
+  end
+end

data/lib/generators/deepagents/tool/templates/tool_spec.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+require 'rails_helper'
+
+RSpec.describe Deepagents::Tools::<%= class_name %>Tool do
+  describe '.build' do
+    it 'returns a DeepAgents::Tool instance' do
+      tool = described_class.build
+      expect(tool).to be_a(DeepAgents::Tool)
+      expect(tool.name).to eq('<%= file_name %>')
+    end
+    
+    it 'has the correct description' do
+      tool = described_class.build
+      expect(tool.description).to eq('<%= tool_description %>')
+    end
+    
+    it 'executes the tool functionality' do
+      tool = described_class.build
+      # Test with sample parameters - adjust based on your tool's parameters
+      result = tool.execute(<% tool_parameters.each_with_index do |param, i| %><%= i > 0 ? ', ' : '' %>'test_<%= param %>'<% end %>)
+      expect(result).to be_a(String)
+      expect(result).to include('Result of <%= file_name %> operation')
+    end
+    
+    it 'handles errors gracefully' do
+      tool = described_class.build
+      
+      # Mock an error in the tool execution
+      allow_any_instance_of(DeepAgents::Tool).to receive(:execute).and_raise(StandardError.new('Test error'))
+      
+      # The tool should catch the error and return an error message
+      expect { tool.execute(<% tool_parameters.each_with_index do |param, i| %><%= i > 0 ? ', ' : '' %>'test_<%= param %>'<% end %>) }.not_to raise_error
+    end
+  end
+end

data/lib/generators/deepagents/tool/tool_generator.rb

@@ -0,0 +1,58 @@
+module Deepagents
+  module Generators
+    class ToolGenerator < Rails::Generators::NamedBase
+      source_root File.expand_path('templates', __dir__)
+      
+      desc "Creates a new DeepAgents tool for your Rails application"
+      
+      class_option :description, type: :string, default: "", desc: "Description of what the tool does"
+      class_option :parameters, type: :array, default: [], desc: "Parameters for the tool"
+      
+      def create_tool_file
+        template "tool.rb", "app/deepagents/tools/#{file_name}_tool.rb"
+      end
+      
+      def create_tool_spec_file
+        template "tool_spec.rb", "spec/deepagents/tools/#{file_name}_tool_spec.rb"
+      end
+      
+      def display_next_steps
+        say "\n"
+        say "Tool #{file_name} has been created! 🔧", :green
+        say "\n"
+        say "Next steps:", :yellow
+        say "  1. Edit app/deepagents/tools/#{file_name}_tool.rb to implement your tool's functionality"
+        say "  2. Add this tool to your agents by including it in the tools array"
+        say "\n"
+      end
+      
+      private
+      
+      def tool_description
+        options[:description].present? ? options[:description] : "Performs #{file_name} operations"
+      end
+      
+      def tool_parameters
+        options[:parameters].empty? ? ["query"] : options[:parameters]
+      end
+      
+      def parameters_as_args
+        params = tool_parameters
+        if params.size == 1
+          params.first
+        else
+          params.join(", ")
+        end
+      end
+      
+      def parameters_with_defaults
+        params = tool_parameters
+        if params.size == 1
+          "#{params.first}"
+        else
+          params.map { |p| "#{p}" }.join(", ")
+        end
+      end
+    end
+  end
+end

data/lib/generators/deepagents/view/templates/_conversation.html.erb

@@ -0,0 +1,10 @@
+<%# DeepAgents %> <%= class_name %> <%# Conversation Partial %>
+<% if conversation.messages.any? %>
+  <% conversation.messages.order(:created_at).each do |message| %>
+    <%= render partial: "message", locals: { message: message } %>
+  <% end %>
+<% else %>
+  <div class="deepagents-empty-state">
+    <p>No messages yet. Start the conversation by typing a message below.</p>
+  </div>
+<% end %>

data/lib/generators/deepagents/view/templates/_form.html.erb

@@ -0,0 +1,33 @@
+<%# DeepAgents %> <%= class_name %> <%# Form Partial %>
+<%= form_with(model: #{file_name}, local: true, class: "deepagents-form") do |form| %>
+  <% if #{file_name}.errors.any? %>
+    <div class="deepagents-error-messages">
+      <h2><%= pluralize(#{file_name}.errors.count, "error") %> prohibited this conversation from being saved:</h2>
+      <ul>
+        <% #{file_name}.errors.full_messages.each do |message| %>
+          <li><%= message %></li>
+        <% end %>
+      </ul>
+    </div>
+  <% end %>
+
+  <div class="deepagents-form-group">
+    <%= form.label :title %>
+    <%= form.text_field :title, class: "deepagents-form-control" %>
+  </div>
+
+  <div class="deepagents-form-group">
+    <%= form.label :agent_name %>
+    <%= form.select :agent_name, DeepagentsRails.service.available_agents.map { |name| [name.titleize, name] }, { include_blank: "Default Agent" }, class: "deepagents-form-control" %>
+  </div>
+
+  <div class="deepagents-form-group">
+    <%= form.label :system_message, "System Instructions" %>
+    <%= form.text_area :system_message, class: "deepagents-form-control", rows: 3 %>
+    <small class="deepagents-form-text">Optional instructions to guide the agent's behavior</small>
+  </div>
+
+  <div class="deepagents-form-actions">
+    <%= form.submit "Create Conversation", class: "deepagents-button" %>
+  </div>
+<% end %>

data/lib/generators/deepagents/view/templates/_message.html.erb

@@ -0,0 +1,31 @@
+<%# DeepAgents %> <%= class_name %> <%# Message Partial %>
+<div class="deepagents-message <%= message.role %>">
+  <div class="deepagents-message-avatar">
+    <% if message.role == 'user' %>
+      <svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" fill="currentColor" viewBox="0 0 16 16">
+        <path d="M8 8a3 3 0 1 0 0-6 3 3 0 0 0 0 6zm2-3a2 2 0 1 1-4 0 2 2 0 0 1 4 0zm4 8c0 1-1 1-1 1H3s-1 0-1-1 1-4 6-4 6 3 6 4zm-1-.004c-.001-.246-.154-.986-.832-1.664C11.516 10.68 10.289 10 8 10c-2.29 0-3.516.68-4.168 1.332-.678.678-.83 1.418-.832 1.664h10z"/>
+      </svg>
+    <% elsif message.role == 'assistant' %>
+      <svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" fill="currentColor" viewBox="0 0 16 16">
+        <path d="M6 12.5a.5.5 0 0 1 .5-.5h3a.5.5 0 0 1 0 1h-3a.5.5 0 0 1-.5-.5ZM3 8.062C3 6.76 4.235 5.765 5.53 5.886a26.58 26.58 0 0 0 4.94 0C11.765 5.765 13 6.76 13 8.062v1.157a.933.933 0 0 1-.765.935c-.845.147-2.34.346-4.235.346-1.895 0-3.39-.2-4.235-.346A.933.933 0 0 1 3 9.219V8.062Zm4.542-.827a.25.25 0 0 0-.217.068l-.92.9a24.767 24.767 0 0 1-1.871-.183.25.25 0 0 0-.068.495c.55.076 1.232.149 2.02.193a.25.25 0 0 0 .189-.071l.754-.736.847 1.71a.25.25 0 0 0 .404.062l.932-.97a25.286 25.286 0 0 0 1.922-.188.25.25 0 0 0-.068-.495c-.538.074-1.207.145-1.98.189a.25.25 0 0 0-.166.076l-.754.785-.842-1.7a.25.25 0 0 0-.182-.135Z"/>
+        <path d="M8.5 1.866a1 1 0 1 0-1 0V3h-2A4.5 4.5 0 0 0 1 7.5V8a1 1 0 0 0-1 1v2a1 1 0 0 0 1 1v1a2 2 0 0 0 2 2h10a2 2 0 0 0 2-2v-1a1 1 0 0 0 1-1V9a1 1 0 0 0-1-1v-.5A4.5 4.5 0 0 0 10.5 3h-2V1.866ZM14 7.5V13a1 1 0 0 1-1 1H3a1 1 0 0 1-1-1V7.5A3.5 3.5 0 0 1 5.5 4h5A3.5 3.5 0 0 1 14 7.5Z"/>
+      </svg>
+    <% else %>
+      <svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" fill="currentColor" viewBox="0 0 16 16">
+        <path d="M8 15A7 7 0 1 1 8 1a7 7 0 0 1 0 14zm0 1A8 8 0 1 0 8 0a8 8 0 0 0 0 16z"/>
+        <path d="M7.002 11a1 1 0 1 1 2 0 1 1 0 0 1-2 0zM7.1 4.995a.905.905 0 1 1 1.8 0l-.35 3.507a.552.552 0 0 1-1.1 0L7.1 4.995z"/>
+      </svg>
+    <% end %>
+  </div>
+  
+  <div class="deepagents-message-content">
+    <div class="deepagents-message-header">
+      <span class="deepagents-message-role"><%= message.role.capitalize %></span>
+      <span class="deepagents-message-timestamp"><%= time_ago_in_words(message.created_at) %> ago</span>
+    </div>
+    
+    <div class="deepagents-message-body">
+      <%= simple_format(message.content) %>
+    </div>
+  </div>
+</div>

data/lib/generators/deepagents/view/templates/index.html.erb

@@ -0,0 +1,35 @@
+<%# DeepAgents %> <%= class_name %> <%# Index View %>
+<div class="deepagents-container">
+  <h1>Conversations</h1>
+  
+  <div class="deepagents-actions">
+    <%= link_to "New Conversation", new_#{file_name}_path, class: "deepagents-button" %>
+  </div>
+  
+  <div class="deepagents-conversations-list">
+    <% if @conversations.any? %>
+      <% @conversations.each do |conversation| %>
+        <div class="deepagents-conversation-card">
+          <h3><%= link_to conversation.title.presence || "Conversation ##{conversation.id}", #{file_name}_path(conversation) %></h3>
+          <div class="deepagents-conversation-meta">
+            <span class="deepagents-timestamp"><%= time_ago_in_words(conversation.created_at) %> ago</span>
+            <% if conversation.agent_name.present? %>
+              <span class="deepagents-agent-name"><%= conversation.agent_name %></span>
+            <% end %>
+          </div>
+          <div class="deepagents-conversation-preview">
+            <% if conversation.messages.any? %>
+              <%= truncate(conversation.messages.last.content, length: 100) %>
+            <% else %>
+              <em>No messages yet</em>
+            <% end %>
+          </div>
+        </div>
+      <% end %>
+    <% else %>
+      <div class="deepagents-empty-state">
+        <p>No conversations yet. <%= link_to "Start a new conversation", new_#{file_name}_path %>.</p>
+      </div>
+    <% end %>
+  </div>
+</div>