deepagents_rails 0.1.0

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

@@ -0,0 +1,178 @@
+<%# DeepAgents <%= class_name %> Agent Interface %>
+
+<div class="deepagents-container">
+  <div class="deepagents-header">
+    <h1><%= class_name %> Agent</h1>
+    <p>Ask a question or provide instructions to interact with the agent.</p>
+  </div>
+  
+  <div class="deepagents-conversation" id="conversation">
+    <% if @messages.present? %>
+      <% @messages.each do |message| %>
+        <div class="message <%= message[:role] %>">
+          <div class="message-content">
+            <%= simple_format(message[:content]) %>
+          </div>
+          <div class="message-timestamp">
+            <%= message[:timestamp].strftime("%H:%M") %>
+          </div>
+        </div>
+      <% end %>
+    <% else %>
+      <div class="empty-state">
+        <p>No messages yet. Start a conversation with the agent.</p>
+      </div>
+    <% end %>
+  </div>
+  
+  <%= form_with url: deepagents_<%= file_name %>_index_path, method: :post, class: "deepagents-input-form", id: "agent-form", data: { remote: true } do |f| %>
+    <div class="input-container">
+      <%= f.text_area :input, placeholder: "Type your message here...", rows: 3, class: "deepagents-input", id: "agent-input" %>
+      
+      <div class="file-upload-container">
+        <%= f.file_field :files, multiple: true, class: "file-input", id: "file-upload" %>
+        <label for="file-upload" class="file-upload-button">
+          <i class="fa fa-paperclip"></i> Attach Files
+        </label>
+      </div>
+      
+      <%= f.submit "Send", class: "send-button", id: "send-button" %>
+    </div>
+  <% end %>
+</div>
+
+<style>
+  .deepagents-container {
+    max-width: 800px;
+    margin: 0 auto;
+    padding: 20px;
+    font-family: system-ui, -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, "Helvetica Neue", Arial, sans-serif;
+  }
+  
+  .deepagents-header {
+    margin-bottom: 20px;
+    text-align: center;
+  }
+  
+  .deepagents-conversation {
+    border: 1px solid #e0e0e0;
+    border-radius: 8px;
+    padding: 15px;
+    height: 400px;
+    overflow-y: auto;
+    margin-bottom: 20px;
+    background-color: #f9f9f9;
+  }
+  
+  .message {
+    margin-bottom: 15px;
+    padding: 10px 15px;
+    border-radius: 8px;
+    max-width: 80%;
+  }
+  
+  .message.user {
+    background-color: #e1f5fe;
+    margin-left: auto;
+  }
+  
+  .message.assistant {
+    background-color: #f0f0f0;
+    margin-right: auto;
+  }
+  
+  .message-timestamp {
+    font-size: 0.8em;
+    color: #888;
+    text-align: right;
+    margin-top: 5px;
+  }
+  
+  .input-container {
+    display: flex;
+    flex-direction: column;
+    gap: 10px;
+  }
+  
+  .deepagents-input {
+    width: 100%;
+    padding: 10px;
+    border: 1px solid #ddd;
+    border-radius: 8px;
+    resize: none;
+  }
+  
+  .file-upload-container {
+    display: flex;
+    align-items: center;
+  }
+  
+  .file-input {
+    display: none;
+  }
+  
+  .file-upload-button {
+    display: inline-block;
+    padding: 8px 12px;
+    background-color: #f0f0f0;
+    border-radius: 4px;
+    cursor: pointer;
+    font-size: 0.9em;
+  }
+  
+  .send-button {
+    padding: 10px 20px;
+    background-color: #2196f3;
+    color: white;
+    border: none;
+    border-radius: 8px;
+    cursor: pointer;
+    font-weight: bold;
+  }
+  
+  .send-button:hover {
+    background-color: #1976d2;
+  }
+  
+  .empty-state {
+    text-align: center;
+    color: #888;
+    padding: 40px 0;
+  }
+</style>
+
+<script>
+  document.addEventListener('DOMContentLoaded', function() {
+    const form = document.getElementById('agent-form');
+    const input = document.getElementById('agent-input');
+    const conversation = document.getElementById('conversation');
+    
+    form.addEventListener('ajax:success', function(event) {
+      const [data, status, xhr] = event.detail;
+      
+      // Add user message
+      const userMessage = document.createElement('div');
+      userMessage.className = 'message user';
+      userMessage.innerHTML = `
+        <div class="message-content">${input.value.replace(/\n/g, '<br>')}</div>
+        <div class="message-timestamp">${new Date().toLocaleTimeString([], {hour: '2-digit', minute:'2-digit'})}</div>
+      `;
+      conversation.appendChild(userMessage);
+      
+      // Add assistant message
+      const assistantMessage = document.createElement('div');
+      assistantMessage.className = 'message assistant';
+      assistantMessage.innerHTML = `
+        <div class="message-content">${data.response.replace(/\n/g, '<br>')}</div>
+        <div class="message-timestamp">${new Date().toLocaleTimeString([], {hour: '2-digit', minute:'2-digit'})}</div>
+      `;
+      conversation.appendChild(assistantMessage);
+      
+      // Clear input
+      input.value = '';
+      
+      // Scroll to bottom
+      conversation.scrollTop = conversation.scrollHeight;
+    });
+  });
+</script>

