mokkku 0.0.3 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 04c69427e72610fa1be06c75060a255eb0ab45c64b3e1a65a5649aa2420552f7
-  data.tar.gz: efc88cd532cd54f28ae0dc4fb2c8cfba8fb0d3e674183494fdc47ef8adcfef15
+  metadata.gz: 9dd97441b73b6ba8795909cfc1115ff90ca6b63a16fe0a2e9ea00a3d75f44933
+  data.tar.gz: 79041506c510a3c642c8cf0e9b0db97117df43f6b8a3e09257dcd8e6ded08721
 SHA512:
-  metadata.gz: afc7eb5280d025becdd44e0bd65412645a15bea68812ccc3c09a9179f79b2327faed3937c326c5da56f484a29cafc65e2d1f7bdf98550fff3d8a40995698f864
-  data.tar.gz: e07d24c250a66e1550da05e37a4961c3d4ad0e389e905aa662ca514ac9b96af48c263d3c24fab1e5281884f267381c5fbe588d8c67cd1540981388e81d707e81
+  metadata.gz: aa603c3e83ee372173558dfeabbb0969f9eb6cd45a313c447555e71961446fa97416c9a563f54bd3f810a7e5aa984881a419d01208e163a166b59dbcacf0d186
+  data.tar.gz: 94d08b8ea75f384f78f473794a5862e614c53fa50388fee8b51bdc4283fdc869f81394e66d0873740fad795a007f119fb4231c6311d78c39c97e5848eee5a4ee

data/bin/mokkku

@@ -38,4 +38,18 @@ OptionParser.new do |opts|
   end
 end.parse!
 
+# Load Rails project
+begin
+  require File.join(base_dir, 'config', 'boot')
+  require File.join(base_dir, 'config', 'environment')
+rescue LoadError
+  puts "\nUh-oh! You must be in the root directory of a Rails project.\n"
+  raise
+end
+
+# Load Rails models
+Rails.application.eager_load! if defined?(Rails)
+# Rails 6 support
+Zeitwerk::Loader.eager_load_all if defined?(Zeitwerk)
+
 Mokkku::Cli::Runner.start(options)

data/lib/mokkku/generators/builder.rb

@@ -1,5 +1,6 @@
 require_relative 'gpt'
 require_relative 'claude'
+require_relative 'gemini'
 
 module Mokkku
   module Generators
@@ -12,6 +13,8 @@ module Mokkku
           Mokkku::Generators::Gpt
         when 'claude'
           Mokkku::Generators::Claude
+        when 'gemini'
+          Mokkku::Generators::Gemini
         else
           raise UnsupportedGenerator
         end

data/lib/mokkku/generators/gemini.rb

@@ -0,0 +1,23 @@
+require 'gemini-ai'
+
+module Mokkku
+  module Generators
+    class Gemini
+      def self.call(api_key:, prompt:)
+        client = ::Gemini.new(
+          credentials: {
+            service: 'generative-language-api',
+            api_key: api_key
+          },
+          options: { model: 'gemini-pro', server_sent_events: false }
+        )
+
+        result = client.generate_content({
+          contents: { role: 'user', parts: { text: prompt } }
+        })
+
+        result['candidates'].first['content']['parts'].first['text']
+      end
+    end
+  end
+end

data/lib/mokkku/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mokkku
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.5
 platform: ruby
 authors:
 - Paweł Dąbrowski
@@ -72,6 +72,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 0.3.0
+- !ruby/object:Gem::Dependency
+  name: gemini-ai
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.2'
 description: Generate mock data that is unique for your application
 email: contact@paweldabrowski.com
 executables:
@@ -86,6 +100,7 @@ files:
 - lib/mokkku/cli/validator.rb
 - lib/mokkku/generators/builder.rb
 - lib/mokkku/generators/claude.rb
+- lib/mokkku/generators/gemini.rb
 - lib/mokkku/generators/gpt.rb
 - lib/mokkku/mocked_object.rb
 - lib/mokkku/utils.rb