ruby_llm-sequel 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: b0cdecd16447df344e6e618c6e0883890348068cfa185085304e6bd26bffbb9a
+  data.tar.gz: 2984799b793177bf89fef310270444aa3728c7ad4f9fee0558df599065cebc4a
+SHA512:
+  metadata.gz: 91a0ad1ef357c27909b064abcbbd6b5cc83b256d1de8b280f8b0d60e6d7fa71c9b9bda191a8d80ded993eb100cb3ad9b68f2483fd9fe9dd15ace65e7d17bc834
+  data.tar.gz: 85d122fcad93f5c14af4e5897c9173a69b0c1725a231f60bd8d0b606da31f696cf8f837d9653e2973540230ae8273079fbdde15fe92f132a759aa90c509b8976

data/LICENSE

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2025 Julian Pasquale
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.md

@@ -0,0 +1,266 @@
+# RubyLLM::Sequel
+
+Sequel ORM integration for [RubyLLM](https://github.com/contextco/ruby_llm), providing the same ActiveRecord-style `acts_as_*` methods for Sequel models to interact with LLM providers.
+
+This gem enables your Sequel models to serve as:
+- **Chat interfaces** - Direct LLM conversations with persistent message history
+- **Message storage** - User, assistant, system, and tool messages
+- **Tool call tracking** - Function/tool invocations and results
+- **Model registry** - Track available models, pricing, and capabilities
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'ruby_llm-sequel'
+```
+
+And then execute:
+
+```bash
+bundle install
+```
+
+Or install it yourself as:
+
+```bash
+gem install ruby_llm-sequel
+```
+
+## Usage
+
+### Setup
+
+First, require the gem in your application:
+
+```ruby
+require 'ruby_llm/sequel'
+```
+
+### Database Schema
+
+Your database needs tables for models, chats, messages, and tool calls. Here's an example schema:
+
+```ruby
+DB.create_table :models do
+  primary_key :id
+  String :model_id, null: false
+  String :name, null: false
+  String :provider, null: false
+  String :family
+  Integer :context_window
+  Integer :max_output_tokens
+
+  String :capabilities, text: true
+  String :pricing, text: true
+  # or use jsonb for Postgres
+  jsonb :capabilities, text: true
+  jsonb :pricing, text: true
+
+  DateTime :created_at
+  DateTime :updated_at
+
+  index [:provider, :model_id], unique: true
+end
+
+DB.create_table :chats do
+  primary_key :id
+  foreign_key :model_id, :models
+  TrueClass :active, default: true
+  DateTime :created_at
+  DateTime :updated_at
+end
+
+DB.create_table :messages do
+  primary_key :id
+  foreign_key :chat_id, :chats, null: false
+  String :role, null: false
+  String :content, text: true
+  Integer :input_tokens
+  Integer :output_tokens
+
+  String :content_raw, text: true
+  # or use jsonb for Postgres
+  jsonb :content_raw
+
+  foreign_key :model_id, :models
+  foreign_key :tool_call_id, :tool_calls
+  DateTime :created_at
+  DateTime :updated_at
+end
+
+DB.create_table :tool_calls do
+  primary_key :id
+  foreign_key :message_id, :messages, null: false
+  String :tool_call_id, null: false, unique: true
+  String :name, null: false
+  String :arguments, text: true
+  DateTime :created_at
+  DateTime :updated_at
+end
+```
+
+### Model Setup
+
+Define your Sequel models with the RubyLLM plugin and appropriate `acts_as_*` methods:
+
+```ruby
+class Model < Sequel::Model
+  plugin ::Sequel::Plugins::RubyLLM
+
+  acts_as_model
+end
+
+class Chat < Sequel::Model
+  plugin ::Sequel::Plugins::RubyLLM
+
+  acts_as_chat(model: :llm_model,model_class: 'ApiFr::Agents::Model')
+end
+
+class Message < Sequel::Model
+  plugin ::Sequel::Plugins::RubyLLM
+
+  acts_as_message(model: :llm_model, model_class: 'ApiFr::Agents::Model',)
+end
+
+class ToolCall < Sequel::Model
+  plugin ::Sequel::Plugins::RubyLLM
+
+  acts_as_tool_call
+end
+```
+
+### Basic Chat Usage
+
+```ruby
+# Configure RubyLLM
+RubyLLM.configure do |config|
+  config.openai_api_key = ENV['OPENAI_API_KEY']
+end
+
+# Create a chat
+chat = Chat.create
+
+# Set instructions
+chat.with_instructions("You are a helpful assistant")
+
+# Send a message and get a response
+chat.create_user_message("Hello!")
+response = chat.ask  # Calls the LLM and stores the response
+
+puts response.content
+# => "Hello! How can I help you today?"
+
+# Access message history
+chat.messages.each do |message|
+  puts "#{message.role}: #{message.content}"
+end
+```
+
+### Tool Calls
+
+```ruby
+# Define a tool
+class Weather < RubyLLM::Tool
+  description "Gets current weather for a location"
+
+  params do  # the params DSL is only available in v1.9+. older versions should use the param helper instead
+    string :latitude, description: "Latitude (e.g., 52.5200)"
+    string :longitude, description: "Longitude (e.g., 13.4050)"
+  end
+
+  def execute(latitude:, longitude:)
+    url = "https://api.open-meteo.com/v1/forecast?latitude=#{latitude}&longitude=#{longitude}&current=temperature_2m,wind_speed_10m"
+
+    response = Faraday.get(url)
+    data = JSON.parse(response.body)
+  rescue => e
+    { error: e.message }
+  end
+end
+
+# Use the tool in a chat
+chat.with_tool(Weather.new)
+response = chat.ask("What's the weather in San Francisco?")
+
+# Tool calls are automatically tracked
+response.tool_calls.each do |tool_call|
+  puts "Called: #{tool_call.name}"
+  puts "Arguments: #{tool_call.arguments}"
+end
+```
+
+### Model Registry
+
+```ruby
+# Refresh model information from providers
+Model.refresh! # Fetches latest models from all configured providers
+
+# Query models
+gpt4 = Model.first(provider: 'openai', model_id: 'gpt-4')
+
+# Check capabilities
+gpt4.function_calling?   # => true
+gpt4.streaming?          # => true
+gpt4.supports?('vision') # => false
+
+# Get pricing
+gpt4.input_price_per_million  # => 30.0
+gpt4.output_price_per_million # => 60.0
+```
+
+## Important Notes
+
+### Database Compatibility
+
+The gem supports both PostgreSQL (with jsonb) and other databases (JSON as text):
+- PostgreSQL: Uses `jsonb` columns automatically
+- SQLite/MySQL: Stores JSON as text strings
+- JSON parsing/serialization is handled transparently
+
+### The model association
+
+The Sequel gem already defines the `model` instance method for subclasses of `Sequel::Model` so it's recommended to use a different association name like `llm_model` to avoid name conflicts
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake test` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+### Running Tests
+
+The test suite uses Minitest and runs against an in-memory SQLite database. To run the tests:
+
+```bash
+bundle exec rake test
+```
+
+To run a specific test file:
+
+```bash
+ruby -Ilib:spec spec/ruby_llm/sequel/chat_methods_spec.rb
+```
+
+### Test Configuration
+
+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)
+- **Transaction rollback** to isolate tests and maintain a clean database state
+
+All test configuration is in `spec/spec_helper.rb`.
+
+### Contributing to Tests
+
+When adding new features, please include corresponding tests. Test files are located in `spec/ruby_llm/sequel/` and follow the naming convention `*_methods_spec.rb`.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and the created tag, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/ruby_llm-sequel.
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).

