boxcars 0.3.2 → 0.3.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 463f57f1436cfea29e0a60bbfce071afd355a5d0cadbe1d69f96adc7393eacd6
-  data.tar.gz: 282e528fb8cb8b532b621db5c2e7ce2b82c4607057c01f0feb1eaacfe98ba09b
+  metadata.gz: 4c6fc81621333c6663dd04a49215f01a1fbd5fa247551b80b652676a4c06f155
+  data.tar.gz: a956727b4b986d42cc9f7b2b609dba561473f1976cce5f16ce2db641de778dfc
 SHA512:
-  metadata.gz: c76d2772db0925f71779c0bac1a2e92af20bcf398cc6e7cf943edeadb2014426554fba607b034fd68ac88bd26f2d42eef114de57332e83a4fe76558a9b2f0cea
-  data.tar.gz: '0489fffef87c32fbf6a7a9af4fbca77ef991437353211174617b6ea8e229e49accc85c39af93414ce1628a4472cc208d8075570abba5e8269b30cfb66000a40e'
+  metadata.gz: cac0c355836d68387a0fe66e7c65efe341c4ec746627eb275411198a0ba0092291e7add9cb8fbd93aee00da1d61420e36eaa1c90d97621629f3db5f56c0b134d
+  data.tar.gz: 799149ab51920e7d95d136ad5b4e57ac98c7e344ce277bcbd6e0ee8d89e8cc0d8eea23f37d821712818b16eacf8f0b236bb8c5730633250dc02c209b647f7b8f

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    boxcars (0.3.2)
+    boxcars (0.3.3)
       google_search_results (~> 2.2)
       gpt4all (~> 0.0.4)
       hnswlib (~> 0.8)

data/lib/boxcars/boxcar/url_text.rb

@@ -18,11 +18,20 @@ module Boxcars
     # @return [String] The text for the url.
     def run(url)
       url = URI.parse(url)
-      get_answer(url)
+      do_encoding(get_answer(url))
     end
 
     private
 
+    def do_encoding(answer)
+      if answer.is_a?(Result)
+        answer.explanation = answer.explanation.encode(xml: :text)
+        answer
+      else
+        answer.encode(xml: :text)
+      end
+    end
+
     def html_to_text(url, response)
       Nokogiri::HTML(response.body).css(%w[h1 h2 h3 h4 h5 h6 p a].join(",")).map do |e|
         itxt = e.inner_text.strip

data/lib/boxcars/observation.rb

@@ -29,7 +29,12 @@ module Boxcars
 
     # @return [String] An explanation of the result
     def to_s
-      note
+      note.to_s
+    end
+
+    # @return [String] An explanation of the result
+    def to_text
+      to_s
     end
 
     # create a new Observaton from a text string with a status of :ok

data/lib/boxcars/result.rb

@@ -3,7 +3,7 @@
 module Boxcars
   # used by Boxcars to return structured result and additional context
   class Result
-    attr_reader :status, :answer, :explanation, :suggestions, :added_context
+    attr_accessor :status, :answer, :explanation, :suggestions, :added_context
 
     # @param status [Symbol] :ok or :error
     # @param answer [String] The answer to the question

data/lib/boxcars/train/xml_train.rb

@@ -13,6 +13,7 @@ module Boxcars
     # @param kwargs [Hash] Additional arguments including: name, description, top_k, return_direct, and stop
     # @abstract
     def initialize(boxcars:, prompt:, engine: nil, **kwargs)
+      @using_xml = true
       super
     end
 
@@ -41,7 +42,9 @@ module Boxcars
     end
 
     def build_output(text)
-      if text =~ /#{close_tag(thought_prefix)}/
+      if text.end_with?("</usetool>")
+        "<data>#{engine_prefix}#{text}</output></data>"
+      elsif text =~ /#{close_tag(thought_prefix)}/
         "<data>#{engine_prefix}#{text}</data>"
       else
         "<data>#{text}</data>"

data/lib/boxcars/train.rb

@@ -3,7 +3,7 @@
 module Boxcars
   # @abstract
   class Train < EngineBoxcar
-    attr_reader :boxcars, :return_values, :return_intermediate_steps,
+    attr_reader :boxcars, :return_values, :return_intermediate_steps, :using_xml,
                 :max_iterations, :early_stopping_method, :name_to_boxcar_map,
                 :observation_prefix, :thought_prefix, :final_answer_prefix, :answer_prefix, :question_prefix, :engine_prefix
 
@@ -184,6 +184,18 @@ module Boxcars
       end
     end
 
+    def get_boxcar_result(boxcar, boxcar_input)
+      boxcar_result = boxcar.run(boxcar_input)
+      return boxcar_result unless using_xml
+
+      if boxcar_result.is_a?(Result)
+        boxcar_result.answer = boxcar_result.answer.encode(xml: :text)
+      else
+        boxcar_result = boxcar_result.encode(xml: :text)
+      end
+      boxcar_result
+    end
+
     # execute the train train
     # @param inputs [Hash] The inputs.
     # @return [Hash] The output.
@@ -197,7 +209,7 @@ module Boxcars
 
         if (boxcar = name_to_boxcar_map[output.boxcar])
           begin
-            observation = Observation.ok(boxcar.run(output.boxcar_input))
+            observation = Observation.ok(get_boxcar_result(boxcar, output.boxcar_input))
             return_direct = boxcar.return_direct
           rescue Boxcars::ConfigurationError, Boxcars::SecurityError => e
             raise e
@@ -212,9 +224,7 @@ module Boxcars
           observation = Observation.err("Error - #{output.boxcar} is not a valid action, try again.")
           return_direct = false
         end
-        # rubocop:disable Lint/RedundantStringCoercion
-        Boxcars.debug "Observation: #{observation.to_s}", :green
-        # rubocop:enable Lint/RedundantStringCoercion
+        Boxcars.debug "Observation: #{observation}", :green
         intermediate_steps.append([output, observation])
         if return_direct
           output = TrainFinish.new({ return_values[0] => observation }, "")

data/lib/boxcars/version.rb

@@ -2,5 +2,5 @@
 
 module Boxcars
   # The current version of the gem.
-  VERSION = "0.3.2"
+  VERSION = "0.3.3"
 end

data/lib/boxcars/x_node.rb

@@ -10,7 +10,6 @@ module Boxcars
       @node = node
       @valid_names = []
       @children = {}
-      # @attributes = node.attributes.transform_values(&:value)
       @attributes = node.attributes.values.to_h { |a| [a.name.to_sym, a.value] }
 
       node.children.each do |child|
@@ -30,8 +29,13 @@ module Boxcars
 
     def self.from_xml(xml)
       doc = Nokogiri::XML.parse(xml)
-      raise XmlError, "XML is not valid: #{doc.errors.map { |e| "#{e.line}:#{e.column} #{e.message}" }}" if doc.errors.any?
-
+      if doc.errors.any?
+        Boxcars.debug("XML: #{xml}", :yellow)
+        # rubocop:disable Lint/Debugger
+        debugger if ENV.fetch("DEBUG_XML", false)
+        # rubocop:enable Lint/Debugger
+        raise XmlError, "XML is not valid: #{doc.errors.map { |e| "#{e.line}:#{e.column} #{e.message}" }}"
+      end
       XNode.new(doc.root)
     end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: boxcars
 version: !ruby/object:Gem::Version
-  version: 0.3.2
+  version: 0.3.3
 platform: ruby
 authors:
 - Francis Sullivan