langchainrb 0.4.1 → 0.5.0

data/lib/langchain/vectorsearch/qdrant.rb

@@ -2,13 +2,21 @@
 
 module Langchain::Vectorsearch
   class Qdrant < Base
+    #
+    # Wrapper around Qdrant
+    #
+    # Gem requirements: gem "qdrant-ruby", "~> 0.9.0"
+    #
+    # Usage:
+    # qdrant = Langchain::Vectorsearch::Qdrant.new(url:, api_key:, index_name:, llm:, llm_api_key:)
+    #
+
     # Initialize the Qdrant client
     # @param url [String] The URL of the Qdrant server
     # @param api_key [String] The API key to use
     # @param index_name [String] The name of the index to use
-    # @param llm [Symbol] The LLM to use
-    # @param llm_api_key [String] The API key for the LLM
-    def initialize(url:, api_key:, index_name:, llm:, llm_api_key:)
+    # @param llm [Object] The LLM client to use
+    def initialize(url:, api_key:, index_name:, llm:)
       depends_on "qdrant-ruby"
       require "qdrant"
 
@@ -18,7 +26,7 @@ module Langchain::Vectorsearch
       )
       @index_name = index_name
 
-      super(llm: llm, llm_api_key: llm_api_key)
+      super(llm: llm)
     end
 
     # Add a list of texts to the index
@@ -29,7 +37,7 @@ module Langchain::Vectorsearch
 
       Array(texts).each do |text|
         batch[:ids].push(SecureRandom.uuid)
-        batch[:vectors].push(llm_client.embed(text: text))
+        batch[:vectors].push(llm.embed(text: text))
         batch[:payloads].push({content: text})
       end
 
@@ -59,7 +67,7 @@ module Langchain::Vectorsearch
       query:,
       k: 4
     )
-      embedding = llm_client.embed(text: query)
+      embedding = llm.embed(text: query)
 
       similarity_search_by_vector(
         embedding: embedding,
@@ -96,7 +104,7 @@ module Langchain::Vectorsearch
 
       prompt = generate_prompt(question: question, context: context)
 
-      llm_client.chat(prompt: prompt)
+      llm.chat(prompt: prompt)
     end
   end
 end

data/lib/langchain/vectorsearch/weaviate.rb

@@ -2,13 +2,21 @@
 
 module Langchain::Vectorsearch
   class Weaviate < Base
+    #
+    # Wrapper around Weaviate
+    #
+    # Gem requirements: gem "weaviate-ruby", "~> 0.8.0"
+    #
+    # Usage:
+    # weaviate = Langchain::Vectorsearch::Weaviate.new(url:, api_key:, index_name:, llm:, llm_api_key:)
+    #
+
     # Initialize the Weaviate adapter
     # @param url [String] The URL of the Weaviate instance
     # @param api_key [String] The API key to use
     # @param index_name [String] The name of the index to use
-    # @param llm [Symbol] The LLM to use
-    # @param llm_api_key [String] The API key for the LLM
-    def initialize(url:, api_key:, index_name:, llm:, llm_api_key:)
+    # @param llm [Object] The LLM client to use
+    def initialize(url:, api_key:, index_name:, llm:)
       depends_on "weaviate-ruby"
       require "weaviate"
 
@@ -18,7 +26,7 @@ module Langchain::Vectorsearch
       )
       @index_name = index_name
 
-      super(llm: llm, llm_api_key: llm_api_key)
+      super(llm: llm)
     end
 
     # Add a list of texts to the index
@@ -29,7 +37,7 @@ module Langchain::Vectorsearch
         {
           class: index_name,
           properties: {content: text},
-          vector: llm_client.embed(text: text)
+          vector: llm.embed(text: text)
         }
       end
 
@@ -58,7 +66,7 @@ module Langchain::Vectorsearch
     # @param k [Integer|String] The number of results to return
     # @return [Hash] The search results
     def similarity_search(query:, k: 4)
-      embedding = llm_client.embed(text: query)
+      embedding = llm.embed(text: query)
 
       similarity_search_by_vector(embedding: embedding, k: k)
     end
@@ -91,7 +99,7 @@ module Langchain::Vectorsearch
 
       prompt = generate_prompt(question: question, context: context)
 
-      llm_client.chat(prompt: prompt)
+      llm.chat(prompt: prompt)
     end
   end
 end

data/lib/{version.rb → langchain/version.rb}

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

data/lib/langchain.rb

@@ -4,8 +4,7 @@ require "logger"
 require "pathname"
 require "colorize"
 
-require_relative "./version"
-require_relative "./dependency_helper"
+require_relative "./langchain/version"
 
 module Langchain
   class << self
@@ -20,10 +19,12 @@ module Langchain
 
   autoload :Loader, "langchain/loader"
   autoload :Data, "langchain/data"
+  autoload :DependencyHelper, "langchain/dependency_helper"
 
   module Agent
     autoload :Base, "langchain/agent/base"
     autoload :ChainOfThoughtAgent, "langchain/agent/chain_of_thought_agent/chain_of_thought_agent.rb"
+    autoload :SQLQueryAgent, "langchain/agent/sql_query_agent/sql_query_agent.rb"
   end
 
   module Tool
@@ -32,6 +33,7 @@ module Langchain
     autoload :RubyCodeInterpreter, "langchain/tool/ruby_code_interpreter"
     autoload :SerpApi, "langchain/tool/serp_api"
     autoload :Wikipedia, "langchain/tool/wikipedia"
