openai-term 0.1.3 → 1.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9ceff9ceafa5d50d99036675fc0d85415da18a56e80bb0d6c0b408d4c6ca5325
-  data.tar.gz: d3b401c41cbb07d81d00877e4425c69c4630c5ccecd3f82be4e1082cb50ee408
+  metadata.gz: cc2ef0acdef82bc39ba989179889cbeda29564a8e2d01b23f39ce1e502f43929
+  data.tar.gz: 95691aa7a8fe4a5c20973d2421fd82e0f64de72b81aa1af6eddb189e2d2a89b6
 SHA512:
-  metadata.gz: 2c2796bf28a6de76dc9bde4db90772a1a360fa68c86c64b46c09fbe398177457574db59b4b0641fbee2ae9298734cf6d97e0ed8cb42c0a79eac84a1d452915ae
-  data.tar.gz: 3c95325c3e3d4103fb2bcfe2b4390cc491e4b603c7398059a5eacca5eb8e8ea243cbd56cb2644c7a432e2dda2e0a5d6de9c237d7ff674c2652263a2cbeab8343
+  metadata.gz: 4d297e98de6a0b0f124dccd1d9f0db9a71c80d271a1dc3dc7eb336ef82539be668cb878e9cf91736d6083f6cefeb89141e5bc34d118708fa6abe998c1272c9f4
+  data.tar.gz: 4e5366d42797d2d6cba8357a0b1fbf067d469b3579d457675f2b99ce3590ee7b914f77885a5225562047cc7b996cb0363f108030f4a5a61c782fcaa4f3e97580

data/bin/openai

@@ -13,13 +13,8 @@ require "ruby/openai"
 
 def model
   c = %w(text-davinci-003 code-cushman-001 text-curie-001 text-ada-001)
-  m = @prompt.select("What AI model do you want to use?") do |menu|
-    menu.choice c[0], 0
-    menu.choice c[1], 1
-    menu.choice c[2], 2
-    menu.choice c[3], 3
-  end
-  return c[m]
+  m = @prompt.select("What AI model do you want to use? (see https://beta.openai.com/docs/models/codex for details)", c, cycle: true)
+  return m
 end
 
 # HANDLE COMMAND LINE OPTIONS
@@ -33,6 +28,7 @@ optparse = OptionParser.new do |opts|
   opts.on('-t', '--text text', 'The text to process')                                 { |t| @t = t }
   opts.on('-x', '--max max_tokens', 'Specify max number of words in response')        { |x| @x = x.to_i }
   opts.on('-m', '--model', 'The AI model to use (default = text-davinci-003i)')       { @m = model } 
+  opts.on('-i', '--image', 'Create an image with the text supplied by -t or -f')      { @i = true } 
   opts.on('-h', 'Display SHORT help text')                                            { puts opts; exit }
   opts.on('-v', '--version', 'Display the version number')                            { puts "Version: 0.1"; exit }
 end
@@ -57,21 +53,25 @@ else
   exit
 end
 
-# REQUEST RESPONSE
+# REQUEST AND PRINT RESPONSE
 client = OpenAI::Client.new(access_token: @ai)
 
 begin
-  response = client.completions( parameters: { model: @m, prompt: @q, max_tokens: @x })
+  if @i
+    response = client.images.generate(parameters: { prompt: @q })
+    puts "Image url:"
+    puts response.dig("data", 0, "url")
+  else
+    response = client.completions( parameters: { model: @m, prompt: @q, max_tokens: @x })
+    begin
+      output = response["choices"][0]["text"]
+    rescue
+      output = "No OpenAI response"
+    end
+    puts output.strip + "\n\n"
+  end
 rescue => error
   p error
 end
 
-#PRINT RESPONSE
-begin
-  output = response["choices"][0]["text"]
-rescue
-  output = "No OpenAI response"
-end
-puts output.strip + "\n\n"
-
 # vim: set sw=2 sts=2 et ft=ruby fdm=syntax fdn=2 fcs=fold\:\ :

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: openai-term
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 1.0.0
 platform: ruby
 authors:
 - Geir Isene
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-01-26 00:00:00.000000000 Z
+date: 2023-01-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-openai
@@ -41,8 +41,9 @@ dependencies:
 description: 'This is a pretty straight forward interface to OpenAI with the option
   to select the AI model and the maximum token length (number of maximum words in
   the AI''s response). You will use the -t option to supply the query to OpenAI or
-  the -f option to read the query from a text file instead. New in 0.1.3: Rescue error
-  on OpenAI connection.'
+  the -f option to read the query from a text file instead. New in 1.0.0: Added the
+  functionality to create images via the -i option (in conjunction with either the
+  -t or the -f option).'
 email: g@isene.com
 executables:
 - openai