chatgpt2023 0.1.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8f614928641b93d4fbc86a6df977e4bc52c42e66a46958cd88b0c086ef0432fa
-  data.tar.gz: 433beb2a5d8bab7ffe05fd01d615b3c32e0691aac2bebb8c121940f6290fbaff
+  metadata.gz: 4728e79c0d12f0dd5a5ce7a7970f507d6fd79471727fb205f7416b3326fe8985
+  data.tar.gz: 7ca136ca869d4908c9fa70827b4a2a549cff92633026c1c70e8843886c026b59
 SHA512:
-  metadata.gz: 8f21a74de8195a255865f8f60be43291c13170196dbb1c4fb4f6337118da47c3decc4aab4352f74de2ff16ead582b8565703f127c034f6257602e5ad786a7d52
-  data.tar.gz: 64ea34173dceb5d96f1dc98b2ba22d729c1376839363ebf8b88a957b436780c12c353ad1877c7a26ba6a23735109ffc251ab8079a63b5c37d68213cd92c5d112
+  metadata.gz: f403b18eec392d2b1958b1168536ad716f02ddcd7445a3c67fa46916193190ee920020a2902e63c00621135392f580a00e12948b6dfa56a40f5316dd42d591d2
+  data.tar.gz: bda567ca1b04f4e4350c475d62cc4576887c36d7762bae951dd918269f149419411eb76e309a6190ad5e34cea71db375dd55a2a0b6cddff86ab0495a74506faf

data/lib/chatgpt2023.rb

@@ -5,6 +5,7 @@
 require 'net/http'
 require 'uri'
 require 'json'
+require 'down'
 
 
 # description: 1st experiment at playing with the ChatGPT API.
@@ -31,42 +32,120 @@ require 'json'
 #  chat.completion 'Say this is a test'
 #  #=> This is indeed a test 
 
+class ChatGpt2023Error < Exception
+end
+
 class ChatGpt2023
   
   def initialize(apikey: nil, debug: false)
     
-    @apiurl = "https://api.openai.com/v1/completions"
+    @apiurl = "https://api.openai.com/v1"
+    
     raise 'You must supply an API key!' unless apikey
     @apikey, @debug = apikey, debug
     
   end
   
-  def completions(s)
-    r = self.submit(s)
+  def completions(s, temperature: 0, max_tokens: 7)
+    
+    r = go_completions(s, temperature: temperature, max_tokens: max_tokens)
+    raise ChatGpt2023Error, r[:error][:message].inspect if r.has_key? :error
+    
+    puts 'completions r: ' + r.inspect if @debug
     r[:choices]
+    
   end
   
-  def completion(s)
-    self.completions(s).first[:text].strip
+  def completion(s, temperature: 0, max_tokens: 7)
+    go_completions(s, temperature: temperature, max_tokens: max_tokens)\
+        .first[:text].strip
   end
   
   alias complete completion
+  alias ask completion
+  
+  def edits(s, s2)
+    r = go_edits(s, s2)
+  end
+  
+  def images(s)
+    go_images_generations(s)
+  end
+  
+  def image(s)
+    r = images(s)
+    Down.download(r[:data].first[:url] )
+  end  
+  
+  alias imagine image
+  
+  def images_edit(s, image, mask: nil)
+    go_images_edits(s, image, mask: mask)
+  end
+  
+  private
 
-  def submit(promptx='Say this is a test', prompt: promptx, type: :completions)
+  def go_completions(s, temperature: 0, max_tokens: 7)
+
+    h = {
+      "model" => 'text-davinci-003',
+      "prompt" => s,
+      "temperature" => temperature,
+      "max_tokens" => max_tokens
+    }    
+    
+    submit('completions', h)    
     
-    uri = URI.parse(@apiurl)
+  end
+  
+  def go_edits(s, s2)
+
+    h = {
+      "model" => 'text-davinci-edit-001',
+      "input" => s,
+      "instruction" => s2
+    }
+    
+    submit('edits', h)
+    
+  end
+  
+  def go_images_generations(s, n: 1, size: '1024x1024')
+    
+    h = {
+      "prompt" => s,
+      "n" => n,
+      "size" => size
+    }
+    
+    submit('images/generations', h)
+    
+  end
+  
+  def go_images_edits(s, image, mask: nil, n: 1, size: '1024x1024')
+    
+    h = {
+      "image" => image,
+      "prompt" => s,
+      "n" => n,
+      "size" => size
+    }
+    
+    h['mask'] = mask if mask
+    
+    submit('images/edits', h)
+    
+  end  
+  
+  def submit(uri2, h)
+        
+    uri = URI.parse(@apiurl + '/' + uri2)
     request = Net::HTTP::Post.new(uri)
     request.content_type = "application/json"
     request["Authorization"] = 'Bearer ' + @apikey
-    
-    model = {completions: 'text-davinci-003'}
-    
-    request.body = JSON.dump({
-      "model" => model[type],
-      "prompt" => prompt,
-      "temperature" => 0,
-      "max_tokens" => 7
-    })
+        
+    puts 'h: ' + h.inspect if @debug
+    request.body = JSON.dump(h)
 
     req_options = {
       use_ssl: uri.scheme == "https",

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: chatgpt2023
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.1
 platform: ruby
 authors:
 - James Robertson
@@ -36,8 +36,28 @@ cert_chain:
   6FQD1/GISew7VvxUJdptXeuVNIsdNKxvL3RpfLCuFsi1WXyJ4k3odRMTmS0kAfTy
   J4sZZW9RNfabTMQQY7DIs3tUAn6i+O0r9lo=
   -----END CERTIFICATE-----
-date: 2023-01-25 00:00:00.000000000 Z
-dependencies: []
+date: 2023-02-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: down
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.4'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 5.4.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.4'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 5.4.0
 description: 
 email: digital.robertson@gmail.com
 executables: []