ruby_bots 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 31af013343283ed662eea6678279251ccbd7bda130b8633f05c173cb6186fa62
+  data.tar.gz: b273bf6290b5c5b2cc00b2abf139b70393f5ee804a9b1e8e161a6f302a8d5c52
+SHA512:
+  metadata.gz: cecf81c9e2d26a1f07fca2188c5f30230fd75b0040f5632b0f63416d4d177db97694ad0a66a11e084f3306772987ffa14785025ec6761503a7bfbd62413294b6
+  data.tar.gz: f2007a6dfbace71b3747cbe503c36fc45337806e71aa213a2fd00595f494bb431da266aa6320356801f8bc475fe13291cfb7936fc4fdb7b51946c8ab9394a7f8

data/lib/ruby_bots/bot.rb

@@ -0,0 +1,14 @@
+module RubyBots
+  class Bot < Tool
+    attr_accessor :tools
+
+    def initialize(tools:, **kwargs)
+      @tools = tools
+      super(**kwargs)
+    end
+
+    def operate(inputs)
+      raise NotImplementedError
+    end
+  end
+end

data/lib/ruby_bots/bots/openai_bot.rb

@@ -0,0 +1,25 @@
+module RubyBots
+  class OpenAIBot < OpenAITool
+    attr_accessor :tools
+
+  DEFAULT_DESCRIPTION = "This bot will use OpenAI to determine the appropriate tool."
+  
+  def initialize(name: "OpenAI bot", description: DEFAULT_DESCRIPTION, tools:)
+    @name = name
+    @description = description
+    @tools = tools
+  end
+  
+  def operate(inputs)
+    params = {
+      messages: [
+        { role: :system, content: system_instructions },
+        { role: :user, content: inputs }
+      ]
+    }.merge(default_params) 
+
+    response = client.chat(parameters: params)
+
+    response.dig("choices", 0, "message", "content")
+  end
+end

data/lib/ruby_bots/bots/pipeline_bot.rb

@@ -0,0 +1,19 @@
+module RubyBots
+  class PipelineBot < Bot
+    DEFAULT_DESCRIPTION = "This bot will utilize all of its tools in a syncronouse pipeline based on the order the tools are provided."
+    
+    def initialize(name: "Pipeline bot", decription: DEFAULT_DESCRIPTION, tools:)
+      @name = name
+      @description = description
+      @tools = tools
+    end
+      
+    def operate(inputs)
+      tools.each do |tool|
+        inputs = tool.operate(inputs)
+      end
+
+      inputs
+    end
+  end
+end

data/lib/ruby_bots/bots/router_bot.rb

@@ -0,0 +1,41 @@
+module RubyBots
+  class RouterBot < OpenAIBot
+    DEFAULT_DESCRIPTION = "This bot will route the user's input to the appropriate tool. It will only select and use one tool."
+    
+    def initialize(name: "Router bot", decription: DEFAULT_DESCRIPTION, tools:)
+      @name = name
+      @description = description
+      @tools = tools
+    end
+
+    def system_instructions
+      <<~PROMPT
+      You are an assistant helping to route a user's input to the correct tool.
+      You can use 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 "I'm sorry, I am still learning." and nothing more.
+      PROMPT
+    end
+      
+    def operate(inputs)
+      params = {
+        messages: [
+          { role: :system, content: system_instructions },
+          { role: :user, content: inputs }
+        ]
+      }.merge(default_params) 
+
+      response = client.chat(parameters: params)
+        
+      response_text = response.dig("choices", 0, "message", "content")
+      selected_tool = tools.find{|t| t.name == response_text}
+      if selected_tool
+        selected_tool.operate(inputs)
+      else
+        response_text
+      end
+    end
+  end
+end

data/lib/ruby_bots/tool.rb

@@ -0,0 +1,23 @@
+module RubyBots
+  class Tool
+    attr_accessor :name, :description
+    
+    def initialize(name:, description:)
+      @name = name
+      @description = description
+    end
+
+    def validate_inputs(inputs)
+      raise NotImplementedError
+    end
+
+    def run(inputs)
+      raise NotImplementedError
+    end
+
+    def response(inputs)
+      validate_inputs(inputs)
+      run(inputs)
+    end
+  end
+end

data/lib/ruby_bots/tools/openai_tool.rb

@@ -0,0 +1,42 @@
+require 'openai'
+
+module RubyBots
+  class OpenAITool < Tool
+    attr_accessor :name, :description
+
+    DEFAULT_DESCRIPTION = "This tool will use open ai to determine the output."
+  
+    def initialize(name: "OpenAI Tool", description: DEFAULT_DESCRIPTION)
+      @name = name
+      @description = description
+    end
+  
+    def client
+      @client ||= OpenAI::Client.new(access_token: ENV["OPENAI_ACCESS_TOKEN"])
+    end
+  
+    def default_params
+      {
+        model: 'gpt-4',
+        temperature: 0
+      }
+    end
+  
+    def system_instructions
+      "You are a helpful assistant."
+    end
+    
+    def operate(inputs)
+      params = {
+        messages: [
+          { role: :system, content: system_instructions },
+          { role: :user, content: inputs }
+        ]
+      }.merge(default_params) 
+  
+      response = client.chat(parameters: params)
+      
+      response.dig("choices", 0, "message", "content")
+    end
+  end
+end

data/lib/ruby_bots/version.rb

@@ -0,0 +1,3 @@
+module RubyBots
+  VERSION = "0.0.1".freeze
+end

data/lib/ruby_bots.rb

@@ -0,0 +1,20 @@
+module RubyBots
+  class << self
+  end
+
+  def self.bot
+    RubyBots::Bot
+  end
+
+  def self.tool
+    RubyBots::Tool
+  end
+end
+
+require "ruby_bots/tool"
+require "ruby_bots/bot"
+# require "ruby_bots/version"
+# require "ruby_bots/bots/openai_bot.rb"
+# require "ruby_bots/bots/pipeline_bot.rb"
+# require "ruby_bots/bots/router_bot.rb"
+# require "ruby_bots/tools/openai_tool.rb"

metadata

@@ -0,0 +1,127 @@
+--- !ruby/object:Gem::Specification
+name: ruby_bots
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Justin Paulson
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2023-03-31 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: debug
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.1'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.1'
+- !ruby/object:Gem::Dependency
+  name: dotenv
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.8'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.8'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.2'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.2'
+- !ruby/object:Gem::Dependency
+  name: google_search_results
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.2'
+- !ruby/object:Gem::Dependency
+  name: ruby-openai
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+description: Bots can use different tools to solve any number of tasks. Bots can be
+  powered by OpenAI.
+email:
+- jpaulson@aha.io
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/ruby_bots.rb
+- lib/ruby_bots/bot.rb
+- lib/ruby_bots/bots/openai_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/version.rb
+homepage: https://github.com/aha-app/ruby_bots
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/aha-app/ruby_bots
+  source_code_uri: https://github.com/aha-app/ruby_bots
+  changelog_uri: https://github.com/aha-app/ruby_bots
+  rubygems_mfa_required: 'true'
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.6.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.3.26
+signing_key:
+specification_version: 4
+summary: Ruby Bots lets you create tools and bots to complete tasks. Inspired by python
+  langchain.
+test_files: []