easy_ml 0.2.0.pre.rc26 → 0.2.0.pre.rc28

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3f577a0c71824e4041031e46e9e32126930814b5a3f5e808a21d5c20475520b1
-  data.tar.gz: 44d57d7d320b24705dbe85acc645f8570125c441f7fd3634bbf66f821a37860d
+  metadata.gz: 759592a8ae36ec3898d0acf83ff58ef0b0da9fd3cc3cfc492ac151cf8cc22360
+  data.tar.gz: d24de00abea52b9fe4d2d911075924134f66315d2e6a8fb9d0b28cbbbd7236ee
 SHA512:
-  metadata.gz: ff805d949b60ec0504efcf4c3c105e655278c460e1f308a27cfa0adcb17bee170480f7b1e4489e050deb7877e48f8ee4d0e2252a593da9dbef77ca340071523b
-  data.tar.gz: 4b39bc1cd971a8c1b9d2fea4f66c7cf3e6068471dad233a2762812a89cf9fa74ae659a9ca2df72e6913fdc3518fbd033969eddab8b02c4f0b060bdc3470ed3e0
+  metadata.gz: f9743d17f8f429e86d7b5635f8dc72179fcd928f0a1cf394bc2b8ff750e7349a8383f4bd58e2d1e5b4ef96da670578034493c28fdd863b88a0022cddc87013eb
+  data.tar.gz: 22e5a0ebe5aaa8a53e8927b8caae83c5badfe1deea04885db5f2081ad0e30bfe3166279d0208b8b7363cfecf95a572afb3132a94217beda99bab9a1781090a68

data/app/controllers/easy_ml/predictions_controller.rb

@@ -0,0 +1,18 @@
+module EasyML
+  class PredictionsController < ApplicationController
+    skip_before_action :verify_authenticity_token, only: [:create]
+
+    def create
+      model_name = params[:model]
+      input = params[:input]
+
+      predictions = EasyML::Predict.predict(model_name, input)
+
+      render json: { prediction: predictions.first }
+    rescue ActiveRecord::RecordNotFound
+      render json: { error: "Model not found" }, status: :not_found
+    rescue StandardError => e
+      render json: { error: e.message }, status: :unprocessable_entity
+    end
+  end
+end

data/app/jobs/easy_ml/clean_job.rb

@@ -0,0 +1,9 @@
+module EasyML
+  class CleanJob < ApplicationJob
+    @queue = :easy_ml
+
+    def self.perform
+      EasyML::Cleaner.clean
+    end
+  end
+end

data/config/initializers/zhong.rb

@@ -0,0 +1,17 @@
+require "zhong"
+
+Zhong.setup do |config|
+  config.logger = Rails.logger
+  config.time_zone = "UTC"
+
+  # Use Redis to persist the scheduler state
+  config.redis = Redis.new(
+    url: ENV.fetch("REDIS_URL") { "redis://localhost:6379/0" },
+  )
+end
+
+Zhong.schedule do
+  every 1.hour, "cleanup" do
+    CleanJob.perform_later
+  end
+end

data/config/routes.rb

@@ -1,6 +1,13 @@
 EasyML::Engine.routes.draw do
   root to: "models#index"
   get "healthcheck", to: "health#up"
+
+  mount Zhong::Web, at: "/zhong"
+  mount Resque::Server.new, at: "/resque"
+
+  # Predictions API
+  resources :predictions, only: [:create]
+
   resources :models, as: :easy_ml_models do
     member do
       post :train

data/lib/easy_ml/engine.rb

@@ -49,7 +49,7 @@ module EasyML
       Polars.enable_string_cache
     end
 
-    if %w[db:migrate db:migrate:status db:setup db:drop].include?(ARGV.first)
+    if %w[db:migrate db:migrate:status db:setup db:drop assets:precompile].include?(ARGV.first)
       config.eager_load_paths = config.eager_load_paths.without(config.eager_load_paths.map(&:to_s).grep(/easy_ml/).map { |p| Pathname.new(p) })
     else
       config.after_initialize do

data/lib/easy_ml/version.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 module EasyML
-  VERSION = "0.2.0-rc26"
+  VERSION = "0.2.0-rc28"
 
   module Version
   end

data/lib/tasks/zhong.rake

@@ -0,0 +1,6 @@
+namespace :zhong do
+  desc "Run Zhong"
+  task start: :environment do
+    Zhong.start
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: easy_ml
 version: !ruby/object:Gem::Version
-  version: 0.2.0.pre.rc26
+  version: 0.2.0.pre.rc28
 platform: ruby
 authors:
 - Brett Shollenberger
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-01-17 00:00:00.000000000 Z
+date: 2025-01-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -447,6 +447,7 @@ files:
 - app/controllers/easy_ml/deploys_controller.rb
 - app/controllers/easy_ml/health_controller.rb
 - app/controllers/easy_ml/models_controller.rb
+- app/controllers/easy_ml/predictions_controller.rb
 - app/controllers/easy_ml/retraining_runs_controller.rb
 - app/controllers/easy_ml/settings_controller.rb
 - app/easy_ml/.vite/manifest-assets.json
@@ -520,6 +521,7 @@ files:
 - app/helpers/easy_ml/application_helper.rb
 - app/jobs/easy_ml/application_job.rb
 - app/jobs/easy_ml/batch_job.rb
+- app/jobs/easy_ml/clean_job.rb
 - app/jobs/easy_ml/compute_feature_job.rb
 - app/jobs/easy_ml/deploy_job.rb
 - app/jobs/easy_ml/finalize_feature_job.rb
@@ -590,6 +592,7 @@ files:
 - bin/vite
 - config/initializers/inflections.rb
 - config/initializers/resque.rb
+- config/initializers/zhong.rb
 - config/resque-pool.yml
 - config/routes.rb
 - config/spring.rb
@@ -662,6 +665,7 @@ files:
 - lib/easy_ml/version.rb
 - lib/tasks/resque.rake
 - lib/tasks/vite.rake
+- lib/tasks/zhong.rake
 - public/easy_ml/assets/.vite/manifest-assets.json
 - public/easy_ml/assets/.vite/manifest.json
 - public/easy_ml/assets/assets/Application-Cu7lNJmG.css