data/lib/generators/deepagents/agent/templates/show.html.erb

@@ -0,0 +1,131 @@
+<%# DeepAgents <%= class_name %> Agent Conversation View %>
+
+<div class="deepagents-container">
+  <div class="deepagents-header">
+    <h1>Conversation #<%= @conversation.id %></h1>
+    <p><%= @conversation.created_at.strftime("%B %d, %Y at %H:%M") %></p>
+    <%= link_to "Back to All Conversations", deepagents_<%= file_name %>_index_path, class: "back-link" %>
+  </div>
+  
+  <div class="deepagents-conversation" id="conversation">
+    <% @conversation.messages.each do |message| %>
+      <div class="message <%= message.role %>">
+        <div class="message-content">
+          <%= simple_format(message.content) %>
+        </div>
+        <div class="message-timestamp">
+          <%= message.created_at.strftime("%H:%M") %>
+        </div>
+      </div>
+    <% end %>
+  </div>
+  
+  <% if @conversation.files.any? %>
+    <div class="files-section">
+      <h3>Files</h3>
+      <div class="files-container">
+        <% @conversation.files.each do |file| %>
+          <div class="file-item">
+            <div class="file-name"><%= file.filename %></div>
+            <div class="file-actions">
+              <%= link_to "Download", file.url, class: "file-download" %>
+            </div>
+          </div>
+        <% end %>
+      </div>
+    </div>
+  <% end %>
+</div>
+
+<style>
+  .deepagents-container {
+    max-width: 800px;
+    margin: 0 auto;
+    padding: 20px;
+    font-family: system-ui, -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, "Helvetica Neue", Arial, sans-serif;
+  }
+  
+  .deepagents-header {
+    margin-bottom: 20px;
+  }
+  
+  .back-link {
+    display: inline-block;
+    margin-top: 10px;
+    color: #2196f3;
+    text-decoration: none;
+  }
+  
+  .back-link:hover {
+    text-decoration: underline;
+  }
+  
+  .deepagents-conversation {
+    border: 1px solid #e0e0e0;
+    border-radius: 8px;
+    padding: 15px;
+    max-height: 600px;
+    overflow-y: auto;
+    margin-bottom: 20px;
+    background-color: #f9f9f9;
+  }
+  
+  .message {
+    margin-bottom: 15px;
+    padding: 10px 15px;
+    border-radius: 8px;
+    max-width: 80%;
+  }
+  
+  .message.user {
+    background-color: #e1f5fe;
+    margin-left: auto;
+  }
+  
+  .message.assistant {
+    background-color: #f0f0f0;
+    margin-right: auto;
+  }
+  
+  .message-timestamp {
+    font-size: 0.8em;
+    color: #888;
+    text-align: right;
+    margin-top: 5px;
+  }
+  
+  .files-section {
+    margin-top: 30px;
+  }
+  
+  .files-container {
+    display: flex;
+    flex-wrap: wrap;
+    gap: 10px;
+  }
+  
+  .file-item {
+    border: 1px solid #e0e0e0;
+    border-radius: 4px;
+    padding: 10px;
+    width: calc(50% - 5px);
+    display: flex;
+    justify-content: space-between;
+    align-items: center;
+  }
+  
+  .file-name {
+    font-size: 0.9em;
+    word-break: break-all;
+  }
+  
+  .file-download {
+    color: #2196f3;
+    text-decoration: none;
+    font-size: 0.9em;
+  }
+  
+  .file-download:hover {
+    text-decoration: underline;
+  }
+</style>

