great_axe 0.1.0 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '09a7644ba52342d037616585df1059a8d8de14e7dca0e3075375223353badd60'
-  data.tar.gz: 0beb4ef373ce10e7e0e0aa7928b6a5a27aeb8c033b29375858bd8beb04ced099
+  metadata.gz: 1b003919a105be25d6ef0dbf50a2e831ebd3ae65542b96ad95c793c4f068580e
+  data.tar.gz: a009c9fec6d4a0c0e2c7179cb9127e4c4854622c55ac7aff135961df98dc2a18
 SHA512:
-  metadata.gz: e0b911b97e8e068fb0eb32624e79cfe99c696bcb0b40d502069700b7070f40afaccb7047c0f4dc2b94f0cd7e80f9650db27a89330aefbbc40aef2793ae8d29af
-  data.tar.gz: 75f0be4ae5a3f2c6631534349c012ee791180705df2462c4c1c1ed3d8ec821ef4230061d5b86a3721e75ab002842a16c75078cddf44fce129976d1a4587be232
+  metadata.gz: b8caefcb6eb71b358e8fc970e66370055d15d4d191532e4c1c30cf27edb14fad3c036914fbb241643174f7009a8d9a347abf03fabcb1f6eaa8c5cdecab65d6bd
+  data.tar.gz: a99a99b0f0c0e950e32a13e0567839789e0e03df19673e92e1d50acacb3895bf120adca231a485db0090c88f2a42e3a1988708bd767aa708bc3c6f79a5dce9db

data/lib/great_axe/ai_support/open_ai_helper.rb

@@ -0,0 +1,102 @@
+require 'openai'
+require 'yaml'
+
+module GreatAxe
+  module AiSupport
+    module OpenAiHelper
+      class << self
+        def client
+          @client ||= create_client
+        end
+
+        def create_client
+          configure_client
+          OpenAI::Client.new
+        end
+
+        def configure_client
+          OpenAI.configure do |config|
+            config.access_token = ENV.fetch('OPENAI_ACCESS_TOKEN')
+            config.organization_id = ENV.fetch('OPENAI_ORGANIZATION_ID', nil)
+          end
+        end
+
+        def input(request)
+          client.chat(
+            parameters: {
+              model: model,
+              messages: [{ role: role, content: request }],
+              temperature: temperature
+            })
+        end
+
+        def create_file(options)
+          path, request, choice = options.values_at(:path, :request, :choice)
+          File.write(path, output(request: request, choice: choice))
+        end
+
+        def output(options)
+          request, choice = options.values_at(:request, :choice)
+          choice ||= 0
+          extract_text(input(request), 'choices', choice, 'message', 'content')
+        end
+
+        def edit_file(options)
+          path, request, choice = options.values_at(:path, :request, :choice)
+          content = File.read(path)
+          response = edit(content: content, request: request)
+          File.write(path, extract_text(response, 'choices', choice, 'text'))
+        end
+
+        def edit(options)
+          content, request = options.values_at(:content, :request)
+          client.edits(
+            parameters: {
+              model: @model,
+              input: content,
+              instruction: request
+            }
+          )
+        end
+
+        def update_setting(key, value)
+          raise 'Invalid key' unless %w[model temperature role].include?(key)
+
+          load_settings[key] = value
+          overwrite_settings
+        end
+
+        private
+
+        SETTINGS_PATH = './ai_settings.yml'
+
+        def model
+          @model ||= load_settings['model']
+        end
+
+        def temperature
+          @temperature ||= load_settings['temperature']
+          @temperature.is_a?(String) ? @temperature.to_f : @temperature
+        end
+
+        def role
+          @role ||= load_settings['role']
+        end
+
+        def extract_text(response, *keys)
+          response.dig(*keys)
+        end
+
+        def load_settings
+          return pp 'Please create a ai_settings.yml file' unless File.exist?(SETTINGS_PATH)
+
+          @load_settings ||= YAML.load_file(SETTINGS_PATH)
+        end
+
+        def overwrite_settings
+          File.open(SETTINGS_PATH, 'w') { |file| YAML.dump(load_settings, file) }
+        end
+      end
+    end
+  end
+end

