langchainrb 0.4.0 → 0.4.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ce8728ec2208577809174e154642db161121cb9dd49e0ec5d190d080e68b1d78
-  data.tar.gz: bb0e0ccc4558ca849549f495a4adfacc5f7851c786869974afdaef29f0cde3ca
+  metadata.gz: 100c9989f1a6787168064f6d810f009c6ded8129949f0040a7c1ed05ebff9d20
+  data.tar.gz: 0cf01f053841e9a45353d72b62df9a29c0daf7268fb69aff794aae830f7db448
 SHA512:
-  metadata.gz: b2b4c27e31d730563aeca70a0aa3c4cf129e69773e34f397ba057faa8298a4368c1b9f66f925188f867f1feb47b4e07f77df702fa7c6cb76ad1e1a8464b895f6
-  data.tar.gz: 55dd3fbc21e2cdf9bd84afcd6bb4de0f72c960dec0c6b1d2efff1f9492b3d5c7399f2d14c323597045e64eafb6f2f20992348d640317c64721fb0556f8a64126
+  metadata.gz: becea5f089e7455e8dd9437cb3647c921eebb3155a9685697190e0e757d31ec8d9f4572faa3e299911b44b371c097bd9b5edf3da35772562027d1b12cd743b9e
+  data.tar.gz: 44cfe6f8cb2056d6a0a5c619ded7828a9cdad0e5f4e8876242adad2f13251ea1c3dbb64c9feffea25b61e91f163245954c1f272fadb9f4f3a2a06cadd64bd9a0

data/CHANGELOG.md

@@ -1,5 +1,10 @@
 ## [Unreleased]
 
+## [0.4.1] - 2023-06-02
+- Beautiful colored log messages
+- 🛠️ Tools
+  - Introducing `Langchain::Tool::RubyCodeInterpreter`, a tool executes sandboxed Ruby code
+
 ## [0.4.0] - 2023-06-01
 - [BREAKING] Everything is namespaced under `Langchain::` now
 - Pgvector similarity search uses the cosine distance by default now

data/Gemfile.lock

@@ -1,7 +1,8 @@
 PATH
   remote: .
   specs:
-    langchainrb (0.4.0)
+    langchainrb (0.4.1)
+      colorize (~> 0.8.1)
       tiktoken_ruby (~> 0.0.5)
 
 GEM
@@ -32,6 +33,7 @@ GEM
     ast (2.4.2)
     builder (3.2.4)
     byebug (11.1.3)
+    childprocess (4.1.0)
     chroma-db (0.3.0)
       dry-monads (~> 1.6)
       ruby-next-core (>= 0.15.0)
@@ -39,6 +41,7 @@ GEM
     cohere-ruby (0.9.4)
       faraday (>= 1.0.0)
       faraday_middleware (>= 1.0.0)
+    colorize (0.8.1)
     concurrent-ruby (1.2.2)
     crass (1.0.6)
     diff-lcs (1.5.0)
@@ -250,6 +253,8 @@ GEM
     ruby-rc4 (0.1.5)
     ruby2_keywords (0.0.5)
     rubyzip (2.3.2)
+    safe_ruby (1.0.4)
+      childprocess (>= 0.3.9)
     standard (1.28.2)
       language_server-protocol (~> 3.17.0.2)
       lint_roller (~> 1.0)
@@ -313,6 +318,7 @@ DEPENDENCIES
   replicate-ruby (~> 0.2.2)
   rspec (~> 3.0)
   ruby-openai (~> 4.0.0)
+  safe_ruby (~> 1.0.4)
   standardrb
   weaviate-ruby (~> 0.8.0)
   wikipedia-client (~> 1.17.0)

data/README.md

