ruby_bots 0.1.1 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 661bc338ef1cc7a60d6a4b3076c194faaf1c71d21db937d1de05fbfcaa9bc899
-  data.tar.gz: b721ca417c4bf4762b1462b2f1672452a7e27ee46b1fec3b35ec369167de7e5d
+  metadata.gz: f6340ca62c9070462527e782f8003e099a278da7e0c70beb4d3bc87da7609a89
+  data.tar.gz: 0a9207838ae9ce3ac5b1e00cb8ab1e0a32e591dd5cf03efddf11feb5a4921681
 SHA512:
-  metadata.gz: cf2275c793c811c194e4c25e5497ba16a8eb54898b487e53159f966352ea71041d3324717b2f99b4fd1611c1244ed5344422fe15779a8317759b3142ba8097cf
-  data.tar.gz: 5532942135eea17d71566e52609f30327da76de6a1be7e3a1e108d7cab31e9576d452edda5ff92c6c829fb25bc87dcddf7f5eb57c5c1c1eeee8b0617d0afcbb6
+  metadata.gz: 28cb37f96735cb4341ff5f58260db3e568b15933b39b423d9c5b862ddd73f6d8f42a2e4d73aa17e2e74c6643a1d3e377926bcdf01fa70214212618ad6c63f30a
+  data.tar.gz: efa7b38c6cf1ce93e9dd225eb6d7dae0f58f57b8951b8b2287c188e5aba529ca813ca8af5c8a614a59598a499669e6610b84dacaca55d7bde4918b70122514a1

data/lib/ruby_bots/bots/openai_chat_bot.rb

@@ -0,0 +1,60 @@
+module RubyBots
+  class OpenAIChatBot < OpenAIBot
+    include RubyBots::Chattable
+
+    DEFAULT_DESCRIPTION = 'This bot will use OpenAI to determine the appropriate tool and use it. It will also chat responses to the user to clarify their request.'.freeze
+
+    def initialize(tools:, name: 'OpenAI chat bot', description: DEFAULT_DESCRIPTION)
+      super(tools:, name:, description:)
+    end
+
+    def system_instructions
+      <<~PROMPT
+        You are an assistant that is chatting with a user and also using tools.
+        You can use the following tools (name - description):
+        #{tools.map { |t| "#{t.name} - #{t.description}" }.join('\n')}
+
+        Select from the following tools (name - description):
+        #{tools.map { |t| "#{t.name} - #{t.description}" }.join('\n')}
+
+        If it is clear that a tool best fits the user's request, return only the tool name and nothing more.
+        If no tool clearly matches the user's request respond with questions to help you clarify the user's response.
+      PROMPT
+    end
+
+    private
+
+    def run(input, &block)
+      @messages = [
+        { role: :system, content: system_instructions },
+        { role: :user, content: input }
+      ]
+
+      bot_output = ''
+
+      until ['exit', 'quit', 'q', 'stop', 'bye'].include?(input.chomp.downcase)
+        response = client.chat(parameters: params)
+
+        bot_output = response.dig('choices', 0, 'message', 'content')
+
+        @messages << { role: :assistant, content: bot_output }
+
+        if tools.map(&:name).include?(bot_output)
+          return tools.find { |t| t.name == bot_output }.response(input)
+        end
+
+        input = yield bot_output
+
+        @messages << { role: :user, content: input }
+      end
+
+      bot_output
+    end
+
+    def params
+      {
+        messages: @messages
+      }.merge(default_params)
+    end
+  end
+end

data/lib/ruby_bots/chattable.rb

@@ -0,0 +1,17 @@
+module RubyBots
+  module Chattable
+    def response(input, &block)
+      run_input_validations(input)
+
+      raise RubyBots::InvalidInputError, { errors: @errors } if @errors.any?
+
+      run(input, &block)
+    end
+
+    private
+
+    def run(input, &block)
+      raise NotImplementedError
+    end
+  end
+end

data/lib/ruby_bots/tools/openai_chat_tool.rb

