ragnar-cli 0.1.0.pre.3 → 0.1.0.pre.4

data/lib/ragnar/topic_modeling.rb

@@ -1,24 +1,27 @@
-# Main entry point for topic modeling functionality
-# Designed for future extraction into a separate gem
+# frozen_string_literal: true
 
-require_relative 'topic_modeling/topic'
-require_relative 'topic_modeling/term_extractor'
-require_relative 'topic_modeling/metrics'
-require_relative 'topic_modeling/topic_labeler'
-require_relative 'topic_modeling/engine'
+# Topic modeling wrapper that delegates to the Topical gem
+# This maintains backward compatibility while using the extracted library
+
+require 'topical'
 
 module Ragnar
   module TopicModeling
+    # Re-export Topical classes for backward compatibility
+    Topic = Topical::Topic
+    Engine = Topical::Engine
+    
+    # Re-export metrics module
+    Metrics = Topical::Metrics
     
     # Convenience method to create a new topic modeling engine
     def self.new(**options)
-      Engine.new(**options)
+      Topical::Engine.new(**options)
     end
     
     # Extract topics from embeddings and documents (simple interface)
     def self.extract(embeddings:, documents:, **options)
-      engine = Engine.new(**options)
-      engine.fit(embeddings: embeddings, documents: documents)
+      Topical.extract(embeddings: embeddings, documents: documents, **options)
     end
   end
 end

data/lib/ragnar/umap_processor.rb

@@ -1,4 +1,5 @@
 require 'json'
+require 'time'
 
 module Ragnar
   class UmapProcessor
@@ -196,10 +197,10 @@ module Ragnar
     end
     
     def apply(batch_size: 100)
-      # Load the trained UMAP model (reduced embeddings)
-      reduced_embeddings = load_model
+      # Load the trained UMAP model
+      umap_model = load_umap_model
       
-      puts "Applying saved UMAP embeddings to database..."
+      puts "Applying UMAP transformation to database documents..."
       
       # Get all embeddings from database
       all_docs = @database.get_embeddings
@@ -214,84 +215,95 @@ module Ragnar
       end
       
       puts "Found #{all_docs.size} documents in database"
-      puts "Loaded #{reduced_embeddings.size} reduced embeddings from model"
       
-      if all_docs.size != reduced_embeddings.size
-        puts "⚠️  Warning: Mismatch between database documents (#{all_docs.size}) and model embeddings (#{reduced_embeddings.size})"
-        puts "   This suggests the model was trained on a different dataset."
-        puts "   Please retrain the UMAP model after indexing all your documents."
-        return {
-          processed: 0,
-          skipped: 0,
-          errors: 1
-        }
-      end
-      
-      # Prepare updates - match document IDs to reduced embeddings
-      updates = all_docs.each_with_index.map do |doc, idx|
-        {
-          id: doc[:id],
-          reduced_embedding: reduced_embeddings[idx]
-        }
+      # Process in batches for memory efficiency
+      processed_count = 0
+      error_count = 0
+      skipped_count = 0
+      
+      all_docs.each_slice(batch_size) do |batch|
+        begin
+          # Extract embeddings
+          embeddings = batch.map { |d| d[:embedding] }
+          
+          # Validate embeddings
+          valid_indices = []
+          embeddings_to_transform = []
+          
+          embeddings.each_with_index do |emb, idx|
+            if emb.nil? || !emb.is_a?(Array) || emb.empty?
+              skipped_count += 1
+              next
+            end
+            
+            if emb.any? { |v| !v.is_a?(Numeric) || v.nan? || !v.finite? }
+              skipped_count += 1
+              next
+            end
+            
+            valid_indices << idx
+            embeddings_to_transform << emb
+          end
+          
+          next if embeddings_to_transform.empty?
+          
+          # Transform using the loaded UMAP model
+          reduced_embeddings = umap_model.transform(embeddings_to_transform)
+          
+          # Prepare updates for valid documents
+          updates = valid_indices.map.with_index do |batch_idx, transform_idx|
+            {
+              id: batch[batch_idx][:id],
+              reduced_embedding: reduced_embeddings[transform_idx]
+            }
+          end
+          
+          # Update database
+          @database.update_reduced_embeddings(updates)
+          processed_count += updates.size
+          
+          puts "  Processed batch: #{updates.size} documents transformed"
+        rescue => e
+          puts "  ⚠️  Error processing batch: #{e.message}"
+          error_count += batch.size
+        end
       end
       