data/lib/generators/deepagents/controller/controller_generator.rb

@@ -0,0 +1,51 @@
+module Deepagents
+  module Generators
+    class ControllerGenerator < Rails::Generators::NamedBase
+      source_root File.expand_path('templates', __dir__)
+      
+      desc "Creates DeepAgents API controllers for your Rails application"
+      
+      class_option :skip_routes, type: :boolean, default: false, desc: "Skip routes generation"
+      class_option :api_version, type: :string, default: "v1", desc: "API version"
+      
+      def create_controller_file
+        template "api_controller.rb", "app/controllers/api/#{options[:api_version]}/#{file_name}_controller.rb"
+      end
+      
+      def create_serializer_file
+        template "serializer.rb", "app/serializers/#{file_name}_serializer.rb"
+      end
+      
+      def add_routes
+        return if options[:skip_routes]
+        
+        route_file = "config/routes.rb"
+        route_content = <<~ROUTE
+          namespace :api do
+            namespace :#{options[:api_version]} do
+              resources :#{plural_name}, only: [:index, :show, :create] do
+                member do
+                  post :run
+                  post :upload
+                end
+              end
+            end
+          end
+        ROUTE
+        
+        inject_into_file route_file, route_content, after: "Rails.application.routes.draw do\n"
+      end
+      
+      def display_next_steps
+        say "\n"
+        say "DeepAgents API controller for #{file_name} has been created! 🎮", :green
+        say "\n"
+        say "Next steps:", :yellow
+        say "  1. Ensure you have the required models (use the model generator if needed)"
+        say "  2. Add any custom API endpoints to your controller"
+        say "  3. Test your API endpoints with curl or Postman"
+        say "\n"
+      end
+    end
+  end
+end

data/lib/generators/deepagents/controller/templates/api_controller.rb

@@ -0,0 +1,87 @@
+# frozen_string_literal: true
+
+module Api
+  module <%= options[:api_version].camelize %>
+    class <%= class_name %>Controller < ApplicationController
+      before_action :set_conversation, only: [:show, :run, :upload]
+      
+      # GET /api/<%= options[:api_version] %>/<%= plural_name %>
+      def index
+        @conversations = Deepagents::<%= class_name %>Conversation.recent.limit(20)
+        
+        render json: @conversations, each_serializer: <%= class_name %>Serializer
+      end
+      
+      # GET /api/<%= options[:api_version] %>/<%= plural_name %>/:id
+      def show
+        render json: @conversation, serializer: <%= class_name %>Serializer, include: ['messages', 'files']
+      end
+      
+      # POST /api/<%= options[:api_version] %>/<%= plural_name %>
+      def create
+        @conversation = Deepagents::<%= class_name %>Conversation.new(conversation_params)
+        
+        if @conversation.save
+          # Add a system message if provided
+          if params[:system_message].present?
+            @conversation.<%= file_name %>_messages.create!(
+              role: 'system',
+              content: params[:system_message]
+            )
+          end
+          
+          render json: @conversation, serializer: <%= class_name %>Serializer, status: :created
+        else
+          render json: { errors: @conversation.errors }, status: :unprocessable_entity
+        end
+      end
+      
+      # POST /api/<%= options[:api_version] %>/<%= plural_name %>/:id/run
+      def run
+        input = params[:input]
+        agent_name = params[:agent_name]
+        
+        if input.blank?
+          render json: { error: 'Input is required' }, status: :unprocessable_entity
+          return
+        end
+        
+        result = @conversation.run_agent(input, agent_name)
+        
+        render json: {
+          message: result[:message],
+          files: result[:files]
+        }
+      end
+      
+      # POST /api/<%= options[:api_version] %>/<%= plural_name %>/:id/upload
+      def upload
+        if params[:file].blank?
+          render json: { error: 'File is required' }, status: :unprocessable_entity
+          return
+        end
+        
+        file = params[:file]
+        
+        @file = @conversation.<%= file_name %>_files.create!(
+          filename: file.original_filename,
+          content: file.read
+        )
+        
+        render json: @file
+      end
+      
+      private
+      
+      def set_conversation
+        @conversation = Deepagents::<%= class_name %>Conversation.find(params[:id])
+      rescue ActiveRecord::RecordNotFound
+        render json: { error: 'Conversation not found' }, status: :not_found
+      end
+      
+      def conversation_params
+        params.require(:<%= file_name %>).permit(:title, :agent_name, metadata: {})
+      end
+    end
+  end
+end

