langchainrb 0.3.2 → 0.3.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e751a38f6a248db9aabfac3f1b9bc547f61304dd366be999005941045a5adcd6
-  data.tar.gz: 56db69c0a578bdf198bfc12cb53e92f0e1d15654a1dfa29715d5d7833550da33
+  metadata.gz: 735fb9932ea28c3d6f3938b964e4edd5d030ed4c12c46a9f35293d35469cf952
+  data.tar.gz: 66a8030ea90c19c1950c4136c48677b6490df948dc2693562eca7b8a370dc3ae
 SHA512:
-  metadata.gz: bdd3060863a967b48a6123ea379d9a98632730a726b8da9924bb0061511b8bf88fa05f36c49ac4b8f136472e9aa204c4bd09929d31e0f2c0700ad16361a1a8cf
-  data.tar.gz: 88f04804d10f51d639b8643bd683b8a420920fc4ad2d26ef6c45f4fac034ec3f8b54787de51c4799f4a074029ad5649efca6e950b29fef62b96ce9452e5ec8cc
+  metadata.gz: 426ae91fc1c2d297b84758f6fe87d62a6dbaf4614776ab7d247c994ce849a979d1752d1241963f7fd3b8d71a68dc6f576660daabcacf19159b63acdda65fa7d9
+  data.tar.gz: 4acd1a02be734cccc431499c0e1659694004f8ee30b27b99d51b580373c4a410e87f8bf96edfeec38e67f1066d34441ccbb4ce6e4b6597623f606876e3fcb16f

data/CHANGELOG.md

@@ -1,5 +1,10 @@
 ## [Unreleased]
 
+## [0.3.3] - 2023-05-16
+- Dependencies are now optionally loaded and required at runtime
+- Start using `standardrb` for linting
+- Use the Ruby logger
+
 ## [0.3.2] - 2023-05-15
 - Agents
   - Fix Chain of Thought prompt loader

data/Gemfile

@@ -8,3 +8,5 @@ gemspec
 gem "rake", "~> 13.0"
 
 gem "rspec", "~> 3.0"
+
+gem "standardrb"

data/Gemfile.lock

@@ -1,8 +1,7 @@
 PATH
   remote: .
   specs:
-    langchainrb (0.3.2)
-      cohere-ruby (~> 0.9.3)
+    langchainrb (0.3.3)
       eqn (~> 1.6.5)
       google_search_results (~> 2.0.0)
       milvus (~> 0.9.0)
@@ -35,6 +34,7 @@ GEM
       tzinfo (~> 2.0)
     addressable (2.8.4)
       public_suffix (>= 2.0.2, < 6.0)
+    ast (2.4.2)
     builder (3.2.4)
     byebug (11.1.3)
     coderay (1.1.3)
@@ -128,6 +128,9 @@ GEM
     i18n (1.13.0)
       concurrent-ruby (~> 1.0)
     ice_nine (0.11.2)
+    json (2.6.3)
+    language_server-protocol (3.17.0.3)
+    lint_roller (1.0.0)
     loofah (2.21.1)
       crass (~> 1.0.2)
       nokogiri (>= 1.5.9)
@@ -138,10 +141,15 @@ GEM
     minitest (5.18.0)
     multi_xml (0.6.0)
     multipart-post (2.3.0)
+    nokogiri (1.14.3-arm64-darwin)
+      racc (~> 1.4)
     nokogiri (1.14.3-x86_64-darwin)
       racc (~> 1.4)
     nokogiri (1.14.3-x86_64-linux)
       racc (~> 1.4)
+    parallel (1.23.0)
+    parser (3.2.2.1)
+      ast (~> 2.4.1)
     pinecone (0.1.71)
       dry-struct (~> 1.6.0)
       dry-validation (~> 1.10.0)
@@ -173,7 +181,10 @@ GEM
       rake (>= 12.2)
       thor (~> 1.0)
       zeitwerk (~> 2.5)
+    rainbow (3.1.1)
     rake (13.0.6)
+    regexp_parser (2.8.0)
+    rexml (3.2.5)
     rspec (3.12.0)
       rspec-core (~> 3.12.0)
       rspec-expectations (~> 3.12.0)
@@ -187,15 +198,45 @@ GEM
       diff-lcs (>= 1.2.0, < 2.0)
       rspec-support (~> 3.12.0)
     rspec-support (3.12.0)
