tiny_chat_gpt 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 65e3af15974d0088f1cc53eacb0957e600a324ce7862d4e4656bb9022b097773
+  data.tar.gz: 5a01788191a33bccc7cdb77b2183f249ea3c576c7f2ecb11667ed636a9447d8a
+SHA512:
+  metadata.gz: 1f4dbb55d0e27478b28619db2ec83f2a1bd801d3631686c2d11ac27ef14c5dd03fe8d88037da07988de1c8203fbaef20aaeb57b8b0401cec86ebd16bfd4f9efc
+  data.tar.gz: 12b646bc6db077e7407e7499bab7b6d0f33102a92b81f888d252a8e38a2b0effaf8123e09132e63167f786248cd07a4e78e18ef1d4ce92736967bb18e8016bee

data/lib/tiny_chat_gpt.rb

@@ -0,0 +1,61 @@
+require 'json'
+require 'net/http'
+
+# 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.
+#
+# 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.
+class TinyChatGpt
+  API_URL = "https://api.openai.com/v1/engines/davinci/jobs".freeze
+
+  def initialize(model, api_key)
+    @model = model
+    @api_key = api_key
+  end
+
+  def ask(prompt)
+    request = {
+      prompt: prompt,
+      max_tokens: 100,
+      n: 1,
+      stop: "",
+      temperature: 0.5,
+      model: @model
+    }
+    response = _send_request(request)
+
+    if response.code == 200
+      _parse_response(response)
+    else
+      raise TinyChatGpt::APIError.new("ERR: non-200 HTTP status to ChatGPT: #{response.code}")
+    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)
+  end
+
+  def _parse_response(response)
+    response = JSON.parse(response)
+    response["choices"].first["text"]
+  end
+end
+
+class TinyChatGpt::APIError < RuntimeError; end

metadata

@@ -0,0 +1,44 @@
+--- !ruby/object:Gem::Specification
+name: tiny_chat_gpt
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Jeff Lunt
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2023-02-10 00:00:00.000000000 Z
+dependencies: []
+description: a tiny ChatGPT client
+email: jefflunt@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/tiny_chat_gpt.rb
+homepage: https://github.com/jefflunt/tiny_chat_gpt
+licenses:
+- MIT
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.4.1
+signing_key:
+specification_version: 4
+summary: this library provides a ChatGPT client in Ruby that can be used to interact
+  with ChatGPT programmatically
+test_files: []