mokkku 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 234a86d569711e52239d937a81d56f24f2b15df2fc70d3bdc534cb95775e1d7b
-  data.tar.gz: 2f25efb4d4351948b99991f698621cb9a44ba512146ecf2834bb60cf2c9db818
+  metadata.gz: 04c69427e72610fa1be06c75060a255eb0ab45c64b3e1a65a5649aa2420552f7
+  data.tar.gz: efc88cd532cd54f28ae0dc4fb2c8cfba8fb0d3e674183494fdc47ef8adcfef15
 SHA512:
-  metadata.gz: b4f7e2e15f31417967cd38fa3141197e8515b9e58fe867a79a5a8e0fbe826579002afb16676dfdaba0aa3fca8d3d75dac162ba516b14036b4d12c87800d0ebad
-  data.tar.gz: 9bf25b64bcc41d563c8b5f64f395b6d2325f42fb2599b5c2d16c6ed5810544f7ea461646e61b12d7b02b22436216680a1b50310b8a65dae25be37b6096447a4f
+  metadata.gz: afc7eb5280d025becdd44e0bd65412645a15bea68812ccc3c09a9179f79b2327faed3937c326c5da56f484a29cafc65e2d1f7bdf98550fff3d8a40995698f864
+  data.tar.gz: e07d24c250a66e1550da05e37a4961c3d4ad0e389e905aa662ca514ac9b96af48c263d3c24fab1e5281884f267381c5fbe588d8c67cd1540981388e81d707e81

data/lib/mokkku/cli/runner.rb

@@ -1,6 +1,7 @@
 require_relative 'validator'
 require_relative 'prompt'
 require_relative '../generators/builder'
+require_relative '../yaml_sanitizer'
 
 module Mokkku
   module Cli
@@ -21,7 +22,10 @@ module Mokkku
           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"), yaml_mock)
+          File.write(
+            File.join(options.fetch(:mocks_path, './spec/mocks'), "#{mock_file_name}.yml"),
+            ::Mokkku::YamlSanitizer.sanitize(yaml_mock)
+          )
         end
 
         puts 'Done!'

data/lib/mokkku/generators/builder.rb

@@ -1,4 +1,5 @@
 require_relative 'gpt'
+require_relative 'claude'
 
 module Mokkku
   module Generators
@@ -9,6 +10,8 @@ module Mokkku
         case generator
         when 'gpt'
           Mokkku::Generators::Gpt
+        when 'claude'
+          Mokkku::Generators::Claude
         else
           raise UnsupportedGenerator
         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/version.rb

@@ -3,7 +3,7 @@ module Mokkku
     module_function
 
     def to_s
-      '0.0.2'
+      '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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mokkku
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Paweł Dąbrowski
@@ -58,6 +58,20 @@ dependencies:
     - - "~>"
       - !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:
@@ -71,10 +85,12 @@ files:
 - 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: {}