ruby_bots 0.0.21 → 0.0.22

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 76d6f4b77a8ef6bc0f8124475142c2fb90460e9c9f602381eca42488aea99ea3
-  data.tar.gz: bf11eed995aec29f80a3f97f86fc1856f6c4733cdbe7d716cf705af0f9733ad4
+  metadata.gz: 9cb9f98a4334909eaa93b2c4f0ed5dcb3c92c85b32ecaeafcf17c2f1e8219a70
+  data.tar.gz: 978d6e24d899760d5c79c585ebbf6ee4bb93086c8e809a2d6befa145a74ddf03
 SHA512:
-  metadata.gz: 7c052e9aad0d5de43e0ab7644b15ded54a84b1c87429a7780c455d80e93239e15a878a4d015eb9dde9383e300132d661e6e72218dff79b2ac8855117ab2aade3
-  data.tar.gz: 7fd242669b9332dec10d886ece3145ea3b76420459b5fb532e502f3050803d90cb839e6bb5e65711e6d79dc221a6dc6e5d9b5c115e019ead934bc8a877d8fd2d
+  metadata.gz: 44a634ac21766cabd85022c83c96d0672b1cbea6fea831932cf954b891cd310741e618b23cd4a0837e053bf4fb1c706fdc28a414ba7af2ccea451c57ad699d47
+  data.tar.gz: 8bbfe05922e5dd8e3016427e60aee9f10e443c811d6dfc1a57fb555d8b134eedd2f41bde6fe2b446315b0b4cfddd65e76d4b408932445190c042da4e4810a5db

data/lib/ruby_bots/bot.rb

@@ -9,7 +9,7 @@ module RubyBots
 
     private
 
-    def run(inputs)
+    def run(input)
       raise NotImplementedError
     end
   end

data/lib/ruby_bots/bots/openai_bot.rb

@@ -9,13 +9,24 @@ module RubyBots
       super(name: name, description: description)
     end
 
+    def system_instructions
+      <<~PROMPT
+      You are an assistant designed to select a tool for a user to use. You are provided with the user's input.
+      Select from the following tools (name - description):
+      #{tools.map{|t| t.name + " - " + t.description}.join("\n")}
+
+      Return only the name of the tool that best fits the user's request.
+      If no tools match the user's request respond with "no tool" and nothing more.
+      PROMPT
+    end
+
     private
     
-    def run(inputs)
+    def run(input)
       params = {
         messages: [
           { role: :system, content: system_instructions },
-          { role: :user, content: inputs }
+          { role: :user, content: input }
         ]
       }.merge(default_params) 
 

data/lib/ruby_bots/bots/openai_react_bot.rb