data/lib/ruby_llm/sequel/chat_methods.rb

@@ -0,0 +1,333 @@
+# frozen_string_literal: true
+
+module RubyLLM
+  module Sequel
+    module ChatMethods
+      attr_accessor :assume_model_exists, :context
+
+      def before_save
+        super
+        resolve_model_from_strings
+      end
+
+      def model=(value)
+        @model_string = value if value.is_a?(String)
+        return if value.is_a?(String)
+
+        send("#{self.class.model_association_name}=", value)
+      end
+
+      def model_id=(value)
+        if value.is_a?(Integer) || value.nil?
+          super
+        else
+          @model_string = value
+        end
+      end
+
+      def provider=(value)
+        @provider_string = value
+      end
+
+      def provider
+        model_record = send(self.class.model_association_name)
+        model_record&.provider
+      end
+
+      def to_llm
+        model_record = send(self.class.model_association_name)
+        @chat ||= (context || ::RubyLLM).chat(
+          model: model_record.model_id,
+          provider: model_record.provider.to_sym,
+          assume_model_exists: assume_model_exists || false
+        )
+        @chat.reset_messages!
+
+        message_class = self.class.message_class.constantize
+        tool_calls_assoc = message_class.tool_calls_association_name
+
+        messages_association_dataset.eager(
+          tool_calls_assoc, :parent_tool_call, self.class.model_association_name
+        ).all.each do |msg|
+          @chat.add_message(msg.to_llm)
+        end
+
+        setup_persistence_callbacks
+      end
+
+      def with_instructions(instructions, replace: false)
+        db.transaction do
+          messages_association_dataset.where(role: 'system').destroy if replace
+          send("add_#{self.class.messages_association_name.to_s.singularize}", role: 'system', content: instructions)
+        end
+        to_llm.with_instructions(instructions)
+        self
+      end
+
+      def with_tool(...)
+        to_llm.with_tool(...)
+        self
+      end
+
+      def with_tools(...)
+        to_llm.with_tools(...)
+        self
+      end
+
+      def with_model(model_name, provider: nil, assume_exists: false)
+        self.model = model_name
+        self.provider = provider if provider
+        self.assume_model_exists = assume_exists
+        resolve_model_from_strings
+        save
+        model_record = send(self.class.model_association_name)
+        to_llm.with_model(model_record.model_id, provider: model_record.provider.to_sym, assume_exists: assume_exists)
+        self
+      end
+
+      def with_temperature(...)
+        to_llm.with_temperature(...)
+        self
+      end
+
+      def with_params(...)
+        to_llm.with_params(...)
+        self
+      end
+
+      def with_headers(...)
+        to_llm.with_headers(...)
+        self
+      end
+
+      def with_schema(...)
+        to_llm.with_schema(...)
+        self
+      end
+
+      def on_new_message(&block)
+        to_llm
+
+        existing_callback = @chat.instance_variable_get(:@on)[:new_message]
+
+        @chat.on_new_message do
+          existing_callback&.call
+          block&.call
+        end
+        self
+      end
+
+      def on_end_message(&block)
+        to_llm
+
+        existing_callback = @chat.instance_variable_get(:@on)[:end_message]
+
+        @chat.on_end_message do |msg|
+          existing_callback&.call(msg)
+          block&.call(msg)
+        end
+        self
+      end
+
+      def on_tool_call(...)
+        to_llm.on_tool_call(...)
+        self
+      end
+
+      def on_tool_result(...)
+        to_llm.on_tool_result(...)
+        self
+      end
+
+      def create_user_message(content, with: nil)
+        raise UnsupportedFeatureError, "Cannot use 'with' parameter with Sequel integration" if with
+
+        content_text, _, content_raw = prepare_content_for_storage(content)
+
+        message_record = send("add_#{self.class.messages_association_name.to_s.singularize}", role: 'user',
+                                                                                              content: content_text)
+        message_record.update(content_raw: content_raw) if message_record.columns.include?(:content_raw) && content_raw
+
+        message_record
+      end
+
+      def ask(message, with: nil, &)
+        create_user_message(message, with: with)
+        complete(&)
+      end
+
+      alias say ask
+
+      def complete(...)
+        to_llm.complete(...)
+      rescue ::RubyLLM::Error => e
+        cleanup_failed_messages if !@message.id.nil? && @message.content && @message.content != ''
+        cleanup_orphaned_tool_results
+        raise e
+      end
+
+      private
+
+      # Helper to handle JSON data for both PostgreSQL (jsonb) and other databases (text)
+      def jsonb_or_string(value)
+        return nil if value.nil?
+
+        # If the database supports pg_jsonb (PostgreSQL), use it
+        if defined?(::Sequel::Postgres) && db.database_type == :postgres
+          ::Sequel.pg_jsonb(value)
+        else
+          # For other databases (like SQLite), store as JSON string
+          value.is_a?(String) ? value : JSON.generate(value)
+        end
+      end
+
+      def resolve_model_from_strings
+        config = context&.config || ::RubyLLM.config
+        @model_string ||= config.default_model unless send(self.class.model_association_name)
+        return unless @model_string
+
+        model_info, _provider = ::RubyLLM::Models.resolve(
+          @model_string,
+          provider: @provider_string,
+          assume_exists: assume_model_exists || false,
+          config: config
+        )
+
+        model_record = self.class.model_class.constantize.find_or_create(
+          model_id: model_info.id,
+          provider: model_info.provider
+        ) do |m|
+          m.name = model_info.name || model_info.id
+          m.family = model_info.family
+          m.context_window = model_info.context_window
+          m.max_output_tokens = model_info.max_output_tokens
+          m.capabilities = jsonb_or_string(model_info.capabilities || [])
+          m.modalities = jsonb_or_string(model_info.modalities.to_h)
+          m.pricing = jsonb_or_string(model_info.pricing.to_h)
+          m.metadata = jsonb_or_string(model_info.metadata || {})
+        end
+
+        send("#{self.class.model_association_name}=", model_record)
+        @model_string = nil
+        @provider_string = nil
+      end
+
+      def setup_persistence_callbacks
+        return @chat if @chat.instance_variable_get(:@_persistence_callbacks_setup)
+
+        @chat.on_new_message { persist_new_message }
+        @chat.on_end_message { |msg| persist_message_completion(msg) }
+
+        @chat.instance_variable_set(:@_persistence_callbacks_setup, true)
+        @chat
+      end
+
+      def persist_new_message
+        @message = send("add_#{self.class.messages_association_name.to_s.singularize}", role: 'assistant', content: '')
+      end
+
+      def persist_message_completion(message)
+        return unless message
+
+        tool_call_id = find_tool_call_id(message.tool_call_id) if message.tool_call_id
+
+        db.transaction do
+          content_text, _attachments_to_persist, content_raw = prepare_content_for_storage(message.content)
+
+          attrs = {
+            role: message.role,
+            content: content_text,
+            input_tokens: message.input_tokens,
+            output_tokens: message.output_tokens
+          }
+          attrs[:cached_tokens] = message.cached_tokens if @message.columns.include?(:cached_tokens)
+          if @message.columns.include?(:cache_creation_tokens)
+            attrs[:cache_creation_tokens] = message.cache_creation_tokens
+          end
+
+          attrs[self.class.model_association_name] = send(self.class.model_association_name)
+
+          if tool_call_id
+            parent_tool_call_assoc = @message.class.association_reflection(:parent_tool_call)
+            attrs[parent_tool_call_assoc[:key]] = tool_call_id if parent_tool_call_assoc
+          end
+
+          @message.update(attrs)
+          if @message.columns.include?(:content_raw) && content_raw
+            @message.update(content_raw: ::Sequel.pg_jsonb(content_raw))
+          end
+
+          persist_tool_calls(message.tool_calls) unless message.tool_calls.nil? || message.tool_calls.empty?
+        end
+      end
+
+      def persist_tool_calls(tool_calls)
+        tool_calls.each_value do |tool_call|
+          attributes = tool_call.to_h
+          attributes[:tool_call_id] = attributes.delete(:id)
+          @message.send("add_#{@message.class.tool_calls_association_name.to_s.singularize}", **attributes)
+        end
+      end
+
+      def find_tool_call_id(tool_call_id)
+        message_class = self.class.message_class.constantize
+        tool_call_class = message_class.tool_call_class.constantize
+
+        # Find the tool_call record by its string tool_call_id within this chat's messages
+        tool_call = tool_call_class
+                    .where(tool_call_id: tool_call_id)
+                    .where(message_id: messages_association_dataset.select(:id))
+                    .first
+
+        tool_call&.id
+      end
+
+      def prepare_content_for_storage(content)
+        attachments = nil
+        content_raw = nil
+        content_text = content
+
+        case content
+        when ::RubyLLM::Content::Raw
+          content_raw = content.value
+          content_text = nil
+        when ::RubyLLM::Content
+          attachments = content.attachments if content.attachments.any?
+          content_text = content.text
+        when Hash, Array
+          content_raw = content
+          content_text = nil
+        end
+
+        [content_text, attachments, content_raw]
+      end
+
+      def cleanup_failed_messages
+        ::RubyLLM.logger.warn "RubyLLM: API call failed, destroying message: #{@message.id}"
+        @message.destroy
+      end
+
+      def cleanup_orphaned_tool_results
+        reload
+        last = messages_association_dataset.order(:id).last
+
+        return unless last&.tool_call? || last&.tool_result?
+
+        if last.tool_call?
+          last.destroy
+        elsif last.tool_result?
+          tool_call_message = last.parent_tool_call.message_association
+          expected_results = tool_call_message.tool_calls_association.map(&:id)
+          actual_results = tool_call_message.tool_calls_association.select(&:result_association).map(&:id)
+
+          if expected_results.sort != actual_results.sort
+            tool_call_message.tool_calls_association.each do |tc|
+              tc.result_association&.destroy
+            end
+            tool_call_message.destroy
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ruby_llm/sequel/message_methods.rb

