chatgpt2023 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8f614928641b93d4fbc86a6df977e4bc52c42e66a46958cd88b0c086ef0432fa
-  data.tar.gz: 433beb2a5d8bab7ffe05fd01d615b3c32e0691aac2bebb8c121940f6290fbaff
+  metadata.gz: 88c1cfbb0ea81bad341b25450770d3f9cd5142047a76f9650ed165cfc4295941
+  data.tar.gz: f155c706875520f4ffb3d25549b7a8cbde8b1b6c88919eb6b9e85f252486b979
 SHA512:
-  metadata.gz: 8f21a74de8195a255865f8f60be43291c13170196dbb1c4fb4f6337118da47c3decc4aab4352f74de2ff16ead582b8565703f127c034f6257602e5ad786a7d52
-  data.tar.gz: 64ea34173dceb5d96f1dc98b2ba22d729c1376839363ebf8b88a957b436780c12c353ad1877c7a26ba6a23735109ffc251ab8079a63b5c37d68213cd92c5d112
+  metadata.gz: cb5037feeaeb3181994e6d669bbf0d97c01e3dbaf06ca1f47754e9b9ab136a79c877e5ffb4c477319658fd2bb8594940831278fa02309a0b887ffab8aa0b3692
+  data.tar.gz: 2c597fa1448e6552e5ddfe58ff008b42de82b3413f05f020df304e6ff3ace7df88890a3a2d25e687ec98de9c988eaf415515d6fecca2ae4b3f27868c0c9ea3fa

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.
@@ -35,38 +36,108 @@ 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 = self.go_completions(s, temperature: temperature, 
+                            max_tokens: max_tokens)
     r[:choices]
   end
   
   def completion(s)
-    self.completions(s).first[:text].strip
+    self.go_completions(s).first[:text].strip
   end
   
   alias complete completion
+  
+  def edits(s, s2)
+    r = self.go_edits(s, s2)
+  end
+  
+  def images(s)
+    self.go_images_generations(s)
+  end
+  
+  def image(s)
+    r = self.images(s)
+    img = r[:data].first[:url]    
+    Down.download(img)
+  end  
+  
+  alias imagine image
+  
+  def images_edit(s, image, mask: nil)
+    self.go_images_edits(s, image, mask: mask)
+  end
+  
+  private
+
+  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)    
+    
+  end
+  
+  def go_edits(s, s2)
 
-  def submit(promptx='Say this is a test', prompt: promptx, type: :completions)
+    h = {
+      "model" => 'text-davinci-edit-001',
+      "input" => s,
+      "instruction" => s2
+    }
     
-    uri = URI.parse(@apiurl)
+    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(uri, h)
+        
+    uri = URI.parse(@apiurl + '/' + uri)
     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.0
 platform: ruby
 authors:
 - James Robertson
@@ -37,7 +37,27 @@ cert_chain:
   J4sZZW9RNfabTMQQY7DIs3tUAn6i+O0r9lo=
   -----END CERTIFICATE-----
 date: 2023-01-25 00:00:00.000000000 Z
-dependencies: []
+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: []