active_cortex 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ca05346fe4261874164f6fbf48a473b4e310f2cafa206721b2f0ddb2dcfdff5c
-  data.tar.gz: 2a0163d3e56f87d416a56a105ad3c29af488de955445180d0adafa347d7d1c99
+  metadata.gz: c1020861a0627f8c3df3db06459b51287e067cb8ecee1a3c9ee412e80aa374d9
+  data.tar.gz: 6e6c36fff9beec9b256f8bae8b72e118da33b4b6a4535cbe6574c09d8561489f
 SHA512:
-  metadata.gz: 5f4b74059e5887c179f7c330dd1a47aa87e68521b3e63a1283359ae4067cac187097890a52bd4b931b5b0cbcb6bdbecae16e985a9c16a708ecc3b78b77be6554
-  data.tar.gz: c585a12f113acf4086389fe67f20a1558e054c9f1ad3aa6e87ffb22de42f31e385ce9c883b2041a3a619bfdb95feab2d4ada8f54e0ea6b43f53713742d3d24f8
+  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/model.rb

@@ -53,7 +53,7 @@ module ActiveCortex::Model
 
     def validate_arguments!(field_name, prompt, max_results, model)
       raise ArgumentError, "field_name must be a symbol or string" unless field_name.is_a?(Symbol) || field_name.is_a?(String)
-      raise ArgumentError, "prompt must be a proc" unless prompt.is_a?(Proc)
+      raise ArgumentError, "prompt must be a proc" unless prompt.is_a?(Proc) || prompt.is_a?(Symbol)
       raise ArgumentError, "max_results must be a number" unless max_results.nil? || max_results.is_a?(Integer)
       raise ArgumentError, "model must be a string" unless model.is_a?(String)
     end

data/lib/active_cortex/version.rb

@@ -1,3 +1,3 @@
 module ActiveCortex
-  VERSION = "0.2.0"
+  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.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Ori Marash
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-11-15 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