langchainrb 0.9.2 → 0.9.4

metadata

@@ -1,15 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: langchainrb
 version: !ruby/object:Gem::Version
-  version: 0.9.2
+  version: 0.9.4
 platform: ruby
 authors:
 - Andrei Bondarev
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-02-11 00:00:00.000000000 Z
+date: 2024-02-28 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 7.0.8
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 7.0.8
 - !ruby/object:Gem::Dependency
   name: baran
   requirement: !ruby/object:Gem::Requirement
@@ -30,42 +44,42 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.8.1
+        version: 1.1.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.8.1
+        version: 1.1.0
 - !ruby/object:Gem::Dependency
   name: tiktoken_ruby
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.0.5
+        version: 0.0.7
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.0.5
+        version: 0.0.7
 - !ruby/object:Gem::Dependency
   name: json-schema
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 4.0.0
+        version: '4'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 4.0.0
+        version: '4'
 - !ruby/object:Gem::Dependency
   name: zeitwerk
   requirement: !ruby/object:Gem::Requirement
@@ -178,6 +192,34 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 2.2.7
+- !ruby/object:Gem::Dependency
+  name: vcr
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: webmock
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: ai21
   requirement: !ruby/object:Gem::Requirement
@@ -416,6 +458,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '1.13'
+- !ruby/object:Gem::Dependency
+  name: mail
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.8'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.8'
 - !ruby/object:Gem::Dependency
   name: open-weather-ruby-client
   requirement: !ruby/object:Gem::Requirement
@@ -612,7 +668,7 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-description: Build LLM-backed Ruby applications with Ruby's LangChain
+description: Build LLM-backed Ruby applications with Ruby's Langchain.rb
 email:
 - andrei.bondarev13@gmail.com
 executables: []
@@ -670,6 +726,7 @@ files:
 - lib/langchain/llm/llama_cpp.rb
 - lib/langchain/llm/ollama.rb
 - lib/langchain/llm/openai.rb
+- lib/langchain/llm/prompts/ollama/summarize_template.yaml
 - lib/langchain/llm/prompts/summarize_template.yaml
 - lib/langchain/llm/replicate.rb
 - lib/langchain/llm/response/ai21_response.rb
@@ -692,6 +749,7 @@ files:
 - lib/langchain/processors/base.rb
 - lib/langchain/processors/csv.rb
 - lib/langchain/processors/docx.rb
+- lib/langchain/processors/eml.rb
 - lib/langchain/processors/html.rb
 - lib/langchain/processors/json.rb
 - lib/langchain/processors/jsonl.rb
@@ -705,12 +763,18 @@ files:
 - lib/langchain/prompt/loading.rb
 - lib/langchain/prompt/prompt_template.rb
 - lib/langchain/tool/base.rb
-- lib/langchain/tool/calculator.rb
-- lib/langchain/tool/database.rb
-- lib/langchain/tool/google_search.rb
-- lib/langchain/tool/ruby_code_interpreter.rb
-- lib/langchain/tool/weather.rb
-- lib/langchain/tool/wikipedia.rb
+- lib/langchain/tool/calculator/calculator.json
+- lib/langchain/tool/calculator/calculator.rb
+- lib/langchain/tool/database/database.json
+- lib/langchain/tool/database/database.rb
+- lib/langchain/tool/google_search/google_search.json
+- lib/langchain/tool/google_search/google_search.rb
+- lib/langchain/tool/ruby_code_interpreter/ruby_code_interpreter.json
+- lib/langchain/tool/ruby_code_interpreter/ruby_code_interpreter.rb
+- lib/langchain/tool/weather/weather.json
+- lib/langchain/tool/weather/weather.rb
+- lib/langchain/tool/wikipedia/wikipedia.json
+- lib/langchain/tool/wikipedia/wikipedia.rb
 - lib/langchain/utils/cosine_similarity.rb
 - lib/langchain/utils/token_length/ai21_validator.rb
 - lib/langchain/utils/token_length/base_validator.rb
@@ -758,5 +822,5 @@ requirements: []
 rubygems_version: 3.4.1
 signing_key:
 specification_version: 4
-summary: Build LLM-backed Ruby applications with Ruby's LangChain
+summary: Build LLM-backed Ruby applications with Ruby's Langchain.rb
 test_files: []

data/lib/langchain/tool/database.rb

@@ -1,90 +0,0 @@
-module Langchain::Tool
-  class Database < Base
-    #
-    # Connects to a database, executes SQL queries, and outputs DB schema for Agents to use
-    #
-    # Gem requirements: gem "sequel", "~> 5.68.0"
-    #
-
-    NAME = "database"
-
-    description <<~DESC
-      Useful for getting the result of a database query.
-
-      The input to this tool should be valid SQL.
-    DESC
-
-    attr_reader :db, :requested_tables, :excluded_tables
-
-    #
-    # Establish a database connection
-    #
-    # @param connection_string [String] Database connection info, e.g. 'postgres://user:password@localhost:5432/db_name'
-    # @param tables [Array<Symbol>] The tables to use. Will use all if empty.
-    # @param except_tables [Array<Symbol>] The tables to exclude. Will exclude none if empty.
-
-    # @return [Database] Database object
-    #
-    def initialize(connection_string:, tables: [], exclude_tables: [])
-      depends_on "sequel"
-
-      raise StandardError, "connection_string parameter cannot be blank" if connection_string.empty?
-
-      @db = Sequel.connect(connection_string)
-      @requested_tables = tables
-      @excluded_tables = exclude_tables
-    end
-
-    #
-    # Returns the database schema
-    #
-    # @return [String] schema
-    #
-    def dump_schema
-      Langchain.logger.info("Dumping schema tables and keys", for: self.class)
-      schema = ""
-      db.tables.each do |table|
-        next if excluded_tables.include?(table)
-        next unless requested_tables.empty? || requested_tables.include?(table)
-
-        primary_key_columns = []
-        primary_key_column_count = db.schema(table).count { |column| column[1][:primary_key] == true }
-
-        schema << "CREATE TABLE #{table}(\n"
-        db.schema(table).each do |column|
-          schema << "#{column[0]} #{column[1][:type]}"
-          if column[1][:primary_key] == true
-            schema << " PRIMARY KEY" if primary_key_column_count == 1
-          else
-            primary_key_columns << column[0]
-          end
-          schema << ",\n" unless column == db.schema(table).last && primary_key_column_count == 1
-        end
-        if primary_key_column_count > 1
-          schema << "PRIMARY KEY (#{primary_key_columns.join(",")})"
-        end
-        db.foreign_key_list(table).each do |fk|
-          schema << ",\n" if fk == db.foreign_key_list(table).first
-          schema << "FOREIGN KEY (#{fk[:columns][0]}) REFERENCES #{fk[:table]}(#{fk[:key][0]})"
-          schema << ",\n" unless fk == db.foreign_key_list(table).last
-        end
-        schema << ");\n"
-      end
-      schema
-    end
-
-    #
-    # Evaluates a sql expression
-    #
-    # @param input [String] sql expression
-    # @return [Array] results
-    #
-    def execute(input:)
-      Langchain.logger.info("Executing \"#{input}\"", for: self.class)
-
-      db[input].to_a
-    rescue Sequel::DatabaseError => e
-      Langchain.logger.error(e.message, for: self.class)
-    end
-  end
-end