+    autoload :Database, "langchain/tool/database"
   end
 
   module Processors
@@ -43,6 +45,7 @@ module Langchain
     autoload :JSONL, "langchain/processors/jsonl"
     autoload :PDF, "langchain/processors/pdf"
     autoload :Text, "langchain/processors/text"
+    autoload :Xlsx, "langchain/processors/xlsx"
   end
 
   module Utils
@@ -60,6 +63,7 @@ module Langchain
   end
 
   module LLM
+    autoload :AI21, "langchain/llm/ai21"
     autoload :Base, "langchain/llm/base"
     autoload :Cohere, "langchain/llm/cohere"
     autoload :GooglePalm, "langchain/llm/google_palm"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: langchainrb
 version: !ruby/object:Gem::Version
-  version: 0.4.1
+  version: 0.5.0
 platform: ruby
 authors:
 - Andrei Bondarev
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-06-02 00:00:00.000000000 Z
+date: 2023-06-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: tiktoken_ruby
@@ -66,6 +66,48 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 3.10.0
+- !ruby/object:Gem::Dependency
+  name: yard
+  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: rdiscount
+  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
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.2.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.2.0
 - !ruby/object:Gem::Dependency
   name: chroma-db
   requirement: !ruby/object:Gem::Requirement
@@ -276,6 +318,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 0.9.0
+- !ruby/object:Gem::Dependency
+  name: roo
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.10.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.10.0
 - !ruby/object:Gem::Dependency
   name: ruby-openai
   requirement: !ruby/object:Gem::Requirement
@@ -304,6 +360,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 1.0.4
+- !ruby/object:Gem::Dependency
+  name: sequel
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 5.68.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 5.68.0
 - !ruby/object:Gem::Dependency
   name: weaviate-ruby
   requirement: !ruby/object:Gem::Requirement
@@ -341,6 +411,7 @@ extra_rdoc_files: []
 files:
 - ".env.example"
 - ".rspec"
+- ".rubocop.yml"
 - CHANGELOG.md
 - Gemfile
 - Gemfile.lock
@@ -353,12 +424,17 @@ files:
 - examples/store_and_query_with_pinecone.rb
 - examples/store_and_query_with_qdrant.rb
 - examples/store_and_query_with_weaviate.rb
-- lib/dependency_helper.rb
+- lefthook.yml
 - lib/langchain.rb
 - lib/langchain/agent/base.rb
 - lib/langchain/agent/chain_of_thought_agent/chain_of_thought_agent.rb
 - lib/langchain/agent/chain_of_thought_agent/chain_of_thought_agent_prompt.json
+- lib/langchain/agent/sql_query_agent/sql_query_agent.rb
+- lib/langchain/agent/sql_query_agent/sql_query_agent_answer_prompt.json
+- lib/langchain/agent/sql_query_agent/sql_query_agent_sql_prompt.json
 - lib/langchain/data.rb
+- lib/langchain/dependency_helper.rb
+- lib/langchain/llm/ai21.rb
 - lib/langchain/llm/base.rb
 - lib/langchain/llm/cohere.rb
 - lib/langchain/llm/google_palm.rb
@@ -375,12 +451,14 @@ files:
 - lib/langchain/processors/jsonl.rb
 - lib/langchain/processors/pdf.rb
 - lib/langchain/processors/text.rb
+- lib/langchain/processors/xlsx.rb
 - lib/langchain/prompt/base.rb
 - lib/langchain/prompt/few_shot_prompt_template.rb
 - 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/ruby_code_interpreter.rb
 - lib/langchain/tool/serp_api.rb
 - lib/langchain/tool/wikipedia.rb
@@ -392,8 +470,8 @@ files:
 - lib/langchain/vectorsearch/pinecone.rb
 - lib/langchain/vectorsearch/qdrant.rb
 - lib/langchain/vectorsearch/weaviate.rb
+- lib/langchain/version.rb
 - lib/langchainrb.rb
-- lib/version.rb
 - sig/langchain.rbs
 homepage: https://rubygems.org/gems/langchainrb
 licenses:

data/lib/dependency_helper.rb

@@ -1,30 +0,0 @@
-# frozen_string_literal: true
-
-VersionError = Class.new(ScriptError)
-
-# This method requires and loads the given gem, and then checks to see if the version of the gem meets the requirements listed in `langchain.gemspec`
-# This solution was built to avoid auto-loading every single gem in the Gemfile when the developer will mostly likely be only using a few of them.
-#
-# @param gem_name [String] The name of the gem to load
-# @return [Boolean] Whether or not the gem was loaded successfully
-# @raise [LoadError] If the gem is not installed
-# @raise [VersionError] If the gem is installed, but the version does not meet the requirements
-#
-def depends_on(gem_name)
-  gem(gem_name) # require the gem
-
-  return(true) unless defined?(Bundler) # If we're in a non-bundler environment, we're no longer able to determine if we'll meet requirements
-
-  gem_version = Gem.loaded_specs[gem_name].version
-  gem_requirement = Bundler.load.dependencies.find { |g| g.name == gem_name }&.requirement
-
-  raise LoadError unless gem_requirement
-
-  unless gem_requirement.satisfied_by?(gem_version)
-    raise VersionError, "The #{gem_name} gem is installed, but version #{gem_requirement} is required. You have #{gem_version}."
-  end
-
-  true
-rescue LoadError
-  raise LoadError, "Could not load #{gem_name}. Please ensure that the #{gem_name} gem is installed."
-end