langchainrb 0.3.0 → 0.3.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1d43d7b8fab03608be188730dd7e67947e161176ea11ad29a3d6a3b3469045da
-  data.tar.gz: 2a0cf937f20dcb620fde4fac5bfdd1cbafee92d1212ab61fc25db352c3bce79f
+  metadata.gz: e751a38f6a248db9aabfac3f1b9bc547f61304dd366be999005941045a5adcd6
+  data.tar.gz: 56db69c0a578bdf198bfc12cb53e92f0e1d15654a1dfa29715d5d7833550da33
 SHA512:
-  metadata.gz: 2eb42e6ea5ee796bd7b78addd9c3cc75449ca84a680c0b50668eb678e14cb4991fe38577a323acbd628b30b3753a2197da8a34ef908dac15402277e078a8c287
-  data.tar.gz: 1a1d51903ef37908b2f0f5a2ded0542a04f5e812ff4c2fa37b67dbaa80636d50a70d7cf8bf05bfceedbe21a016d6dcb6e98a9fa3e25981feb05f4a4cec2ca52f
+  metadata.gz: bdd3060863a967b48a6123ea379d9a98632730a726b8da9924bb0061511b8bf88fa05f36c49ac4b8f136472e9aa204c4bd09929d31e0f2c0700ad16361a1a8cf
+  data.tar.gz: 88f04804d10f51d639b8643bd683b8a420920fc4ad2d26ef6c45f4fac034ec3f8b54787de51c4799f4a074029ad5649efca6e950b29fef62b96ce9452e5ec8cc

data/CHANGELOG.md

@@ -1,4 +1,13 @@
 ## [Unreleased]
+
+## [0.3.2] - 2023-05-15
+- Agents
+  - Fix Chain of Thought prompt loader
+
+## [0.3.1] - 2023-05-12
+- Tools
+  - Introducing `Tool::Wikipedia`, a tool that looks up Wikipedia entries
+
 ## [0.3.0] - 2023-05-12
 
 - Agents

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    langchainrb (0.3.0)
+    langchainrb (0.3.2)
       cohere-ruby (~> 0.9.3)
       eqn (~> 1.6.5)
       google_search_results (~> 2.0.0)
@@ -10,6 +10,7 @@ PATH
       qdrant-ruby (~> 0.9.0)
       ruby-openai (~> 4.0.0)
       weaviate-ruby (~> 0.8.0)
+      wikipedia-client (~> 1.17.0)
 
 GEM
   remote: https://rubygems.org/
@@ -32,6 +33,8 @@ GEM
       i18n (>= 1.6, < 2)
       minitest (>= 5.1)
       tzinfo (~> 2.0)
+    addressable (2.8.4)
+      public_suffix (>= 2.0.2, < 6.0)
     builder (3.2.4)
     byebug (11.1.3)
     coderay (1.1.3)
@@ -150,6 +153,7 @@ GEM
     pry-byebug (3.10.1)
       byebug (~> 11.0)
       pry (>= 0.13, < 0.15)
+    public_suffix (5.0.1)
     qdrant-ruby (0.9.2)
       faraday (~> 1)
       faraday_middleware (~> 1)
@@ -196,6 +200,8 @@ GEM
       faraday (~> 1)
       faraday_middleware (~> 1)
       graphlient (~> 0.6.0)
+    wikipedia-client (1.17.0)
+      addressable (~> 2.7)
     zeitwerk (2.6.8)
 
 PLATFORMS

data/README.md

@@ -224,8 +224,9 @@ agent.run(question: "How many full soccer fields would be needed to cover the di
 
 | Name     | Description          | Requirements         |
 | -------- | :------------------: | :------------------: |
-| "search" | A wrapper around Google Search | `ENV["SERP_API_KEY"]` (https://serpapi.com/manage-api-key)
 | "calculator" | Useful for getting the result of a math expression | |
+| "search" | A wrapper around Google Search | `ENV["SERPAPI_API_KEY"]` (https://serpapi.com/manage-api-key)
+| "wikipedia" | Calls Wikipedia API to retrieve the summary | |
 
 ## Development
 

data/lib/agent/chain_of_thought_agent/chain_of_thought_agent.rb

@@ -101,7 +101,7 @@ module Agent
     # @return [PromptTemplate] PromptTemplate instance
     def prompt_template
       @template ||= Prompt.load_from_path(
-        file_path: "lib/agent/chain_of_thought_agent/chain_of_thought_agent_prompt.json"
+        file_path: Pathname.new(__dir__).join("chain_of_thought_agent_prompt.json")
       )
     end
   end

data/lib/langchain.rb

@@ -33,4 +33,5 @@ module Tool
   autoload :Base, "tool/base"
   autoload :Calculator, "tool/calculator"
   autoload :SerpApi, "tool/serp_api"
+  autoload :Wikipedia, "tool/wikipedia"
 end

data/lib/tool/base.rb

@@ -11,7 +11,8 @@ module Tool
 
     TOOLS = {
       "calculator" => "Tool::Calculator",
-      "search" => "Tool::SerpApi"
+      "search" => "Tool::SerpApi",
+      "wikipedia" => "Tool::Wikipedia"
     }
 
     # Executes the tool and returns the answer

data/lib/tool/serp_api.rb

@@ -4,6 +4,9 @@ require "google_search_results"
 
 module Tool
   class SerpApi < Base
+    # 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. " +
@@ -28,7 +31,7 @@ module Tool
     def self.execute_search(input:)
       GoogleSearch.new(
         q: input,
-        serp_api_key: ENV["SERP_API_KEY"]
+        serp_api_key: ENV["SERPAPI_API_KEY"]
       )
       .get_hash
     end

data/lib/tool/wikipedia.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+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."
+
+    # Executes Wikipedia API search and returns the answer
+    # @param input [String] search query
+    # @return [String] Answer
+    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 
+    end
+  end
+end

data/lib/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: langchainrb
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.3.2
 platform: ruby
 authors:
 - Andrei Bondarev
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-05-12 00:00:00.000000000 Z
+date: 2023-05-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pry-byebug
@@ -150,6 +150,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 0.8.0
+- !ruby/object:Gem::Dependency
+  name: wikipedia-client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.17.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.17.0
 description: Build ML/AI-powered applications with Ruby's LangChain
 email:
 - andrei.bondarev13@gmail.com
@@ -184,6 +198,7 @@ files:
 - lib/tool/base.rb
 - lib/tool/calculator.rb
 - lib/tool/serp_api.rb
+- lib/tool/wikipedia.rb
 - lib/vectorsearch/base.rb
 - lib/vectorsearch/milvus.rb
 - lib/vectorsearch/pinecone.rb