-      puts "Updating database with reduced embeddings..."
-      @database.update_reduced_embeddings(updates)
+      puts "\nUMAP application complete:"
+      puts "  ✓ Processed: #{processed_count} documents"
+      puts "  ⚠️  Skipped: #{skipped_count} documents (invalid embeddings)" if skipped_count > 0
+      puts "  ❌ Errors: #{error_count} documents" if error_count > 0
       
       {
-        processed: updates.size,
-        skipped: 0,
-        errors: 0
+        processed: processed_count,
+        skipped: skipped_count,
+        errors: error_count
       }
     end
     
     private
     
-    def process_batch(docs)
-      # Extract embeddings
-      embeddings = docs.map { |d| d[:embedding] }
-      
-      # Transform using UMAP
-      # The transform method returns a 2D array where each row is a reduced embedding
-      reduced = @umap_model.transform(embeddings)
-      
-      # Prepare updates
-      updates = docs.each_with_index.map do |doc, idx|
-        {
-          id: doc[:id],
-          reduced_embedding: reduced[idx]
-        }
-      end
-      
-      # Update database
-      @database.update_reduced_embeddings(updates)
-    end
-    
     def save_model
-      return unless @umap_instance && @reduced_embeddings
+      return unless @umap_instance
       
-      # Save the trained UMAP model for transforming new queries
+      # Save the trained UMAP model for transforming new data
       @umap_instance.save_model(@model_path)
       puts "UMAP model saved to: #{@model_path}"
       
-      # Also cache the reduced embeddings separately for the apply method
-      embeddings_path = @model_path.sub(/\.bin$/, '_embeddings.json')
-      ClusterKit::Dimensionality::UMAP.save_data(@reduced_embeddings, embeddings_path)
-      puts "Reduced embeddings cached to: #{embeddings_path}"
-    end
-    
-    def load_model
-      return @reduced_embeddings if @reduced_embeddings
-      
-      # For the apply method, we need the pre-computed embeddings
-      embeddings_path = @model_path.sub(/\.bin$/, '_embeddings.json')
-      unless File.exist?(embeddings_path)
-        raise "Cached embeddings not found at #{embeddings_path}. Please train a model first."
+      # Save metadata about the training if we have params
+      if @model_params
+        metadata_path = @model_path.sub(/\.bin$/, '_metadata.json')
+        metadata = {
+          trained_at: Time.now.iso8601,
+          n_components: @model_params[:n_components],
+          n_neighbors: @model_params[:n_neighbors],
+          min_dist: @model_params[:min_dist],
+          document_count: @database.get_embeddings.size,
+          model_version: 2  # Version 2: proper transform-based approach
+        }
+        File.write(metadata_path, JSON.pretty_generate(metadata))
+        puts "Model metadata saved to: #{metadata_path}"
       end
-      
-      @reduced_embeddings = ClusterKit::Dimensionality::UMAP.load_data(embeddings_path)
-      puts "Cached embeddings loaded from: #{embeddings_path}"
-      @reduced_embeddings
     end
     
     def load_umap_model

data/lib/ragnar/umap_transform_service.rb

@@ -1,124 +1,205 @@
+require 'json'
 require 'clusterkit'
 
 module Ragnar
+  # Service for applying UMAP transformations to embeddings
+  # Separates transformation logic from training (UmapProcessor)
   class UmapTransformService
-    include Singleton
+    attr_reader :model_path, :database
     
-    def initialize
+    def initialize(model_path: "umap_model.bin", database:)
+      @model_path = model_path
+      @database = database
       @umap_model = nil
-      @model_path = "umap_model.bin"
+      @model_metadata = nil
     end
     
-    # Transform a query embedding to reduced space using saved UMAP model
-    def transform_query(query_embedding, model_path = nil)
-      # Use the real UMAP model's transform capability
-      model_path ||= @model_path
+    # Transform embeddings for specific documents
+    # @param document_ids [Array<Integer>] IDs of documents to transform
+    # @return [Hash] Results with :processed, :skipped, :errors counts
+    def transform_documents(document_ids)
+      return { processed: 0, skipped: 0, errors: 0 } if document_ids.empty?
       
-      # Load the model if not already loaded
-      load_model(model_path) unless @umap_model
+      load_model!
       
-      # Transform the query embedding using the trained UMAP model
-      # The transform method expects a 2D array (even for a single embedding)
-      result = @umap_model.transform([query_embedding])
+      # Fetch documents
+      documents = @database.get_documents_by_ids(document_ids)
       
-      # Return the first (and only) transformed embedding
-      result.first
-    rescue => e
-      # Fall back to k-NN approximation if model loading fails
-      puts "Warning: Could not use UMAP model for transform: #{e.message}"
-      puts "Falling back to k-NN approximation..."
-      knn_approximate_transform(query_embedding)
-    end
-    
-    # Check if we can do transforms
-    def model_available?(model_path = nil)
-      model_path ||= @model_path
+      if documents.empty?
+        return { processed: 0, skipped: 0, errors: 0 }
+      end
       
-      # First check if the actual UMAP model file exists
-      if File.exist?(model_path)
-        return true
+      # Extract and validate embeddings
+      valid_docs = []
+      embeddings_to_transform = []
+      skipped_count = 0
+      
+      documents.each do |doc|
+        emb = doc[:embedding]
+        
+        if emb.nil? || !emb.is_a?(Array) || emb.empty?
+          skipped_count += 1
+          next
+        end
+        
+        if emb.any? { |v| !v.is_a?(Numeric) || v.nan? || !v.finite? }
+          skipped_count += 1
+          next
+        end
+        
+        valid_docs << doc
+        embeddings_to_transform << emb
       end
       
-      # Fallback: check if the database has reduced embeddings for k-NN approximation
-      database = Database.new("./rag_database")
-      stats = database.get_stats
-      stats[:with_reduced_embeddings] > 0
+      return { processed: 0, skipped: skipped_count, errors: 0 } if embeddings_to_transform.empty?
+      
+      # Transform using UMAP
+      begin
+        reduced_embeddings = @umap_model.transform(embeddings_to_transform)
+        
+        # Prepare updates
+        updates = valid_docs.zip(reduced_embeddings).map do |doc, reduced_emb|
+          {
+            id: doc[:id],
+            reduced_embedding: reduced_emb,
+            umap_version: model_version
+          }
+        end
+        
+        # Update database
+        @database.update_reduced_embeddings(updates)
+        
+        { processed: updates.size, skipped: skipped_count, errors: 0 }
+      rescue => e
+        puts "Error transforming documents: #{e.message}"
+        { processed: 0, skipped: skipped_count, errors: valid_docs.size }
+      end
     end
     
-    private
-    
-    def load_model(model_path)
-      unless File.exist?(model_path)
-        raise "UMAP model not found at #{model_path}. Please train a model first."
+    # Transform a single query embedding
+    # @param embedding [Array<Numeric>] Query embedding to transform
+    # @return [Array<Float>, nil] Reduced embedding or nil if error
+    def transform_query(embedding)
+      return nil if embedding.nil? || !embedding.is_a?(Array) || embedding.empty?
+      
+      # Validate embedding
+      if embedding.any? { |v| !v.is_a?(Numeric) || v.nan? || !v.finite? }
+        puts "Warning: Invalid query embedding (contains NaN or Infinity)"
+        return nil
       end
       
-      @umap_model = ClusterKit::Dimensionality::UMAP.load_model(model_path)
-      puts "UMAP model loaded for query transformation"
+      load_model!
+      
+      begin
+        # Transform returns array of arrays, get first (and only) result
+        @umap_model.transform([embedding]).first
+      rescue => e
+        puts "Error transforming query: #{e.message}"
+        nil
+      end
     end
     
-    def knn_approximate_transform(query_embedding)
-      # Fallback k-NN approximation method
-      # Get database stats to know dimensions
-      database = Database.new("./rag_database")
-      stats = database.get_stats
-      
-      # If we don't have reduced embeddings, we can't transform
-      if stats[:with_reduced_embeddings] == 0
-        raise "No reduced embeddings available in database"
-      end
+    # Check if a UMAP model exists
+    # @return [Boolean] true if model file exists
+    def model_exists?
+      File.exist?(@model_path)
+    end
+    
+    # Get metadata about the trained model
+    # @return [Hash, nil] Model metadata or nil if not found
+    def model_metadata
+      return @model_metadata if @model_metadata
       
