mokkku 0.0.7 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6bc74f6aa066c531790d02be4d8b461978dd586b0a8c3f4275b0e74d2d12d5c7
-  data.tar.gz: 630c84be5245471760c79f245caacc240b733cded99c11db2761fb9655d70a12
+  metadata.gz: 4cd54dda1bd58c2eb99881690a02c99b7962c018bdf07132122d5240d6885c2f
+  data.tar.gz: 5c61996856695c6f53b8015eaa9c7923bf48a067969e8a9f4337db82cfc58b54
 SHA512:
-  metadata.gz: f738b67f66cc8dac9d73104eda8f01152cd173ecba1787f8b6be2acecf9f9726552cdcf51d0ab91de36db91c925872e93e122bae5a39abecb7a0c47ebea9eabb
-  data.tar.gz: dd6b831f50e46b375dafa703eb165f6be1f4dd6097f50201f7fdef5ab40f6b7364b30371f94acd8a088afe615891ebc37079372508828d06ffc0500c817e5097
+  metadata.gz: 15a72e5e0e60cb7a2ef7b9febcc3dc17dfa01d3393358ee4a147654e2e915eafee3c7d55a9fb174b83433d2e724117fa79c4f33ca23d89d7be4b731b2ed2f0dc
+  data.tar.gz: 76acb0e6fba9015ee604d9f19eaf9834ef66c8d1a752cade074753f4d7d69f19f2f783e1d7846a2c81beb4c9cab4f93a9aa7d44d6b8017440af5141c33775d6d

data/lib/mokkku/cli/runner.rb

@@ -12,20 +12,25 @@ module Mokkku
         options[:models].each do |model_name|
           columns = Object.const_get(model_name).columns
           prompt = ::Mokkku::Cli::Prompt.generate(model_name, columns, options.fetch(:mocks_count, 10), options.fetch(:language, 'english'))
-          generator = ::Mokkku::Generators::Builder.call(
-            generator: options[:llm_model],
-            api_key: options[:llm_api_key],
-            prompt: prompt
-          )
 
-          yaml_mock = generator.call(api_key: options[:llm_api_key], prompt: prompt)
-          mock_file_name = model_name.to_s.gsub(/([a-z\d])([A-Z])/, '\1_\2').downcase
+          if options.key?(:llm_model) && options.key?(:llm_api_key)
+            generator = ::Mokkku::Generators::Builder.call(
+              generator: options[:llm_model],
+              api_key: options[:llm_api_key],
+              prompt: prompt
+            )
 
-          puts "Write mocks for #{model_name} model in #{mock_file_name}.yml"
-          File.write(
-            File.join(options.fetch(:mocks_path, './spec/mocks'), "#{mock_file_name}.yml"),
-            ::Mokkku::YamlSanitizer.sanitize(yaml_mock)
-          )
+            yaml_mock = generator.call(api_key: options[:llm_api_key], prompt: prompt)
+            mock_file_name = model_name.to_s.gsub(/([a-z\d])([A-Z])/, '\1_\2').downcase
+
+            puts "Write mocks for #{model_name} model in #{mock_file_name}.yml"
+            File.write(
+              File.join(options.fetch(:mocks_path, './spec/mocks'), "#{mock_file_name}.yml"),
+              ::Mokkku::YamlSanitizer.sanitize(yaml_mock)
+            )
+          else
+            puts "Prompt for #{model_name} model: \n\n#{prompt}"
+          end
         end
 
         puts 'Done!'

data/lib/mokkku/cli/validator.rb

@@ -7,8 +7,8 @@ module Mokkku
 
       def self.validate_options!(options)
         raise ModelsNotProvided if options.fetch(:models, []).empty?
-        raise InvalidLlmModel unless %w[gpt gemini claude].include?(options[:llm_model].to_s.downcase)
-        raise LlmApiKeyNotProvided if options[:llm_api_key].to_s == ''
+        raise InvalidLlmModel if options.key?(:llm_model) && !%w[gpt gemini claude].include?(options[:llm_model].to_s.downcase)
+        raise LlmApiKeyNotProvided if options[:llm_api_key].to_s == '' && options.key?(:llm_model)
       end
     end
   end