@@ -260,10 +260,10 @@ agent.run(question: "How many full soccer fields would be needed to cover the di
 | Name         | Description                                        | ENV Requirements                                              | Gem Requirements                          |
 | ------------ | :------------------------------------------------: | :-----------------------------------------------------------: | :---------------------------------------: |
 | "calculator" | Useful for getting the result of a math expression |                                                               | `gem "eqn", "~> 1.6.5"`                   |
-| "search"     | A wrapper around Google Search                     | `ENV["SERPAPI_API_KEY"]` (https://serpapi.com/manage-api-key) | `gem "google_search_results", "~> 2.0.0"` |                      |
+| "ruby_code_interpreter" | Interprets Ruby expressions             |                                                               | `gem "safe_ruby", "~> 1.0.4"`             |
+| "search"     | A wrapper around Google Search                     | `ENV["SERPAPI_API_KEY"]` (https://serpapi.com/manage-api-key) | `gem "google_search_results", "~> 2.0.0"` |
 | "wikipedia"  | Calls Wikipedia API to retrieve the summary        |                                                               | `gem "wikipedia-client", "~> 1.17.0"`     |
 
-
 #### Loaders 🚚
 
 Need to read data from various sources? Load it up.

data/lib/langchain/agent/chain_of_thought_agent/chain_of_thought_agent.rb

@@ -42,7 +42,7 @@ module Langchain::Agent
       )
 
       loop do
-        Langchain.logger.info("Agent: Passing the prompt to the #{llm} LLM")
+        Langchain.logger.info("[#{self.class.name}]".red + ": Sending the prompt to the #{llm} LLM")
         response = llm_client.complete(
           prompt: prompt,
           stop_sequences: ["Observation:"],
@@ -59,12 +59,11 @@ module Langchain::Agent
           # Find the input to the action in the "Action Input: [action_input]" format
           action_input = response.match(/Action Input: "?(.*)"?/)&.send(:[], -1)
 
-          Langchain.logger.info("Agent: Using the \"#{action}\" Tool with \"#{action_input}\"")
-
           # Retrieve the Tool::[ToolName] class and call `execute`` with action_input as the input
-          result = Langchain::Tool
-            .const_get(Langchain::Tool::Base::TOOLS[action.strip])
-            .execute(input: action_input)
+          tool = Langchain::Tool.const_get(Langchain::Tool::Base::TOOLS[action.strip])
+          Langchain.logger.info("[#{self.class.name}]".red + ": Invoking \"#{tool}\" Tool with \"#{action_input}\"")
+
+          result = tool.execute(input: action_input)
 
           # Append the Observation to the prompt
           prompt += if prompt.end_with?("Observation:")

data/lib/langchain/tool/calculator.rb

@@ -18,6 +18,8 @@ module Langchain::Tool
     # @param input [String] math expression
     # @return [String] Answer
     def execute(input:)
+      Langchain.logger.info("[#{self.class.name}]".light_blue + ": Executing \"#{input}\"")
+
       Eqn::Calculator.calc(input)
     rescue Eqn::ParseError, Eqn::NoVariableValueError
       # Sometimes the input is not a pure math expression, e.g: "12F in Celsius"

data/lib/langchain/tool/ruby_code_interpreter.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+module Langchain::Tool
+  class RubyCodeInterpreter < Base
+    description <<~DESC
+      A Ruby code interpreter. Use this to execute ruby expressions. Input should be a valid ruby expression. If you want to see the output of the tool, make sure to return a value.
+    DESC
+
+    def initialize(timeout: 30)
+      @timeout = timeout
+      depends_on "safe_ruby"
+      require "safe_ruby"
+    end
+
+    # @param input [String] ruby code expression
+    # @return [String] Answer
+    def execute(input:)
+      Langchain.logger.info("[#{self.class.name}]".light_blue + ": Executing \"#{input}\"")
+
+      safe_eval(input)
+    end
+
+    def safe_eval(code)
+      SafeRuby.eval(code, timeout: @timeout)
+    end
+  end
+end

data/lib/langchain/tool/serp_api.rb

@@ -33,6 +33,8 @@ module Langchain::Tool
     # TODO: Glance at all of the fields that langchain Python looks through: https://github.com/hwchase17/langchain/blob/v0.0.166/langchain/utilities/serpapi.py#L128-L156
     # We may need to do the same thing here.
     def execute(input:)
+      Langchain.logger.info("[#{self.class.name}]".light_blue + ": Executing \"#{input}\"")
+
       hash_results = execute_search(input: input)
 
       hash_results.dig(:answer_box, :answer) ||

data/lib/langchain/tool/wikipedia.rb

@@ -22,6 +22,8 @@ module Langchain::Tool
     # @param input [String] search query
     # @return [String] Answer
     def execute(input:)
+      Langchain.logger.info("[#{self.class.name}]".light_blue + ": Executing \"#{input}\"")
+
       page = ::Wikipedia.find(input)
       # It would be nice to figure out a way to provide page.content but the LLM token limit is an issue
       page.summary

data/lib/langchain.rb

@@ -2,6 +2,7 @@
 
 require "logger"
 require "pathname"
+require "colorize"
 
 require_relative "./version"
 require_relative "./dependency_helper"
@@ -13,7 +14,7 @@ module Langchain
     attr_reader :root
   end
 
-  @logger ||= ::Logger.new($stdout, level: :warn, formatter: ->(severity, datetime, progname, msg) { "[LangChain.rb] #{msg}\n" })
+  @logger ||= ::Logger.new($stdout, level: :warn, formatter: ->(severity, datetime, progname, msg) { "[LangChain.rb]".yellow + " #{msg}\n" })
 
   @root = Pathname.new(__dir__)
 
@@ -28,6 +29,7 @@ module Langchain
   module Tool
     autoload :Base, "langchain/tool/base"
     autoload :Calculator, "langchain/tool/calculator"
+    autoload :RubyCodeInterpreter, "langchain/tool/ruby_code_interpreter"
     autoload :SerpApi, "langchain/tool/serp_api"
     autoload :Wikipedia, "langchain/tool/wikipedia"
   end

data/lib/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Langchain
-  VERSION = "0.4.0"
+  VERSION = "0.4.1"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: langchainrb
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.4.1
 platform: ruby
 authors:
 - Andrei Bondarev
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-06-01 00:00:00.000000000 Z
+date: 2023-06-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: tiktoken_ruby
@@ -24,6 +24,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 0.0.5
+- !ruby/object:Gem::Dependency
+  name: colorize
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.8.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.8.1
 - !ruby/object:Gem::Dependency
   name: dotenv-rails
   requirement: !ruby/object:Gem::Requirement
@@ -276,6 +290,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 4.0.0
+- !ruby/object:Gem::Dependency
+  name: safe_ruby
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.4
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.4
 - !ruby/object:Gem::Dependency
   name: weaviate-ruby
   requirement: !ruby/object:Gem::Requirement
@@ -353,6 +381,7 @@ files:
 - lib/langchain/prompt/prompt_template.rb
 - lib/langchain/tool/base.rb
 - lib/langchain/tool/calculator.rb
+- lib/langchain/tool/ruby_code_interpreter.rb
 - lib/langchain/tool/serp_api.rb
 - lib/langchain/tool/wikipedia.rb
 - lib/langchain/utils/token_length_validator.rb