-      # Get all documents with their embeddings
-      all_docs = database.get_embeddings
+      metadata_path = @model_path.sub(/\.bin$/, '_metadata.json')
+      return nil unless File.exist?(metadata_path)
       
-      # Find k nearest neighbors in full embedding space
-      k = 5
-      neighbors = []
+      @model_metadata = JSON.parse(File.read(metadata_path), symbolize_names: true)
+    rescue => e
+      puts "Error loading model metadata: #{e.message}"
+      nil
+    end
+    
+    # Get the version of the current model
+    # @return [Integer] Model version (timestamp of file modification)
+    def model_version
+      return 0 unless File.exist?(@model_path)
+      File.mtime(@model_path).to_i
+    end
+    
+    # Check if model needs retraining based on staleness
+    # @return [Hash] Staleness info with :needs_retraining, :coverage_percentage
+    def check_model_staleness
+      return { needs_retraining: true, coverage_percentage: 0, reason: "No model exists" } unless model_exists?
       
-      all_docs.each_with_index do |doc, idx|
-        next unless doc[:embedding] && doc[:reduced_embedding]
-        
-        distance = euclidean_distance(query_embedding, doc[:embedding])
-        neighbors << { idx: idx, distance: distance, reduced: doc[:reduced_embedding] }
-      end
+      metadata = model_metadata
+      return { needs_retraining: true, coverage_percentage: 0, reason: "No metadata found" } unless metadata
       
-      # Sort by distance and take k nearest
-      neighbors.sort_by! { |n| n[:distance] }
-      k_nearest = neighbors.first(k)
+      trained_count = metadata[:document_count] || 0
+      current_count = @database.document_count
       
-      # Average the reduced embeddings of k nearest neighbors
-      # This is a simple approximation of the transform
-      if k_nearest.empty?
-        raise "No neighbors found for transform"
+      if current_count == 0
+        return { needs_retraining: false, coverage_percentage: 100, reason: "No documents" }
       end
       
-      reduced_dims = k_nearest.first[:reduced].size
-      averaged = Array.new(reduced_dims, 0.0)
+      coverage = (trained_count.to_f / current_count * 100).round(1)
+      staleness = 100 - coverage
+      
+      {
+        needs_retraining: staleness > 30,
+        coverage_percentage: coverage,
+        trained_documents: trained_count,
+        current_documents: current_count,
+        staleness_percentage: staleness,
+        reason: staleness > 30 ? "Model covers only #{coverage}% of documents" : "Model is up to date"
+      }
+    end
+    
+    private
+    
+    def load_model!
+      return if @umap_model
       
-      # Weighted average based on inverse distance
-      total_weight = 0.0
-      k_nearest.each do |neighbor|
-        # Use inverse distance as weight (closer = higher weight)
-        weight = 1.0 / (neighbor[:distance] + 0.001) # Add small epsilon to avoid division by zero
-        total_weight += weight
-        
-        neighbor[:reduced].each_with_index do |val, idx|
-          averaged[idx] += val * weight
-        end
+      unless File.exist?(@model_path)
+        raise "UMAP model not found at #{@model_path}. Please train a model first using 'ragnar train-umap'."
       end
       
-      # Normalize by total weight
-      averaged.map { |val| val / total_weight }
+      @umap_model = ClusterKit::Dimensionality::UMAP.load_model(@model_path)
     end
+  end
+  
+  # Singleton service for backwards compatibility
+  # This allows the old UmapTransformService.instance pattern to work
+  class UmapTransformServiceSingleton
+    include Singleton
     
-    def euclidean_distance(vec1, vec2)
-      return Float::INFINITY if vec1.size != vec2.size
-      
-      sum = 0.0
-      vec1.each_with_index do |val, idx|
-        diff = val - vec2[idx]
-        sum += diff * diff
+    def initialize
+      @database = Database.new(Config.instance.database_path)
+      @service = UmapTransformService.new(database: @database)
+    end
+    
+    def transform_query(embedding, model_path = nil)
+      if model_path && model_path != @service.model_path
+        # Create a new service with different model path
+        service = UmapTransformService.new(model_path: model_path, database: @database)
+        service.transform_query(embedding)
+      else
+        @service.transform_query(embedding)
       end