data/lib/mokkku/mocked_object.rb

@@ -1,6 +1,10 @@
 module Mokkku
   class MockedObject
-    def initialize(mocks)
+    attr_reader :mocked_class
+    attr_accessor :selected_object
+
+    def initialize(mocked_class, mocks)
+      @mocked_class = mocked_class
       @mocks = mocks
       @selected_object = nil
     end
@@ -18,12 +22,7 @@ module Mokkku
     end
 
     def method_missing(method_name, *args, &block)
-      if @selected_object.nil?
-        @selected_object = mocked_objects.sample(random: Mokkku::Random)
-        @selected_object.public_send(method_name)
-      else
-        @selected_object.public_send(method_name)
-      end
+      @selected_object.public_send(method_name)
     end
   end
 end

data/lib/mokkku/uniq_utils.rb

@@ -0,0 +1,61 @@
+require 'set'
+
+module Mokkku
+  class UniqueUtils
+    RetryLimitExceeded = Class.new(StandardError)
+
+    class << self
+      def add_instance(generator, max_retries)
+        instances[generator.mocked_class] ||= Mokkku::UniqueUtils.new(generator, max_retries)
+      end
+
+      def instances
+        Thread.current[:mokkku_unique_utils] ||= {}
+      end
+
+      def clear
+        instances.each_value(&:clear)
+        instances.clear
+      end
+    end
+
+    def initialize(generator, max_retries)
+      @generator = generator
+      @max_retries = max_retries
+    end
+
+    def clear
+      previous_results.clear
+    end
+
+    private
+
+    def method_missing(name, *args, **kwargs)
+      if @generator.selected_object.nil? && @generator.send(:mocked_objects).first.to_h.keys.include?(name)
+        @max_retries.times do
+          next_object = @generator.send(:mocked_objects).sample(random: Mokkku::Random)
+        
+          next if previous_results.include?(next_object)
+
+          previous_results << next_object
+          @generator.instance_variable_set(:@selected_object, next_object)
+          break
+        end
+
+        if @generator.selected_object.nil?
+          raise RetryLimitExceeded, "Retry limit exceeded for #{name}"
+        end
+      end
+
+      @generator.public_send(name, *args, **kwargs)
+    end
+
+    def respond_to_missing?(name, *args)
+      @generator.respond_to?(name, *args) || super
+    end
+
+    def previous_results
+      @previous_results ||= Set.new
+    end
+  end
+end

data/lib/mokkku/utils.rb

@@ -1,5 +1,6 @@
 require 'yaml'
 require_relative 'mocked_object'
+require_relative 'uniq_utils'
 
 module Mokkku
   module Utils
@@ -11,11 +12,12 @@ module Mokkku
         mock_path = File.join(Mokkku.configuration.mocks_path, "#{mock_file_name}.yml")
         data = File.read(mock_path)
         parsed_data = YAML.safe_load(data, symbolize_names: true)
-        mocked_object = Mokkku::MockedObject.new(parsed_data)
+        mocked_object = Mokkku::MockedObject.new(const_name.to_s, parsed_data)
+        unique_mocked_object = Mokkku::UniqueUtils.add_instance(mocked_object, 100)
 
-        const_set const_name, mocked_object
+        const_set const_name, unique_mocked_object
 
-        mocked_object
+        unique_mocked_object
       end
     end
   end

data/lib/mokkku/version.rb

@@ -3,7 +3,7 @@ module Mokkku
     module_function
 
     def to_s
-      '0.0.7'
+      '0.1.0'
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mokkku
 version: !ruby/object:Gem::Version
-  version: 0.0.7
+  version: 0.1.0
 platform: ruby
 authors:
 - Paweł Dąbrowski
@@ -103,6 +103,7 @@ files:
 - lib/mokkku/generators/gemini.rb
 - lib/mokkku/generators/gpt.rb
 - lib/mokkku/mocked_object.rb
+- lib/mokkku/uniq_utils.rb
 - lib/mokkku/utils.rb
 - lib/mokkku/version.rb
 - lib/mokkku/yaml_sanitizer.rb