ruby_bots 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 31af013343283ed662eea6678279251ccbd7bda130b8633f05c173cb6186fa62
-  data.tar.gz: b273bf6290b5c5b2cc00b2abf139b70393f5ee804a9b1e8e161a6f302a8d5c52
+  metadata.gz: 97bf25557fb288a1416c47c9e3e566e437c04c5d2c19fedab06ae83837e24408
+  data.tar.gz: 26bb0781e3d5c2580b385539ae041eda38fb6ce2d323df20e80260780df78b88
 SHA512:
-  metadata.gz: cecf81c9e2d26a1f07fca2188c5f30230fd75b0040f5632b0f63416d4d177db97694ad0a66a11e084f3306772987ffa14785025ec6761503a7bfbd62413294b6
-  data.tar.gz: f2007a6dfbace71b3747cbe503c36fc45337806e71aa213a2fd00595f494bb431da266aa6320356801f8bc475fe13291cfb7936fc4fdb7b51946c8ab9394a7f8
+  metadata.gz: 7cefa94ef3b53ea79d46d39fb8256a3d96e96db852f45ae51feed2c58e0826fd2ba536acb43a1c801fdae83de8694be7f30e9e6541c344228d4bda1cac4dcbf7
+  data.tar.gz: 2b2afd2c55c6878d52a5d9675bbcfc39b90a6df8374482fc06e762c68dec48afd88bf5282a36aa48859e77ec5697b7911e8a755fe5367d29734424225da701b7

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.2".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.2
 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