data/lib/generators/deepagents/controller/templates/serializer.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+class <%= class_name %>Serializer < ActiveModel::Serializer
+  attributes :id, :title, :agent_name, :metadata, :created_at, :updated_at
+  
+  has_many :<%= file_name %>_messages, serializer: <%= class_name %>MessageSerializer
+  has_many :<%= file_name %>_files, serializer: <%= class_name %>FileSerializer
+  
+  # Add any custom serialization logic here
+end
+
+class <%= class_name %>MessageSerializer < ActiveModel::Serializer
+  attributes :id, :role, :content, :metadata, :created_at
+end
+
+class <%= class_name %>FileSerializer < ActiveModel::Serializer
+  attributes :id, :filename, :content, :metadata, :created_at, :url
+  
+  def url
+    object.url if object.respond_to?(:url)
+  end
+end

data/lib/generators/deepagents/install/install_generator.rb

@@ -0,0 +1,41 @@
+module Deepagents
+  module Generators
+    class InstallGenerator < Rails::Generators::Base
+      source_root File.expand_path('templates', __dir__)
+      
+      desc "Creates a DeepAgents initializer and necessary files for your Rails application"
+      
+      def create_initializer_file
+        template "initializer.rb", "config/initializers/deepagents.rb"
+      end
+      
+      def create_configuration_file
+        template "deepagents.yml", "config/deepagents.yml"
+      end
+      
+      def mount_engine
+        route "mount DeepagentsRails::Engine => '/deepagents'"
+      end
+      
+      def add_environment_variables
+        append_to_file '.env.example', <<~ENV
+          
+          # DeepAgents configuration
+          # DEEPAGENTS_API_KEY=your_api_key_here
+          # DEEPAGENTS_PROVIDER=claude # or openai
+        ENV
+      end
+      
+      def display_post_install_message
+        say "\n"
+        say "DeepAgents Rails has been installed! 🎉", :green
+        say "\n"
+        say "Next steps:", :yellow
+        say "  1. Configure your API keys in config/deepagents.yml or through environment variables"
+        say "  2. Generate your first agent with: rails g deepagents:agent NAME"
+        say "  3. Check out the documentation at https://github.com/cdaviis/deepagents_rails"
+        say "\n"
+      end
+    end
+  end
+end

data/lib/generators/deepagents/install/templates/deepagents.yml

