active_cortex 0.2.1 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d69240425c3faa7beaf76c5fc9b48beea6d9b5d3ec032ac01d874b5b240b1754
-  data.tar.gz: 2507aff2c4a6ed7b8d32cbcb2372d9fc3a759e5d43ad568df0633553a0825f66
+  metadata.gz: c1020861a0627f8c3df3db06459b51287e067cb8ecee1a3c9ee412e80aa374d9
+  data.tar.gz: 6e6c36fff9beec9b256f8bae8b72e118da33b4b6a4535cbe6574c09d8561489f
 SHA512:
-  metadata.gz: 15c2ba257de168c3aeedcc494ae796575d881e4dade4affc4017075bb0dc1678e544c98d058f5750606b64cfcd1f9ba24f274784e2b988c8c950af85e530f3fc
-  data.tar.gz: f78d0887c9f2890089fea953a4a199b31720fee26e603521022552035e2969b07631c8c52baced8b70b30639f70b3d5d4fbada1266b324254a2c6aff17c7c88d
+  metadata.gz: 6536b4dc37ff820006cc77d9d0f15255640dc6be9732bdbcbf1e7e1262f33b650f42c174f593592db979cc0fc8a9e47fbdbc6733c824a9fd50676a44bb42d21e
+  data.tar.gz: ea9d43ad9662b87f6262d2df69fedfbead2e5a171a6d7e8effd1f64d8043e8f3ab185eefc34e6d43dfd63c7c2fd94883c4f9137ec531ea1082156fbf71b9cd95

data/lib/active_cortex/generator/boolean.rb

@@ -0,0 +1,45 @@
+class ActiveCortex::Generator::Boolean < ActiveCortex::Generator
+  def self.accepts?(record:, field_name:)
+    record.class.attribute_types[field_name.to_s].type == :boolean
+  end
+
+  def self.generate(record:, field_name:)
+    record.send("#{field_name}=", generation)
+  end
+
+  def generation
+    openai_content || raise(ActiveCortex::Error, openai_error_message)
+  end
+
+  private
+
+  def convert_to_boolean(content)
+    case content
+    when "Yes", "yes", "True", "true", "1"
+      true
+    when "No", "no", "False", "false", "0"
+      false
+    else
+      raise ActiveCortex::Error, "Could not convert content to boolean: #{content}"
+    end
+  end
+
+  def openai_content
+    convert_to_boolean(openai_response["choices"][0]["message"]["content"])
+  rescue
+    nil
+  end
+
+  def openai_error_message
+    "Error from OpenAI. " + { response: openai_response }.to_json
+  end
+
+  def openai_response
+    @openai_response ||= openai_client.chat(parameters: {
+      model: model,
+      messages: [
+        { role: "user", content: prompt }
+      ],
+    })
+  end
+end

data/lib/active_cortex/version.rb

@@ -1,3 +1,3 @@
 module ActiveCortex
-  VERSION = "0.2.1"
+  VERSION = "0.3.0"
 end

data/lib/active_cortex.rb

@@ -4,6 +4,7 @@ require "active_cortex/railtie"
 require "active_cortex/generator"
 require "active_cortex/generator/text"
 require "active_cortex/generator/has_many"
+require "active_cortex/generator/boolean"
 require "active_cortex/model"
 
 module ActiveCortex

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: active_cortex
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.3.0
 platform: ruby
 authors:
 - Ori Marash
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-12-20 00:00:00.000000000 Z
+date: 2024-02-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -65,6 +65,7 @@ files:
 - lib/active_cortex.rb
 - lib/active_cortex/config.rb
 - lib/active_cortex/generator.rb
+- lib/active_cortex/generator/boolean.rb
 - lib/active_cortex/generator/has_many.rb
 - lib/active_cortex/generator/text.rb
 - lib/active_cortex/model.rb