tiny_chat_gpt 1.0.0 → 1.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 65e3af15974d0088f1cc53eacb0957e600a324ce7862d4e4656bb9022b097773
-  data.tar.gz: 5a01788191a33bccc7cdb77b2183f249ea3c576c7f2ecb11667ed636a9447d8a
+  metadata.gz: 76e6d7ac36335701339ae739c5f4b4e01359f2e50d613acc9e2575bfcd406a8a
+  data.tar.gz: c28f7e4e0ad73791bc343fc0baa1f0aa986b09d1a211113e8203f7cf16d68265
 SHA512:
-  metadata.gz: 1f4dbb55d0e27478b28619db2ec83f2a1bd801d3631686c2d11ac27ef14c5dd03fe8d88037da07988de1c8203fbaef20aaeb57b8b0401cec86ebd16bfd4f9efc
-  data.tar.gz: 12b646bc6db077e7407e7499bab7b6d0f33102a92b81f888d252a8e38a2b0effaf8123e09132e63167f786248cd07a4e78e18ef1d4ce92736967bb18e8016bee
+  metadata.gz: ba1d9bd365ca7f33733c5d5fdab9697e56b26a1fd92b9894355a9f55bf4f817762ea192514990d2065ee05df4dcc2515402acff55b99dc1ecc63150783998309
+  data.tar.gz: 7d1da302b9fd7e3575062de5444ee9c0a5571cba2ac809319f3e82d62c95a55ec1be90910cce274b1fb164d1880f6112536330c61579a1759c3ff7bae55ce4af

data/lib/tiny_chat_gpt.rb

@@ -1,60 +1,90 @@
 require 'json'
+require 'uri'
 require 'net/http'
+require 'tiny_color'
+require 'tty-markdown'
 
-# the TinyChatGPT class is an API adapter for OpenAI's GPT-3 based ChatGPT
-# model.  this class makes it easy to send prompts to the ChatGPT API and
-# receive responses.
+# TinyChatGpt provides an interface to OpenAI GPT API - every instance of this
+# class starts a new conversation, which is necessary in order to carry on a
+# conversation.
 #
 # usage:
-#   chatbot = ChatGPT.new("davinci", "your_api_key")
-#   puts chatbot.ask("Hello, how are you today?")
 #
-# if there is any non-200 HTTP response from the API then an instance of
-# TinyChatGpt::APIError will be raised with a helpful error message.
-#
-# NOTE: this version does not maintain any context from one prompt to the next,
-# so having a longer conversation with ChatGPT via this client is not yet
-# supported. every use of the #ask method is sent as a separate API request,
-# and does not include the context of previous prompts and replies.
+#   api_key = 'your_api_key'
+#   model = TinyChatGpt::MODEL_3_5_TURBO
+#   chat = TinyChatGpt.new(model, api_key)
+#   response = chat.prompt('Hello, how are you?')
+#   puts response
 class TinyChatGpt
-  API_URL = "https://api.openai.com/v1/engines/davinci/jobs".freeze
+  URI = URI('https://api.openai.com/v1/chat/completions')
+  MODEL_3_5_TURBO = 'gpt-3.5-turbo'
+  MODEL_4 = 'gpt-4-0613'
+
+  attr_reader :prompt_tokens,
+              :completion_tokens
 
   def initialize(model, api_key)
     @model = model
     @api_key = api_key
+    @msgs = []
+
+    @prompt_tokens = 0
+    @completion_tokens = 0
   end
 
-  def ask(prompt)
+  def total_tokens
+    prompt_tokens + completion_tokens
+  end
+
+  def prompt(prompt)
+    @msgs << {
+      role: 'user',
+      content: prompt
+    }
+
     request = {
-      prompt: prompt,
-      max_tokens: 100,
-      n: 1,
-      stop: "",
-      temperature: 0.5,
-      model: @model
+      model: @model,
+      messages: @msgs
     }
+
     response = _send_request(request)
 
-    if response.code == 200
-      _parse_response(response)
+    if response.code == '200'
+      resp = _parse_ok(response)
+      completion = resp["choices"].first.dig('message', 'content')
+      @msgs << {
+        role: 'assistant',
+        content: completion
+      }
+
+      @prompt_tokens += resp.dig('usage', 'prompt_tokens')
+      @completion_tokens += resp.dig('usage', 'completion_tokens')
+
+      TTY::Markdown.parse(completion)
     else
-      raise TinyChatGpt::APIError.new("ERR: non-200 HTTP status to ChatGPT: #{response.code}")
+      "#{'ERR'.light_red}: HTTP status code #{response.code}, #{_parse_error(response)}"
     end
   end
 
   def _send_request(request)
-    uri = URI(API_URL)
-    http = Net::HTTP.new(uri.host, uri.port)
-    http.use_ssl = true
-    request = Net::HTTP::Post.new(uri, 'Content-Type' => 'application/json')
-    request.basic_auth(@api_key, '')
-    request.body = request.to_json
-    http.request(request)
+    Net::HTTP.post(
+      URI,
+      request.to_json,
+      {
+        'Content-Type' => 'application/json',
+        'Authorization' => "Bearer #{KEY}"
+      }
+    )
+  end
+
+  def _parse_ok(response)
+    JSON.parse(response.body)
   end
 
-  def _parse_response(response)
-    response = JSON.parse(response)
-    response["choices"].first["text"]
+  def _parse_error(response)
+    JSON
+      .parse(response.body)
+      .dig('error', 'message')
   end
 end
 

metadata

@@ -1,15 +1,43 @@
 --- !ruby/object:Gem::Specification
 name: tiny_chat_gpt
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.2.0
 platform: ruby
 authors:
 - Jeff Lunt
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-02-10 00:00:00.000000000 Z
-dependencies: []
+date: 2023-10-29 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: tiny_color
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: tty-markdown
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: a tiny ChatGPT client
 email: jefflunt@gmail.com
 executables: []