great_axe 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1b003919a105be25d6ef0dbf50a2e831ebd3ae65542b96ad95c793c4f068580e
-  data.tar.gz: a009c9fec6d4a0c0e2c7179cb9127e4c4854622c55ac7aff135961df98dc2a18
+  metadata.gz: d270b833b6dfb6ce3a7c341830007bea1bcca75933a636ac496524573289e9c9
+  data.tar.gz: bab9c39186f5793eb424eced4f600dd898139cea69b22b68e6df01acdd798189
 SHA512:
-  metadata.gz: b8caefcb6eb71b358e8fc970e66370055d15d4d191532e4c1c30cf27edb14fad3c036914fbb241643174f7009a8d9a347abf03fabcb1f6eaa8c5cdecab65d6bd
-  data.tar.gz: a99a99b0f0c0e950e32a13e0567839789e0e03df19673e92e1d50acacb3895bf120adca231a485db0090c88f2a42e3a1988708bd767aa708bc3c6f79a5dce9db
+  metadata.gz: 6e74485966e9a1493c9e19273b8fd03029d1cdb308f786606fa1a61061b4d6df64467e263beb51fcd7cc32590b009cbf5a77dee25229bfa81bc50f4a7c0a4480
+  data.tar.gz: 311f400e26707d1760e003a8bfb4c46011d0abba1f6d6224ab7132f1e35ccacfb2541873de5d242af6ecac0f08937970e715f3658e03ed84e2a869f4e9a3ae22

data/lib/great_axe/ai_support/open_ai_helper.rb

@@ -66,6 +66,14 @@ module GreatAxe
           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'

data/lib/great_axe/version.rb

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

data/lib/great_axe.rb

@@ -9,6 +9,7 @@ module GreatAxe
     include GreatAxe::AiSupport
 
     desc 'setup', 'creates a ai_settings.yml file in the root of your project'
+
     def setup
       SettingsGenerator.start
     end
@@ -61,15 +62,29 @@ module GreatAxe
     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,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: great_axe
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - aguspe