@@ -0,0 +1,53 @@
+# frozen_string_literal: true
+
+module RubyLLM
+  module Sequel
+    module MessageMethods
+      def to_llm
+        cached = columns.include?(:cached_tokens) ? self[:cached_tokens] : nil
+        cache_creation = columns.include?(:cache_creation_tokens) ? self[:cache_creation_tokens] : nil
+
+        ::RubyLLM::Message.new(
+          role: role.to_sym,
+          content: extract_content,
+          tool_calls: extract_tool_calls,
+          tool_call_id: extract_tool_call_id,
+          input_tokens: input_tokens,
+          output_tokens: output_tokens,
+          cached_tokens: cached,
+          cache_creation_tokens: cache_creation,
+          model_id: model_association&.model_id
+        )
+      end
+
+      def tool_call?
+        role == 'assistant' && tool_calls_association.any?
+      end
+
+      def tool_result?
+        role == 'tool' && !tool_call_id.nil?
+      end
+
+      private
+
+      def extract_content
+        return ::RubyLLM::Content::Raw.new(content_raw) if columns.include?(:content_raw) && !content_raw.nil?
+
+        content
+      end
+
+      def extract_tool_calls
+        tool_calls_association.to_h do |tool_call|
+          [
+            tool_call.tool_call_id,
+            tool_call.to_llm
+          ]
+        end
+      end
+
+      def extract_tool_call_id
+        parent_tool_call&.tool_call_id
+      end
+    end
+  end
+end

