ai-chat 0.2.2 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1b59ea831d946cd885c9df7c8524d0bd4cf26361ad9190780969698ac8db8b36
-  data.tar.gz: e21d404bc9c5e0ab5e1f0951053a6c159393f81f743460fdafbca0d032542895
+  metadata.gz: c2397a7d43d950bf70abd84c83d0e57b2cadb70fc127985fae4da4a9ecf142b8
+  data.tar.gz: d44a9d1999ce48cd0af92f367f5f0e2f586b8c19d1fc9e95ce8c7aa7e3fffa84
 SHA512:
-  metadata.gz: a94b918c1feecb3e9983a0d3c0411469084f3529167a45119ff77609adad1efc1c7ae2907817a6737ae865507d6e84eeeb8e24387b4da27fede223d3ed20f5a4
-  data.tar.gz: c5f9508ac39c6b952724ebf9ac7dbef96418e5f59f8ce520f314a5255dad4065b500537daa3f108fe8640f9f015b7c2c6b48722dfaa1ddf641bbee8b4a4d5164
+  metadata.gz: fdb13f4e405f46a21591f71ae818544bc60f7bc083e4b3eb9a8a6d57b9eba03995fe8b75828f3972e564dfa3ed3ed09aed4a8b7ae2260b3602cbede9d80afd8e
+  data.tar.gz: 427b5db79e006d0f6f9b20b83472a78c3234118807e4d43d472304426ca2d80d5b189d029924cd548d7004206914a4e63fd52ca93b1944e55e81276309a501d9

data/README.md

@@ -88,7 +88,7 @@ a.generate! # => "Matz is nice and so we are nice" (or similar)
 pp a.messages
 # => [
 #      {:role=>"user", :content=>"If the Ruby community had an official motto, what might it be?"},
-#      {:role=>"assistant", :content=>"Matz is nice and so we are nice", :response => #<AI::Chat::Response id=resp_abc... model=gpt-4.1-nano tokens=12>}
+#      {:role=>"assistant", :content=>"Matz is nice and so we are nice", :response => { id=resp_abc... model=gpt-4.1-nano tokens=12 } }
 #    ]
 
 # Continue the conversation
@@ -108,7 +108,7 @@ That's it! You're building something like this:
 [
   {:role => "system", :content => "You are a helpful assistant"},
   {:role => "user", :content => "Hello!"},
-  {:role => "assistant", :content => "Hi there! How can I help you today?", :response => #<AI::Chat::Response id=resp_abc... model=gpt-4.1-nano tokens=12>}
+  {:role => "assistant", :content => "Hi there! How can I help you today?", :response => { id=resp_abc... model=gpt-4.1-nano tokens=12 } }
 ]
 ```
 
@@ -577,14 +577,14 @@ pp t.messages.last
 # => {
 #      :role => "assistant",
 #      :content => "Hello! How can I help you today?",
-#      :response => #<AI::Response id=resp_abc... model=gpt-4.1-nano tokens=12>
+#      :response => { id=resp_abc... model=gpt-4.1-nano tokens=12 }
 #    }
 
 # Access detailed information
 response = t.last[:response]
-response.id           # => "resp_abc123..."
-response.model        # => "gpt-4.1-nano"
-response.usage        # => {:prompt_tokens=>5, :completion_tokens=>7, :total_tokens=>12}
+response[:id]           # => "resp_abc123..."
+response[:model]        # => "gpt-4.1-nano"
+response[:usage]        # => {:prompt_tokens=>5, :completion_tokens=>7, :total_tokens=>12}
 ```
 
 This information is useful for:
@@ -599,7 +599,7 @@ You can also, if you know a response ID, continue an old conversation by setting
 t = AI::Chat.new
 t.user("Hello!")
 t.generate!
-old_id = t.last[:response].id # => "resp_abc123..."
+old_id = t.last[:response][:id] # => "resp_abc123..."
 
 # Some time in the future...
 

data/ai-chat.gemspec

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

data/lib/ai/amazing_print.rb

@@ -0,0 +1,71 @@
+require "amazing_print"
+
+module AmazingPrint
+  module AI
+    def self.included(base)
+      base.send :alias_method, :cast_without_ai, :cast
+      base.send :alias_method, :cast, :cast_with_ai
+    end
+
+    def cast_with_ai(object, type)
+      case object
+      when ::AI::Chat
+        :ai_object
+      else
+        cast_without_ai(object, type)
+      end
+    end
+
+    private
+
+    def awesome_ai_object(object)
+      case object
+      when ::AI::Chat
+        format_ai_chat(object)
+      else
+        awesome_object(object)
+      end
+    end
+
+    def format_ai_chat(chat)
+      vars = []
+      
+      # Format messages with truncation
+      if chat.instance_variable_defined?(:@messages)
+        messages = chat.instance_variable_get(:@messages).map do |msg|
+          truncated_msg = msg.dup
+          if msg[:content].is_a?(String) && msg[:content].length > 80
+            truncated_msg[:content] = msg[:content][0..77] + "..."
+          end
+          truncated_msg
+        end
+        vars << ["@messages", messages]
+      end
+      
+      # Add other variables (except sensitive ones)
+      skip_vars = [:@api_key, :@client, :@messages]
+      chat.instance_variables.sort.each do |var|
+        next if skip_vars.include?(var)
+        value = chat.instance_variable_get(var)
+        vars << [var.to_s, value] unless value.nil?
+      end
+
+      format_object(chat, vars)
+    end
+
+    def format_object(object, vars)
+      data = vars.map do |(name, value)|
+        name = colorize(name, :variable) unless @options[:plain]
+        "#{name}: #{inspector.awesome(value)}"
+      end
+
+      if @options[:multiline]
+        "#<#{object.class}\n#{data.map { |line| "  #{line}" }.join("\n")}\n>"
+      else
+        "#<#{object.class} #{data.join(', ')}>"
+      end
+    end
+  end
+end
+
+AmazingPrint::Formatter.send(:include, AmazingPrint::AI)

data/lib/ai/chat.rb

@@ -8,8 +8,6 @@ require "pathname"
 require "stringio"
 require "fileutils"
 
-require_relative "response"
-
 module AI
   # :reek:MissingSafeMethod { exclude: [ generate! ] }
   # :reek:TooManyMethods
@@ -101,13 +99,18 @@ module AI
     def generate!
       response = create_response
 
-      chat_response = Response.new(response)
-
       text_response = extract_text_from_response(response)
 
       image_filenames = extract_and_save_images(response)
-
-      chat_response.images = image_filenames
+      response_usage = response.usage.to_h.slice(:input_tokens, :output_tokens, :total_tokens)
+
+      chat_response = {
+        id: response.id,
+        model: response.model,
+        usage: response_usage,
+        total_tokens: response_usage[:total_tokens],
+        images: image_filenames
+      }
 
       message = if schema
         if text_response.nil? || text_response.empty?
@@ -131,7 +134,7 @@ module AI
         )
       end
 
