ai-chat 0.5.1 → 0.5.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 77efcbc43e3402b184ce49ff870671796904a0935a37d55dd9eeaaea93572b53
-  data.tar.gz: 8996a14db31e7455815d10acac9827658fcbaf5358e1a461d2b83a946b8f3cc9
+  metadata.gz: 9012bf87a54da64ccab695a2d440249cf691f2f5943272c918065b24137fa6fb
+  data.tar.gz: 9d3931249dc62145f8eb3cc3fc44382a038c901b8bfc4a2ccda0d01d8653489d
 SHA512:
-  metadata.gz: b1993b6e812a75387b53ade7f29785a80aa0c70c7b6aa97d5ac9eed7e73fc27a87f1a76676e9878416849af3c9ae3967d6b78838be0c681f077fa63c81e32993
-  data.tar.gz: 0f463e8beef43965db7b85b94335624322f46bc39cc4dd2c1a02ad44993ae972bdbfb4a670890b3f55ddf9ce5ebf57fd8b2e92f85fd3340a2be1d7aa6d1ca549
+  metadata.gz: d9447afcbff626605c7b865675cd28d598e2c3cee19010aa86a9955b5a3cf109ea68ea40765bfe9aa92f43d823710eb91f7d4b3cfd92d4105c02f6ad9e45bdc9
+  data.tar.gz: 5ae2565d62ae358f62938a5b888e55d73bc581449e0f0e1c4de836f713832c1bf19fe0e9ba6d8ae0c4b108323532ab5e905dc22079e0aa4ecc833b6a9b884386

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.4"
   spec.authors = ["Raghu Betina", "Jelani Woods"]
   spec.email = ["raghu@firstdraft.com", "jelani@firstdraft.com"]
   spec.homepage = "https://github.com/firstdraft/ai-chat"
@@ -18,18 +18,15 @@ Gem::Specification.new do |spec|
     "source_code_uri" => "https://github.com/firstdraft/ai-chat"
   }
 
-  spec.required_ruby_version = "~> 3.2"
-  spec.add_runtime_dependency "openai", "~> 0.34"
+  spec.required_ruby_version = ">= 3.2"
+  spec.add_runtime_dependency "openai", "~> 0.43"
   spec.add_runtime_dependency "marcel", "~> 1.0"
   spec.add_runtime_dependency "base64", "~> 0.1", "> 0.1.1"
   spec.add_runtime_dependency "json", "~> 2.0"
-  spec.add_runtime_dependency "ostruct", "~> 0.2"
   spec.add_runtime_dependency "tty-spinner", "~> 0.9.3"
-  spec.add_runtime_dependency "amazing_print", "~> 1.8"
+  spec.add_runtime_dependency "amazing_print", "~> 2.0"
 
   spec.add_development_dependency "dotenv", ">= 1.0.0"
-  spec.add_development_dependency "refinements", "~> 11.1"
-
   spec.extra_rdoc_files = Dir["README*", "LICENSE*"]
   spec.files = Dir["*.gemspec", "lib/**/*"]
 end

data/lib/ai/chat.rb

@@ -4,7 +4,6 @@ require "base64"
 require "json"
 require "marcel"
 require "openai"
-require "ostruct"
 require "pathname"
 require "stringio"
 require "fileutils"
@@ -157,15 +156,14 @@ module AI
 
     def proxy=(value)
       @proxy = value
-      if value
-        @client = OpenAI::Client.new(
+      @client = if value
+        OpenAI::Client.new(
           api_key: @api_key,
           base_url: BASE_PROXY_URL
         )
       else
-        @client = OpenAI::Client.new(api_key: @api_key)
+        OpenAI::Client.new(api_key: @api_key)
       end
-      value
     end
 
     def schema=(value)
@@ -208,8 +206,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 +226,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
@@ -525,8 +523,8 @@ module AI
       image_filenames = []
 
       image_outputs = response.output.select { |output|
-          output.respond_to?(:type) && output.type == :image_generation_call
-        }
+        output.respond_to?(:type) && output.type == :image_generation_call
+      }
 
       return image_filenames if image_outputs.empty?
 
@@ -574,7 +572,7 @@ module AI
       @api_key_validated = true
     rescue OpenAI::Errors::AuthenticationError
       message = if proxy
-         <<~STRING
+        <<~STRING
           It looks like you're using an invalid API key. Proxying is enabled, so you must use an OpenAI API key from prepend.me. Please disable proxy or update your API key before generating a response.
         STRING
       else

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.4
 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
@@ -16,14 +16,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.34'
+        version: '0.43'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.34'
+        version: '0.43'
 - !ruby/object:Gem::Dependency
   name: marcel
   requirement: !ruby/object:Gem::Requirement
@@ -72,20 +72,6 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '2.0'
-- !ruby/object:Gem::Dependency
-  name: ostruct
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '0.2'
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '0.2'
 - !ruby/object:Gem::Dependency
   name: tty-spinner
   requirement: !ruby/object:Gem::Requirement
@@ -106,14 +92,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.8'
+        version: '2.0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.8'
+        version: '2.0'
 - !ruby/object:Gem::Dependency
   name: dotenv
   requirement: !ruby/object:Gem::Requirement
@@ -128,28 +114,14 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 1.0.0
-- !ruby/object:Gem::Dependency
-  name: refinements
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '11.1'
-  type: :development
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '11.1'
 email:
 - raghu@firstdraft.com
 - jelani@firstdraft.com
 executables: []
 extensions: []
 extra_rdoc_files:
-- README.md
 - LICENSE
+- README.md
 files:
 - LICENSE
 - README.md
@@ -175,7 +147,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 +156,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: []