ruby_bots 0.0.3 → 0.0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 656073bfe550e5be358a7189a137acd2fab90279f46a1d9c44899dddd022f844
-  data.tar.gz: 28bf1c1005c95aed8cb5f319c012a6ea4c54c8c1e2b5d9941af88bc6841eb9e3
+  metadata.gz: b09beb21193f74ab2940c6b117e245bc916a74aaf888d56d70d682aea97198e8
+  data.tar.gz: 7e72bb00d11cb236555a3b7cd187e850e68cdd62d3d44fefc2775b139df283f8
 SHA512:
-  metadata.gz: 3338bbd983626a62f6448ba4631a940932c2639abb00fb4e109ef47b2ad7b5f777371321f94b42f1ee6be510964a951f8ac8d181c6f953023ae2fb34c1ab334c
-  data.tar.gz: 1fb67357ceac7f6ceafeb0737a54e138e25350b79fc2aefc08278a9d1b83c4db391e4da669c53a390928cf56289c200ad943a86b8482a6362735857c8e04eb9b
+  metadata.gz: ab42068df4967ba2c05c9e4a9869a09e95ec09c633690873242ed157adba7beb798111a45a03d3eedfa45f28f8ce433ef5c148a1528c3cd72fcc578c3963154f
+  data.tar.gz: 59a8a478faa42a654a3abc9f37cd4ff6d103dc3e8049783b8bc0bf4748a99824f6b0f38bb6621361b16fd3c67aec66bb9f9e67ff6b5ff7219612b6dbcd7b59bc

data/lib/ruby_bots/bot.rb

@@ -7,7 +7,7 @@ module RubyBots
       super(**kwargs)
     end
 
-    def operate(inputs)
+    def run(inputs)
       raise NotImplementedError
     end
   end

data/lib/ruby_bots/bots/openai_bot.rb

@@ -10,7 +10,7 @@ module RubyBots
       @tools = tools
     end
     
-    def operate(inputs)
+    def run(inputs)
       params = {
         messages: [
           { role: :system, content: system_instructions },

data/lib/ruby_bots/bots/pipeline_bot.rb

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

data/lib/ruby_bots/bots/router_bot.rb

@@ -19,7 +19,7 @@ module RubyBots
       PROMPT
     end
       
-    def operate(inputs)
+    def run(inputs)
       params = {
         messages: [
           { role: :system, content: system_instructions },
@@ -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.operate(inputs)
+        selected_tool.response(inputs)
       else
         response_text
       end

data/lib/ruby_bots/tool.rb

@@ -1,20 +1,22 @@
 module RubyBots
   class Tool
     attr_accessor :name, :description, :errors
-    
+
     def initialize(name:, description:)
       @name = name
       @description = description
       @errors = []
-      @input_validators = []
-      @output_validators = []
+      @input_validators ||= []
+      @output_validators ||= []
     end
 
-    def validate_inputs(method)
+    def self.validate_inputs(method)
+      @input_validators ||= []
       @input_validators << method
     end
 
-    def validate_outputs(method)
+    def self.validate_outputs(method)
+      @output_validators ||= []
       @output_validators << method
     end
 

data/lib/ruby_bots/tools/openai_tool.rb

@@ -7,12 +7,11 @@ module RubyBots
     DEFAULT_DESCRIPTION = "This tool will use open ai to determine the output."
   
     def initialize(name: "OpenAI Tool", description: DEFAULT_DESCRIPTION)
-      @name = name
-      @description = description
+      super(name: name, description: description)
     end
   
     def client
-      @client ||= OpenAI::Client.new(access_token: ENV["OPENAI_ACCESS_TOKEN"])
+      @client ||= OpenAI::Client.new(access_token: RubyBots.config.openai_api_key)
     end
   
     def default_params
@@ -26,7 +25,7 @@ module RubyBots
       "You are a helpful assistant."
     end
     
-    def operate(inputs)
+    def run(inputs)
       params = {
         messages: [
           { role: :system, content: system_instructions },

data/lib/ruby_bots/version.rb

@@ -1,3 +1,3 @@
 module RubyBots
-  VERSION = "0.0.3".freeze
+  VERSION = "0.0.6".freeze
 end

data/lib/ruby_bots.rb

@@ -1,5 +1,19 @@
 module RubyBots
   class << self
+    def config
+      @config ||= Configuration.new
+    end
+
+    def configure
+      yield config
+    end
+  end
+
+  class Configuration
+    attr_accessor :openai_api_key
+    def initialize()
+      @openai_api_key = ENV["OPENAI_ACCESS_TOKEN"]
+    end
   end
 
   def self.bot
@@ -17,7 +31,7 @@ end
 
 require "ruby_bots/tool"
 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/tools/openai_tool.rb"
+require "ruby_bots/bots/router_bot.rb"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_bots
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.6
 platform: ruby
 authors:
 - Justin Paulson