easy_ml 0.2.0.pre.rc83 → 0.2.0.pre.rc85

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: easy_ml
 version: !ruby/object:Gem::Version
-  version: 0.2.0.pre.rc83
+  version: 0.2.0.pre.rc85
 platform: ruby
 authors:
 - Brett Shollenberger
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-02-20 00:00:00.000000000 Z
+date: 2025-02-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -719,9 +719,6 @@ files:
 - lib/easy_ml/data/dataset_manager/writer/partitioned.rb
 - lib/easy_ml/data/dataset_manager/writer/partitioned/partition_reasons.rb
 - lib/easy_ml/data/date_converter.rb
-- lib/easy_ml/data/embeddings.rb
-- lib/easy_ml/data/embeddings/adapters.rb
-- lib/easy_ml/data/embeddings/compression.rb
 - lib/easy_ml/data/partition.rb
 - lib/easy_ml/data/partition/boundaries.rb
 - lib/easy_ml/data/polars_column.rb
@@ -802,8 +799,8 @@ files:
 - public/easy_ml/assets/.vite/manifest-assets.json
 - public/easy_ml/assets/.vite/manifest.json
 - public/easy_ml/assets/assets/Application-nnn_XLuL.css
-- public/easy_ml/assets/assets/entrypoints/Application.tsx-Bbf3mD_b.js
-- public/easy_ml/assets/assets/entrypoints/Application.tsx-Bbf3mD_b.js.map
+- public/easy_ml/assets/assets/entrypoints/Application.tsx-CD8voxfL.js
+- public/easy_ml/assets/assets/entrypoints/Application.tsx-CD8voxfL.js.map
 homepage: https://github.com/brettshollenberger/easy_ml
 licenses:
 - MIT

data/lib/easy_ml/data/embeddings/adapters.rb

@@ -1,56 +0,0 @@
-module EasyML
-  module Data
-    class Embeddings
-      class Adapters
-        attr_accessor :model, :config
-
-        ADAPTERS = {
-          anthropic: Langchain::LLM::Anthropic,
-          gemini: Langchain::LLM::GoogleGemini,
-          openai: Langchain::LLM::OpenAI,
-          ollama: Langchain::LLM::Ollama,
-        }
-
-        DEFAULTS = {
-          api_key: {
-            anthropic: ENV["ANTHROPIC_API_KEY"],
-            gemini: ENV["GEMINI_API_KEY"],
-            openai: ENV["OPENAI_API_KEY"],
-            ollama: ENV["OLLAMA_API_KEY"],
-          },
-        }
-
-        def initialize(model, config = {})
-          @model = model.to_sym
-          @config = config.symbolize_keys
-          apply_defaults
-        end
-
-        def embed(df, col)
-          pick
-          texts = df[col].to_a
-          df = df.with_column(
-            embeddings: adapter.embed(text: texts),
-          )
-        end
-
-        private
-
-        def pick
-          @adapter ||= ADAPTERS[@model].new(config)
-          self
-        end
-
-        def apply_defaults
-          @config = @config.deep_symbolize_keys
-
-          DEFAULTS.each do |k, v|
-            unless @config.key?(k)
-              @config[k] = v[@model]
-            end
-          end
-        end
-      end
-    end
-  end
-end

data/lib/easy_ml/data/embeddings/compression.rb

@@ -1 +0,0 @@
-

data/lib/easy_ml/data/embeddings.rb

@@ -1,43 +0,0 @@
-module EasyML
-  module Data
-    class Embeddings
-      COMPRESSION_DEFAULT = {
-        present: :balanced,
-      }
-
-      attr_reader :df, :column, :model, :adapter, :compression,
-                  :embeddings, :compressed_embeddings
-
-      def initialize(options = {})
-        @df = options[:df]
-        @column = options[:column]
-        @model = options[:model]
-        @config = options[:config] || {}
-        @compression = options[:compression] || COMPRESSION_DEFAULT
-      end
-
-      def create
-        embed
-        compress
-      end
-
-      def embed
-        @embeddings ||= adapter.embed(df, column)
-      end
-
-      def compress
-        @compressed_embeddings ||= compression_adapter.compress(embeddings)
-      end
-
-      private
-
-      def adapter
-        @adapter ||= EasyML::Data::Embeddings::Adapters.new(model, config)
-      end
-
-      def compression_adapter
-        @compression_adapter ||= EasyML::Data::Embeddings::Compression.new(compression)
-      end
-    end
-  end
-end