ruby_conversations 1.0.0

data/lib/generators/ruby_conversations/install/templates/initializer.rb.erb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+RubyConversations.configure do |config|
+  # Storage mode: :local or :remote
+  config.storage_mode = :local
+  
+  # Remote API settings (only needed for remote mode)
+  # config.api_url = ENV['RUBY_CONVERSATIONS_API_URL']
+  # config.jwt_secret = ENV['RUBY_CONVERSATIONS_JWT_SECRET']
+  
+  # Default LLM settings
+  config.default_llm_model = 'gpt-4'
+  
+  # Customize model behaviors
+  config.conversation_class = 'AiConversation'
+  config.message_class = 'AiMessage'
+  config.prompt_class = 'Prompt'
+end 

data/lib/generators/ruby_conversations/install/templates/migrations/create_ai_conversations.rb.erb

@@ -0,0 +1,13 @@
+class CreateAiConversations < ActiveRecord::Migration[7.0]
+  def change
+    create_table :ai_conversations do |t|
+      t.string :conversationable_type
+      t.bigint :conversationable_id
+
+      t.timestamps
+    end
+
+    add_index :ai_conversations, [:conversationable_type, :conversationable_id], 
+              name: 'index_ai_conversations_on_conversationable'
+  end
+end 

data/lib/generators/ruby_conversations/install/templates/migrations/create_ai_message_inputs.rb.erb

@@ -0,0 +1,13 @@
+class CreateAiMessageInputs < ActiveRecord::Migration[7.0]
+  def change
+    create_table :ai_message_inputs do |t|
+      t.string :placeholder_name
+      t.text :value
+      t.bigint :ai_message_prompt_id, null: false
+
+      t.timestamps
+    end
+
+    add_index :ai_message_inputs, :ai_message_prompt_id
+  end
+end 

data/lib/generators/ruby_conversations/install/templates/migrations/create_ai_message_prompts.rb.erb

@@ -0,0 +1,18 @@
+class CreateAiMessagePrompts < ActiveRecord::Migration[7.0]
+  def change
+    create_table :ai_message_prompts do |t|
+      t.bigint :ai_message_id, null: false
+      t.bigint :prompt_id
+      t.text :draft
+      t.bigint :prompt_version_id
+      t.string :name
+      t.string :role
+
+      t.timestamps
+    end
+
+    add_index :ai_message_prompts, :ai_message_id
+    add_index :ai_message_prompts, :prompt_id
+    add_index :ai_message_prompts, :prompt_version_id
+  end
+end 

data/lib/generators/ruby_conversations/install/templates/migrations/create_ai_messages.rb.erb

@@ -0,0 +1,18 @@
+class CreateAiMessages < ActiveRecord::Migration[7.0]
+  def change
+    create_table :ai_messages do |t|
+      t.bigint :ai_conversation_id, null: false
+      t.jsonb :request
+      t.text :response
+      t.string :model_identifier
+      t.string :tool
+      t.text :change_description
+      t.string :llm, default: 'us.anthropic.claude-3-7-sonnet-20250219-v1:0', null: false
+
+      t.timestamps
+    end
+
+    add_index :ai_messages, :ai_conversation_id
+    add_index :ai_messages, :llm
+  end
+end 

data/lib/generators/ruby_conversations/install/templates/migrations/create_ai_tool_calls.rb.erb

@@ -0,0 +1,15 @@
+class CreateAiToolCalls < ActiveRecord::Migration[7.0]
+  def change
+    create_table :ai_tool_calls do |t|
+      t.references :ai_message, null: false, foreign_key: true
+      t.string :name, null: false
+      t.jsonb :arguments, null: false, default: {}
+      t.text :result
+      t.text :error
+      
+      t.timestamps
+    end
+    
+    add_index :ai_tool_calls, :name
+  end
+end 

data/lib/generators/ruby_conversations/install/templates/migrations/create_prompt_versions.rb.erb

@@ -0,0 +1,14 @@
+class CreatePromptVersions < ActiveRecord::Migration[7.0]
+  def change
+    create_table :prompt_versions do |t|
+      t.bigint :prompt_id, null: false
+      t.text :message
+      t.float :temperature
+      t.text :change_description
+
+      t.timestamps
+    end
+
+    add_index :prompt_versions, :prompt_id
+  end
+end 

data/lib/generators/ruby_conversations/install/templates/migrations/create_prompts.rb.erb

