ruby_bots 0.0.1 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 31af013343283ed662eea6678279251ccbd7bda130b8633f05c173cb6186fa62
-  data.tar.gz: b273bf6290b5c5b2cc00b2abf139b70393f5ee804a9b1e8e161a6f302a8d5c52
+  metadata.gz: 656073bfe550e5be358a7189a137acd2fab90279f46a1d9c44899dddd022f844
+  data.tar.gz: 28bf1c1005c95aed8cb5f319c012a6ea4c54c8c1e2b5d9941af88bc6841eb9e3
 SHA512:
-  metadata.gz: cecf81c9e2d26a1f07fca2188c5f30230fd75b0040f5632b0f63416d4d177db97694ad0a66a11e084f3306772987ffa14785025ec6761503a7bfbd62413294b6
-  data.tar.gz: f2007a6dfbace71b3747cbe503c36fc45337806e71aa213a2fd00595f494bb431da266aa6320356801f8bc475fe13291cfb7936fc4fdb7b51946c8ab9394a7f8
+  metadata.gz: 3338bbd983626a62f6448ba4631a940932c2639abb00fb4e109ef47b2ad7b5f777371321f94b42f1ee6be510964a951f8ac8d181c6f953023ae2fb34c1ab334c
+  data.tar.gz: 1fb67357ceac7f6ceafeb0737a54e138e25350b79fc2aefc08278a9d1b83c4db391e4da669c53a390928cf56289c200ad943a86b8482a6362735857c8e04eb9b

data/lib/ruby_bots/bots/openai_bot.rb

@@ -2,24 +2,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) 
+    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 = client.chat(parameters: params)
 
-    response.dig("choices", 0, "message", "content")
+      response.dig("choices", 0, "message", "content")
+    end
   end
 end

data/lib/ruby_bots/tool.rb

@@ -1,23 +1,37 @@
 module RubyBots
   class Tool
-    attr_accessor :name, :description
+    attr_accessor :name, :description, :errors
     
     def initialize(name:, description:)
       @name = name
       @description = description
+      @errors = []
+      @input_validators = []
+      @output_validators = []
     end
 
-    def validate_inputs(inputs)
-      raise NotImplementedError
+    def validate_inputs(method)
+      @input_validators << method
+    end
+
+    def validate_outputs(method)
+      @output_validators << method
     end
 
     def run(inputs)
       raise NotImplementedError
     end
 
-    def response(inputs)
-      validate_inputs(inputs)
-      run(inputs)
+    def response(input)
+      @input_validators.each do |validator|
+        validator.call(input)
+      end
+      
+      output = run(input)
+
+      @output_validators.each do |validator|
+        validator.call(output)
+      end
     end
   end
 end

data/lib/ruby_bots/tools/openai_tool.rb

@@ -2,7 +2,7 @@ require 'openai'
 
 module RubyBots
   class OpenAITool < Tool
-    attr_accessor :name, :description
+    validate_inputs :input_is_json
 
     DEFAULT_DESCRIPTION = "This tool will use open ai to determine the output."
   
@@ -38,5 +38,11 @@ module RubyBots
       
       response.dig("choices", 0, "message", "content")
     end
+
+    def input_is_json(input)
+      JSON.parse(input)
+    rescue JSON::ParserError
+      errors.add(:input, "must be valid JSON")
+    end
   end
 end

data/lib/ruby_bots/version.rb

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

data/lib/ruby_bots.rb

@@ -9,12 +9,15 @@ module RubyBots
   def self.tool
     RubyBots::Tool
   end
+  
+  class Error < StandardError; end
+  class InvalidInputError < Error; end
+  class InvalidOutputError < Error; 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"
+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

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby_bots
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.3
 platform: ruby
 authors:
 - Justin Paulson
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-03-31 00:00:00.000000000 Z
+date: 2023-04-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: debug