chatgpt2023 0.2.1 → 0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4728e79c0d12f0dd5a5ce7a7970f507d6fd79471727fb205f7416b3326fe8985
-  data.tar.gz: 7ca136ca869d4908c9fa70827b4a2a549cff92633026c1c70e8843886c026b59
+  metadata.gz: 9ec634a55d06fe1dbbeec12535c076677953903146f57aad99bdb1090fdb1c5c
+  data.tar.gz: 53b6db1a22dabbc4d21948e9d257d93dae12814bf13ef98e6d795f45b9a14ab8
 SHA512:
-  metadata.gz: f403b18eec392d2b1958b1168536ad716f02ddcd7445a3c67fa46916193190ee920020a2902e63c00621135392f580a00e12948b6dfa56a40f5316dd42d591d2
-  data.tar.gz: bda567ca1b04f4e4350c475d62cc4576887c36d7762bae951dd918269f149419411eb76e309a6190ad5e34cea71db375dd55a2a0b6cddff86ab0495a74506faf
+  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'
@@ -46,18 +47,40 @@ class ChatGpt2023
     
   end
   
-  def completions(s, temperature: 0, max_tokens: 7)
+  # 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]
     
-    r = go_completions(s, temperature: temperature, max_tokens: max_tokens)
-    raise ChatGpt2023Error, r[:error][:message].inspect if r.has_key? :error
+  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, temperature: 0, max_tokens: 7)
-    go_completions(s, temperature: temperature, max_tokens: max_tokens)\
+  def completion(s, temperature: 1, max_tokens: 32)
+    completions(s, temperature: temperature, max_tokens: max_tokens)\
         .first[:text].strip
   end
   
@@ -84,14 +107,29 @@ class ChatGpt2023
   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)    
@@ -155,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.1
+  version: 0.2.2
 platform: ruby
 authors:
 - James Robertson
@@ -36,7 +36,7 @@ cert_chain:
   6FQD1/GISew7VvxUJdptXeuVNIsdNKxvL3RpfLCuFsi1WXyJ4k3odRMTmS0kAfTy
   J4sZZW9RNfabTMQQY7DIs3tUAn6i+O0r9lo=
   -----END CERTIFICATE-----
-date: 2023-02-07 00:00:00.000000000 Z
+date: 2023-02-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: down