-      self.previous_response_id = response.id
+      self.previous_response_id = chat_response[:id]
 
       message
     end
@@ -173,6 +176,38 @@ module AI
       "#<#{self.class.name} @messages=#{messages.inspect} @model=#{@model.inspect} @schema=#{@schema.inspect} @reasoning_effort=#{@reasoning_effort.inspect}>"
     end
 
+    # Support for Ruby's pp (pretty print)
+    def pretty_print(q)
+      q.group(1, "#<#{self.class}", '>') do
+        q.breakable
+        
+        # Show messages with truncation
+        q.text "@messages="
+        truncated_messages = @messages.map do |msg|
+          truncated_msg = msg.dup
+          if msg[:content].is_a?(String) && msg[:content].length > 80
+            truncated_msg[:content] = msg[:content][0..77] + "..."
+          end
+          truncated_msg
+        end
+        q.pp truncated_messages
+        
+        # Show other instance variables (except sensitive ones)
+        skip_vars = [:@messages, :@api_key, :@client]
+        instance_variables.sort.each do |var|
+          next if skip_vars.include?(var)
+          value = instance_variable_get(var)
+          unless value.nil?
+            q.text ","
+            q.breakable
+            q.text "#{var}="
+            q.pp value
+          end
+        end
+      end
+    end
+
+
     private
 
     class InputClassificationError < StandardError; end
@@ -210,7 +245,7 @@ module AI
     def prepare_messages_for_api
       return messages unless previous_response_id
 
-      previous_response_index = messages.find_index { |message| message[:response]&.id == previous_response_id }
+      previous_response_index = messages.find_index { |message| message.dig(:response, :id) == previous_response_id }
 
       if previous_response_index
         messages[(previous_response_index + 1)..] || []
@@ -357,6 +392,8 @@ module AI
       tools_list
     end
 
+    # :reek:UtilityFunction
+    # :reek:ManualDispatch
     def extract_text_from_response(response)
       response.output.flat_map { |output|
         output.respond_to?(:content) ? output.content : []
@@ -366,6 +403,7 @@ module AI
     end
 
     # :reek:FeatureEnvy
+    # :reek:UtilityFunction
     def wrap_schema_if_needed(schema)
       if schema.key?(:format) || schema.key?("format")
         schema

data/lib/ai-chat.rb

@@ -1 +1,8 @@
 require_relative "ai/chat"
+
+# Load amazing_print extension if amazing_print is available
+begin
+  require_relative "ai/amazing_print"
+rescue LoadError
+  # amazing_print not available, skip custom formatting
+end

metadata

@@ -1,13 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ai-chat
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 0.2.3
 platform: ruby
 authors:
 - Raghu Betina
+autorequire: 
 bindir: bin
 cert_chain: []
-date: 1980-01-02 00:00:00.000000000 Z
+date: 2025-08-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: openai
@@ -93,20 +94,21 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '11.1'
+description: 
 email:
 - raghu@firstdraft.com
 executables: []
 extensions: []
 extra_rdoc_files:
-- LICENSE
 - README.md
+- LICENSE
 files:
 - LICENSE
 - README.md
 - ai-chat.gemspec
 - lib/ai-chat.rb
+- lib/ai/amazing_print.rb
 - lib/ai/chat.rb
-- lib/ai/response.rb
 homepage: https://github.com/firstdraft/ai-chat
 licenses:
 - MIT
@@ -117,6 +119,7 @@ metadata:
   label: AI Chat
   rubygems_mfa_required: 'true'
   source_code_uri: https://github.com/firstdraft/ai-chat
+post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
@@ -131,7 +134,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.7.1
+rubygems_version: 3.4.6
+signing_key: 
 specification_version: 4
 summary: A beginner-friendly Ruby interface for OpenAI's API
 test_files: []

data/lib/ai/response.rb

@@ -1,17 +0,0 @@
-module AI
-  # :reek:IrresponsibleModule
-  # :reek:TooManyInstanceVariables
-  class Response
-    attr_reader :id, :model, :usage, :total_tokens
-    # :reek:Attribute
-    attr_accessor :images
-
-    def initialize(response)
-      @id = response.id
-      @model = response.model
-      @usage = response.usage.to_h.slice(:input_tokens, :output_tokens, :total_tokens)
-      @total_tokens = @usage[:total_tokens]
-      @images = []
-    end
-  end
-end