chatgpt2023 0.2.0 → 0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 88c1cfbb0ea81bad341b25450770d3f9cd5142047a76f9650ed165cfc4295941
-  data.tar.gz: f155c706875520f4ffb3d25549b7a8cbde8b1b6c88919eb6b9e85f252486b979
+  metadata.gz: 9ec634a55d06fe1dbbeec12535c076677953903146f57aad99bdb1090fdb1c5c
+  data.tar.gz: 53b6db1a22dabbc4d21948e9d257d93dae12814bf13ef98e6d795f45b9a14ab8
 SHA512:
-  metadata.gz: cb5037feeaeb3181994e6d669bbf0d97c01e3dbaf06ca1f47754e9b9ab136a79c877e5ffb4c477319658fd2bb8594940831278fa02309a0b887ffab8aa0b3692
-  data.tar.gz: 2c597fa1448e6552e5ddfe58ff008b42de82b3413f05f020df304e6ff3ace7df88890a3a2d25e687ec98de9c988eaf415515d6fecca2ae4b3f27868c0c9ea3fa
+  metadata.gz: 35a5f4aa64030d07af1b053a3adaeb827aaa2ad8d0b545df5cef516ab07d6574cf4715029481ba283dbe40a83cb8ed2b09738c1f916656818f42ec97c3e9a303
+  data.tar.gz: 96a6868c366028eacaf522b8d8a795034b56f473f0b3aab801b672d779f2de1b694ce295232f2dc2e9c4b2a49e02f55dd56e5a16ab3c537652b9b48ae3bc8a93

data/lib/chatgpt2023.rb

@@ -24,6 +24,7 @@ require 'down'
 #  -d '{"model": "text-davinci-003", "prompt": "Say this is a test", "temperature": 0, "max_tokens": 7}'
 
 # ChatGpt documentation: https://beta.openai.com/docs/introduction/overview
+# ChatGpt web page: https://chat.openai.com/chat
 
 # Usage:
 #  require 'chatgpt2023'
@@ -32,6 +33,9 @@ require 'down'
 #  chat.completion 'Say this is a test'
 #  #=> This is indeed a test 
 
+class ChatGpt2023Error < Exception
+end
+
 class ChatGpt2023
   
   def initialize(apikey: nil, debug: false)
@@ -43,47 +47,89 @@ class ChatGpt2023
     
   end
   
-  def completions(s, temperature: 0, max_tokens: 7)
-    r = self.go_completions(s, temperature: temperature, 
-                            max_tokens: max_tokens)
+  # Example
+  # c = ChatGpt2023.new(apikey: 'yourapikey')
+  # s = '
+  # # Ruby
+  # # Ask the user for their name and say "Hello"
+  # '
+  # r = c.code_completions s, temperature: 0.2
+  # puts r.first[:text]
+  
+  def code_completions(s, temperature: 1, max_tokens: 32, n: 1)
+    
+    r = go_code(s, temperature: temperature, 
+                       max_tokens: max_tokens, n: n)
+    puts 'code r: ' + r.inspect if @debug
+    r[:choices]
+    
+  end
+
+  def code_completion(s, temperature: 1, max_tokens: 32)
+    code_completions(s, temperature: temperature, max_tokens: max_tokens)\
+        .first[:text].strip
+  end  
+  
+  def completions(s, temperature: 1, max_tokens: 32, n: 1)
+    
+    r = go_completions(s, temperature: temperature, 
+                       max_tokens: max_tokens, n: n)
+    puts 'completions r: ' + r.inspect if @debug
     r[:choices]
+    
   end
   
-  def completion(s)
-    self.go_completions(s).first[:text].strip
+  def completion(s, temperature: 1, max_tokens: 32)
+    completions(s, temperature: temperature, max_tokens: max_tokens)\
+        .first[:text].strip
   end
   
   alias complete completion
+  alias ask completion
   
   def edits(s, s2)
-    r = self.go_edits(s, s2)
+    r = go_edits(s, s2)
   end
   
   def images(s)
-    self.go_images_generations(s)
+    go_images_generations(s)
   end
   
   def image(s)
-    r = self.images(s)
-    img = r[:data].first[:url]    
-    Down.download(img)
+    r = images(s)
+    Down.download(r[:data].first[:url] )
   end  
   
   alias imagine image
   
   def images_edit(s, image, mask: nil)
-    self.go_images_edits(s, image, mask: mask)
+    go_images_edits(s, image, mask: mask)
   end
   
   private
+  
+  def go_code(s, temperature: 0, max_tokens: 7, n: 1)
 
-  def go_completions(s, temperature: 0, max_tokens: 7)
+    h = {
+      "model" => 'code-davinci-002',
+      "prompt" => s,
+      "temperature" => temperature,
+      "max_tokens" => max_tokens,
+      "n" => n
+    }    
+    
+    submit('completions', h)    
+    
+  end  
+
+  def go_completions(s, temperature: 0, max_tokens: 7, n: 1)
 
     h = {
       "model" => 'text-davinci-003',
       "prompt" => s,
       "temperature" => temperature,
-      "max_tokens" => max_tokens
+      "max_tokens" => max_tokens,
+      "n" => n
     }    
     
     submit('completions', h)    
@@ -129,9 +175,9 @@ class ChatGpt2023
     
   end  
   
-  def submit(uri, h)
+  def submit(uri2, h)
         
-    uri = URI.parse(@apiurl + '/' + uri)
+    uri = URI.parse(@apiurl + '/' + uri2)
     request = Net::HTTP::Post.new(uri)
     request.content_type = "application/json"
     request["Authorization"] = 'Bearer ' + @apikey
@@ -147,7 +193,10 @@ class ChatGpt2023
       http.request(request)
     end
     
-    JSON.parse(response.body, symbolize_names: true)
+    h = JSON.parse(response.body, symbolize_names: true)
+    raise ChatGpt2023Error, h[:error][:message].inspect if h.has_key? :error
+    
+    return h
   end
   
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: chatgpt2023
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.2
 platform: ruby
 authors:
 - James Robertson
@@ -36,7 +36,7 @@ cert_chain:
   6FQD1/GISew7VvxUJdptXeuVNIsdNKxvL3RpfLCuFsi1WXyJ4k3odRMTmS0kAfTy
   J4sZZW9RNfabTMQQY7DIs3tUAn6i+O0r9lo=
   -----END CERTIFICATE-----
-date: 2023-01-25 00:00:00.000000000 Z
+date: 2023-02-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: down