chatgpt2023 0.2.2 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9ec634a55d06fe1dbbeec12535c076677953903146f57aad99bdb1090fdb1c5c
-  data.tar.gz: 53b6db1a22dabbc4d21948e9d257d93dae12814bf13ef98e6d795f45b9a14ab8
+  metadata.gz: 9b89dececa8f5b5da00c97e7e745168c9a0526481f61138a03d5d2b7ff7b4576
+  data.tar.gz: b8d72dc15a1a051493786a53d22ed3a0bf135234bddf5d7d72c1db77a264ed6c
 SHA512:
-  metadata.gz: 35a5f4aa64030d07af1b053a3adaeb827aaa2ad8d0b545df5cef516ab07d6574cf4715029481ba283dbe40a83cb8ed2b09738c1f916656818f42ec97c3e9a303
-  data.tar.gz: 96a6868c366028eacaf522b8d8a795034b56f473f0b3aab801b672d779f2de1b694ce295232f2dc2e9c4b2a49e02f55dd56e5a16ab3c537652b9b48ae3bc8a93
+  metadata.gz: 31f3e1e57bbd60a724dff54a8317bd8501d939f797d5dd1bc673b9814f581f313d475d2e43900757057e64e5ab2d6e8106cdf055719c16ae935895f9283a2d81
+  data.tar.gz: 96e1288e2a79d823d18aaeeb6716f7239c26bc956d1a807530cf64a78ead4a162055103256ab7f6352ef56218198b4de01f9e7e6cf2bc83ce3236729d7861862

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.
@@ -189,14 +190,130 @@ class ChatGpt2023
       use_ssl: uri.scheme == "https",
     }
 
-    response = Net::HTTP.start(uri.hostname, uri.port, req_options) do |http|
-      http.request(request)
-    end
+    attempts = 0
     
-    h = JSON.parse(response.body, symbolize_names: true)
+    begin
+    
+      attempts += 1
+      
+      response = Net::HTTP.start(uri.hostname, uri.port, req_options) do |http|
+        http.request(request)
+      end
+      
+      h = JSON.parse(response.body, symbolize_names: true)
+      
+      if h[:error] then
+        puts 'warning:' + h[:error][:message] 
+        sleep 5
+      end
+    
+    end while h.has_key?(:error) and attempts <= 5
+        
     raise ChatGpt2023Error, h[:error][:message].inspect if h.has_key? :error
     
     return h
   end
   
 end
+
+class CGRecorder <  ChatGpt2023
+  
+  attr_reader :index
+  
+  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)
+    title = 'ChatGPT prompt log'
+    @index.title = title    
+    
+  end
+  
+  def code_completion(s, tags=nil, temperature: 1, max_tokens: 2000)
+    
+    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: 1000)
+    
+    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
+
+class ChatAway
+  
+  # statement below used for debugging
+  #attr_reader :dx, :prompts
+  
+  def initialize(questions, apikey: nil, filepath: '/tmp/chatgpt', debug: false)
+    
+    @debug = debug
+    
+    FileUtils.mkdir_p filepath
+    idxfile = File.join(filepath, 'index.xml')
+    cgfile =  File.join(filepath, 'chatgpt.xml')
+    
+    @dx = questions.is_a?(Dynarex) ? questions : Dynarex.new(questions)
+    @chat = CGRecorder.new(apikey: apikey, indexfile: idxfile, logfile: cgfile)
+    @prompts = @chat.index.all.map(&:prompt)
+    
+  end
+  
+  def start()
+    
+    @dx.all.each do |rx|
+      
+        if @prompts.include?(rx.prompt) and rx.redo != 'true'
+          next 
+        end
+        
+        type = rx.type == 'code' ? :code_completion : :completion
+        
+        prompt = rx.prompt
+        
+        puts 'prompt: ' + prompt
+
+        attempts = 0
+        
+        begin
+          r = @chat.method(type).call prompt
+        rescue
+          puts 'Something not working! ' + ($!).inspect
+          sleep 2
+          attempts += 1
+          retry if attempts < 4 
+        end
+        
+        sleep 2
+        
+    end    
+    
+  end
+  
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: chatgpt2023
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 0.4.0
 platform: ruby
 authors:
 - James Robertson
@@ -36,7 +36,7 @@ cert_chain:
   6FQD1/GISew7VvxUJdptXeuVNIsdNKxvL3RpfLCuFsi1WXyJ4k3odRMTmS0kAfTy
   J4sZZW9RNfabTMQQY7DIs3tUAn6i+O0r9lo=
   -----END CERTIFICATE-----
-date: 2023-02-13 00:00:00.000000000 Z
+date: 2023-02-14 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: []