mokkku 0.0.1 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: dea0bbaeb26c51558aa530f63a7a57ea1118d5d858c46b284b0aa0eefa43fe1c
-  data.tar.gz: 63c3ff3581a61bae6f77c9f20a11a56cd7bca662c4722f7284c8ac1523b0b4fb
+  metadata.gz: 04c69427e72610fa1be06c75060a255eb0ab45c64b3e1a65a5649aa2420552f7
+  data.tar.gz: efc88cd532cd54f28ae0dc4fb2c8cfba8fb0d3e674183494fdc47ef8adcfef15
 SHA512:
-  metadata.gz: dc4da5ff3401d84a7c6340da54caa95a334fb548b7d33eb7194487492276c202c61bf17dc9eea471d982ea134076d5f821de7d4d23f248929cdbf67fb7a67369
-  data.tar.gz: a8a65b0ce61a7317d43261ec44e83a47344524c4226fa8f6d78d69f27544d703b0f8b501fbb6001765ff555aec60b27f1b34d637de1960cdc8fe697ce2902f76
+  metadata.gz: afc7eb5280d025becdd44e0bd65412645a15bea68812ccc3c09a9179f79b2327faed3937c326c5da56f484a29cafc65e2d1f7bdf98550fff3d8a40995698f864
+  data.tar.gz: e07d24c250a66e1550da05e37a4961c3d4ad0e389e905aa662ca514ac9b96af48c263d3c24fab1e5281884f267381c5fbe588d8c67cd1540981388e81d707e81

data/bin/mokkku

@@ -0,0 +1,41 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require 'mokkku'
+require 'optionparser'
+
+options = {}
+
+OptionParser.new do |opts|
+  opts.banner = 'Usage: mokkku [options]'
+
+  opts.on('-m', '--model Model', 'Provide model name to generate mocks for') do |opt|
+    options[:models] = [opt]
+  end
+
+  opts.on('--models Model1,Model2', 'Provide model names to generate mocks for') do |opt|
+    options[:models] = opt.split(',')
+  end
+
+  opts.on('-a', '--api-key value', 'API key for LLM') do |opt|
+    options[:llm_api_key] = opt
+  end
+
+  opts.on('-llm', '--llm-model value', 'Type of LLM: GPT, Gemini, or Claude') do |opt|
+    options[:llm_model] = opt
+  end
+
+  opts.on('-p', '--mocks-path value', 'Location where files will be created, default to ./spec/mocks') do |opt|
+    options[:mocks_path] = opt
+  end
+
+  opts.on('-c', '--count value', 'Number of entries to generate in mock file, 10 by default') do |opt|
+    options[:mocks_count] = opt
+  end
+
+  opts.on('-l', '--language value', 'The language of mocks, english by default') do |opt|
+    options[:language] = opt
+  end
+end.parse!
+
+Mokkku::Cli::Runner.start(options)

data/lib/mokkku/cli/prompt.rb

@@ -0,0 +1,27 @@
+module Mokkku
+  module Cli
+    class Prompt
+      IGNORED_COLUMNS = %w[id created_at updated_at].freeze
+
+      def self.generate(model_name, columns, mocks_count, language)
+        columns_for_prompt = columns.reject { |col| IGNORED_COLUMNS.include?(col.name) || col.name.match?(/.*\_id$/) }
+                                    .inject({}) { |hash, col| hash[col.name] = col.sql_type_metadata.type; hash }
+
+        prompt_start = "Act like a virtual assistant that generates YAML file with the test data for a given Ruby on Rails model. Generate #{mocks_count.to_i} entries for the following model named #{model_name}:"
+
+        yaml_attributes = columns_for_prompt.map do |name, type|
+          "  #{name}: #{type}\n"
+        end
+
+        prompt_yaml_part = "- #{yaml_attributes.join}"
+        prompt_summary = 'Reply only with the content of the YAML file.'
+
+        if language != 'english'
+          prompt_summary += " Translate test data to #{language} language."
+        end
+
+        "#{prompt_start}\n\n#{prompt_yaml_part}\n\n#{prompt_summary}"
+      end
+    end
+  end
+end

data/lib/mokkku/cli/runner.rb

@@ -0,0 +1,35 @@
+require_relative 'validator'
+require_relative 'prompt'
+require_relative '../generators/builder'
+require_relative '../yaml_sanitizer'
+
+module Mokkku
+  module Cli
+    class Runner
+      def self.start(options = {})
+        ::Mokkku::Cli::Validator.validate_options!(options)
+
+        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
+
+          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)
+          )
+        end
+
+        puts 'Done!'
+      end
+    end
+  end
+end