data/lib/ruby_llm/sequel/model_methods.rb

@@ -0,0 +1,188 @@
+# frozen_string_literal: true
+
+module RubyLLM
+  module Sequel
+    module ModelMethods
+      def self.included(base)
+        base.extend(ClassMethods)
+      end
+
+      module ClassMethods
+        # Helper to handle JSON data for both PostgreSQL (jsonb) and other databases (text)
+        def jsonb_or_string(value)
+          return nil if value.nil?
+
+          # If the database supports pg_jsonb (PostgreSQL), use it
+          if defined?(::Sequel::Postgres) && db.database_type == :postgres
+            ::Sequel.pg_jsonb(value)
+          else
+            # For other databases (like SQLite), store as JSON string
+            value.is_a?(String) ? value : JSON.generate(value)
+          end
+        end
+
+        def refresh!
+          ::RubyLLM.models.refresh!
+          save_to_database
+        end
+
+        def save_to_database
+          ::Sequel::Model.db.transaction do
+            ::RubyLLM.models.all.each do |model_info|
+              from_llm(model_info)
+            end
+          end
+        end
+
+        def from_llm(model_info)
+          modalities_hash =
+            if model_info.modalities.respond_to?(:to_h)
+              model_info.modalities.to_h
+            else
+              model_info.modalities || {}
+            end
+          capabilities_array = model_info.capabilities || []
+          pricing_hash = model_info.pricing.respond_to?(:to_h) ? model_info.pricing.to_h : (model_info.pricing || {})
+          metadata_hash =
+            if model_info.metadata.respond_to?(:to_h)
+              model_info.metadata.to_h
+            else
+              model_info.metadata || {}
+            end
+
+          model_data = {
+            model_id: model_info.id, # RubyLLM::Model::Info uses 'id' not 'model_id'
+            name: model_info.name,
+            provider: model_info.provider,
+            family: model_info.family,
+            model_created_at: model_info.created_at,
+            context_window: model_info.context_window,
+            max_output_tokens: model_info.max_output_tokens,
+            knowledge_cutoff: model_info.knowledge_cutoff,
+            modalities: jsonb_or_string(modalities_hash),
+            capabilities: jsonb_or_string(capabilities_array),
+            pricing: jsonb_or_string(pricing_hash),
+            metadata: jsonb_or_string(metadata_hash)
+          }
+
+          existing = first(provider: model_info.provider, model_id: model_info.id)
+          if existing
+            existing.update(model_data)
+            existing
+          else
+            create(model_data)
+          end
+        end
+      end
+
+      def to_llm
+        parse_jsonb = lambda do |value|
+          return nil if value.nil?
+
+          value.is_a?(String) ? ::JSON.parse(value) : value
+        end
+
+        deep_symbolize = lambda do |obj|
+          case obj
+          when Hash
+            obj.transform_keys(&:to_sym).transform_values { |v| deep_symbolize.call(v) }
+          when Array
+            obj.map { |item| deep_symbolize.call(item) }
+          else
+            obj
+          end
+        end
+
+        parsed_pricing = parse_jsonb.call(pricing)
+        symbolized_pricing = parsed_pricing ? deep_symbolize.call(parsed_pricing) : nil
+
+        ::RubyLLM::Model::Info.new(
+          id: model_id, # RubyLLM::Model::Info expects 'id' not 'model_id'
+          name: name,
+          provider: provider,
+          family: family,
+          created_at: model_created_at,
+          context_window: context_window,
+          max_output_tokens: max_output_tokens,
+          knowledge_cutoff: knowledge_cutoff,
+          modalities: parse_jsonb.call(modalities),
+          capabilities: parse_jsonb.call(capabilities),
+          pricing: symbolized_pricing,
+          metadata: parse_jsonb.call(metadata)
+        )
+      end
+
+      def supports?(capability)
+        return false unless capabilities
+
+        caps_array = capabilities.is_a?(String) ? ::JSON.parse(capabilities) : capabilities
+
+        caps_array.include?(capability.to_s)
+      end
+
+      def supports_vision?
+        supports?('vision')
+      end
+
+      def supports_functions?
+        supports?('function_calling') || supports?('tools')
+      end
+
+      def input_price_per_million
+        return nil unless pricing
+
+        price_hash = pricing.is_a?(String) ? JSON.parse(pricing) : pricing
+
+        if price_hash.dig('text_tokens', 'standard', 'input_per_million')
+          price_hash.dig('text_tokens', 'standard', 'input_per_million').to_f
+        elsif price_hash.dig('text_tokens', 'input')
+          price_hash.dig('text_tokens', 'input').to_f
+        elsif price_hash['input'] || price_hash[:input]
+          (price_hash['input'] || price_hash[:input]).to_f
+        end
+      end
+
+      def output_price_per_million
+        return nil unless pricing
+
+        price_hash = pricing.is_a?(String) ? JSON.parse(pricing) : pricing
+
+        if price_hash.dig('text_tokens', 'standard', 'output_per_million')
+          price_hash.dig('text_tokens', 'standard', 'output_per_million').to_f
+        elsif price_hash.dig('text_tokens', 'output')
+          price_hash.dig('text_tokens', 'output').to_f
+        elsif price_hash['output'] || price_hash[:output]
+          (price_hash['output'] || price_hash[:output]).to_f
+        end
+      end
+
+      def type
+        to_llm.type
+      end
+
+      def function_calling?
+        to_llm.function_calling?
+      end
+
+      def structured_output?
+        to_llm.structured_output?
+      end
+
+      def batch?
+        to_llm.batch?
+      end
+
+      def reasoning?
+        to_llm.reasoning?
+      end
+
+      def citations?
+        to_llm.citations?
+      end
+
+      def streaming?
+        to_llm.streaming?
+      end
+    end
+  end
+end