@@ -0,0 +1,16 @@
+class CreatePrompts < ActiveRecord::Migration[7.0]
+  def change
+    create_table :prompts do |t|
+      t.text :message
+      t.string :name
+      t.string :role
+      t.float :temperature, default: 0.0
+      t.string :valid_placeholders
+
+      t.timestamps
+    end
+    
+    add_index :prompts, :name, unique: true
+    add_index :prompts, :role
+  end
+end 

data/lib/ruby_conversations/configuration.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+module RubyConversations
+  # Configuration options for RubyConversations
+  class Configuration
+    attr_accessor :storage_mode, :api_url, :jwt_secret, :default_llm_model, :default_llm_provider,
+                  :conversation_class, :message_class, :message_input_class, :message_prompt_class,
+                  :prompt_class, :prompt_version_class
+
+    def initialize
+      @storage_mode = :local
+      @default_llm_model = 'claude-3-7-sonnet'
+      @default_llm_provider = 'bedrock'
+      @conversation_class = 'AiConversation'
+      @message_class = 'AiMessage'
+      @message_input_class = 'AiMessageInput'
+      @message_prompt_class = 'AiMessagePrompt'
+      @prompt_class = 'Prompt'
+      @prompt_version_class = 'PromptVersion'
+    end
+
+    def validate!
+      return unless storage_mode == :remote
+      raise ConfigurationError, 'api_url is required for remote storage' unless api_url
+      raise ConfigurationError, 'jwt_secret is required for remote storage' unless jwt_secret
+    end
+  end
+end

data/lib/ruby_conversations/engine.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+module RubyConversations
+  # Defines the Rails engine for RubyConversations.
+  class Engine < ::Rails::Engine
+    config.generators do |g|
+      g.test_framework :rspec
+    end
+
+    config.autoload_paths += Dir[
+      root.join('app', 'models', 'concerns', '**', '*.rb'),
+      root.join('app', 'models', '**', '*.rb')
+    ]
+
+    initializer 'ruby_conversations.zeitwerk' do |_app|
+      Rails.autoloaders.main.push_dir(root.join('app', 'models', 'concerns'))
+      Rails.autoloaders.main.push_dir(root.join('app', 'models'))
+    end
+
+    initializer 'ruby_conversations.assets' do |app|
+      # Future UI components will be added here
+    end
+
+    config.to_prepare do
+      Dir.glob(Engine.root.join('app', 'models', 'concerns', 'ruby_conversations', '*.rb')).each do |c|
+        require_dependency(c)
+      end
+
+      Dir.glob(Engine.root.join('app', 'models', 'ruby_conversations', '*.rb')).each do |m|
+        require_dependency(m)
+      end
+    end
+
+    initializer 'ruby_conversations.factories', after: 'factory_bot.set_factory_paths' do
+      if defined?(FactoryBot)
+        factory_path = File.expand_path('../../spec/factories', __dir__)
+        FactoryBot.definition_file_paths << factory_path unless FactoryBot.definition_file_paths.include?(factory_path)
+      end
+    end
+  end
+end

data/lib/ruby_conversations/jwt_client.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+module RubyConversations
+  # Handles JWT encoding and decoding for secure communication.
+  class JwtClient
+    def initialize(secret)
+      @secret = secret
+    end
+
+    def token
+      JWT.encode(payload, @secret, 'HS256')
+    end
+
+    private
+
+    def payload
+      {
+        exp: 5.minutes.from_now.to_i,
+        iat: Time.current.to_i
+      }
+    end
+  end
+end

data/lib/ruby_conversations/storage/base.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+module RubyConversations
+  module Storage
+    # Abstract base class for storage adapters.
+    class Base
+      def remote?
+        raise NotImplementedError
+      end
+
+      def store_conversation(_conversation)
+        raise NotImplementedError
+      end
+
+      def fetch_conversation(id)
+        raise NotImplementedError
+      end
+
+      def fetch_prompt(name)
+        raise NotImplementedError
+      end
+
+      def store_prompt(prompt)
+        raise NotImplementedError
+      end
+    end
+  end
+end

data/lib/ruby_conversations/storage/local.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+module RubyConversations
+  module Storage
+    # Storage adapter for handling conversations locally using ActiveRecord.
+    class Local < Base
+      def remote?
+        false
+      end
+
+      def store_conversation(conversation)
+        # No-op for local storage - ActiveRecord handles it
+        conversation
+      end
+
+      def fetch_conversation(id)
+        RubyConversations.configuration.conversation_class.constantize.find(id)
+      end
+
+      def fetch_prompt(name)
+        RubyConversations.configuration.prompt_class.constantize.find_by!(name: name)
+      end
+
+      def store_prompt(prompt)
+        # No-op for local storage - ActiveRecord handles it
+        prompt
+      end
+    end
+  end
+end