+    rubocop (1.50.2)
+      json (~> 2.3)
+      parallel (~> 1.10)
+      parser (>= 3.2.0.0)
+      rainbow (>= 2.2.2, < 4.0)
+      regexp_parser (>= 1.8, < 3.0)
+      rexml (>= 3.2.5, < 4.0)
+      rubocop-ast (>= 1.28.0, < 2.0)
+      ruby-progressbar (~> 1.7)
+      unicode-display_width (>= 2.4.0, < 3.0)
+    rubocop-ast (1.28.1)
+      parser (>= 3.2.1.0)
+    rubocop-performance (1.16.0)
+      rubocop (>= 1.7.0, < 2.0)
+      rubocop-ast (>= 0.4.0)
     ruby-openai (4.0.0)
       faraday (>= 1)
       faraday-multipart (>= 1)
+    ruby-progressbar (1.13.0)
     ruby2_keywords (0.0.5)
+    standard (1.28.2)
+      language_server-protocol (~> 3.17.0.2)
+      lint_roller (~> 1.0)
+      rubocop (~> 1.50.2)
+      standard-custom (~> 1.0.0)
+      standard-performance (~> 1.0.1)
+    standard-custom (1.0.0)
+      lint_roller (~> 1.0)
+    standard-performance (1.0.1)
+      lint_roller (~> 1.0)
+      rubocop-performance (~> 1.16.0)
+    standardrb (1.0.1)
+      standard
     thor (1.2.1)
     treetop (1.6.12)
       polyglot (~> 0.3)
     tzinfo (2.0.6)
       concurrent-ruby (~> 1.0)
+    unicode-display_width (2.4.2)
     weaviate-ruby (0.8.1)
       faraday (~> 1)
       faraday_middleware (~> 1)
@@ -205,15 +246,18 @@ GEM
     zeitwerk (2.6.8)
 
 PLATFORMS
+  arm64-darwin-22
   x86_64-darwin-19
   x86_64-linux
 
 DEPENDENCIES
+  cohere-ruby (~> 0.9.3)
   dotenv-rails (~> 2.7.6)
   langchainrb!
   pry-byebug (~> 3.10.0)
   rake (~> 13.0)
   rspec (~> 3.0)
+  standardrb
 
 BUNDLED WITH
    2.4.0

data/README.md

@@ -104,6 +104,8 @@ openai.complete(prompt: "What is the meaning of life?")
 ```
 
 #### Cohere
+Add `gem "cohere-ruby", "~> 0.9.3"` to your Gemfile.
+
 ```ruby
 cohere = LLM::Cohere.new(api_key: ENV["COHERE_API_KEY"])
 ```
@@ -211,7 +213,7 @@ agent.tools
 # => ["search", "calculator"]
 ```
 ```ruby
-agent.run(question: "How many full soccer fields would be needed to cover the distance between NYC and DC in a straight line?", logging: true)
+agent.run(question: "How many full soccer fields would be needed to cover the distance between NYC and DC in a straight line?")
 #=> "Approximately 2,945 soccer fields would be needed to cover the distance between NYC and DC in a straight line."
 ```
 
@@ -228,6 +230,16 @@ agent.run(question: "How many full soccer fields would be needed to cover the di
 | "search" | A wrapper around Google Search | `ENV["SERPAPI_API_KEY"]` (https://serpapi.com/manage-api-key)
 | "wikipedia" | Calls Wikipedia API to retrieve the summary | |
 
+
+## Logging
+
+LangChain.rb uses standard logging mechanisms and defaults to `:debug` level. Most messages are at info level, but we will add debug or warn statements as needed.
+To show all log messages:
+
+```ruby
+Lanchgain.logger.level = :info
+```
+
 ## Development
 
 After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.

data/Rakefile

@@ -2,7 +2,12 @@
 
 require "bundler/gem_tasks"
 require "rspec/core/rake_task"
+require "standard/rake"
 
 RSpec::Core::RakeTask.new(:spec)
 
 task default: :spec
+
+Rake::Task["spec"].enhance do
+  Rake::Task["standard:fix"].invoke
+end

data/examples/create_and_manage_few_shot_prompt_templates.rb

@@ -9,10 +9,10 @@ prompt = Prompt::FewShotPromptTemplate.new(
     template: "Input: {input}\nOutput: {output}"
   ),
   examples: [
-    { "input": "happy", "output": "sad" },
-    { "input": "tall", "output": "short" }
+    {input: "happy", output: "sad"},
+    {input: "tall", output: "short"}
   ],
-   input_variables: ["adjective"]
+  input_variables: ["adjective"]
 )
 
 prompt.format(adjective: "good")

data/examples/store_and_query_with_pinecone.rb

@@ -35,9 +35,9 @@ pinecone.ask(
 )
 
 # Generate your an embedding and search by it
-openai = LLM::OpenAI.new(api_key: ENV['OPENAI_API_KEY'])
+openai = LLM::OpenAI.new(api_key: ENV["OPENAI_API_KEY"])
 embedding = openai.embed(text: "veggie")
 
 pinecone.similarity_search_by_vector(
   embedding: embedding
-)
+)

