langchainrb 0.6.9 → 0.6.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4d046a7cae545da3092694fee60b0fa3caa0852492c9bf7dbbe35001d96106e4
-  data.tar.gz: 8ed841d2359400f9a4700a49b82030fbe3dd3563545eca31b487914e8fb236f7
+  metadata.gz: ba1fb0e3fbc05e4279fe3a698ad8fb1a25e02788991a8e6a7b27b411771096f3
+  data.tar.gz: 309cda1c8c7a4982b22c6ad2f82c20fb12ca3bdfdc3e8c0ebeaa9687a7f71ce0
 SHA512:
-  metadata.gz: eed2f03cbdd74ef42fafd07524dc6bf5c7ac2278e5e55dd1aad5b2cd87f319a91c8719a6e9ac43ac50cda5525f3defab14cfecc749bdec81075d52b3935b23dd
-  data.tar.gz: cd4fce88b0f9f545f6d32660879e6fa5044ec74b41dcb41bc29eafb78b307d8b2cb8204856098657d8c9707331c8629c8a707a227fe4d20a5816347f8a66991c
+  metadata.gz: 3457cbad7efbc5504f4cb3b684e3837984be5d486c1ee21a718508d606dd63ccc2223dffc55f3cc9c52f5fd0a533b364407bbb4c3208515e2bab8ca2af9ea60a
+  data.tar.gz: 1ff3cded239c286ee87d7a2f0a1cfa45c734b185e35b6e6fbc63fd2951f7c7397e8562fa145a16f545ef07b2e983f40715010e9c96653b5f2bf128325a5a7577

data/.env.example

@@ -1,4 +1,5 @@
 AI21_API_KEY=
+ANTHROPIC_API_KEY=
 CHROMA_URL=
 COHERE_API_KEY=
 GOOGLE_PALM_API_KEY=

data/CHANGELOG.md

@@ -1,5 +1,9 @@
 ## [Unreleased]
 
+## [0.6.10] - 2023-08-01
+- 🗣️ LLMs
+  - Introducing Anthropic support
+
 ## [0.6.9] - 2023-07-29
 
 ## [0.6.8] - 2023-07-21

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    langchainrb (0.6.9)
+    langchainrb (0.6.10)
       baran (~> 0.1.6)
       colorize (~> 0.8.1)
       json-schema (~> 4.0.0)
@@ -33,6 +33,9 @@ GEM
       public_suffix (>= 2.0.2, < 6.0)
     afm (0.2.2)
     ai21 (0.2.1)
+    anthropic (0.1.0)
+      faraday (>= 1)
+      faraday-multipart (>= 1)
     ast (2.4.2)
     baran (0.1.7)
     builder (3.2.4)
@@ -318,6 +321,7 @@ PLATFORMS
 
 DEPENDENCIES
   ai21 (~> 0.2.1)
+  anthropic (~> 0.1.0)
   chroma-db (~> 0.3.0)
   cohere-ruby (~> 0.9.5)
   docx (~> 0.8.0)

data/README.md

@@ -203,6 +203,12 @@ Add `gem "ai21", "~> 0.2.1"` to your Gemfile.
 ai21 = Langchain::LLM::AI21.new(api_key: ENV["AI21_API_KEY"])
 ```
 
+#### Anthropic
+Add `gem "anthropic", "~> 0.1.0"` to your Gemfile.
+```ruby
+anthropic = Langchain::LLM::Anthropic.new(api_key: ENV["ANTHROPIC_API_KEY"])
+```
+
 ### Using Prompts 📋
 
 #### Prompt Templates
@@ -537,7 +543,7 @@ Join us in the [Langchain.rb](https://discord.gg/WDARp7J2n8) Discord server.
 
 ## Contributing
 
-Bug reports and pull requests are welcome on GitHub at https://github.com/andreibondarev/langchain.
+Bug reports and pull requests are welcome on GitHub at https://github.com/andreibondarev/langchainrb.
 
 ## License
 

data/lib/langchain/llm/anthropic.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+module Langchain::LLM
+  #
+  # Wrapper around Anthropic APIs.
+  #
+  # Gem requirements:
+  #   gem "anthropic", "~> 0.1.0"
+  #
+  # Usage:
+  #     anthorpic = Langchain::LLM::Anthropic.new(api_key:)
+  #
+  class Anthropic < Base
+    DEFAULTS = {
+      temperature: 0.0,
+      completion_model_name: "claude-2"
+    }.freeze
+
+    # TODO: Implement token length validator for Anthropic
+    # LENGTH_VALIDATOR = Langchain::Utils::TokenLength::AnthropicValidator
+
+    def initialize(api_key:, llm_options: {}, default_options: {})
+      depends_on "anthropic"
+      require "anthropic"
+
+      @client = ::Anthropic::Client.new(access_token: api_key, **llm_options)
+      @defaults = DEFAULTS.merge(default_options)
+    end
+
+    #
+    # Generate a completion for a given prompt
+    #
+    # @param prompt [String] The prompt to generate a completion for
+    # @param params [Hash] extra parameters passed to Anthropic::Client#complete
+    # @return [String] The completion
+    #
+    def complete(prompt:, **params)
+      parameters = compose_parameters @defaults[:completion_model_name], params
+
+      parameters[:prompt] = prompt
+
+      # TODO: Implement token length validator for Anthropic
+      # parameters[:max_tokens_to_sample] = validate_max_tokens(prompt, parameters[:completion_model_name])
+
+      response = client.complete(parameters: parameters)
+      response.dig("completion")
+    end
+
+    private
+
+    def compose_parameters(model, params)
+      default_params = {model: model, temperature: @defaults[:temperature]}
+
+      default_params.merge(params)
+    end
+
+    # TODO: Implement token length validator for Anthropic
+    # def validate_max_tokens(messages, model)
+    #   LENGTH_VALIDATOR.validate_max_tokens!(messages, model)
+    # end
+  end
+end

data/lib/langchain/vectorsearch/milvus.rb

@@ -8,7 +8,7 @@ module Langchain::Vectorsearch
     # Gem requirements: gem "milvus", "~> 0.9.0"
     #
     # Usage:
-    # milvus = Langchain::Vectorsearch::Milvus.new(url:, index_name:, llm:, llm_api_key:)
+    # milvus = Langchain::Vectorsearch::Milvus.new(url:, index_name:, llm:, api_key:)
     #
 
     def initialize(url:, index_name:, llm:, api_key: nil)

data/lib/langchain/version.rb

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

data/lib/langchain.rb

@@ -131,6 +131,7 @@ module Langchain
 
   module LLM
     autoload :AI21, "langchain/llm/ai21"
+    autoload :Anthropic, "langchain/llm/anthropic"
     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.6.9
+  version: 0.6.10
 platform: ruby
 authors:
 - Andrei Bondarev
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-07-29 00:00:00.000000000 Z
+date: 2023-08-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: baran
@@ -136,6 +136,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 0.2.1
+- !ruby/object:Gem::Dependency
+  name: anthropic
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.1.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.1.0
 - !ruby/object:Gem::Dependency
   name: chroma-db
   requirement: !ruby/object:Gem::Requirement
@@ -518,6 +532,7 @@ files:
 - lib/langchain/data.rb
 - lib/langchain/dependency_helper.rb
 - lib/langchain/llm/ai21.rb
+- lib/langchain/llm/anthropic.rb
 - lib/langchain/llm/base.rb
 - lib/langchain/llm/cohere.rb
 - lib/langchain/llm/google_palm.rb