llm_classifier 0.1.0

data/lib/llm_classifier/rails/generators/templates/classifier.rb.erb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+class <%= class_name %> < LlmClassifier::Classifier
+  categories <%= categories_array.map { |c| ":#{c}" }.join(", ") %>
+
+  # Uncomment to enable multi-label classification
+  # multi_label true
+
+  # Uncomment to override the default model
+  # model "gpt-4o-mini"
+
+  # Uncomment to override the default adapter
+  # adapter :openai
+
+  system_prompt <<~PROMPT
+    You are a classifier. Analyze the given input and classify it into the appropriate category.
+
+    Categories:
+<% categories_array.each do |cat| -%>
+    - <%= cat %>: [describe what this category means]
+<% end -%>
+
+    Respond with ONLY a JSON object:
+    {
+      "categories": ["category"],
+      "confidence": 0.0-1.0,
+      "reasoning": "Brief explanation"
+    }
+  PROMPT
+
+  # Uncomment to add domain knowledge
+  # knowledge do
+  #   important_terms %w[term1 term2 term3]
+  #   category_hints({
+  #     <%= categories_array.first %>: "hint for identifying this category"
+  #   })
+  # end
+
+  # Uncomment to preprocess input
+  # before_classify do |input|
+  #   input.downcase.strip
+  # end
+
+  # Uncomment to log or process results
+  # after_classify do |result|
+  #   Rails.logger.info("Classification result: #{result.category}")
+  # end
+end

data/lib/llm_classifier/rails/generators/templates/classifier_spec.rb.erb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+require "rails_helper"
+
+RSpec.describe <%= class_name %> do
+  describe ".classify" do
+    it "classifies input into valid categories" do
+      # TODO: Add VCR cassette or mock the LLM response
+      result = described_class.classify("sample input text")
+
+      expect(result).to be_success
+      expect(result.category).to be_in(<%= categories_array %>)
+    end
+  end
+end

data/lib/llm_classifier/rails/railtie.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module LlmClassifier
+  module Rails
+    # Rails integration hook
+    class Railtie < ::Rails::Railtie
+      initializer "llm_classifier.configure_rails_initialization" do
+        # Set Rails logger as default
+        LlmClassifier.configuration.logger = ::Rails.logger
+      end
+
+      generators do
+        require_relative "generators/install_generator"
+        require_relative "generators/classifier_generator"
+      end
+    end
+  end
+end

data/lib/llm_classifier/result.rb

@@ -0,0 +1,65 @@
+# frozen_string_literal: true
+
+module LlmClassifier
+  # Result object returned from classification operations
+  class Result
+    attr_reader :categories, :confidence, :reasoning, :raw_response, :metadata, :error
+
+    def initialize(categories: [], confidence: nil, reasoning: nil, raw_response: nil, error: nil, metadata: {})
+      @categories = Array(categories)
+      @confidence = confidence
+      @reasoning = reasoning
+      @raw_response = raw_response
+      @metadata = metadata
+      @error = error
+    end
+
+    def success?
+      @error.nil?
+    end
+
+    def failure?
+      !success?
+    end
+
+    def category
+      @categories.first
+    end
+
+    def multi_label?
+      @categories.size > 1
+    end
+
+    def to_h
+      {
+        success: success?,
+        categories: @categories,
+        category: category,
+        confidence: @confidence,
+        reasoning: @reasoning,
+        metadata: @metadata,
+        error: @error
+      }
+    end
+
+    class << self
+      def success(categories:, confidence: nil, reasoning: nil, raw_response: nil, metadata: {})
+        new(
+          categories: categories,
+          confidence: confidence,
+          reasoning: reasoning,
+          raw_response: raw_response,
+          metadata: metadata
+        )
+      end
+
+      def failure(error:, raw_response: nil, metadata: {})
+        new(
+          error: error,
+          raw_response: raw_response,
+          metadata: metadata
+        )
+      end
+    end
+  end
+end

data/lib/llm_classifier/version.rb

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

data/lib/llm_classifier.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+require "zeitwerk"
+
+loader = Zeitwerk::Loader.for_gem
+loader.inflector.inflect(
+  "openai" => "OpenAI",
+  "ruby_llm" => "RubyLlm"
+)
+loader.ignore("#{__dir__}/llm_classifier/rails")
+loader.setup
+
+# LlmClassifier provides LLM-powered classification with pluggable adapters and Rails integration
+module LlmClassifier
+  class Error < StandardError; end
+  class ConfigurationError < Error; end
+  class AdapterError < Error; end
+  class ClassificationError < Error; end
+
+  class << self
+    def configuration
+      @configuration ||= Configuration.new
+    end
+
+    def configure
+      yield(configuration)
+    end
+
+    def reset_configuration!
+      @configuration = Configuration.new
+    end
+  end
+end
+
+loader.eager_load
+
+# Load Rails integration if Rails is present
+if defined?(Rails::Railtie)
+  require_relative "llm_classifier/rails/railtie"
+  require_relative "llm_classifier/rails/concerns/classifiable"
+end

metadata

@@ -0,0 +1,88 @@
+--- !ruby/object:Gem::Specification
+name: llm_classifier
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Dmitry Sychev
+bindir: exe
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: zeitwerk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.6'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.6'
+description: A flexible Ruby gem for building LLM-based classifiers. Define categories,
+  system prompts, and domain knowledge using a clean DSL. Supports multiple LLM backends
+  (ruby_llm, OpenAI, Anthropic) and integrates seamlessly with Rails.
+email:
+- dmitry.sychev@axiumfoundry.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".devcontainer/Dockerfile"
+- ".devcontainer/compose.yaml"
+- ".devcontainer/devcontainer.json"
+- ".rspec"
+- ".rubocop.yml"
+- CHANGELOG.md
+- LICENSE.txt
+- README.md
+- Rakefile
+- lib/llm_classifier.rb
+- lib/llm_classifier/adapters/anthropic.rb
+- lib/llm_classifier/adapters/base.rb
+- lib/llm_classifier/adapters/openai.rb
+- lib/llm_classifier/adapters/ruby_llm.rb
+- lib/llm_classifier/classifier.rb
+- lib/llm_classifier/configuration.rb
+- lib/llm_classifier/content_fetchers/base.rb
+- lib/llm_classifier/content_fetchers/null.rb
+- lib/llm_classifier/content_fetchers/web.rb
+- lib/llm_classifier/knowledge.rb
+- lib/llm_classifier/rails/concerns/classifiable.rb
+- lib/llm_classifier/rails/generators/classifier_generator.rb
+- lib/llm_classifier/rails/generators/install_generator.rb
+- lib/llm_classifier/rails/generators/templates/classifier.rb.erb
+- lib/llm_classifier/rails/generators/templates/classifier_spec.rb.erb
+- lib/llm_classifier/rails/railtie.rb
+- lib/llm_classifier/result.rb
+- lib/llm_classifier/version.rb
+homepage: https://github.com/AxiumFoundry/llm_classifier
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/AxiumFoundry/llm_classifier
+  source_code_uri: https://github.com/AxiumFoundry/llm_classifier
+  changelog_uri: https://github.com/AxiumFoundry/llm_classifier/blob/main/CHANGELOG.md
+  rubygems_mfa_required: 'true'
+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.6.9
+specification_version: 4
+summary: LLM-powered classification for Ruby with pluggable adapters and Rails integration
+test_files: []