data/examples/store_and_query_with_qdrant.rb

@@ -9,7 +9,6 @@ qdrant = Vectorsearch::Qdrant.new(
   llm_api_key: ENV["COHERE_API_KEY"]
 )
 
-
 # Create the default schema.
 qdrant.create_default_schema
 
@@ -33,4 +32,4 @@ qdrant.similarity_search(
 # Interact with your index through Q&A
 qdrant.ask(
   question: "What is the best recipe for chicken?"
-)
+)

data/examples/store_and_query_with_weaviate.rb

@@ -27,4 +27,4 @@ weaviate.add_texts(
 weaviate.similarity_search(
   query: "chicken",
   k: 1
-)
+)

data/lib/agent/chain_of_thought_agent/chain_of_thought_agent.rb

@@ -5,7 +5,7 @@ module Agent
     attr_reader :llm, :llm_api_key, :llm_client, :tools
 
     # Initializes the Agent
-    # 
+    #
     # @param llm [Symbol] The LLM to use
     # @param llm_api_key [String] The API key for the LLM
     # @param tools [Array] The tools to use
@@ -22,7 +22,7 @@ module Agent
     end
 
     # Validate tools when they're re-assigned
-    # 
+    #
     # @param value [Array] The tools to use
     # @return [Array] The tools that will be used
     def tools=(value)
@@ -31,11 +31,10 @@ module Agent
     end
 
     # Run the Agent!
-    # 
+    #
     # @param question [String] The question to ask
-    # @param logging [Boolean] Whether or not to log the Agent's actions
     # @return [String] The answer to the question
-    def run(question:, logging: false)
+    def run(question:)
       question = question.strip
       prompt = create_prompt(
         question: question,
@@ -43,24 +42,25 @@ module Agent
       )
 
       loop do
-        puts("Agent: Passing the prompt to the #{llm} LLM") if logging
+        Langchain.logger.info("Agent: Passing the prompt to the #{llm} LLM")
         response = llm_client.generate_completion(
           prompt: prompt,
           stop_sequences: ["Observation:"],
           max_tokens: 500
         )
 
+        binding.pry
         # Append the response to the prompt
-        prompt += response;
-    
+        prompt += response
+
         # Find the requested action in the "Action: search" format
         action = response.match(/Action: (.*)/)&.send(:[], -1)
-        
+
         if action
           # Find the input to the action in the "Action Input: [action_input]" format
           action_input = response.match(/Action Input: "?(.*)"?/)&.send(:[], -1)
 
-          puts("Agent: Using the \"#{action}\" Tool with \"#{action_input}\"") if logging
+          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 = Tool
@@ -68,10 +68,10 @@ module Agent
             .execute(input: action_input)
 
           # Append the Observation to the prompt
-          if prompt.end_with?("Observation:")
-            prompt += " #{result}\nThought:"
+          prompt += if prompt.end_with?("Observation:")
+            " #{result}\nThought:"
           else
-            prompt += "\nObservation: #{result}\nThought:"
+            "\nObservation: #{result}\nThought:"
           end
         else
           # Return the final answer
@@ -92,7 +92,7 @@ module Agent
         question: question,
         tool_names: "[#{tools.join(", ")}]",
         tools: tools.map do |tool|
-          "#{tool}: #{Tool.const_get(Tool::Base::TOOLS[tool]).const_get("DESCRIPTION")}"
+          "#{tool}: #{Tool.const_get(Tool::Base::TOOLS[tool]).const_get(:DESCRIPTION)}"
         end.join("\n")
       )
     end

data/lib/dependency_helper.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+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
+
+  if !gem_requirement.satisfied_by?(gem_version)
+    raise "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

data/lib/langchain.rb

@@ -1,6 +1,8 @@
 # frozen_string_literal: true
 
 require_relative "./version"