@@ -0,0 +1,131 @@
+module RubyBots
+  class OpenAIReactBot < OpenAIBot
+    attr_accessor :tools
+
+    DEFAULT_DESCRIPTION = "This bot will use the ReAct framework to determine the appropriate response. It is powered by OpenAI and the ReAct framework."
+    
+    DEFAULT_TOOLS = [
+      RubyBots::Tool.new(name:"search", description:"Search the web for information. Input should be the string to search for."),
+      RubyBots::Tool.new(name:"calculate", description:"Solve math problems. Input should be a mathematical expression with no additional details or context."),
+    ]
+
+    def initialize(name: "OpenAI ReAct bot", description: DEFAULT_DESCRIPTION, tools: DEFAULT_TOOLS)
+      @tools = tools
+      super(name: name, description: description)
+    end
+
+    def examples
+      [
+        <<~EXAMPLE
+        User: What is the current temperature in the city Julia Roberts was born in?
+        Thought: I need to know where Julia Roberts was born.
+        Action: search[Julia Roberts birthplace]
+        Observation: Smyrna, Georgia
+        Thought: I need to know the current temperature in Smyrna, Georgia.
+        Action: search[current temperature Smyrna, Georgia]
+        Observation: 72 degrees Fahrenheit
+        Thought: I need to tell the user the current temperature in the city Julia Roberts was born in.
+        Answer: 72 degrees Fahrenheit
+        EXAMPLE
+        ,
+        <<~EXAMPLE
+        User: What is the square of the age of Albert Einstein at his death?
+        Thought: I need to know the age of Albert Einstein at his death.
+        Action: search[Albert Einstein age at death]
+        Observation: 76 years old
+        Thought: I need to know the square of 76.
+        Action: calculate[76^2]
+        Observation: 5776
+        Thought: I need to tell the user the square of the age of Albert Einstein at his death.
+        Answer: 5776
+        EXAMPLE
+        ,
+        <<~EXAMPLE
+        User: What is half of the amount of years that have passed since the year the first airplane flew?
+        Thought: I need to know the year the first airplane flew.
+        Action: search[first airplane flight year]
+        Observation: 1903
+        Thought: I need to know the current year.
+        Action: search[current year]
+        Observation: 2023
+        Thought: I need to know the amount of years that have passed since 1903.
+        Action: calculate[2023 - 1903]
+        Observation: 120
+        Thought: I need to know half of 120.
+        Action: calculate[120 / 2]
+        Observation: 60
+        Thought: I need to tell the user half of the amount of years that have passed since the year the first airplane flew.
+        Answer: 60
+        EXAMPLE
+      ]
+    end
+
+    def system_instructions
+      <<~PROMPT
+      You are an assistant designed to provide solutions for a user. You are provided with the user's input.
+      You run in a loop of thought, action, observation, and reflection until you have a solution for the user.
+      You can utilize the following actions to gather more information (name - description):
+      #{ tools.map{ |t| t.name + " - " + t.description }.join("\n") }
+
+      You should begin by providing a thought and an action with the necessary input for the action. The action will be
+      executed externally, and then you will be called again with the observation returned from the action.
+      
+      You should then begin the loop again and provide a thought and action.
+
+      If you have a solution for the user, return the solution instead of a new action.
+      The final answer should only answer the question without additional information about reasoning or process.
+
+      Examples:
+      #{ examples.join("\n\n") }
+      PROMPT
+    end
+
+    private
+    
+    def run(input)
+      params = {
+        messages: [
+          { role: :system, content: system_instructions },
+          { role: :user, content: input }
+        ]
+      }.merge(default_params) 
+
+      response = client.chat(parameters: params)
+      response = response.dig("choices", 0, "message", "content")
+
+      while !is_answer?(response)
+        params[:messages] << { role: :assistant, content: response }
+
+        observation = get_observation_from_action(response)
+
+        params[:messages] << { role: :assistant, content: "Observation: #{observation}" }
+
+        response = client.chat(parameters: params)
+
+        response = response.dig("choices", 0, "message", "content")
+      end
+
+      get_answer_from_response(response)
+    end
+
+    def is_answer?(input)
+      input.match(/Answer: /)
+    end
+
+    def get_observation_from_action(response)
+      t = response.match(/Action: ([\s\S]*)/)[1]
+      tool_name = t.match(/([a-zA-Z\-\_]*)\[([\s\S]*)\]/)[1]
+      tool_input = t.match(/([a-zA-Z\-\_]*)\[([\s\S]*)\]/)[2]
+
+      notify_observers(:action, tool_name, tool_input)
+
+      tool = tools.find{|t| t.name == tool_name}
+
+      tool.response(tool_input)
+    end
+
+    def get_answer_from_response(response)
+      response.match(/Answer: ([\s\S]*)/)[1]
+    end
+  end
+end

data/lib/ruby_bots/bots/pipeline_bot.rb

@@ -8,12 +8,12 @@ module RubyBots
 
     private
       
-    def run(inputs)
+    def run(input)
       tools.each do |tool|
-        inputs = tool.run(inputs)
+        input = tool.run(input)
       end
 