data/lib/ruby_llm/sequel/tool_call_methods.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+module RubyLLM
+  module Sequel
+    module ToolCallMethods
+      def to_llm
+        ::RubyLLM::ToolCall.new(
+          id: tool_call_id,
+          name: name,
+          arguments: parse_arguments
+        )
+      end
+
+      def result_message
+        message_association.chat_association.messages_association_dataset.where(tool_call_id: id).first
+      end
+
+      private
+
+      def parse_arguments
+        if defined?(::Sequel::Postgres::JSONBHash) && arguments.is_a?(::Sequel::Postgres::JSONBHash)
+          return arguments.to_h
+        end
+        return arguments if arguments.is_a?(Hash)
+        return {} if arguments.nil? || arguments.empty?
+
+        JSON.parse(arguments)
+      end
+    end
+  end
+end

data/lib/ruby_llm/sequel/version.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+module RubyLLM
+  module Sequel
+    VERSION = '0.1.0'
+  end
+end

data/lib/ruby_llm/sequel.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+require 'sequel'
+
+Sequel.extension :inflector
+Sequel::Model.plugin :validation_helpers
+
+require_relative 'sequel/version'
+require_relative 'sequel/model_methods'
+require_relative 'sequel/chat_methods'
+require_relative 'sequel/message_methods'
+require_relative 'sequel/tool_call_methods'
+require_relative '../sequel/plugins/ruby_llm'
+
+module RubyLLM
+  module Sequel
+    class UnsupportedFeatureError < StandardError; end
+  end
+end

