Fix boolean generation saving

lib/active_cortex/generator/boolean.rb

@@ -3,7 +3,7 @@ def self.accepts?(record:, field_name:)
     record.class.attribute_types[field_name.to_s].type == :boolean
   end
 
-  def self.generate(record:, field_name:)
+  def save_generation(record:, field_name:)
     record.send("#{field_name}=", generation)
   end
 
@@ -14,10 +14,13 @@ def generation
   private
 
   def convert_to_boolean(content)
+    content = content.downcase
+      .gsub(/[^a-z]/, "")
+
     case content
-    when "Yes", "yes", "True", "true", "1"
+    when "yes", "true", "1"
       true
-    when "No", "no", "False", "false", "0"
+    when "no", "false", "0"
       false
     else
       raise ActiveCortex::Error, "Could not convert content to boolean: #{content}"
@@ -26,6 +29,8 @@ def convert_to_boolean(content)
 
   def openai_content
     convert_to_boolean(openai_response["choices"][0]["message"]["content"])
+  rescue ActiveCortex::Error => e
+    raise e
   rescue
     nil
   end

spec/active_cortex/generator/boolean_spec.rb

@@ -17,4 +17,11 @@
       expect(result).to be_a(TrueClass).or be_a(FalseClass)
     end
   end
+
+  context "when saving the generation" do
+    it "saves the generation to the record", :vcr do
+      generator.save_generation(record: book, field_name: :completed)
+      expect(book.completed).to be_a(TrueClass).or be_a(FalseClass)
+    end
+  end
 end

spec/dummy/db/migrate/20231026194012_create_books.rb

@@ -3,6 +3,7 @@ def change
     create_table :books do |t|
       t.string :title, null: false
       t.string :summary
+      t.boolean :completed, default: false
 
       t.timestamps
     end

spec/dummy/db/schema.rb

@@ -14,6 +14,7 @@
   create_table "books", force: :cascade do |t|
     t.string "title", null: false
     t.string "summary"
+    t.boolean "completed", default: false
     t.datetime "created_at", null: false
     t.datetime "updated_at", null: false
   end