#!/usr/bin/env ruby

require 'time'
require 'term/ansicolor'
include Term::ANSIColor
require 'ollama'
include Ollama

base_url = ENV['OLLAMA_URL'] || 'http://%s' % ENV.fetch('OLLAMA_HOST')
ollama = Client.new(base_url:)
ollama.tags.models.each do |model|
  name, modified_at = model.name, Time.parse(model.modified_at)
  infobar.puts(
    "Updating model #{bold {name}} (last modified at #{modified_at.iso8601}):"
  )
  ollama.pull(name:)
end