-      Math.sqrt(sum)
+    end
+    
+    def model_available?(model_path = nil)
+      if model_path
+        File.exist?(model_path)
+      else
+        @service.model_exists?
+      end
+    end
+  end
+  
+  # For backwards compatibility - old code uses UmapTransformService.instance
+  class << UmapTransformService
+    def instance
+      UmapTransformServiceSingleton.instance
     end
   end
 end

data/lib/ragnar/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Ragnar
-  VERSION = "0.1.0.pre.3"
+  VERSION = "0.1.0.pre.4"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ragnar-cli
 version: !ruby/object:Gem::Version
-  version: 0.1.0.pre.3
+  version: 0.1.0.pre.4
 platform: ruby
 authors:
 - Chris Petersen
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-08-23 00:00:00.000000000 Z
+date: 2025-09-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor
@@ -30,14 +30,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.2'
+        version: 1.2.3
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.2'
+        version: 1.2.3
 - !ruby/object:Gem::Dependency
   name: lancelot
   requirement: !ruby/object:Gem::Requirement
@@ -47,7 +47,7 @@ dependencies:
         version: '0.3'
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.3.2
+        version: 0.3.3
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
@@ -57,21 +57,27 @@ dependencies:
         version: '0.3'
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.3.2
+        version: 0.3.3
 - !ruby/object:Gem::Dependency
-  name: clusterkit
+  name: topical
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.1.0.pre.2
+        version: 0.1.0
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.1.1
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.1.0.pre.2
+        version: 0.1.0
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.1.1
 - !ruby/object:Gem::Dependency
   name: baran
   requirement: !ruby/object:Gem::Requirement
@@ -92,14 +98,20 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.1.0.pre.1
+        version: '0.1'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.1.2
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.1.0.pre.1
+        version: '0.1'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.1.2
 - !ruby/object:Gem::Dependency
   name: tty-progressbar
   requirement: !ruby/object:Gem::Requirement
@@ -114,6 +126,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '0.18'
+- !ruby/object:Gem::Dependency
+  name: thor-interactive
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.1.0.pre.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.1.0.pre.3
 - !ruby/object:Gem::Dependency
   name: rake
   requirement: !ruby/object:Gem::Requirement
@@ -186,6 +212,8 @@ files:
 - lib/ragnar.rb
 - lib/ragnar/chunker.rb
 - lib/ragnar/cli.rb
+- lib/ragnar/cli_visualization.rb
+- lib/ragnar/config.rb
 - lib/ragnar/context_repacker.rb
 - lib/ragnar/database.rb
 - lib/ragnar/embedder.rb
@@ -194,24 +222,17 @@ files:
 - lib/ragnar/query_processor.rb
 - lib/ragnar/query_rewriter.rb
 - lib/ragnar/topic_modeling.rb
-- lib/ragnar/topic_modeling/engine.rb
-- lib/ragnar/topic_modeling/labeling_strategies.rb
-- lib/ragnar/topic_modeling/llm_adapter.rb
-- lib/ragnar/topic_modeling/metrics.rb
-- lib/ragnar/topic_modeling/term_extractor.rb
-- lib/ragnar/topic_modeling/topic.rb
-- lib/ragnar/topic_modeling/topic_labeler.rb
 - lib/ragnar/umap_processor.rb
 - lib/ragnar/umap_transform_service.rb
 - lib/ragnar/version.rb
 - lib/ragnar_cli.rb
-homepage: https://github.com/cpetersen/ragnar
+homepage: https://github.com/scientist-labs/ragnar
 licenses:
 - MIT
 metadata:
-  homepage_uri: https://github.com/cpetersen/ragnar
-  source_code_uri: https://github.com/cpetersen/ragnar
-  changelog_uri: https://github.com/cpetersen/ragnar/blob/main/CHANGELOG.md
+  homepage_uri: https://github.com/scientist-labs/ragnar
+  source_code_uri: https://github.com/scientist-labs/ragnar
+  changelog_uri: https://github.com/scientist-labs/ragnar/blob/main/CHANGELOG.md
 post_install_message:
 rdoc_options: []
 require_paths: