rails_ai_promptable 0.1.0

data/lib/rails_ai_promptable/providers/mistral_provider.rb

@@ -0,0 +1,52 @@
+# frozen_string_literal: true
+
+require 'net/http'
+require 'json'
+
+module RailsAIPromptable
+  module Providers
+    class MistralProvider < BaseProvider
+      def initialize(configuration)
+        super
+        @api_key = configuration.mistral_api_key || configuration.api_key
+        @base_url = configuration.mistral_base_url || 'https://api.mistral.ai/v1'
+        @timeout = configuration.timeout
+      end
+
+      def generate(prompt:, model:, temperature:, format:)
+        uri = URI.parse("#{@base_url}/chat/completions")
+        http = Net::HTTP.new(uri.host, uri.port)
+        http.use_ssl = uri.scheme == 'https'
+        http.read_timeout = @timeout
+
+        request = Net::HTTP::Post.new(uri.request_uri, {
+          'Content-Type' => 'application/json',
+          'Authorization' => "Bearer #{@api_key}"
+        })
+
+        body = {
+          model: model,
+          messages: [{ role: 'user', content: prompt }],
+          temperature: temperature,
+          max_tokens: 2048
+        }
+
+        request.body = body.to_json
+
+        response = http.request(request)
+        parsed = JSON.parse(response.body)
+
+        if response.code.to_i >= 400
+          error_message = parsed.dig('message') || 'Unknown error'
+          raise "Mistral API error: #{error_message}"
+        end
+
+        # Extract content (OpenAI-compatible format)
+        parsed.dig('choices', 0, 'message', 'content')
+      rescue => e
+        RailsAIPromptable.configuration.logger.error("[rails_ai_promptable] mistral error: #{e.message}")
+        nil
+      end
+    end
+  end
+end

data/lib/rails_ai_promptable/providers/openai_provider.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+require 'net/http'
+require 'json'
+
+module RailsAIPromptable
+  module Providers
+    class OpenAIProvider < BaseProvider
+      def initialize(configuration)
+        super
+        @api_key = configuration.api_key
+        @base_url = configuration.openai_base_url
+        @timeout = configuration.timeout
+      end
+
+      def generate(prompt:, model:, temperature:, format:)
+        uri = URI.parse("#{@base_url}/chat/completions")
+        http = Net::HTTP.new(uri.host, uri.port)
+        http.use_ssl = uri.scheme == 'https'
+        request = Net::HTTP::Post.new(uri.request_uri, initheader = {
+          'Content-Type' => 'application/json',
+          'Authorization' => "Bearer #{@api_key}"
+        })
+
+        body = {
+          model: model,
+          messages: [{ role: 'user', content: prompt }],
+          temperature: temperature
+        }
+
+        request.body = body.to_json
+
+        response = http.request(request)
+        parsed = JSON.parse(response.body)
+
+        # naive extraction
+        parsed.dig('choices', 0, 'message', 'content')
+      rescue => e
+        RailsAIPromptable.configuration.logger.error("[rails_ai_promptable] openai error: #{e.message}")
+        nil
+      end
+    end
+  end
+end

data/lib/rails_ai_promptable/providers/openrouter_provider.rb

@@ -0,0 +1,60 @@
+# frozen_string_literal: true
+
+require 'net/http'
+require 'json'
+
+module RailsAIPromptable
+  module Providers
+    class OpenRouterProvider < BaseProvider
+      def initialize(configuration)
+        super
+        @api_key = configuration.openrouter_api_key || configuration.api_key
+        @base_url = configuration.openrouter_base_url || 'https://openrouter.ai/api/v1'
+        @timeout = configuration.timeout
+        @app_name = configuration.openrouter_app_name
+        @site_url = configuration.openrouter_site_url
+      end
+
+      def generate(prompt:, model:, temperature:, format:)
+        uri = URI.parse("#{@base_url}/chat/completions")
+        http = Net::HTTP.new(uri.host, uri.port)
+        http.use_ssl = uri.scheme == 'https'
+        http.read_timeout = @timeout
+
+        headers = {
+          'Content-Type' => 'application/json',
+          'Authorization' => "Bearer #{@api_key}"
+        }
+
+        # OpenRouter requires these headers for tracking/attribution
+        headers['HTTP-Referer'] = @site_url if @site_url
+        headers['X-Title'] = @app_name if @app_name
+
+        request = Net::HTTP::Post.new(uri.request_uri, headers)
+
+        body = {
+          model: model,
+          messages: [{ role: 'user', content: prompt }],
+          temperature: temperature,
+          max_tokens: 2048
+        }
+
+        request.body = body.to_json
+
+        response = http.request(request)
+        parsed = JSON.parse(response.body)
+
+        if response.code.to_i >= 400
+          error_message = parsed.dig('error', 'message') || 'Unknown error'
+          raise "OpenRouter API error: #{error_message}"
+        end
+
+        # Extract content (OpenAI-compatible format)
+        parsed.dig('choices', 0, 'message', 'content')
+      rescue => e
+        RailsAIPromptable.configuration.logger.error("[rails_ai_promptable] openrouter error: #{e.message}")
+        nil
+      end
+    end
+  end
+end

