chatgpt2023 0.2.1 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4728e79c0d12f0dd5a5ce7a7970f507d6fd79471727fb205f7416b3326fe8985
-  data.tar.gz: 7ca136ca869d4908c9fa70827b4a2a549cff92633026c1c70e8843886c026b59
+  metadata.gz: 9e1847e43e89e75aec42261b773b2f46909463c0a05be57bb532ac2682421088
+  data.tar.gz: 751d498aeaaa3eb3b68d977ecbb415385d606282584afa9612fdaf03349f7e59
 SHA512:
-  metadata.gz: f403b18eec392d2b1958b1168536ad716f02ddcd7445a3c67fa46916193190ee920020a2902e63c00621135392f580a00e12948b6dfa56a40f5316dd42d591d2
-  data.tar.gz: bda567ca1b04f4e4350c475d62cc4576887c36d7762bae951dd918269f149419411eb76e309a6190ad5e34cea71db375dd55a2a0b6cddff86ab0495a74506faf
+  metadata.gz: 9c3f02bfa4ba351dbfde5d60bc0d74cb8970c46a11d5632ab79b221445549e075cc4abcb03b8a479d7912c919e4a8257239d4f8f117144ea0eede97f7a26df80
+  data.tar.gz: 5cbd01a39b78cf3b927ce5a9729c6aabbcf4ff98c5e4e02b90acad8a414102e3c5910887395837bcc76f678ebf5fbd9db094ae3003d80f3f3bd1b887930214d0

data/lib/chatgpt2023.rb

@@ -6,6 +6,7 @@ require 'net/http'
 require 'uri'
 require 'json'
 require 'down'
+require 'dynarex-daily'
 
 
 # description: 1st experiment at playing with the ChatGPT API.
@@ -24,6 +25,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 +48,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_completions(s, temperature: temperature, max_tokens: max_tokens)
-    raise ChatGpt2023Error, r[:error][:message].inspect if r.has_key? :error
+    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, 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 +108,29 @@ class ChatGpt2023
   end
   
   private
+  
+  def go_code(s, temperature: 0, max_tokens: 7, n: 1)
+
+    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)
+  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 +194,57 @@ 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
+
+class CGRecorder <  ChatGpt2023
+  
+  def initialize(apikey: nil, indexfile: 'cgindex.xml', 
+                 logfile: 'chatgpt.xml', debug: false)
+    
+    super(apikey: apikey, debug: debug)
+    @dx = DynarexDaily.new filename: logfile, fields: %i(prompt result), 
+        autosave: true, order: 'descending', debug: false
+    @index = Dynarex.new(indexfile, schema: 'entries[title]/entry(prompt, ' \
+      + 'tags)', order: 'descending', autosave: true)
+    @index.title = 'ChatGPT prompt log'
+    
+  end
+  
+  def code_completion(s, tags=nil, temperature: 1, max_tokens: 32)
+    
+    r = code_completions(s, temperature: temperature, max_tokens: max_tokens)\
+        .first[:text].strip
+    log(s, r, tags)
+    
+    return r
+    
+  end   
+
+  def completion(s, tags=nil, temperature: 1, max_tokens: 32)
+    
+    r = completions(s, temperature: temperature, max_tokens: max_tokens)\
+        .first[:text].strip
+    log(s, r, tags)
+    
+    return r
+    
   end
   
+  alias complete completion
+  alias ask completion  
+  
+  private
+  
+  def log(prompt, result, tags)
+    
+    @index.create({prompt: prompt, tags: tags})
+    @dx.create({prompt: prompt, result: result})    
+    
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: chatgpt2023
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.3.0
 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
@@ -58,6 +58,26 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 5.4.0
+- !ruby/object:Gem::Dependency
+  name: dynarex-daily
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.7'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.7.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.7'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.7.0
 description: 
 email: digital.robertson@gmail.com
 executables: []
@@ -87,5 +107,5 @@ requirements: []
 rubygems_version: 3.4.4
 signing_key: 
 specification_version: 4
-summary: 1st experiment at playing with the ChatGPT API
+summary: 'A ChatGPT API wrapper. #experimental'
 test_files: []