langchainrb 0.3.0 → 0.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1d43d7b8fab03608be188730dd7e67947e161176ea11ad29a3d6a3b3469045da
-  data.tar.gz: 2a0cf937f20dcb620fde4fac5bfdd1cbafee92d1212ab61fc25db352c3bce79f
+  metadata.gz: a179004be8e547124fe81922b2cf7190d34f94f7775c28df7986d17644dffa31
+  data.tar.gz: 0eaf80695379a5038175d561dfa348d4146073b008aeb662b3b523e8bd04d54d
 SHA512:
-  metadata.gz: 2eb42e6ea5ee796bd7b78addd9c3cc75449ca84a680c0b50668eb678e14cb4991fe38577a323acbd628b30b3753a2197da8a34ef908dac15402277e078a8c287
-  data.tar.gz: 1a1d51903ef37908b2f0f5a2ded0542a04f5e812ff4c2fa37b67dbaa80636d50a70d7cf8bf05bfceedbe21a016d6dcb6e98a9fa3e25981feb05f4a4cec2ca52f
+  metadata.gz: 152fd7a35c9df7e9541b50059a27f5e76d67263ceb4e25dab07224bd80bffe70936ab712ab3345a401b4be5016a4165c8a7a6cf1b4b1673c36275d926d7ce17d
+  data.tar.gz: 848984972f6ab3f1dd4078d303872be1e70ae78a1ceedf2f0b7663f2144d12831c690e1ad48cbd579e779fbe6da8ad6339ee42130716bf4bb137917f5edf3e30

data/CHANGELOG.md

@@ -1,4 +1,9 @@
 ## [Unreleased]
+
+## [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.1)
       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/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/llm/cohere.rb

@@ -34,7 +34,7 @@ module LLM
       default_params = {
         prompt: prompt,
         temperature: DEFAULTS[:temperature],
-        model: DEFAULTS[:completion_model_name]
+        model: "xlarge" #DEFAULTS[:completion_model_name]
       }
 
       if params[:stop_sequences]

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.1"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: langchainrb
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.3.1
 platform: ruby
 authors:
 - Andrei Bondarev
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-05-12 00:00:00.000000000 Z
+date: 2023-05-13 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