ruby_bots 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 661bc338ef1cc7a60d6a4b3076c194faaf1c71d21db937d1de05fbfcaa9bc899
-  data.tar.gz: b721ca417c4bf4762b1462b2f1672452a7e27ee46b1fec3b35ec369167de7e5d
+  metadata.gz: df9f251c6730b2873d3ceb1584d2a82fa9e6a563d2b760706343a60f69de0017
+  data.tar.gz: a9f2522f666a0185019fafd8aa757d2c61a2565d8b2f2bd6c0bfc75577dcd36c
 SHA512:
-  metadata.gz: cf2275c793c811c194e4c25e5497ba16a8eb54898b487e53159f966352ea71041d3324717b2f99b4fd1611c1244ed5344422fe15779a8317759b3142ba8097cf
-  data.tar.gz: 5532942135eea17d71566e52609f30327da76de6a1be7e3a1e108d7cab31e9576d452edda5ff92c6c829fb25bc87dcddf7f5eb57c5c1c1eeee8b0617d0afcbb6
+  metadata.gz: d3bdb00120855fefd228f0fc1c35b9aa4f148fdfa8704f66f3828ef188ef8c67247224be1758aa0241da6c771d755284ad30acba283a4909913c9b075bdd3d46
+  data.tar.gz: 3e07cdb73ee2fb5ad46af11512ce588c4013f2d4fc7ec55bcbcebcccae1f4bfbd43cf60821a68686a4ef1d796e3cb1921cd60b75fd6b89fb42666fa15ad6f998

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/version.rb

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

data/lib/ruby_bots.rb

@@ -39,9 +39,14 @@ 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'
 
 # bots
 require_relative 'ruby_bots/bot'
@@ -49,6 +54,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.2
 platform: ruby
 authors:
 - Justin Paulson
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-04-24 00:00:00.000000000 Z
+date: 2023-04-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: debug
@@ -91,11 +91,14 @@ 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_tool.rb
 - lib/ruby_bots/version.rb
 homepage: https://github.com/aha-app/ruby_bots