great_axe 0.1.1 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ef52211499f7510ab55e0422a50deedeec536b704f2eb28a8c8d15180ce4e87b
-  data.tar.gz: 0b50a6271a1d86fea00964414de14c188703c795e4e692dbbc2da3701b73214c
+  metadata.gz: d270b833b6dfb6ce3a7c341830007bea1bcca75933a636ac496524573289e9c9
+  data.tar.gz: bab9c39186f5793eb424eced4f600dd898139cea69b22b68e6df01acdd798189
 SHA512:
-  metadata.gz: 885c091e4159b2f5e874ce3ecb11cd4e457330cefcf9b80d278ce698c4678328b41943172df1189d7880a91d1054b5bcc73dda34b8d9c49c3a5e6c5c7a17608b
-  data.tar.gz: e04aad5cd8d1ef2441a340e0302cb9635cc410c3f9ed07430d78d64a335427c59e86a1bab4a4b46eed2c7919f3dfafb182b408cfc9375564861116bcc165cff0
+  metadata.gz: 6e74485966e9a1493c9e19273b8fd03029d1cdb308f786606fa1a61061b4d6df64467e263beb51fcd7cc32590b009cbf5a77dee25229bfa81bc50f4a7c0a4480
+  data.tar.gz: 311f400e26707d1760e003a8bfb4c46011d0abba1f6d6224ab7132f1e35ccacfb2541873de5d242af6ecac0f08937970e715f3658e03ed84e2a869f4e9a3ae22

data/lib/great_axe/ai_support/open_ai_helper.rb

@@ -0,0 +1,110 @@
+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
+
+        def list_models
+          client.models.list
+        end
+
+        def model_info(model)
+          client.models.retrieve(id: model)
+        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.1'
+  VERSION = '0.1.3'
 end

data/lib/great_axe.rb

@@ -1,10 +1,19 @@
 # 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 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 +36,55 @@ 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
+
+    desc 'list_models', 'Lists the available open ai models'
+
+    def list_models
+      pp OpenAiHelper.list_models
+    end
+
+    desc 'model_info [MODEL]', 'Gets information on a specific model'
+
+    def model_info(model)
+      pp OpenAiHelper.model_info(model)
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: great_axe
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.3
 platform: ruby
 authors:
 - aguspe
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-07-28 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