ai-chat 0.8.0 → 0.8.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 177299f56fc875c484f3b0552e93858c3f8da072ded5b81a67d891d9b7f7df3e
-  data.tar.gz: c31a180346f098d9c70fc135bdd6556ae12d23f4a85a5618a1907a8f660c0996
+  metadata.gz: b938b2b70b76afcd57c639ccc5ed37db9fb49bf42edc422801264d980a7b1d69
+  data.tar.gz: a9fa2ad8737b7e3fcd6279b55569302be657272050a766e23729cd2af0eb504d
 SHA512:
-  metadata.gz: a4adf79c5e126ee642511bd8a8d2fdbab92fef5d9a596cae6a416bed82861764c61ed80f17f5762e879dae4e8c604475e9c7f5bf82c70224b9bdc60552b936ad
-  data.tar.gz: ca02e9d88b13569899ce3787159862b8f55d2c822e54d41851f43b7a94d22a35bf25794e6dbff25e71d04dc6cdf0d4da6b74c53b9b40476d68d42d8645a97268
+  metadata.gz: d97939df71ccd5093a4b2d7b972eb4cbcdf6f71d1aab4e585ee738f4f005b55123f44aa2a3275d5d35f7693671072cf88a98afedfd8b7d6a38b81529e0142bab
+  data.tar.gz: c1655ee6eae4657c0e7cf727719a236fbea11a593993c4876b91d2e611d6757ce740bae7c77ad7bcdcd016071862be0744ba7fa3aff7ce750d13f62769cf0bb7

data/ai-chat.gemspec

@@ -2,7 +2,7 @@
 
 Gem::Specification.new do |spec|
   spec.name = "ai-chat"
-  spec.version = "0.8.0"
+  spec.version = "0.8.2"
   spec.authors = ["Raghu Betina", "Jelani Woods"]
   spec.email = ["raghu@firstdraft.com", "jelani@firstdraft.com"]
   spec.homepage = "https://github.com/firstdraft/ai-chat"

data/lib/ai/chat.rb

@@ -262,7 +262,13 @@ module AI
     end
 
     def inspect
-      AI.amazing_print(self, plain: !$stdout.tty?, multiline: true)
+      AI.amazing_print(self, plain: true, multiline: true)
+    end
+
+    # Same direct write as AI::Message#pretty_print, for the same reasons.
+    def pretty_print(q)
+      q.flush
+      q.output << AI.amazing_print(self, plain: !AI.color_printer?(q), multiline: true)
     end
 
     def to_html

data/lib/ai/items.rb

@@ -28,15 +28,17 @@ module AI
     end
 
     def inspect
-      build_output(html: false, plain: !$stdout.tty?)
+      build_output(html: false, plain: true)
     end
 
     def pretty_inspect
       "#{inspect}\n"
     end
 
+    # Same direct write as AI::Message#pretty_print, for the same reasons.
     def pretty_print(q)
-      q.output << inspect
+      q.flush
+      q.output << build_output(html: false, plain: !AI.color_printer?(q))
     end
 
     def method_missing(method_name, *args, &block)

data/lib/ai/message.rb

@@ -3,17 +3,21 @@
 module AI
   class Message < Hash
     def inspect
-      AI.amazing_print(display_hash, plain: !$stdout.tty?, index: false)
+      formatted(plain: true)
     end
 
     def pretty_inspect
       "#{inspect}\n"
     end
 
-    # IRB's ColorPrinter calls pretty_print and re-colorizes text,
-    # which escapes our ANSI codes. Write directly to output to bypass.
+    # The console's path, and so the only place color is added (see
+    # AI.color_printer?). IRB's ColorPrinter re-colorizes text, which
+    # escapes our ANSI codes, so write directly to output to bypass it,
+    # but flush first: inside an Array, PP is still buffering the "[" and
+    # earlier separators, so an unflushed write lands ahead of them.
     def pretty_print(q)
-      q.output << inspect
+      q.flush
+      q.output << formatted(plain: !AI.color_printer?(q))
     end
 
     def to_html
@@ -22,6 +26,10 @@ module AI
 
     private
 
+    def formatted(plain:)
+      AI.amazing_print(display_hash, plain: plain, index: false)
+    end
+
     def display_hash
       AI.truncate_data_uris(to_h)
     end

data/lib/ai-chat.rb

@@ -14,6 +14,25 @@ module AI
     AmazingPrint::Inspector.new(**options).awesome(object)
   end
 
+  # Whether pretty_print is drawing into a console that shows color: IRB
+  # (and so rails console) or Pry, with that console's color setting on.
+  # IRB always hands its printer colorize: true and leaves the real decision
+  # (a color terminal, and no --nocolorize) to IRB::Color.colorable?.
+  #
+  # inspect never adds color, because its String also ends up in places that
+  # can't read terminal escape codes: the Better Errors console in the
+  # browser, log files, ERB. Asking $stdout.tty? doesn't tell those apart: a
+  # Rails server started from a terminal has a TTY on $stdout, too.
+  def self.color_printer?(q)
+    if defined?(IRB::ColorPrinter) && q.is_a?(IRB::ColorPrinter)
+      q.instance_variable_get(:@colorize) != false && IRB::Color.colorable?
+    elsif defined?(Pry::ColorPrinter) && q.is_a?(Pry::ColorPrinter)
+      Pry.config.color
+    else
+      false
+    end
+  end
+
   # Recursively truncate base64 data URIs in nested structures for cleaner output
   def self.truncate_data_uris(obj)
     case obj

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ai-chat
 version: !ruby/object:Gem::Version
-  version: 0.8.0
+  version: 0.8.2
 platform: ruby
 authors:
 - Raghu Betina
@@ -156,7 +156,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 4.0.3
+rubygems_version: 4.0.10
 specification_version: 4
 summary: A beginner-friendly Ruby interface for OpenAI's API
 test_files: []