data/lib/rails_ai_promptable/providers.rb

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+module RailsAIPromptable
+  module Providers
+    autoload :BaseProvider, 'rails_ai_promptable/providers/base_provider'
+    autoload :OpenAIProvider, 'rails_ai_promptable/providers/openai_provider'
+    autoload :AnthropicProvider, 'rails_ai_promptable/providers/anthropic_provider'
+    autoload :GeminiProvider, 'rails_ai_promptable/providers/gemini_provider'
+    autoload :CohereProvider, 'rails_ai_promptable/providers/cohere_provider'
+    autoload :AzureOpenAIProvider, 'rails_ai_promptable/providers/azure_openai_provider'
+    autoload :MistralProvider, 'rails_ai_promptable/providers/mistral_provider'
+    autoload :OpenRouterProvider, 'rails_ai_promptable/providers/openrouter_provider'
+
+    def self.for(provider_sym, configuration)
+      case provider_sym.to_sym
+      when :openai
+        OpenAIProvider.new(configuration)
+      when :anthropic, :claude
+        AnthropicProvider.new(configuration)
+      when :gemini, :google
+        GeminiProvider.new(configuration)
+      when :cohere
+        CohereProvider.new(configuration)
+      when :azure_openai, :azure
+        AzureOpenAIProvider.new(configuration)
+      when :mistral
+        MistralProvider.new(configuration)
+      when :openrouter
+        OpenRouterProvider.new(configuration)
+      else
+        raise ArgumentError, "Unknown provider: #{provider_sym}. Supported providers: :openai, :anthropic, :gemini, :cohere, :azure_openai, :mistral, :openrouter"
+      end
+    end
+
+    # Helper method to list all available providers
+    def self.available_providers
+      [:openai, :anthropic, :gemini, :cohere, :azure_openai, :mistral, :openrouter]
+    end
+  end
+end

data/lib/rails_ai_promptable/template_registry.rb

@@ -0,0 +1,73 @@
+# frozen_string_literal: true
+
+require 'yaml'
+
+module RailsAIPromptable
+  class TemplateRegistry
+    class << self
+      def templates
+        @templates ||= {}
+      end
+
+      # Register a template with a name
+      def register(name, template)
+        templates[name.to_sym] = template
+      end
+
+      # Get a template by name
+      def get(name)
+        templates[name.to_sym]
+      end
+
+      # Load templates from a YAML file
+      def load_from_file(file_path)
+        return unless File.exist?(file_path)
+
+        begin
+          loaded_templates = YAML.load_file(file_path)
+          return unless loaded_templates.is_a?(Hash)
+
+          loaded_templates.each do |name, template|
+            register(name, template)
+          end
+        rescue Psych::SyntaxError => e
+          # Log the error but don't raise - invalid YAML files are silently skipped
+          if defined?(Rails) && Rails.respond_to?(:logger) && Rails.logger
+            Rails.logger.warn("[rails_ai_promptable] Failed to load templates from #{file_path}: #{e.message}")
+          end
+        end
+      end
+
+      # Load templates from a directory
+      # Each file should be named <template_name>.yml or <template_name>.txt
+      def load_from_directory(directory_path)
+        return unless Dir.exist?(directory_path)
+
+        Dir.glob(File.join(directory_path, '*')).each do |file_path|
+          next unless File.file?(file_path)
+
+          name = File.basename(file_path, '.*')
+
+          if file_path.end_with?('.yml', '.yaml')
+            content = YAML.load_file(file_path)
+            template = content.is_a?(Hash) ? content['template'] : content.to_s
+          else
+            template = File.read(file_path)
+          end
+
+          register(name, template) if template
+        end
+      end
+
+      # Clear all registered templates
+      def clear!
+        @templates = {}
+      end
+
+      # List all registered template names
+      def list
+        templates.keys
+      end
+    end
+  end
+end