data/lib/sequel/plugins/ruby_llm.rb

@@ -0,0 +1,204 @@
+# frozen_string_literal: true
+
+module Sequel
+  module Plugins
+    # RubyLLM plugin for Sequel models
+    # Provides acts_as_model, acts_as_chat, acts_as_message, and acts_as_tool_call methods
+    module RubyLLM
+      module ClassMethods
+        def acts_as_model(chats: :chats, chat_class: nil)
+          Plugins.inherited_instance_variables(singleton_class,
+                                               :@chats_association_name => nil,
+                                               :@chat_class => nil)
+
+          class << self
+            attr_accessor :chats_association_name, :chat_class
+          end
+          include ::RubyLLM::Sequel::ModelMethods
+
+          chat_class ||= chats.to_s.classify
+
+          self.chats_association_name = chats
+          self.chat_class = chat_class
+
+          one_to_many chats, class: chat_class, key: :model_id
+
+          define_method(:chats_association) do
+            send(self.class.chats_association_name)
+          end
+          define_method(:chats_association_dataset) do
+            public_send(:"#{self.class.chats_association_name}_dataset")
+          end
+
+          define_method(:validate) do
+            super()
+            validates_presence %i[model_id provider name]
+            validates_unique %i[model_id provider]
+          end
+        end
+
+        def acts_as_chat(messages: :messages, message_class: nil, model: :llm_model, model_class: 'Model',
+                         model_key: :model_id)
+          Plugins.inherited_instance_variables(singleton_class,
+                                               :@messages_association_name => nil,
+                                               :@message_class => nil,
+                                               :@model_association_name => nil,
+                                               :@model_class => nil)
+
+          class << self
+            attr_accessor :messages_association_name, :message_class,
+                          :model_association_name, :model_class
+          end
+
+          include ::RubyLLM::Sequel::ChatMethods
+
+          message_class ||= messages.to_s
+          model_class ||= model.to_s
+
+          self.messages_association_name = messages
+          self.message_class = message_class.classify
+          self.model_association_name = model
+          self.model_class = model_class.classify
+
+          define_method(:messages_association) { public_send(self.class.messages_association_name) }
+          define_method(:messages_association_dataset) do
+            public_send(:"#{self.class.messages_association_name}_dataset")
+          end
+
+          define_method(:model_association) { public_send(self.class.model_association_name) }
+          define_method(:model_association_dataset) do
+            public_send(:"#{self.class.model_association_name}_dataset")
+          end
+
+          association_options = { class: model_class }
+          association_options[:key] = model_key if model_key
+          many_to_one model, **association_options
+
+          one_to_many messages, class: message_class, key: :"#{table_name.to_s.singularize}_id",
+                                order: ::Sequel.qualify(message_class.constantize.table_name.to_sym, :created_at).asc
+        end
+
+        def acts_as_message(chat: :chat, chat_class: nil, tool_calls: :tool_calls,
+                            tool_call_class: nil, model: :llm_model, model_class: 'Model',
+                            model_key: :model_id, touch_chat: false)
+          Plugins.inherited_instance_variables(singleton_class,
+                                               :@chat_association_name => nil,
+                                               :@chat_class => nil,
+                                               :@tool_calls_association_name => nil,
+                                               :@tool_call_class => nil,
+                                               :@model_association_name => nil,
+                                               :@model_class => nil)
+
+          class << self
+            attr_accessor :chat_association_name, :chat_class,
+                          :tool_calls_association_name, :tool_call_class,
+                          :model_association_name, :model_class
+          end
+
+          include ::RubyLLM::Sequel::MessageMethods
+
+          chat_class ||= chat.to_s.classify
+          tool_call_class ||= tool_calls.to_s.singularize.classify
+          model_class ||= model.to_s.classify
+
+          self.chat_association_name = chat
+          self.chat_class = chat_class
+          self.tool_calls_association_name = tool_calls
+          self.tool_call_class = tool_call_class
+          self.model_association_name = model
+          self.model_class = model_class
+
+          many_to_one chat, class: chat_class
+
+          model_association_options = { class: model_class }
+          model_association_options[:key] = model_key if model_key
+          many_to_one model, **model_association_options
+
+          many_to_one :parent_tool_call, class: tool_call_class, key: :tool_call_id
+          one_to_many tool_calls, class: tool_call_class, key: :"#{table_name.to_s.singularize}_id"
+
+          plugin :touch, column: :updated_at if touch_chat
+
+          define_method(:validate) do
+            super()
+            validates_presence [:role]
+          end
+
+          define_method(:chat_association) do
+            public_send(self.class.chat_association_name)
+          end
+          define_method(:chat_association_dataset) do
+            public_send(:"#{self.class.chat_association_name}_dataset")
+          end
+
+          define_method(:tool_calls_association) do
+            public_send(self.class.tool_calls_association_name)
+          end
+          define_method(:tool_calls_association_dataset) do
+            public_send(:"#{self.class.tool_calls_association_name}_dataset")
+          end
+
+          define_method(:model_association) do
+            public_send(self.class.model_association_name)
+          end
+          define_method(:model_association_dataset) do
+            public_send(:"#{self.class.model_association_name}_dataset")
+          end
+
+          return if chat == :chat
+
+          alias_method :chat, chat
+        end
+
+        def acts_as_tool_call(message: :message, message_class: nil, result: :result, result_class: nil)
+          Plugins.inherited_instance_variables(singleton_class,
+                                               :@message_association_name => nil,
+                                               :@message_class => nil,
+                                               :@result_association_name => nil,
+                                               :@result_class => nil)
+
+          class << self
+            attr_accessor :message_association_name, :message_class,
+                          :result_association_name, :result_class
+          end
+
+          include ::RubyLLM::Sequel::ToolCallMethods
+
+          message_class ||= message.to_s.classify
+          result_class ||= message_class
+
+          self.message_association_name = message
+          self.message_class = message_class
+          self.result_association_name = result
+          self.result_class = result_class
+
+          plugin :association_dependencies
+
+          many_to_one message, class: message_class, key: :"#{message}_id", eager_loader_key: nil
+          one_to_one result, class: result_class, key: :tool_call_id, eager_loader_key: nil
+
+          add_association_dependencies result => :nullify
+
+          define_method(:validate) do
+            super()
+            validates_presence %i[tool_call_id name]
+            validates_unique [:tool_call_id]
+          end
+
+          define_method(:message_association) do
+            send(self.class.message_association_name)
+          end
+          define_method(:messages_association_dataset) do
+            send("#{self.class.message_association_name}_dataset")
+          end
+          define_method(:result_association) do
+            send(self.class.result_association_name)
+          end
+          define_method(:result_association_dataset) do
+            send("#{self.class.result_association_name}_dataset")
+          end
+        end
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,80 @@
+--- !ruby/object:Gem::Specification
+name: ruby_llm-sequel
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Julian Pasquale
+bindir: bin
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: ruby_llm
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.9'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.9'
+- !ruby/object:Gem::Dependency
+  name: sequel
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+description: Provides an adapter to integrate RubyLLM with Sequel ORM.
+email:
+- jpasquale@fu.do
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- LICENSE
+- README.md
+- lib/ruby_llm/sequel.rb
+- lib/ruby_llm/sequel/chat_methods.rb
+- lib/ruby_llm/sequel/message_methods.rb
+- lib/ruby_llm/sequel/model_methods.rb
+- lib/ruby_llm/sequel/tool_call_methods.rb
+- lib/ruby_llm/sequel/version.rb
+- lib/sequel/plugins/ruby_llm.rb
+homepage: https://gitlab.com/fudo/poc/ruby_llm-sequel
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://gitlab.com/fudo/poc/ruby_llm-sequel
+  source_code_uri: https://gitlab.com/fudo/poc/ruby_llm-sequel
+  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.7.2
+specification_version: 4
+summary: Sequel adapter for RubyLLM models.
+test_files: []