ruby_llm-sequel 0.1.0 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b0cdecd16447df344e6e618c6e0883890348068cfa185085304e6bd26bffbb9a
-  data.tar.gz: 2984799b793177bf89fef310270444aa3728c7ad4f9fee0558df599065cebc4a
+  metadata.gz: 3422a45af2abd1286b9762fcbf6efd245af84375e6dde6ab067adb433a80c3e7
+  data.tar.gz: 54ec6f2324de674e4fe112280843115f0b866b5a8edc93776768d57b3b5b2f76
 SHA512:
-  metadata.gz: 91a0ad1ef357c27909b064abcbbd6b5cc83b256d1de8b280f8b0d60e6d7fa71c9b9bda191a8d80ded993eb100cb3ad9b68f2483fd9fe9dd15ace65e7d17bc834
-  data.tar.gz: 85d122fcad93f5c14af4e5897c9173a69b0c1725a231f60bd8d0b606da31f696cf8f837d9653e2973540230ae8273079fbdde15fe92f132a759aa90c509b8976
+  metadata.gz: 4fac8795a86055f72e7197042006de0f14cd6c2d7e9951ffe470721c0a9c8032d062a32b2b98be2ba7ff3ca3c9d98c35e779bc93dccbda057c8b43f3ec06f7e1
+  data.tar.gz: 7a662f9adb13c9570b564fb7268c8230c19e9b1c1bbbebef392a2d0a38f587b5499bc5b6cb04d4e399a870293c2e9c7c1fecd5bcb010878ee0fb3c02434fecf6

data/README.md

@@ -137,8 +137,14 @@ end
 # Configure RubyLLM
 RubyLLM.configure do |config|
   config.openai_api_key = ENV['OPENAI_API_KEY']
+
+  # For custom Model class names (defaults to 'Model')
+  # config.model_registry_class = 'AIModel'
 end
 
+# Opt-in behaviour to use models registry from database
+RubyLLM.Sequel.use_model_registry!
+
 # Create a chat
 chat = Chat.create
 
@@ -246,7 +252,7 @@ ruby -Ilib:spec spec/ruby_llm/sequel/chat_methods_spec.rb
 The tests use:
 - **Minitest** for the test framework
 - **SQLite** in-memory database for fast test execution
-- **VCR + WebMock** for HTTP interaction recording (when testing with real API calls)
+- **SimpleCov** for code coverage reporting
 - **Transaction rollback** to isolate tests and maintain a clean database state
 
 All test configuration is in `spec/spec_helper.rb`.

data/lib/ruby_llm/sequel/chat_methods.rb

@@ -161,7 +161,7 @@ module RubyLLM
       def complete(...)
         to_llm.complete(...)
       rescue ::RubyLLM::Error => e
-        cleanup_failed_messages if !@message.id.nil? && @message.content && @message.content != ''
+        cleanup_failed_messages if !@message&.id.nil? && @message&.content && @message&.content != ''
         cleanup_orphaned_tool_results
         raise e
       end

data/lib/ruby_llm/sequel/version.rb

@@ -2,6 +2,6 @@
 
 module RubyLLM
   module Sequel
-    VERSION = '0.1.0'
+    VERSION = '0.1.2'
   end
 end

data/lib/ruby_llm/sequel.rb

@@ -15,5 +15,29 @@ require_relative '../sequel/plugins/ruby_llm'
 module RubyLLM
   module Sequel
     class UnsupportedFeatureError < StandardError; end
+
+    class << self
+      def use_model_registry!
+        # Monkey-patch Models to use database
+        RubyLLM::Models.class_eval do
+          def self.load_models
+            read_from_database
+          rescue StandardError => e
+            RubyLLM.logger.debug "Failed to load models from database: #{e.message}, falling back to JSON"
+            read_from_json
+          end
+
+          def self.read_from_database
+            model_class = RubyLLM.config.model_registry_class
+            model_class = model_class.constantize if model_class.is_a?(String)
+            model_class.all.map(&:to_llm)
+          end
+
+          def load_from_database!
+            @models = self.class.read_from_database
+          end
+        end
+      end
+    end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm-sequel
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.2
 platform: ruby
 authors:
 - Julian Pasquale