data/lib/ruby_conversations/storage/remote.rb

@@ -0,0 +1,93 @@
+# frozen_string_literal: true
+
+module RubyConversations
+  module Storage
+    # Storage adapter for handling conversations via a remote API.
+    class Remote < Base
+      def initialize(url:, jwt_secret:)
+        super()
+        @url = url
+        @jwt_client = JwtClient.new(jwt_secret)
+      end
+
+      def remote?
+        true
+      end
+
+      def store_conversation(conversation)
+        response = client.post(
+          "#{@url}/api/conversations",
+          conversation.as_json(conversation_json_options)
+        )
+
+        update_conversation_from_response(conversation, response.body)
+      end
+
+      def fetch_conversation(id)
+        response = client.get("#{@url}/api/conversations/#{id}")
+        build_conversation_from_response(response.body)
+      end
+
+      def fetch_prompt(name)
+        response = client.get("#{@url}/api/prompts/#{name}")
+        build_prompt_from_response(response.body)
+      end
+
+      def store_prompt(prompt)
+        response = client.post(
+          "#{@url}/api/prompts",
+          prompt.as_json(include: :versions)
+        )
+        build_prompt_from_response(response.body)
+      end
+
+      private
+
+      def client
+        @client ||= Faraday.new do |f|
+          f.request :json
+          f.response :json
+          f.adapter Faraday.default_adapter
+          f.headers['Authorization'] = "Bearer #{@jwt_client.token}"
+        end
+      end
+
+      def update_conversation_from_response(conversation, response)
+        conversation.assign_attributes(
+          response.except('ai_messages')
+        )
+
+        response['ai_messages']&.each do |msg_data|
+          msg = conversation.ai_messages.find { |m| m.id == msg_data['id'] }
+          msg&.assign_attributes(msg_data.except('ai_message_prompts'))
+        end
+
+        conversation
+      end
+
+      def build_conversation_from_response(response)
+        RubyConversations.configuration.conversation_class.constantize.new(response)
+      end
+
+      def build_prompt_from_response(response)
+        RubyConversations.configuration.prompt_class.constantize.new(response)
+      end
+
+      # rubocop:disable Metrics/MethodLength
+      def conversation_json_options
+        {
+          include: {
+            ai_messages: {
+              include: {
+                ai_message_prompts: {
+                  include: :ai_message_inputs
+                }
+              }
+            }
+          }
+        }
+      end
+      # rubocop:enable Metrics/MethodLength
+    end
+  end
+end

data/lib/ruby_conversations/version.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+module RubyConversations
+  MAJOR = 1
+  MINOR = 0
+  PATCH = 0 # this is automatically incremented by the build process
+
+  VERSION = "#{RubyConversations::MAJOR}.#{RubyConversations::MINOR}.#{RubyConversations::PATCH}".freeze
+end

data/lib/ruby_conversations.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+require 'rails'
+require 'active_record'
+require 'ruby_llm'
+require 'faraday'
+require 'jwt'
+require 'zeitwerk'
+require 'ruby_conversations/configuration'
+
+loader = Zeitwerk::Loader.for_gem
+loader.ignore("#{__dir__}/generators")
+
+# Main module for the RubyConversations engine.
+module RubyConversations
+  class Error < StandardError; end
+  class ConfigurationError < Error; end
+
+  class << self
+    attr_writer :configuration
+
+    def configuration
+      @configuration ||= Configuration.new
+    end
+
+    def configure
+      yield(configuration)
+      configuration.validate!
+    end
+
+    def storage
+      @storage ||= case configuration.storage_mode
+                   when :local
+                     build_local_storage
+                   when :remote
+                     build_remote_storage
+                   else
+                     raise ConfigurationError, "Unknown storage mode: #{configuration.storage_mode}"
+                   end
+    end
+
+    def reset!
+      @configuration = Configuration.new
+      @storage = nil
+    end
+
+    private
+
+    def build_local_storage
+      Storage::Local.new
+    end
+
+    def build_remote_storage
+      Storage::Remote.new(
+        url: configuration.api_url,
+        jwt_secret: configuration.jwt_secret
+      )
+    end
+  end
+end
+
+loader.setup