ai-chat 0.5.1 → 0.5.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 77efcbc43e3402b184ce49ff870671796904a0935a37d55dd9eeaaea93572b53
-  data.tar.gz: 8996a14db31e7455815d10acac9827658fcbaf5358e1a461d2b83a946b8f3cc9
+  metadata.gz: 1693a4828234dc84df802bd990518caf2bb235ce71753b471216b143aa90eff0
+  data.tar.gz: 574821abde32e11bd88a2651d4e0b1ed88a1fc6f76703024f2bbde4622b6306a
 SHA512:
-  metadata.gz: b1993b6e812a75387b53ade7f29785a80aa0c70c7b6aa97d5ac9eed7e73fc27a87f1a76676e9878416849af3c9ae3967d6b78838be0c681f077fa63c81e32993
-  data.tar.gz: 0f463e8beef43965db7b85b94335624322f46bc39cc4dd2c1a02ad44993ae972bdbfb4a670890b3f55ddf9ce5ebf57fd8b2e92f85fd3340a2be1d7aa6d1ca549
+  metadata.gz: 6046630107863b4f79b17b1c95384af5335a3769800fe8dfa8dbb5a5a28dea4ed3797b1d40e391f79c4567fb83eaef18eba3cbb3ef4f5bab796a0cda2af3d393
+  data.tar.gz: f8836dfc5b2c50c2682a865f97e52f5f5e6f6569a1b1a4de09619f3d9a12ee9184c8e9fd06f5004c5f3d62171adef1aaa8fa929336e5c335c29ffe72e25648d2

data/ai-chat.gemspec

@@ -2,7 +2,7 @@
 
 Gem::Specification.new do |spec|
   spec.name = "ai-chat"
-  spec.version = "0.5.1"
+  spec.version = "0.5.3"
   spec.authors = ["Raghu Betina", "Jelani Woods"]
   spec.email = ["raghu@firstdraft.com", "jelani@firstdraft.com"]
   spec.homepage = "https://github.com/firstdraft/ai-chat"
@@ -18,7 +18,7 @@ Gem::Specification.new do |spec|
     "source_code_uri" => "https://github.com/firstdraft/ai-chat"
   }
 
-  spec.required_ruby_version = "~> 3.2"
+  spec.required_ruby_version = ">= 3.2"
   spec.add_runtime_dependency "openai", "~> 0.34"
   spec.add_runtime_dependency "marcel", "~> 1.0"
   spec.add_runtime_dependency "base64", "~> 0.1", "> 0.1.1"

data/lib/ai/chat.rb

@@ -208,8 +208,8 @@ module AI
       attrs << [:@conversation_id, @conversation_id]
       attrs << [:@last_response_id, @last_response_id] if @last_response_id
 
-      # 3. Messages (the main content, without response details)
-      display_messages = @messages.map { |msg| msg.except(:response) }
+      # 3. Messages (the main content, without response details, with truncated data URIs)
+      display_messages = @messages.map { |msg| AI.truncate_data_uris(msg.except(:response)) }
       attrs << [:@messages, display_messages]
 
       # 4. Optional features (only if enabled/changed from default)
@@ -228,11 +228,11 @@ module AI
     end
 
     def inspect
-      ai(plain: !$stdout.tty?, multiline: true)
+      AI.amazing_print(self, plain: !$stdout.tty?, multiline: true)
     end
 
     def to_html
-      AI.wrap_html(ai(html: true, multiline: true))
+      AI.wrap_html(AI.amazing_print(self, html: true, multiline: true))
     end
 
     def pretty_inspect

data/lib/ai/items.rb

@@ -29,7 +29,7 @@ module AI
 
     def build_output(html: false, plain: false)
       box = build_box
-      items_output = data.ai(html: html, plain: plain, limit: 100, indent: 2, index: true)
+      items_output = AI.amazing_print(data, html: html, plain: plain, limit: 100, indent: 2, index: true)
 
       if html
         "<pre>#{box}</pre>\n#{items_output}"

data/lib/ai/message.rb

@@ -3,7 +3,7 @@
 module AI
   class Message < Hash
     def inspect
-      ai(plain: !$stdout.tty?, index: false)
+      AI.amazing_print(display_hash, plain: !$stdout.tty?, index: false)
     end
 
     def pretty_inspect
@@ -17,7 +17,13 @@ module AI
     end
 
     def to_html
-      AI.wrap_html(ai(html: true, index: false))
+      AI.wrap_html(AI.amazing_print(display_hash, html: true, index: false))
+    end
+
+    private
+
+    def display_hash
+      AI.truncate_data_uris(to_h)
     end
   end
 end

data/lib/ai-chat.rb

@@ -1,10 +1,43 @@
 module AI
-  HTML_PRE_STYLE = "background-color: #1e1e1e; color: #d4d4d4; padding: 1em; white-space: pre-wrap; word-wrap: break-word;"
+  HTML_PRE_STYLE = "background-color: #1e1e1e; color: #d4d4d4; padding: 1em; " \
+                   "border-radius: 4px; overflow-x: auto; " \
+                   "white-space: pre-wrap; word-wrap: break-word;"
 
-  def self.wrap_html(html)
-    html = html.gsub("<pre>", "<pre style=\"#{HTML_PRE_STYLE}\">")
+  def self.wrap_html(content)
+    html = "<pre style=\"#{HTML_PRE_STYLE}\">#{content}</pre>"
     html.respond_to?(:html_safe) ? html.html_safe : html
   end
+
+  # Use AmazingPrint::Inspector directly to avoid conflicts with awesome_print gem
+  # which also defines an `ai` method on Object
+  def self.amazing_print(object, **options)
+    AmazingPrint::Inspector.new(**options).awesome(object)
+  end
+
+  # Recursively truncate base64 data URIs in nested structures for cleaner output
+  def self.truncate_data_uris(obj)
+    case obj
+    when Hash
+      obj.transform_values { |v| truncate_data_uris(v) }
+    when Array
+      obj.map { |v| truncate_data_uris(v) }
+    when String
+      truncate_data_uri(obj)
+    else
+      obj
+    end
+  end
+
+  def self.truncate_data_uri(str)
+    return str unless str.is_a?(String) && str.start_with?("data:") && str.include?(";base64,")
+
+    match = str.match(/\A(data:[^;]+;base64,)(.+)\z/)
+    return str unless match
+
+    prefix = match[1]
+    data = match[2]
+    "#{prefix}#{data[0, 20]}... (#{data.length} chars)"
+  end
 end
 
 require_relative "ai/message"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ai-chat
 version: !ruby/object:Gem::Version
-  version: 0.5.1
+  version: 0.5.3
 platform: ruby
 authors:
 - Raghu Betina
 - Jelani Woods
 bindir: bin
 cert_chain: []
-date: 2026-01-29 00:00:00.000000000 Z
+date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: openai
@@ -148,8 +148,8 @@ email:
 executables: []
 extensions: []
 extra_rdoc_files:
-- README.md
 - LICENSE
+- README.md
 files:
 - LICENSE
 - README.md
@@ -175,7 +175,7 @@ require_paths:
 - lib
 required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
-  - - "~>"
+  - - ">="
     - !ruby/object:Gem::Version
       version: '3.2'
 required_rubygems_version: !ruby/object:Gem::Requirement
@@ -184,7 +184,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.2
+rubygems_version: 4.0.3
 specification_version: 4
 summary: A beginner-friendly Ruby interface for OpenAI's API
 test_files: []