chatgpt2023 0.4.2 → 0.5.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 79e85a14caa0d9f7554af7d2df3faf4356d906ffd32eddbeab97a0ce62588aa6
-  data.tar.gz: 7ed03e8ba40941b21431469711325c9eb20efd5c9374ecfb24e0cdd9d497888b
+  metadata.gz: 8f937d13ec5131003c4acf2b9c046cb56e4462bb72e89700cb56fb2c376c3846
+  data.tar.gz: e2252feb992e7c3538ae917d63abc6b4f4743e8faa328fff0ce36f1af221508e
 SHA512:
-  metadata.gz: '0297c587bab1025db945d899437ca0b19a820c2c172a01ea2129de95b09fd773f02bed19e098d8a3c34881c06b4ddfc301a3e38670bce6c35236300494281099'
-  data.tar.gz: f80fc62462f62eeee57b2ed50ff81a76de628fa27bd2733c82ac57b108da6ed5a2b14f6afce5a600353c97d3a3fc8ce574117341fec1b70f38bb29b1fa972f14
+  metadata.gz: 88b1657db3ac774bfae4df8166a2aa26ef520af5194a7ba1718aea6612a2a75350de457b8dabb150bda4e149a4851648f45eecfdc1dd6276794fdb1358c760a5
+  data.tar.gz: dd610445a564e33fde61cdedda227a843d6410457f90b8a85b3255e627462fb51cae5a3767d56e6346f277466df97053fd3f41155fba06d7fbbd34d4e125144c

data/lib/chatgpt2023.rb

@@ -39,15 +39,71 @@ end
 
 class ChatGpt2023
   
+  attr_accessor :assistant_recent
+  
   def initialize(apikey: nil, attempts: 1, debug: false)
     
     @apiurl = "https://api.openai.com/v1"
     
     raise 'You must supply an API key!' unless apikey
     @apikey, @attempts, @debug = apikey, attempts, debug
+    @assistant_recent = nil
     
   end
   
+  
+  # CURL example
+  # curl https://api.openai.com/v1/chat/completions \
+  # -H 'Content-Type: application/json' \
+  # -H 'Authorization: Bearer YOUR_API_KEY' \
+  # -d '{
+  # "model": "gpt-3.5-turbo",
+  # "messages": [{"role": "user", "content": "Hello!"}]
+  # }'
+  
+  # Ruby example
+
+  # require 'chatgpt2023'
+
+  # c = ChatGpt2023.new(apikey: YOUR_API_KEY)
+  # r = c.chat 'who is Burt Reynolds?'
+  # r2 = c.chat 'what age was he?'
+  # r3 = c.chat 'did he have family?'  
+  #
+  
+  def chat(s, temperature: 1, max_tokens: 3900)
+    r = chats(s, temperature: temperature, max_tokens: max_tokens)
+    return r if r.is_a?(Hash) 
+    return {text: r.first[:message][:content].strip}
+  end
+  
+  def chats(s=nil, messages: [], temperature: 1, max_tokens: 3900, n: 1)
+    
+    if @assistant_recent then
+      
+      # We limit what gets passed back because there should enough 
+      # information in the 1st 140 chars to infer the conversation context
+      #
+      @assistant_recent[:content].slice!(140..-1)
+      messages << @assistant_recent
+      max_tokens -= @assistant_recent[:content].split.length
+    end
+    
+    messages << {'role' => 'user', 'content' => s } if s
+    r = go_chat(messages, temperature: temperature, 
+                       max_tokens: max_tokens, n: n)        
+    
+    puts 'chat/completions r: ' + r.inspect if @debug
+    
+    if r[:error] then r
+      r
+    else
+      @assistant_recent = r[:choices].first[:message]
+      r[:choices]
+    end
+    
+  end  
+  
   # Example
   # c = ChatGpt2023.new(apikey: 'yourapikey')
   # s = '
@@ -56,7 +112,7 @@ class ChatGpt2023
   # '
   # 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, 
@@ -88,7 +144,7 @@ class ChatGpt2023
   end
   
   alias complete completion
-  alias ask completion
+  alias ask chat
   
   def edits(s, s2)
     r = go_edits(s, s2)
@@ -111,6 +167,20 @@ class ChatGpt2023
   
   private
   
+  def go_chat(messages=[], temperature: 0, max_tokens: 4096, n: 1)
+
+    h = {      
+      "model" => 'gpt-3.5-turbo',
+      "messages" => messages,
+      "temperature" => temperature,
+      "max_tokens" => max_tokens,
+      "n" => n
+    }    
+    
+    submit('chat/completions', h)    
+    
+  end  
+  
   def go_code(s, temperature: 0, max_tokens: 7, n: 1)
 
     h = {
@@ -227,7 +297,7 @@ class CGRecorder <  ChatGpt2023
 
     super(apikey: apikey, attempts: attempts, debug: debug)
     @dx = DynarexDaily.new filename: logfile, fields: %i(prompt result), 
-        autosave: true, order: 'descending', debug: false
+        autosave: true, order: 'descending', debug: debug
     @index = Dynarex.new(indexfile, schema: 'entries[title]/entry(prompt, ' \
       + 'tags)', order: 'descending', autosave: true)
     title = 'ChatGPT prompt log'
@@ -235,10 +305,19 @@ class CGRecorder <  ChatGpt2023
     
   end
   
+  def chat(s, tags=nil, temperature: 1, max_tokens: 1000)
+    
+    r = super(s, temperature: temperature, max_tokens: max_tokens)
+    puts 'CGRecorder inside chat: ' + r.inspect if @debug
+    log(s, r[:text].strip, tags) unless r[:error]
+    
+    return r
+  end  
+  
   def code_completion(s, tags=nil, temperature: 1, max_tokens: 2000)
     
     r = super(s, temperature: temperature, max_tokens: max_tokens)    
-    log(s, r[:text].strip, tags)
+    log(s, r[:text].strip, tags) unless r[:error]
     
     return r
     
@@ -248,11 +327,11 @@ class CGRecorder <  ChatGpt2023
     
     r = super(s, temperature: temperature, max_tokens: max_tokens)
     puts 'CGRecorder inside completion: ' + r.inspect if @debug
-    log(s, r[:text].strip, tags)
+    log(s, r[:text].strip, tags) unless r[:error]
     
     return r
-    
   end
+    
   
   alias complete completion
   alias ask completion  

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: chatgpt2023
 version: !ruby/object:Gem::Version
-  version: 0.4.2
+  version: 0.5.1
 platform: ruby
 authors:
 - James Robertson
@@ -36,7 +36,7 @@ cert_chain:
   6FQD1/GISew7VvxUJdptXeuVNIsdNKxvL3RpfLCuFsi1WXyJ4k3odRMTmS0kAfTy
   J4sZZW9RNfabTMQQY7DIs3tUAn6i+O0r9lo=
   -----END CERTIFICATE-----
-date: 2023-02-16 00:00:00.000000000 Z
+date: 2023-03-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: down