langchainrb 0.3.4 → 0.3.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a5a782dd2282ab5dd4aed3f1d0e421a4f9b227fa4c5450ed27f2f98a86af74f4
-  data.tar.gz: b47bb5d6789d7abb81f56ee1beb0b52323184f578475aec3e92fcc19b4a1314a
+  metadata.gz: 252bfabecd9100d29bab8be718ba43421369812a6f02d4526970fcef85a7617e
+  data.tar.gz: 468cdbb3a31238b5dfc53d4dbd22d5d00e2d0347a3482660361841f6663bc1da
 SHA512:
-  metadata.gz: 4e918b49b2b04a0e7009db732a5b24ab080a0d8d6b4c3be4084aa3a1492ddd6c1627d467b8bff4b34e1e5160b71622a75772ef92658dce2589b9542b8b0a8137
-  data.tar.gz: 3b465f1a05e614d64d416582aeaf7d998bf51422705331b220e08b00774d4beed4dcbc4e2a71b81bad87cddfe8c8f4c8db421d650e7161f16b26a1259922f71c
+  metadata.gz: deed8fc8b33c590b1c057a53985295a2447e11f0917c3646b03f592035c15f00fe27b852bc6f6e8455e79a604ac699252720cc84d4d011c69b1b1f71a7047d81
+  data.tar.gz: f6681fdd1ffcc468bb515c3bae1231c17d85c59925305a698be4766bb8beb29b0a748b088c8943b012a01da7d6c38744b7e46efb7ad4fd6f9245f9a634dffcf9

data/CHANGELOG.md

@@ -1,5 +1,8 @@
 ## [Unreleased]
 
+## [0.3.5] - 2023-05-16
+- Bug fixes
+
 ## [0.3.4] - 2023-05-16
 - LLMs
   - Introducing support for HuggingFace

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    langchainrb (0.3.4)
+    langchainrb (0.3.5)
 
 GEM
   remote: https://rubygems.org/

data/lib/llm/base.rb

@@ -8,7 +8,8 @@ module LLM
     # TODO: Add support for HuggingFace and other LLMs
     LLMS = {
       openai: "OpenAI",
-      cohere: "Cohere"
+      cohere: "Cohere",
+      huggingface: "HuggingFace"
     }.freeze
 
     def default_dimension

data/lib/llm/hugging_face.rb

@@ -2,16 +2,18 @@
 
 module LLM
   class HuggingFace < Base
-    # The gem does not currently accept other models: 
+    # The gem does not currently accept other models:
     # https://github.com/alchaplinsky/hugging-face/blob/main/lib/hugging_face/inference_api.rb#L32-L34
     DEFAULTS = {
-      embeddings_model_name: "sentence-transformers/all-MiniLM-L6-v2"      
+      temperature: 0.0,
+      embeddings_model_name: "sentence-transformers/all-MiniLM-L6-v2",
+      dimension: 384 # Vector size generated by the above model
     }.freeze
-    
-    # 
+
+    #
     # Intialize the HuggingFace LLM
     # @param api_key [String] The API key to use
-    # 
+    #
     def initialize(api_key:)
       depends_on "hugging-face"
       require "hugging_face"
@@ -24,9 +26,8 @@ module LLM
     # @return [Array] The embedding
     def embed(text:)
       response = client.embedding(
-        input: text,
-        model: DEFAULTS[:embeddings_model_name]
+        input: text
       )
     end
   end
-end
+end

data/lib/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: langchainrb
 version: !ruby/object:Gem::Version
-  version: 0.3.4
+  version: 0.3.5
 platform: ruby
 authors:
 - Andrei Bondarev