+require_relative "./dependency_helper"
+require_relative "./logging"
 
 module Agent
   autoload :Base, "agent/base"
@@ -32,6 +34,7 @@ end
 module Tool
   autoload :Base, "tool/base"
   autoload :Calculator, "tool/calculator"
+  autoload :News, "tool/news"
   autoload :SerpApi, "tool/serp_api"
   autoload :Wikipedia, "tool/wikipedia"
 end

data/lib/llm/base.rb

@@ -12,16 +12,16 @@ module LLM
     }.freeze
 
     def default_dimension
-      self.class.const_get("DEFAULTS").dig(:dimension)
+      self.class.const_get(:DEFAULTS).dig(:dimension)
     end
 
     # Ensure that the LLM value passed in is supported
     # @param llm [Symbol] The LLM to use
     def self.validate_llm!(llm:)
       # TODO: Fix so this works when `llm` value is a string instead of a symbol
-      unless LLM::Base::LLMS.keys.include?(llm)
+      unless LLM::Base::LLMS.key?(llm)
         raise ArgumentError, "LLM must be one of #{LLM::Base::LLMS.keys}"
       end
     end
   end
-end
+end

data/lib/llm/cohere.rb

@@ -1,10 +1,7 @@
 # frozen_string_literal: true
 
-require "cohere"
-
 module LLM
   class Cohere < Base
-
     DEFAULTS = {
       temperature: 0.0,
       completion_model_name: "base",
@@ -13,6 +10,9 @@ module LLM
     }.freeze
 
     def initialize(api_key:)
+      depends_on "cohere-ruby"
+      require "cohere"
+
       @client = ::Cohere::Client.new(api_key: api_key)
     end
 
@@ -22,7 +22,7 @@ module LLM
     def embed(text:)
       response = client.embed(
         texts: [text],
-        model: DEFAULTS[:embeddings_model_name],
+        model: DEFAULTS[:embeddings_model_name]
       )
       response.dig("embeddings").first
     end
@@ -50,4 +50,4 @@ module LLM
     alias_method :generate_completion, :complete
     alias_method :generate_embedding, :embed
   end
-end
+end

data/lib/llm/openai.rb

@@ -4,7 +4,6 @@ require "openai"
 
 module LLM
   class OpenAI < Base
