chatgpt-rb 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 56bfd3ab0a5f56e05e8da9b5f59f7358383ebb1c78214d1afec32c6dd63fa140
+  data.tar.gz: 3361ca43ea603a5634e95793cfff46215751a855e2836a92ba6161c4eb07d3fc
+SHA512:
+  metadata.gz: '09ab2d88f40d4b00eed86b3fbab7ab8fcc9b454f98c6f1ab0ac3b7461f97324f9151563540c9700d6e24d2ba47491ada4e611394453bd9b093ed2c026254f4cb'
+  data.tar.gz: ff8ba450b24f98564a455c5994e8e19a3ad8cc0e85e4e3f14530a65a669e2a964ae8484815c14098fa6160c48e61b24a69f8cb46de4da4b5874d8a191881bb60

data/bin/chatgpt-rb

@@ -0,0 +1,34 @@
+#!/usr/bin/env ruby
+
+require "dotenv/load"
+require "colorize"
+require "reline"
+require_relative "./../lib/chatgpt_rb"
+
+begin
+  stty_save = `stty -g`.chomp
+rescue
+end
+
+begin
+  conversation = ChatgptRb::Conversation.new(api_key: ENV.fetch("OPEN_AI_KEY"), model: ENV.fetch("OPEN_AI_MODEL", "gpt-3.5-turbo"))
+
+  puts "Type any message to talk with ChatGPT. Type 'exit' to quit. Type 'dump' to dump this conversation to JSON."
+
+  while message = Reline.readline("me> ".colorize(:red), true) do
+    case message.chomp
+    when "exit", "quit", "q", "\\q"
+      exit
+    when "dump"
+      puts "dump> ".colorize(:blue) + conversation.messages.to_json
+    else
+      print("ai> ".colorize(:yellow))
+      conversation.ask(message) { |fragment| print(fragment) }
+      puts
+    end
+  end
+rescue Interrupt
+  puts "^C"
+  `stty #{stty_save}` if stty_save
+  exit
+end

data/lib/chatgpt_rb/conversation.rb

@@ -0,0 +1,126 @@
+require "httparty"
+
+module ChatgptRb
+  class Conversation
+    attr_reader :api_key, :model, :functions, :temperature, :max_tokens, :top_p, :frequency_penalty, :presence_penalty, :messages
+
+    # @param api_key [String]
+    # @param model [String]
+    # @param functions [Array<Hash>]
+    # @param temperature [Float]
+    # @param max_tokens [Integer]
+    # @param top_p [Float]
+    # @param frequency_penalty [Float]
+    # @param presence_penalty [Float]
+    # @param messages [Array<Hash>]
+    # @param prompt [String, nil] instructions that the model can use to inform its responses, for example: "Act like a sullen teenager."
+    def initialize(api_key:, model: "gpt-3.5-turbo", functions: [], temperature: 0.7, max_tokens: 1024, top_p: 1.0, frequency_penalty: 0.0, presence_penalty: 0.0, messages: [], prompt: nil)
+      @api_key = api_key
+      @model = model
+      @functions = functions
+      @temperature = temperature
+      @max_tokens = max_tokens
+      @top_p = top_p
+      @frequency_penalty = frequency_penalty
+      @presence_penalty = presence_penalty
+      @messages = messages
+      @messages << { role: "system", content: prompt } if prompt
+    end
+
+    # @param content [String]
+    def ask(content, &block)
+      @messages << { role: "user", content: }
+      get_next_response(&block)
+    end
+
+    private
+
+    def <<(message)
+      @messages << message
+    end
+
+    def get_next_response(&block)
+      streamed_content = ""
+      streamed_arguments = ""
+      streamed_role = ""
+      streamed_function = ""
+      error_buffer = []
+
+      response = HTTParty.post(
+        "https://api.openai.com/v1/chat/completions",
+        steam_body: block_given?,
+        headers: {
+          "Content-Type": "application/json",
+          "Authorization": "Bearer #{api_key}",
+        },
+        body: {
+          model:,
+          messages: @messages,
+          temperature:,
+          max_tokens:,
+          top_p:,
+          frequency_penalty:,
+          presence_penalty:,
+          stream: block_given?,
+        }.tap { |hash| hash[:functions] = functions.map { |hash| hash.except(:implementation) } unless functions.empty? }.to_json,
+      ) do |fragment|
+        if block_given?
+          fragment.each_line do |line|
+            next if line.nil?
+            next if line == "\n"
+            break if line == "data: [DONE]\n"
+
+            line_without_prefix = line.gsub(/^data: /, "")
+            json = JSON.parse(line_without_prefix)
+
+            break if json.dig("choices", 0, "finish_reason")
+
+            if (function_name = json.dig("choices", 0, "delta", "function_call", "name"))
+              streamed_function = function_name
+              next
+            end
+
+            if (role = json.dig("choices", 0, "delta", "role"))
+              streamed_role = role
+              next
+            end
+
+            if content = json.dig("choices", 0, "delta", "content")
+              yield content unless line_without_prefix.empty?
+              streamed_content << content
+            elsif arguments = json.dig("choices", 0, "delta", "function_call", "arguments")
+              streamed_arguments << arguments
+            end
+          rescue => e
+            error_buffer << "Error: #{e}"
+          end
+        end
+      end
+
+      error_buffer.each { |e| $stderr.puts("Error: #{e}") }
+
+      @messages << if block_given? && streamed_content != ""
+                     { "content" => streamed_content, "role" => streamed_role }
+                   elsif block_given? && streamed_arguments != ""
+                     { "role" => "assistant", "content" => nil, "function_call" => { "name" => streamed_function, "arguments" => streamed_arguments } }
+                   else
+                     response.dig("choices", 0, "message")
+                   end
+
+      if @messages.last["content"]
+        @messages.last["content"]
+      elsif @messages.last["function_call"]
+        function_args = @messages.last["function_call"]
+        function_name = function_args.fetch("name")
+        arguments = JSON.parse(function_args.fetch("arguments"))
+
+        function = functions.find { |function| function[:name] == function_name }
+        content = function.fetch(:implementation).call(**arguments.transform_keys(&:to_sym))
+
+        @messages << { role: "function", name: function_name, content: content.to_json }
+
+        get_next_response(functions:, api_key:, model:, temperature:, max_tokens:, top_p:, frequency_penalty:, presence_penalty:, &block)
+      end
+    end
+  end
+end

data/lib/chatgpt_rb.rb

@@ -0,0 +1 @@
+require_relative "./chatgpt_rb/conversation"

metadata

@@ -0,0 +1,104 @@
+--- !ruby/object:Gem::Specification
+name: chatgpt-rb
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Aaron Breckenridge
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2023-07-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: httparty
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.21'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.21'
+- !ruby/object:Gem::Dependency
+  name: colorize
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.6'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.6'
+- !ruby/object:Gem::Dependency
+  name: dotenv
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.8'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.8'
+- !ruby/object:Gem::Dependency
+  name: reline
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.3'
+description: Provides libraries for interacting with the ChatGPT API and a CLI program
+  `chatgpt-rb` for live conversations.
+email:
+- aaron@breckridge.dev
+executables:
+- chatgpt-rb
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/chatgpt-rb
+- lib/chatgpt_rb.rb
+- lib/chatgpt_rb/conversation.rb
+homepage: https://github.com/breckenedge/chatgpt-rb
+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.10
+signing_key:
+specification_version: 4
+summary: A gem for interacting with the ChatGPT API
+test_files: []