chatgpt2023 0.4.2 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 79e85a14caa0d9f7554af7d2df3faf4356d906ffd32eddbeab97a0ce62588aa6
-  data.tar.gz: 7ed03e8ba40941b21431469711325c9eb20efd5c9374ecfb24e0cdd9d497888b
+  metadata.gz: d76a15a250b67bab6e676ed5c18131cde6310d9c5ace84921e8113c21b4c180f
+  data.tar.gz: bbdce3b4c8ba915e55f3b440da3fc368d90ad6a0555ae1ebf8e6065e44061bd3
 SHA512:
-  metadata.gz: '0297c587bab1025db945d899437ca0b19a820c2c172a01ea2129de95b09fd773f02bed19e098d8a3c34881c06b4ddfc301a3e38670bce6c35236300494281099'
-  data.tar.gz: f80fc62462f62eeee57b2ed50ff81a76de628fa27bd2733c82ac57b108da6ed5a2b14f6afce5a600353c97d3a3fc8ce574117341fec1b70f38bb29b1fa972f14
+  metadata.gz: fb2c05d8493193771ae4480dfa340af843c4c73ae497cc03aa9504c5c9fbb5c30456df7c5c37db26f68bba3419ca41447d4e8436b7aab9b060bcce2160fd6254
+  data.tar.gz: b2c4d4292b30a127935de35156cd48b50ae7c5890e2c5cbe0e6e5f37bf56641db73bb136ee6a2a3d88b8227aacc05f8041281347920d0da7a54d2b9f6a899171

data/lib/chatgpt2023.rb

@@ -39,15 +39,62 @@ 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)
+    
+    messages << @assistant_recent if @assistant_recent
+    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 +103,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 +135,7 @@ class ChatGpt2023
   end
   
   alias complete completion
-  alias ask completion
+  alias ask chat
   
   def edits(s, s2)
     r = go_edits(s, s2)
@@ -111,6 +158,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 +288,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'
@@ -238,7 +299,7 @@ class CGRecorder <  ChatGpt2023
   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,7 +309,7 @@ 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
     

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: chatgpt2023
 version: !ruby/object:Gem::Version
-  version: 0.4.2
+  version: 0.5.0
 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-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: down