data/lib/rails_ai_promptable/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module RailsAIPromptable
+  VERSION = "0.1.0"
+end

data/lib/rails_ai_promptable.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+require 'rails_ai_promptable/version'
+require 'rails_ai_promptable/configuration'
+require 'rails_ai_promptable/template_registry'
+require 'rails_ai_promptable/promptable'
+require 'rails_ai_promptable/providers'
+require 'rails_ai_promptable/logger'
+require 'rails_ai_promptable/background_job'
+
+module RailsAIPromptable
+  class << self
+    attr_accessor :configuration
+
+    def configure
+      self.configuration ||= Configuration.new
+      yield(configuration) if block_given?
+    end
+
+    def client
+      @client ||= Providers.for(configuration.provider, configuration)
+    end
+
+    def reset_client!
+      @client = nil
+    end
+  end
+end

data/sig/rails_ai_promptable.rbs

@@ -0,0 +1,4 @@
+module RailsAiPromptable
+  VERSION: String
+  # See the writing guide of rbs: https://github.com/ruby/rbs#guides
+end

metadata

@@ -0,0 +1,104 @@
+--- !ruby/object:Gem::Specification
+name: rails_ai_promptable
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Shoaib Malik
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2025-11-06 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rails
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '7.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '7.0'
+- !ruby/object:Gem::Dependency
+  name: httparty
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0.20'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0.20'
+description: rails_ai_promptable makes it easy to integrate AI-driven features into
+  your Rails application. It allows you to define promptable methods, chain context,
+  and connect with AI APIs like OpenAI, Anthropic, or local LLMs with minimal setup.
+email:
+- shoaib2109@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".rspec"
+- ".rubocop.yml"
+- CHANGELOG.md
+- CODE_OF_CONDUCT.md
+- LICENSE.txt
+- README.md
+- Rakefile
+- lib/generators/rails_ai_promptable/install_generator.rb
+- lib/generators/rails_ai_promptable/templates/POST_INSTALL
+- lib/generators/rails_ai_promptable/templates/rails_ai_promptable.rb.tt
+- lib/rails_ai_promptable.rb
+- lib/rails_ai_promptable/background_job.rb
+- lib/rails_ai_promptable/configuration.rb
+- lib/rails_ai_promptable/logger.rb
+- lib/rails_ai_promptable/promptable.rb
+- lib/rails_ai_promptable/providers.rb
+- lib/rails_ai_promptable/providers/anthropic_provider.rb
+- lib/rails_ai_promptable/providers/azure_openai_provider.rb
+- lib/rails_ai_promptable/providers/base_provider.rb
+- lib/rails_ai_promptable/providers/cohere_provider.rb
+- lib/rails_ai_promptable/providers/gemini_provider.rb
+- lib/rails_ai_promptable/providers/mistral_provider.rb
+- lib/rails_ai_promptable/providers/openai_provider.rb
+- lib/rails_ai_promptable/providers/openrouter_provider.rb
+- lib/rails_ai_promptable/template_registry.rb
+- lib/rails_ai_promptable/version.rb
+- sig/rails_ai_promptable.rbs
+homepage: https://github.com/shoaibmalik786/rails_ai_promptable
+licenses:
+- MIT
+metadata:
+  allowed_push_host: https://rubygems.org
+  homepage_uri: https://github.com/shoaibmalik786/rails_ai_promptable
+  source_code_uri: https://github.com/shoaibmalik786/rails_ai_promptable
+  changelog_uri: https://github.com/shoaibmalik786/rails_ai_promptable/blob/main/CHANGELOG.md
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.1.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.4.10
+signing_key:
+specification_version: 4
+summary: Add AI promptable behavior to your Rails models and classes.
+test_files: []