data/lib/great_axe/ai_support/settings_generator.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+require 'thor'
+
+module GreatAxe
+  module AiSupport
+    class SettingsGenerator < Thor::Group
+      include Thor::Actions
+
+      def self.source_paths
+        base_path = File.dirname(__FILE__)
+        %W[#{base_path}/templates ]
+      end
+
+      def generate_settings
+        template('settings.tt', './ai_settings.yml')
+      end
+    end
+  end
+end

data/lib/great_axe/ai_support/templates/settings.tt

@@ -0,0 +1,3 @@
+model: gpt-4o
+temperature: 0.7
+role: user

data/lib/great_axe/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module GreatAxe
-  VERSION = '0.1.0'
+  VERSION = '0.1.2'
 end

data/lib/great_axe.rb

@@ -1,10 +1,18 @@
 # frozen_string_literal: true
 
 require 'thor'
-require_relative 'great_axe/open_ai_helper'
+require_relative 'great_axe/ai_support/open_ai_helper'
+require_relative 'great_axe/ai_support/settings_generator'
 
 module GreatAxe
-  class OpenAiCommands < Thor
+  class PluginCommands < Thor
+    include GreatAxe::AiSupport
+
+    desc 'setup', 'creates a ai_settings.yml file in the root of your project'
+    def setup
+      SettingsGenerator.start
+    end
+
     desc 'make [REQUEST]', 'Uses open AI to create a file or generate output'
     option :path,
            type: :string, required: false, desc: 'The path where your file will be created', aliases: '-p'
@@ -27,39 +35,41 @@ module GreatAxe
       end
     end
 
-    desc 'cucumber [NAME]', 'Creates feature and step files only using open ai'
-    option :prompt,
+    desc 'cucumber [PROMPT]', 'Creates feature and step files only using open ai'
+    option :path,
            type: :string,
-           required: true, desc: 'The prompt for open ai', aliases: '-p'
+           required: false, desc: 'The path where your steps will be created', aliases: '-p'
 
-    def cucumber(name)
-      feature_path = "features/#{name}.feature"
-      make(options[:prompt], feature_path)
-      prompt_step = "create cucumber steps for the following scenarios in ruby #{File.read(feature_path)}"
-      make(prompt_step, "features/step_definitions/#{name}_steps.rb")
+    def cucumber(prompt)
+      feature_path = "features/#{options[:path]}.feature" if options[:path]
+      feature_prompt = "create cucumber feature file in ruby for #{prompt}"
+      make(feature_prompt, feature_path)
+      prompt_step = "create ruby cucumber step definitions in ruby for #{prompt}"
+      step_path = "features/step_definitions/#{options[:path]}_steps.rb" if options[:path]
+      make(prompt_step, step_path)
     end
 
-    desc 'steps [NAME]', 'Creates a new step definitions file'
+    desc 'steps [PROMPT]', 'Creates a new step definitions file'
     option :path,
            type: :string,
            required: false, desc: 'The path where your steps will be created', aliases: '-p'
-    option :prompt,
-           type: :string, required: false,
-           desc: 'This will create the selected steps based on your prompt using open ai', aliases: '-pr'
-    option :input,
-           type: :string,
-           required: false, desc: 'It uses a file as input to create the steps', aliases: '-i'
 
-    def steps(name)
-      path = 'features/step_definitions'
-      input = options[:input]
-      if input
-        prompt = options[:prompt] || 'create cucumber steps for the following scenarios in ruby'
-        content = "#{prompt} #{File.read(input)}"
-        make(content, "#{path}/#{name}_steps.rb")
-      else
-        make(options[:open_ai], "#{path}/#{name}_steps.rb")
-      end
+    def steps(prompt)
+      prompt_step = "create ruby cucumber step definitions in ruby for #{prompt}"
+      file_path = File.expand_path(options[:path]) if options[:path]
+      make(prompt_step, file_path)
+    end
+
+    desc 'model [MODEL]', 'Sets the model for the open ai helper'
+    def model(model)
+      OpenAiHelper.update_setting('model', model)
+      pp "Model set to #{model}"
+    end
+
+    desc 'temperature [TEMPERATURE]', 'Sets the temperature for the open ai helper'
+    def temperature(temp)
+      OpenAiHelper.update_setting('temperature', temp)
+      pp "Temperature set to #{temp}"
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: great_axe
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.2
 platform: ruby
 authors:
 - aguspe
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-07-26 00:00:00.000000000 Z
+date: 2024-07-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-openai
@@ -51,7 +51,9 @@ files:
 - README.md
 - Rakefile
 - lib/great_axe.rb
-- lib/great_axe/open_ai_helper.rb
+- lib/great_axe/ai_support/open_ai_helper.rb
+- lib/great_axe/ai_support/settings_generator.rb
+- lib/great_axe/ai_support/templates/settings.tt
 - lib/great_axe/version.rb
 - sig/great_axe.rbs
 - sig/lib/great_axe/open_ai_helper.rbs

data/lib/great_axe/open_ai_helper.rb

@@ -1,66 +0,0 @@
-require 'openai'
-
-module OpenAiHelper
-  class << self
-    def client
-      @client ||= create_client
-    end
-
-    def create_client
-      configure_client
-      OpenAI::Client.new
-    end
-
-    def configure_client
-      OpenAI.configure do |config|
-        config.access_token = ENV.fetch('OPENAI_ACCESS_TOKEN')
-        config.organization_id = ENV.fetch('OPENAI_ORGANIZATION_ID', nil)
-      end
-    end
-
-    def input(request, model = 'gpt-3.5-turbo', temperature = 0.7)
-      client.chat(
-        parameters: {
-          model: model,
-          messages: [{ role: 'user', content: request }],
-          temperature: temperature
-        })
-    end
-
-    def create_file(options)
-      path, request, choice = options.values_at(:path, :request, :choice)
-      File.write(path, output(request: request, choice: choice))
-    end
-
-    def output(options)
-      request, choice = options.values_at(:request, :choice)
-      choice ||= 0
-      extract_text(input(request), 'choices', choice, 'message', 'content')
-    end
-
-    def edit_file(options)
-      path, request, choice = options.values_at(:path, :request, :choice)
-      content = File.read(path)
-      response = edit(content: content, request: request)
-      File.write(path, extract_text(response, 'choices', choice, 'text'))
-    end
-
-    def edit(options)
-      content, request, model = options.values_at(:content, :request, :model)
-      model ||= 'text-davinci-edit-001'
-      client.edits(
-        parameters: {
-          model: model,
-          input: content,
-          instruction: request
-        }
-      )
-    end
-
-    private
-
-    def extract_text(response, *keys)
-      response.dig(*keys)
-    end
-  end
-end