@@ -0,0 +1,49 @@
+module RubyBots
+  class OpenAIChatTool < OpenAITool
+    include RubyBots::Chattable
+
+    DEFAULT_DESCRIPTION = 'This tool will use OpenAI to set up a chat interface with the user. It will chat responses to the user to clarify their request.'.freeze
+
+    def initialize(name: 'OpenAI chat tool', description: DEFAULT_DESCRIPTION)
+      super(name:, description:)
+    end
+
+    def system_instructions
+      <<~PROMPT
+        You are a helpful assistant that is chatting with a user. You can ask the user to provide more information
+        if it helps you determine the correct response.
+      PROMPT
+    end
+
+    private
+
+    def run(input, &block)
+      @messages = [
+        { role: :system, content: system_instructions },
+        { role: :user, content: input }
+      ]
+
+      bot_output = ''
+
+      until ['exit', 'quit', 'q', 'stop', 'bye'].include?(input.chomp.downcase)
+        response = client.chat(parameters: params)
+
+        bot_output = response.dig('choices', 0, 'message', 'content')
+
+        @messages << { role: :assistant, content: bot_output }
+
+        input = yield(bot_output)
+
+        @messages << { role: :user, content: input }
+      end
+
+      bot_output
+    end
+
+    def params
+      {
+        messages: @messages
+      }.merge(default_params)
+    end
+  end
+end

data/lib/ruby_bots/tools/openai_streaming_tool.rb

@@ -0,0 +1,27 @@
+module RubyBots
+  class OpenAIStreamingTool < OpenAIChatTool
+    def run(input, &block)
+      @messages = [
+        { role: :system, content: system_instructions },
+        { role: :user, content: input }
+      ]
+
+      client.chat(
+        parameters: {
+          messages: @messages,
+          model: 'gpt-4',
+          temperature: 0.7,
+          stream: stream_proc
+        }
+      ).each do |response|
+        puts response
+      end
+    end
+
+    def stream_proc
+      proc do |chunk, _bytesize|
+        print chunk.dig('choices', 0, 'delta', 'content')
+      end
+    end
+  end
+end

data/lib/ruby_bots/version.rb

@@ -1,3 +1,3 @@
 module RubyBots
-  VERSION = '0.1.1'.freeze
+  VERSION = '0.1.3'.freeze
 end

data/lib/ruby_bots.rb

@@ -39,9 +39,15 @@ class String
   end
 end
 
+require_relative 'ruby_bots/streamable'
+require_relative 'ruby_bots/chattable'
+require_relative 'ruby_bots/version'
+
 # tools
 require_relative 'ruby_bots/tool'
 require_relative 'ruby_bots/tools/openai_tool'
+require_relative 'ruby_bots/tools/openai_chat_tool'
+require_relative 'ruby_bots/tools/openai_streaming_tool'
 
 # bots
 require_relative 'ruby_bots/bot'
@@ -49,6 +55,4 @@ require_relative 'ruby_bots/bots/openai_bot'
 require_relative 'ruby_bots/bots/pipeline_bot'
 require_relative 'ruby_bots/bots/router_bot'
 require_relative 'ruby_bots/bots/openai_react_bot'
-
-require_relative 'ruby_bots/streamable'
-require_relative 'ruby_bots/version'
+require_relative 'ruby_bots/bots/openai_chat_bot'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby_bots
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.3
 platform: ruby
 authors:
 - Justin Paulson
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-04-24 00:00:00.000000000 Z
+date: 2023-05-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: debug
@@ -91,11 +91,15 @@ files:
 - lib/ruby_bots.rb
 - lib/ruby_bots/bot.rb
 - lib/ruby_bots/bots/openai_bot.rb
+- lib/ruby_bots/bots/openai_chat_bot.rb
 - lib/ruby_bots/bots/openai_react_bot.rb
 - lib/ruby_bots/bots/pipeline_bot.rb
 - lib/ruby_bots/bots/router_bot.rb
+- lib/ruby_bots/chattable.rb
 - lib/ruby_bots/streamable.rb
 - lib/ruby_bots/tool.rb
+- lib/ruby_bots/tools/openai_chat_tool.rb
+- lib/ruby_bots/tools/openai_streaming_tool.rb
 - lib/ruby_bots/tools/openai_tool.rb
 - lib/ruby_bots/version.rb
 homepage: https://github.com/aha-app/ruby_bots