drop Colorize due to GPL license

CHANGELOG.md

@@ -1,4 +1,5 @@
 ## [Unreleased]
+- Drop `colorize` which is GPL-licensed in favour of `rainbow`
 
 ## [0.15.0] - 2024-08-14
 - Fix Langchain::Assistant when llm is Anthropic
@@ -135,7 +136,7 @@
 - AWS Bedrock is available as an LLM provider. Available models from AI21, Cohere, AWS, and Anthropic.
 
 ## [0.7.3] - 2023-11-08
-- LLM response passes through the context in RAG cases 
+- LLM response passes through the context in RAG cases
 - Fix gpt-4 token length validation
 
 ## [0.7.2] - 2023-11-02
@@ -180,7 +181,7 @@
 
 ## [0.6.13] - 2023-08-23
 - Add `k:` parameter to all `ask()` vector search methods
-- Bump Faraday to 2.x 
+- Bump Faraday to 2.x
 
 ## [0.6.12] - 2023-08-13
 

Gemfile.lock

@@ -3,10 +3,10 @@ PATH
   specs:
     langchainrb (0.15.0)
       baran (~> 0.1.9)
-      colorize (~> 1.1.0)
       json-schema (~> 4)
       matrix
       pragmatic_segmenter (~> 0.3.0)
+      rainbow (~> 3.1.0)
       to_bool (~> 2.0.0)
       zeitwerk (~> 2.5)
 
@@ -73,7 +73,6 @@ GEM
     coderay (1.1.3)
     cohere-ruby (0.9.10)
       faraday (>= 2.0.1, < 3.0)
-    colorize (1.1.0)
     concurrent-ruby (1.3.1)
     connection_pool (2.4.1)
     crack (1.0.0)

langchain.gemspec

@@ -28,7 +28,7 @@ Gem::Specification.new do |spec|
   # dependencies
   # Not sure if we should require this as it only applies to OpenAI usecase.
   spec.add_dependency "baran", "~> 0.1.9"
-  spec.add_dependency "colorize", "~> 1.1.0"
+  spec.add_dependency "rainbow", "~> 3.1.0"
   spec.add_dependency "json-schema", "~> 4"
   spec.add_dependency "zeitwerk", "~> 2.5"
   spec.add_dependency "pragmatic_segmenter", "~> 0.3.0"

lib/langchain.rb

@@ -2,7 +2,7 @@
 
 require "logger"
 require "pathname"
-require "colorize"
+require "rainbow"
 require "to_bool"
 require "zeitwerk"
 loader = Zeitwerk::Loader.for_gem

lib/langchain/contextual_logger.rb

@@ -42,19 +42,25 @@ def method_missing(method, *args, **kwargs, &block)
       for_class_name = for_class&.name
 
       log_line_parts = []
-      log_line_parts << "[Langchain.rb]".colorize(color: :yellow)
+      log_line_parts << colorize("[Langchain.rb]", color: :yellow)
       log_line_parts << if for_class.respond_to?(:logger_options)
-        "[#{for_class_name}]".colorize(for_class.logger_options) + ":"
+        colorize("[#{for_class_name}]", for_class.logger_options) + ":"
       elsif for_class_name
         "[#{for_class_name}]:"
       end
-      log_line_parts << args.first.colorize(MESSAGE_COLOR_OPTIONS[method])
+      log_line_parts << colorize(args.first, MESSAGE_COLOR_OPTIONS[method])
       log_line = log_line_parts.compact.join(" ")
 
       @logger.send(
         method,
         log_line
       )
     end
+
+    private
+
+    def colorize(line, options)
+      Langchain::Utils::Colorizer.colorize(line, options)
+    end
   end
 end

lib/langchain/utils/colorizer.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+module Langchain
+  module Utils
+    class Colorizer
+      def self.colorize(line, options)
+        decorated_line = Rainbow(line)
+        options.each_pair.each do |modifier, value|
+          decorated_line = if modifier == :mode
+            decorated_line.public_send(value)
+          else
+            decorated_line.public_send(modifier, value)
+          end
+        end
+        decorated_line
+      end
+    end
+  end
+end

spec/langchain/contextual_logger_spec.rb

@@ -9,7 +9,7 @@
     it "#info" do
       expect(logger).to receive(:info).with(
         <<~LINE.strip
-          #{"[Langchain.rb]".yellow} #{"Hello World".white}
+          #{colorize("[Langchain.rb]", color: :yellow)} #{colorize("Hello World", color: :white)}
         LINE
       )
       subject.info("Hello World")
@@ -18,7 +18,7 @@
     it "#warn" do
       expect(logger).to receive(:warn).with(
         <<~LINE.strip
-          #{"[Langchain.rb]".yellow} #{"Hello World".colorize(color: :yellow, mode: :bold)}
+          #{colorize("[Langchain.rb]", color: :yellow)} #{colorize("Hello World", color: :yellow, mode: :bold)}
         LINE
       )
       subject.warn("Hello World")
@@ -27,7 +27,7 @@
     it "#debug" do
       expect(logger).to receive(:debug).with(
         <<~LINE.strip
-          #{"[Langchain.rb]".yellow} #{"Hello World".white}
+          #{colorize("[Langchain.rb]", color: :yellow)} #{colorize("Hello World", color: :white)}
         LINE
       )
       subject.debug("Hello World")
@@ -38,7 +38,7 @@
     it "#info" do
       expect(logger).to receive(:info).with(
         <<~LINE.strip
-          #{"[Langchain.rb]".yellow} #{"[Langchain::Vectorsearch::Pgvector]".blue}: #{"Hello World".white}
+          #{colorize("[Langchain.rb]", color: :yellow)} #{colorize("[Langchain::Vectorsearch::Pgvector]", color: :blue)}: #{colorize("Hello World", color: :white)}
         LINE
       )
       subject.info("Hello World", for: Langchain::Vectorsearch::Pgvector)
@@ -47,7 +47,7 @@
     it "#warn" do
       expect(logger).to receive(:warn).with(
         <<~LINE.strip
-          #{"[Langchain.rb]".yellow} #{"[Langchain::Vectorsearch::Pgvector]".blue}: #{"Hello World".colorize(color: :yellow, mode: :bold)}
+          #{colorize("[Langchain.rb]", color: :yellow)} #{colorize("[Langchain::Vectorsearch::Pgvector]", color: :blue)}: #{colorize("Hello World", color: :yellow, mode: :bold)}
         LINE
       )
       subject.warn("Hello World", for: Langchain::Vectorsearch::Pgvector)
@@ -56,10 +56,14 @@
     it "doesn't have an issue with objects that don't have .logger_options" do
       expect(logger).to receive(:warn).with(
         <<~LINE.strip
-          #{"[Langchain.rb]".yellow} [Object]: #{"Hello World".colorize(color: :yellow, mode: :bold)}
+          #{colorize("[Langchain.rb]", color: :yellow)} [Object]: #{colorize("Hello World", color: :yellow, mode: :bold)}
         LINE
       )
       subject.warn("Hello World", for: Object)
     end
   end
+
+  def colorize(line, options)
+    Langchain::Utils::Colorizer.colorize(line, options)
+  end
 end