-      inputs
+      input
     end
   end
 end

data/lib/ruby_bots/bots/router_bot.rb

@@ -19,11 +19,11 @@ module RubyBots
 
     private
       
-    def run(inputs)
+    def run(input)
       params = {
         messages: [
           { role: :system, content: system_instructions },
-          { role: :user, content: inputs }
+          { role: :user, content: input }
         ]
       }.merge(default_params) 
 
@@ -32,7 +32,7 @@ module RubyBots
       response_text = response.dig("choices", 0, "message", "content")
       selected_tool = tools.find{|t| t.name == response_text}
       if selected_tool
-        selected_tool.response(inputs)
+        selected_tool.response(input)
       else
         response_text
       end

data/lib/ruby_bots/tool.rb

@@ -48,7 +48,7 @@ module RubyBots
     
     private 
 
-    def run(inputs)
+    def run(input)
       raise NotImplementedError
     end
 

data/lib/ruby_bots/tools/openai_tool.rb

@@ -25,11 +25,11 @@ module RubyBots
 
     private
     
-    def run(inputs)
+    def run(input)
       params = {
         messages: [
           { role: :system, content: system_instructions },
-          { role: :user, content: inputs }
+          { role: :user, content: input }
         ]
       }.merge(default_params) 
   

data/lib/ruby_bots/tools/wolfram_tool.rb

@@ -0,0 +1,24 @@
+require 'cgi'
+require 'uri'
+require 'net/http'
+
+module RubyBots
+  class WolframTool < Tool
+    # This tool requires an App ID key from Wolfram Alpha.
+    # add it to your environment variables as WOLFRAM_APP_ID
+
+    DEFAULT_DESCRIPTION = "This tool will use the Wolfram Alpha API to answer questions."
+
+    def initialize(name: "Wolfram tool", description: DEFAULT_DESCRIPTION)
+      super(name: name, description: description)
+    end
+
+    private
+
+    def run(input)
+      uri = URI("http://api.wolframalpha.com/v1/simple?appid=#{ENV['WOLFRAM_APPID']}&i=#{CGI.escape(input)}")
+      resp = Net::HTTP.get_response(uri)
+      resp.body
+    end
+  end
+end

data/lib/ruby_bots/version.rb

@@ -1,3 +1,3 @@
 module RubyBots
-  VERSION = "0.0.21".freeze
+  VERSION = "0.0.22".freeze
 end

data/lib/ruby_bots.rb

@@ -13,6 +13,7 @@ module RubyBots
     attr_accessor :openai_api_key
     def initialize()
       @openai_api_key = ENV["OPENAI_ACCESS_TOKEN"]
+      @wolfram_app_id = ENV["WOLFRAM_APPID"]
     end
   end
 
@@ -34,4 +35,5 @@ require "ruby_bots/bot"
 require "ruby_bots/tools/openai_tool.rb"
 require "ruby_bots/bots/openai_bot.rb"
 require "ruby_bots/bots/pipeline_bot.rb"
-require "ruby_bots/bots/router_bot.rb"
+require "ruby_bots/bots/router_bot.rb"
+require "ruby_bots/bots/openai_react_bot.rb"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby_bots
 version: !ruby/object:Gem::Version
-  version: 0.0.21
+  version: 0.0.22
 platform: ruby
 authors:
 - Justin Paulson
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-04-03 00:00:00.000000000 Z
+date: 2023-04-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: debug
@@ -91,10 +91,12 @@ files:
 - lib/ruby_bots.rb
 - lib/ruby_bots/bot.rb
 - lib/ruby_bots/bots/openai_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/tool.rb
 - lib/ruby_bots/tools/openai_tool.rb
+- lib/ruby_bots/tools/wolfram_tool.rb
 - lib/ruby_bots/version.rb
 homepage: https://github.com/aha-app/ruby_bots
 licenses: