chat_gpt_client 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f7641c9f199e34018c618fda5d669b58a6ca5ae5c727d72aebe1587beb0d7b82
-  data.tar.gz: ad2f1f30e8b3b766310d403646e872f38bdbd823d9110abc15d6cdcea95d32e5
+  metadata.gz: 63a9663535742a12f4720e60dfac2589c7cfcf02ffe9c63813aa06cb7b9335e2
+  data.tar.gz: 6a1d4649a5ef6ba7ed711f1ba15f905e8b24e913f1224360d3070c2a372ac1c9
 SHA512:
-  metadata.gz: 2bd0344f4af641f8fc9d96c8d771df55e9eb7852365c7a8b559e6ab4dfc70f8a9fcd9daf5f8881652753afb1a0bc9ea6495c14e72808312e5fd97686687d691c
-  data.tar.gz: f570d3e7f136eb8e7725e1f1b4ab80d431faddd550526e1ba67ddc782d3af2a169a7bf3775f6fc99fa6bb3dfe12fed65a4cb84f56530600c59ca25e5e66e2401
+  metadata.gz: 6f7331daaa81a07cef793599a7026f4b73898574dde43fd60cc6f69b553b7102fda812fed5a78567ec9100c26a1dff4df4794c5e7cdbe889bf06b8b5eb7428a7
+  data.tar.gz: 3bbb94c0d72d0f9fbf5c83794a827c7470c6fc31e9ca83f0615274388511e070093c834b5668ecffee1272e0d8157985026ae19ed1186c81fc41de5950c699af

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    chat_gpt_client (0.1.0)
+    chat_gpt_client (0.1.1)
 
 GEM
   remote: https://rubygems.org/
@@ -44,6 +44,7 @@ GEM
 
 PLATFORMS
   x86_64-darwin-21
+  x86_64-linux
 
 DEPENDENCIES
   chat_gpt_client!

data/lib/chat_gpt/client.rb

@@ -3,13 +3,14 @@
 module ChatGPT
   # The client for the OpenAI ChatGPT API
   class Client
-    def initialize(api_key: ENV["OPENAI_API_KEY"], system_prompt: nil)
+    def initialize(api_key: ENV["OPENAI_API_KEY"], system_prompt: nil, temperature: nil)
       if api_key.nil? || api_key.empty?
-        raise Error, "You must provide an API token or set the OPENAI_API_KEY environment variable"
+        raise ChatGPT::Error, "You must provide an API token or set the OPENAI_API_KEY environment variable"
       end
 
       @api_key = api_key
       @system_prompt = system_prompt
+      @temperature = temperature
       @history = []
     end
 
@@ -58,7 +59,7 @@ module ChatGPT
     end
 
     def extract_response(response)
-      raise Error, "Error: #{response.code} - #{response.body}" unless response.code == "200"
+      raise ChatGPT::Error, "Error: #{response.code} - #{response.body}" unless response.code == "200"
 
       JSON.parse(response.body)
     end
@@ -66,7 +67,8 @@ module ChatGPT
     def build_body
       {
         model: "gpt-3.5-turbo",
-        messages: history
+        messages: history,
+        temperature: @temperature
       }
     end
 

data/lib/chat_gpt/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ChatGPT
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: chat_gpt_client
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Francesco Boffa