data/lib/mokkku/cli/validator.rb

@@ -0,0 +1,15 @@
+module Mokkku
+  module Cli
+    module Validator
+      ModelsNotProvided = Class.new(StandardError)
+      InvalidLlmModel = Class.new(StandardError)
+      LlmApiKeyNotProvided = Class.new(StandardError)
+
+      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 == ''
+      end
+    end
+  end
+end

data/lib/mokkku/generators/builder.rb

@@ -0,0 +1,21 @@
+require_relative 'gpt'
+require_relative 'claude'
+
+module Mokkku
+  module Generators
+    class Builder
+      UnsupportedGenerator = Class.new(StandardError)
+
+      def self.call(generator:, api_key:, prompt:)
+        case generator
+        when 'gpt'
+          Mokkku::Generators::Gpt
+        when 'claude'
+          Mokkku::Generators::Claude
+        else
+          raise UnsupportedGenerator
+        end
+      end
+    end
+  end
+end

data/lib/mokkku/generators/claude.rb

@@ -0,0 +1,24 @@
+require 'anthropic'
+
+module Mokkku
+  module Generators
+    class Claude
+      def self.call(api_key:, prompt:)
+        client = Anthropic::Client.new(access_token: api_key)
+
+        response = client.messages(
+          parameters: {
+            model: 'claude-3-5-sonnet-20240620',
+            system: 'Respond only with code.',
+            messages: [
+              { 'role': 'user', 'content': prompt },
+            ],
+            max_tokens: 1_000
+          }
+        )
+
+        response['content'].first['text']
+      end
+    end
+  end
+end

data/lib/mokkku/generators/gpt.rb

@@ -0,0 +1,23 @@
+require 'openai'
+
+module Mokkku
+  module Generators
+    class Gpt
+      def self.call(api_key:, prompt:)
+        client = OpenAI::Client.new(access_token: api_key)
+
+        response = client.completions(parameters: {
+          prompt: prompt,
+          model: 'gpt-3.5-turbo-instruct',
+          temperature: 0,
+          max_tokens: 1_000,
+          top_p: 1.0,
+          frequency_penalty: 0.0,
+          presence_penalty: 0.0
+        })
+
+        response['choices'].first['text']
+      end
+    end
+  end
+end

data/lib/mokkku/version.rb

@@ -3,7 +3,7 @@ module Mokkku
     module_function
 
     def to_s
-      '0.0.1'
+      '0.0.3'
     end
   end
 end

data/lib/mokkku/yaml_sanitizer.rb

@@ -0,0 +1,13 @@
+module Mokkku
+  class YamlSanitizer
+    def self.sanitize(yaml_content)
+      # Some LLMs ignore the instructions and return YAML with additions
+      yaml_lines = yaml_content.split("\n")
+      yaml_lines.shift if yaml_lines.first.start_with?("`")
+      yaml_lines.pop if yaml_lines.last.start_with?("`")
+      yaml_lines.shift unless yaml_lines.first.strip.start_with?('-')
+
+      yaml_lines.join("\n")
+    end
+  end
+end

data/lib/mokkku.rb

@@ -1,4 +1,5 @@
 require_relative 'mokkku/utils'
+require_relative 'mokkku/cli/runner'
 
 module Mokkku
   extend Mokkku::Utils

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mokkku
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.3
 platform: ruby
 authors:
 - Paweł Dąbrowski
@@ -44,16 +44,53 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: ruby-openai
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 7.1.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 7.1.0
+- !ruby/object:Gem::Dependency
+  name: anthropic
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.3.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.3.0
 description: Generate mock data that is unique for your application
 email: contact@paweldabrowski.com
-executables: []
+executables:
+- mokkku
 extensions: []
 extra_rdoc_files: []
 files:
+- bin/mokkku
 - lib/mokkku.rb
+- lib/mokkku/cli/prompt.rb
+- lib/mokkku/cli/runner.rb
+- lib/mokkku/cli/validator.rb
+- lib/mokkku/generators/builder.rb
+- lib/mokkku/generators/claude.rb
+- lib/mokkku/generators/gpt.rb
 - lib/mokkku/mocked_object.rb
 - lib/mokkku/utils.rb
 - lib/mokkku/version.rb
+- lib/mokkku/yaml_sanitizer.rb
 homepage:
 licenses: []
 metadata: {}