tiny_chat_gpt 1.0.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 65e3af15974d0088f1cc53eacb0957e600a324ce7862d4e4656bb9022b097773
-  data.tar.gz: 5a01788191a33bccc7cdb77b2183f249ea3c576c7f2ecb11667ed636a9447d8a
+  metadata.gz: 13551022c05de5ff85240752f609c2a282c5bb5a10b283de0d4c63a6d89a7af6
+  data.tar.gz: 1fe62992425f0ea98de5930c62acf26df9ed602907b8e1fdaf0ab2d856eef8c8
 SHA512:
-  metadata.gz: 1f4dbb55d0e27478b28619db2ec83f2a1bd801d3631686c2d11ac27ef14c5dd03fe8d88037da07988de1c8203fbaef20aaeb57b8b0401cec86ebd16bfd4f9efc
-  data.tar.gz: 12b646bc6db077e7407e7499bab7b6d0f33102a92b81f888d252a8e38a2b0effaf8123e09132e63167f786248cd07a4e78e18ef1d4ce92736967bb18e8016bee
+  metadata.gz: d1675f8bd301df9a523fea30a2eb9bb23d2bfff8809dfcd80ab744aad6216337e1a3e9fac2d4d9bbce7507d987ffb4d2af224a3e6b2fe645f73201d19c0354d4
+  data.tar.gz: 6e7743d42bd23c9b48024e6058f80c1725fddc2d0a5223bc9634215c1a2695c93668bcd25a7975b307dcc6e2d5bd922ae15fc8b7518e6051d57843d456351ff1

data/lib/tiny_chat_gpt.rb

@@ -1,60 +1,71 @@
 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.
 #
 # 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'
 
   def initialize(model, api_key)
     @model = model
     @api_key = api_key
+    @msgs = []
   end
 
-  def ask(prompt)
+  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'
+      TTY::Markdown.parse(_parse_ok(response))
     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)["choices"]
+      .first
+      .dig('message', 'content')
   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.1.0
 platform: ruby
 authors:
 - Jeff Lunt
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-02-10 00:00:00.000000000 Z
-dependencies: []
+date: 2023-10-28 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: []