-
     DEFAULTS = {
       temperature: 0.0,
       completion_model_name: "text-davinci-003",
@@ -53,4 +52,4 @@ module LLM
     alias_method :generate_completion, :complete
     alias_method :generate_embedding, :embed
   end
-end
+end

data/lib/logging.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+require "logger"
+
+module Langchain
+  def self.logger
+    @@logger ||= Logger.new(STDOUT, level: :warn, formatter: -> (severity, datetime, progname, msg) { "[LangChain.rb] #{msg}\n" })
+  end
+
+  def self.logger=(instance)
+    @@logger = instance
+  end
+end

data/lib/prompt/base.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require 'strscan'
+require "strscan"
 
 module Prompt
   class Base
@@ -52,14 +52,12 @@ module Prompt
       FileUtils.mkdir_p(directory_path) unless directory_path.directory?
 
       if save_path.extname == ".json"
-        File.open(file_path, "w") { |f| f.write(to_h.to_json) }
+        File.write(file_path, to_h.to_json)
       else
         raise ArgumentError, "#{file_path} must be json"
       end
     end
 
-    private
-
     #
     # Extracts variables from a template string.
     #

data/lib/prompt/loading.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
-require 'strscan'
-require 'pathname'
+require "strscan"
+require "pathname"
 
 module Prompt
   TYPE_TO_LOADER = {
@@ -70,7 +70,7 @@ module Prompt
     def load_from_config(config)
       # If `_type` key is not present in the configuration hash, add it with a default value of `prompt`
       unless config.key?("_type")
-        puts "[WARN] No `_type` key found, defaulting to `prompt`"
+        Langchain.logger.warn "No `_type` key found, defaulting to `prompt`"
         config["_type"] = "prompt"
       end
 

data/lib/tool/base.rb

@@ -12,9 +12,14 @@ module Tool
     TOOLS = {
       "calculator" => "Tool::Calculator",
       "search" => "Tool::SerpApi",
-      "wikipedia" => "Tool::Wikipedia"
+      "wikipedia" => "Tool::Wikipedia",
+      "news" => "Tool::News"
     }
 
+    def self.description(value)
+      const_set(:DESCRIPTION, value.tr("\n", " ").strip)
+    end
+
     # Executes the tool and returns the answer
     # @param input [String] input to the tool
     # @return [String] answer
@@ -22,17 +27,17 @@ module Tool
       raise NotImplementedError, "Your tool must implement the `self.execute(input:)` method that returns a string"
     end
 
-    # 
+    #
     # Validates the list of strings (tools) are all supported or raises an error
     # @param tools [Array<String>] list of tools to be used
-    # 
+    #
     # @raise [ArgumentError] If any of the tools are not supported
-    # 
+    #
     def self.validate_tools!(tools:)
-      unrecognized_tools = tools - Tool::Base::TOOLS.keys 
+      unrecognized_tools = tools - Tool::Base::TOOLS.keys
 
       if unrecognized_tools.any?
-        raise ArgumentError, "Unrecognized Tools: #{unrecognized_tools}" 
+        raise ArgumentError, "Unrecognized Tools: #{unrecognized_tools}"
       end
     end
   end

data/lib/tool/calculator.rb

@@ -4,8 +4,11 @@ require "eqn"
 
 module Tool
   class Calculator < Base
-    DESCRIPTION = "Useful for getting the result of a math expression. " +
-      "The input to this tool should be a valid mathematical expression that could be executed by a simple calculator."
+    description <<~DESC
+      Useful for getting the result of a math expression.
+
+      The input to this tool should be a valid mathematical expression that could be executed by a simple calculator.
+    DESC
 
     # Evaluates a pure math expression or if equation contains non-math characters (e.g.: "12F in Celsius") then
     # it uses the google search calculator to evaluate the expression

data/lib/tool/serp_api.rb

@@ -7,10 +7,14 @@ module Tool
     # Wrapper around SerpAPI
     # Set ENV["SERPAPI_API_KEY"] to use it
 
-    DESCRIPTION = "A wrapper around Google Search. " +
-      "Useful for when you need to answer questions about current events. " +
-      "Always one of the first options when you need to find information on internet. " +
-      "Input should be a search query."
+    description <<~DESC
+      A wrapper around Google Search.
+
+      Useful for when you need to answer questions about current events.
+      Always one of the first options when you need to find information on internet.
+
+      Input should be a search query.
+    DESC
 
     # Executes Google Search and returns hash_results JSON
     # @param input [String] search query
@@ -18,7 +22,7 @@ module 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 self.execute(input:)
-      hash_results = self.execute_search(input: input)
+      hash_results = execute_search(input: input)
 
       hash_results.dig(:answer_box, :answer) ||
         hash_results.dig(:answer_box, :snippet) ||
@@ -33,7 +37,7 @@ module Tool
         q: input,
         serp_api_key: ENV["SERPAPI_API_KEY"]
       )
-      .get_hash
+        .get_hash
     end
   end
 end

data/lib/tool/wikipedia.rb

@@ -1,15 +1,19 @@
 # frozen_string_literal: true
 
-require 'wikipedia'
+require "wikipedia"
 
 module Tool
   class Wikipedia < Base
     # Tool that adds the capability to search using the Wikipedia API
 
-    DESCRIPTION = "A wrapper around Wikipedia. " +
-      "Useful for when you need to answer general questions about " +
-      "people, places, companies, facts, historical events, or other subjects. " +
-      "Input should be a search query."
+    description <<~DESC
+      A wrapper around Wikipedia.
+
+      Useful for when you need to answer general questions about
+      people, places, companies, facts, historical events, or other subjects.
+
+      Input should be a search query.
+    DESC
 
     # Executes Wikipedia API search and returns the answer
     # @param input [String] search query
@@ -17,7 +21,7 @@ module Tool
     def self.execute(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 
+      page.summary
     end
   end
 end

data/lib/vectorsearch/base.rb

@@ -6,7 +6,7 @@ module Vectorsearch
 
     attr_reader :client, :index_name, :llm, :llm_api_key, :llm_client
 
-    DEFAULT_METRIC = "cosine".freeze
+    DEFAULT_METRIC = "cosine"
 
     # @param llm [Symbol] The LLM to use
     # @param llm_api_key [String] The API key for the LLM
@@ -46,7 +46,7 @@ module Vectorsearch
           input_variables: ["context"]
         ),
         examples: [
-          { context: context }
+          {context: context}
         ],
         input_variables: ["question"],
         example_separator: "\n"
@@ -55,4 +55,4 @@ module Vectorsearch
       prompt_template.format(question: question)
     end
   end
-end
+end

data/lib/vectorsearch/milvus.rb

@@ -6,10 +6,7 @@ module Vectorsearch
   class Milvus < Base
     def initialize(
       url:,
-      api_key: nil,
-      index_name:,
-      llm:,
-      llm_api_key:
+      index_name:, llm:, llm_api_key:, api_key: nil
     )
       @client = ::Milvus::Client.new(
         url: url
@@ -96,7 +93,7 @@ module Vectorsearch
       client.search(
         collection_name: index_name,
         top_k: k.to_s,
-        vectors: [ embedding ],
+        vectors: [embedding],
         dsl_type: 1,
         params: "{\"nprobe\": 10}",
         anns_field: "content",

data/lib/vectorsearch/pinecone.rb

@@ -18,10 +18,10 @@ module Vectorsearch
       llm_api_key:
     )
       ::Pinecone.configure do |config|
-        config.api_key  = api_key
+        config.api_key = api_key
         config.environment = environment
       end
-    
+
       @client = ::Pinecone::Client.new
       @index_name = index_name
 
@@ -38,7 +38,7 @@ module Vectorsearch
         {
           # TODO: Allows passing in your own IDs
           id: SecureRandom.uuid,
-          metadata: { content: text },
+          metadata: {content: text},
           values: generate_embedding(text: text)
         }
       end
@@ -109,4 +109,4 @@ module Vectorsearch
       generate_completion(prompt: prompt)
     end
   end
-end
+end

data/lib/vectorsearch/qdrant.rb

@@ -32,12 +32,12 @@ module Vectorsearch
     def add_texts(
       texts:
     )
-      batch = { ids: [], vectors: [], payloads: [] }
+      batch = {ids: [], vectors: [], payloads: []}
 
       texts.each do |text|
         batch[:ids].push(SecureRandom.uuid)
         batch[:vectors].push(generate_embedding(text: text))
-        batch[:payloads].push({ content: text })
+        batch[:payloads].push({content: text})
       end
 
       client.points.upsert(
@@ -106,4 +106,4 @@ module Vectorsearch
       generate_completion(prompt: prompt)
     end
   end
-end
+end

data/lib/vectorsearch/weaviate.rb

@@ -37,7 +37,7 @@ module Vectorsearch
       objects = texts.map do |text|
         {
           class: index_name,
-          properties: { content: text }
+          properties: {content: text}
         }
       end
 
@@ -50,7 +50,7 @@ module Vectorsearch
     def create_default_schema
       client.schema.create(
         class_name: index_name,
-        vectorizer: "text2vec-#{llm.to_s}",
+        vectorizer: "text2vec-#{llm}",
         # TODO: Figure out a way to optionally enable it
         # "module_config": {
         #   "qna-openai": {}
@@ -132,4 +132,4 @@ module Vectorsearch
       end
     end
   end
-end
+end

data/lib/version.rb

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

metadata

@@ -1,29 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: langchainrb
 version: !ruby/object:Gem::Version
-  version: 0.3.2
+  version: 0.3.3
 platform: ruby
 authors:
 - Andrei Bondarev
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-05-15 00:00:00.000000000 Z
+date: 2023-05-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: pry-byebug
+  name: cohere-ruby
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 3.10.0
+        version: 0.9.3
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 3.10.0
+        version: 0.9.3
 - !ruby/object:Gem::Dependency
   name: dotenv-rails
   requirement: !ruby/object:Gem::Requirement
@@ -39,19 +39,19 @@ dependencies:
       - !ruby/object:Gem::Version
         version: 2.7.6
 - !ruby/object:Gem::Dependency
-  name: cohere-ruby
+  name: pry-byebug
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.9.3
-  type: :runtime
+        version: 3.10.0
+  type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.9.3
+        version: 3.10.0
 - !ruby/object:Gem::Dependency
   name: eqn
   requirement: !ruby/object:Gem::Requirement
@@ -187,10 +187,12 @@ files:
 - lib/agent/base.rb
 - lib/agent/chain_of_thought_agent/chain_of_thought_agent.rb
 - lib/agent/chain_of_thought_agent/chain_of_thought_agent_prompt.json
+- lib/dependency_helper.rb
 - lib/langchain.rb
 - lib/llm/base.rb
 - lib/llm/cohere.rb
 - lib/llm/openai.rb
+- lib/logging.rb
 - lib/prompt/base.rb
 - lib/prompt/few_shot_prompt_template.rb
 - lib/prompt/loading.rb