@@ -0,0 +1,38 @@
+# DeepAgents Rails Configuration
+
+default: &default
+  # API key for your LLM provider (Claude or OpenAI)
+  # You can also set this via environment variable DEEPAGENTS_API_KEY
+  api_key: <%= ENV.fetch('DEEPAGENTS_API_KEY', nil) %>
+  
+  # LLM provider to use (claude or openai)
+  provider: <%= ENV.fetch('DEEPAGENTS_PROVIDER', 'claude') %>
+  
+  # Default model to use
+  model: <%= ENV.fetch('DEEPAGENTS_MODEL', 'claude-3-sonnet-20240229') %>
+  
+  # Default temperature for LLM requests (0.0 to 1.0)
+  temperature: <%= ENV.fetch('DEEPAGENTS_TEMPERATURE', '0.7') %>
+  
+  # Maximum tokens to generate in responses
+  max_tokens: <%= ENV.fetch('DEEPAGENTS_MAX_TOKENS', '4096') %>
+  
+  # Default tools available to all agents
+  default_tools:
+    - calculator
+    - file_system
+    - planning
+
+development:
+  <<: *default
+  log_requests: true
+
+test:
+  <<: *default
+  # Use mock model for testing
+  provider: mock
+  model: mock-model
+
+production:
+  <<: *default
+  log_requests: false

data/lib/generators/deepagents/install/templates/initializer.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+# DeepAgents Rails configuration
+DeepagentsRails::Engine.config.deepagents.tap do |config|
+  # API key for your LLM provider (Claude or OpenAI)
+  # You can set this via environment variable DEEPAGENTS_API_KEY
+  config.api_key = ENV.fetch('DEEPAGENTS_API_KEY', nil)
+  
+  # LLM provider to use (:claude or :openai)
+  # You can set this via environment variable DEEPAGENTS_PROVIDER
+  config.provider = ENV.fetch('DEEPAGENTS_PROVIDER', 'claude').to_sym
+  
+  # Default model to use
+  # For Claude: "claude-3-sonnet-20240229", "claude-3-opus-20240229", etc.
+  # For OpenAI: "gpt-4o", "gpt-4-turbo", etc.
+  config.model = ENV.fetch('DEEPAGENTS_MODEL', 'claude-3-sonnet-20240229')
+  
+  # Default temperature for LLM requests (0.0 to 1.0)
+  config.temperature = ENV.fetch('DEEPAGENTS_TEMPERATURE', '0.7').to_f
+  
+  # Maximum tokens to generate in responses
+  config.max_tokens = ENV.fetch('DEEPAGENTS_MAX_TOKENS', '4096').to_i
+  
+  # Configure default tools available to all agents
+  # config.default_tools = [:calculator, :web_search]
+  
+  # Configure logging
+  config.log_requests = Rails.env.development?
+end
+
+# Load custom tools
+# Dir[Rails.root.join('app/deepagents/tools/**/*.rb')].each { |file| require file }
+
+# Load custom agents
+# Dir[Rails.root.join('app/deepagents/agents/**/*.rb')].each { |file| require file }

data/lib/generators/deepagents/model/model_generator.rb

@@ -0,0 +1,43 @@
+module Deepagents
+  module Generators
+    class ModelGenerator < Rails::Generators::NamedBase
+      source_root File.expand_path('templates', __dir__)
+      
+      desc "Creates DeepAgents models for your Rails application"
+      
+      def create_conversation_model
+        template "conversation.rb", "app/models/deepagents/#{file_name}_conversation.rb"
+      end
+      
+      def create_message_model
+        template "message.rb", "app/models/deepagents/#{file_name}_message.rb"
+      end
+      
+      def create_file_model
+        template "file.rb", "app/models/deepagents/#{file_name}_file.rb"
+      end
+      
+      def create_migrations
+        template "create_conversations_migration.rb", "db/migrate/#{timestamp}_create_deepagents_#{file_name}_conversations.rb"
+        template "create_messages_migration.rb", "db/migrate/#{timestamp(1)}_create_deepagents_#{file_name}_messages.rb"
+        template "create_files_migration.rb", "db/migrate/#{timestamp(2)}_create_deepagents_#{file_name}_files.rb"
+      end
+      
+      def display_next_steps
+        say "\n"
+        say "DeepAgents models for #{file_name} have been created! 📚", :green
+        say "\n"
+        say "Next steps:", :yellow
+        say "  1. Run migrations with: rails db:migrate"
+        say "  2. Use the models in your agents and controllers"
+        say "\n"
+      end
+      
+      private
+      
+      def timestamp(offset = 0)
+        Time.now.utc.strftime("%Y%m%d%H%M%S").to_i + offset
+      end
+    end
+  end
+end