Maybe/app/models/assistant/provided.rb

module Assistant::Provided
  extend ActiveSupport::Concern

  def get_model_provider(ai_model)
    registry.providers.find { |provider| provider.supports_model?(ai_model) }
  end

  private
    def registry
      @registry ||= Provider::Registry.for_concept(:llm)
    end
end