chatgpt-ruby 0.2.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 94d035b49642b3f0c7ca90ef2443711f8b50007bde2ab538cb27a79db6d2198e
-  data.tar.gz: 6013d9bf6c37d54da4c1b9c4686b2576605d394db5881c5d6db4b64ff3223bfa
+  metadata.gz: 6e959cc78a29b94543ca4a79c21905e11296cd5933b39397a31bdd3652b5bede
+  data.tar.gz: df28d307836ff2e4827f6fa37cd68839cef2089b8418ced1d85f87b6677b98ea
 SHA512:
-  metadata.gz: fc802fd52f248b26f40352188b5f8876ffedba0dde39fa8a09655cf9716e50dc3f43c4ee1021187ad146aee5612f59411afa66cfd14eb254c69470b4f25b21b7
-  data.tar.gz: 596422eebcc85c58b2c8800c0ddfdb5d00d1397f67c530f3c732523483dfbf90c88e0eeb420becf325a073f8801dc3dd7a390ce7b726b65dabeb009a3f09ab2c
+  metadata.gz: b06065d572303b32d01ce713968d9f92e582309e356061359a2e216f8afef229e7b5c00376ce805ab84d9612d038ec3f60245ed277c8b5ba24395e1ce4b6a020
+  data.tar.gz: 0a251cea4f557f855d3f515e786c875f62c2d2ea76d1c02bfbbcb85b2ed84adcc14cbf4dcd31983b3b8b1d285b3deb32cb9b9ca59dfd5165dfcdfaa3f8d863c9

data/Gemfile

@@ -10,3 +10,5 @@ gem "rake", "~> 13.0"
 gem "minitest", "~> 5.0"
 
 gem "rubocop", "~> 1.21"
+
+gem 'rest-client'

data/Gemfile.lock

@@ -1,20 +1,34 @@
 PATH
   remote: .
   specs:
-    chatgpt-ruby (0.1.0)
+    chatgpt-ruby (0.4.0)
 
 GEM
   remote: https://rubygems.org/
   specs:
     ast (2.4.2)
+    domain_name (0.5.20190701)
+      unf (>= 0.0.5, < 1.0.0)
+    http-accept (1.7.0)
+    http-cookie (1.0.5)
+      domain_name (~> 0.5)
     json (2.6.3)
+    mime-types (3.4.1)
+      mime-types-data (~> 3.2015)
+    mime-types-data (3.2023.0218.1)
     minitest (5.18.0)
+    netrc (0.11.0)
     parallel (1.22.1)
     parser (3.2.1.1)
       ast (~> 2.4.1)
     rainbow (3.1.1)
     rake (13.0.6)
     regexp_parser (2.7.0)
+    rest-client (2.1.0)
+      http-accept (>= 1.7.0, < 2.0)
+      http-cookie (>= 1.0.2, < 2.0)
+      mime-types (>= 1.16, < 4.0)
+      netrc (~> 0.8)
     rexml (3.2.5)
     rubocop (1.48.1)
       json (~> 2.3)
@@ -29,6 +43,9 @@ GEM
     rubocop-ast (1.27.0)
       parser (>= 3.2.1.0)
     ruby-progressbar (1.13.0)
+    unf (0.1.4)
+      unf_ext
+    unf_ext (0.0.8.2)
     unicode-display_width (2.4.2)
 
 PLATFORMS
@@ -38,6 +55,7 @@ DEPENDENCIES
   chatgpt-ruby!
   minitest (~> 5.0)
   rake (~> 13.0)
+  rest-client
   rubocop (~> 1.21)
 
 BUNDLED WITH

data/lib/chatgpt/client.rb

@@ -1,45 +1,23 @@
-# chatgpt_client.rb
-require 'httparty'
-require 'json'
+# lib/chatgpt_api.rb
 
-module ChatGPTRuby
-  class Client
-    include HTTParty
-    base_uri 'https://api.openai.com/v1'
+require 'rest-client'
 
+module ChatgptRuby
+  class Client
     def initialize(api_key)
       @api_key = api_key
-      @headers = {
-        'Authorization' => "Bearer #{@api_key}",
-        'Content-Type' => 'application/json'
-      }
-    end
-
-    # Methods to interact with the API
-    def generate(prompt, options = {})
-      body = {
-        'prompt' => prompt,
-        'max_tokens' => options.fetch(:max_tokens, 100),
-        'n' => options.fetch(:n, 1),
-        'stop' => options.fetch(:stop, nil),
-        'temperature' => options.fetch(:temperature, 1.0),
-        'top_p' => options.fetch(:top_p, 1.0),
-        'frequency_penalty' => options.fetch(:frequency_penalty, 0.0),
-        'presence_penalty' => options.fetch(:presence_penalty, 0.0),
-      }.to_json
-
-      response = self.class.post('/davinci-codex/completions', headers: @headers, body: body)
-      handle_response(response)
+      @endpoint = 'https://api.chatgpt.com'
     end
 
-    private
-
-    def handle_response(response)
-      if response.code == 200
-        JSON.parse(response.body)
-      else
-        raise "Error: #{response.code} - #{response.message}"
-      end
+    # Methods for accessing the API here
+    def generate_response(prompt)
+      url = "#{@endpoint}/generate"
+      params = {
+        prompt: prompt,
+        api_key: @api_key
+      }
+      response = RestClient.get(url, params: params)
+      JSON.parse(response.body)
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: chatgpt-ruby
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.4.0
 platform: ruby
 authors:
 - Nagendra Dhanakeerthi