langchainrb 0.16.1 → 0.17.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b078089a99e9e8d6654a244165ecc9d0f3dfdd8fbc0367623d41fe771a98ac41
-  data.tar.gz: 890c371564ce9188087bed9eb053a59e11f7b734a44b9f753696f8458f8a7b7e
+  metadata.gz: c125e87218ae9ce233711968edf212e128b50910f118951296593d141c9b7a68
+  data.tar.gz: a8e0890d5eef171709068ee19e4e0955929035d732225916e28debd03cfebcbf
 SHA512:
-  metadata.gz: 8f458bfae5af31190f41661a13c24e5cd63d5f88e594e854ee79ea3b8af1f51b20552f178c89c71e454a86e4d827b3facecd97f0fa3ef107b7b6097754fab5e3
-  data.tar.gz: cfe0c684f89c5eef73ceb26b70292fe8fc4f941e13795ac98bd1d3197321a1303250d2584f9e45aa530e311304004911ffe3a6af7f606f6a733baad21ff2b814
+  metadata.gz: ae228197ef8be183cf9f36dc46bb523033fcfd1c563246e3f49bec081822b819d3ca1b93a0b5a780af276cb80ef3ce6ef6734e87f7cf3fad335d0e57e186e965
+  data.tar.gz: fd17a2caabbce6d6b006831e3eb51c6c3c174f178536991f6c0477aadc4d5ea64c2dae317cf59be9691908c0d0908cca6fac681eedc413b69588883cb6494a50

data/CHANGELOG.md

@@ -1,5 +1,12 @@
 ## [Unreleased]
 
+## [0.17.0] - 2024-10-02
+- [BREAKING] Langchain::Vectorsearch::Milvus was rewritten to work with newer milvus 0.10.0 gem
+- [BREAKING] Removing Langchain::LLM::GooglePalm
+- Assistant can now process image_urls in the messages (currently only for OpenAI and Mistral AI)
+- Vectorsearch providers utilize the global Langchain.logger
+- Update required milvus, qdrant and weaviate versions
+
 ## [0.16.1] - 2024-09-30
 - Deprecate Langchain::LLM::GooglePalm
 - Allow setting response_object: {} parameter when initializing supported Langchain::LLM::* classes

data/README.md

@@ -63,7 +63,6 @@ The `Langchain::LLM` module provides a unified interface for interacting with va
 - Azure OpenAI
 - Cohere
 - Google Gemini
-- Google PaLM (deprecated)
 - Google Vertex AI
 - HuggingFace
 - LlamaCpp
@@ -501,6 +500,12 @@ assistant = Langchain::Assistant.new(
 # Add a user message and run the assistant
 assistant.add_message_and_run!(content: "What's the latest news about AI?")
 
+# Supply an image to the assistant
+assistant.add_message_and_run!(
+  content: "Show me a picture of a cat",
+  image: "https://upload.wikimedia.org/wikipedia/commons/thumb/d/dd/Gfp-wisconsin-madison-the-nature-boardwalk.jpg/2560px-Gfp-wisconsin-madison-the-nature-boardwalk.jpg"
+)
+
 # Access the conversation thread
 messages = assistant.messages
 

data/lib/langchain/assistants/assistant.rb

@@ -63,13 +63,14 @@ module Langchain
 
     # Add a user message to the messages array
     #
-    # @param content [String] The content of the message
     # @param role [String] The role attribute of the message. Default: "user"
+    # @param content [String] The content of the message
+    # @param image_url [String] The URL of the image to include in the message
     # @param tool_calls [Array<Hash>] The tool calls to include in the message
     # @param tool_call_id [String] The ID of the tool call to include in the message
     # @return [Array<Langchain::Message>] The messages
-    def add_message(content: nil, role: "user", tool_calls: [], tool_call_id: nil)
-      message = build_message(role: role, content: content, tool_calls: tool_calls, tool_call_id: tool_call_id)
+    def add_message(role: "user", content: nil, image_url: nil, tool_calls: [], tool_call_id: nil)
+      message = build_message(role: role, content: content, image_url: image_url, tool_calls: tool_calls, tool_call_id: tool_call_id)
 
       # Call the callback with the message
       add_message_callback.call(message) if add_message_callback # rubocop:disable Style/SafeNavigation
@@ -145,8 +146,8 @@ module Langchain
     # @param content [String] The content of the message
     # @param auto_tool_execution [Boolean] Whether or not to automatically run tools
     # @return [Array<Langchain::Message>] The messages
-    def add_message_and_run(content:, auto_tool_execution: false)
-      add_message(content: content, role: "user")
+    def add_message_and_run(content: nil, image_url: nil, auto_tool_execution: false)
+      add_message(content: content, image_url: image_url, role: "user")
       run(auto_tool_execution: auto_tool_execution)
     end
 
@@ -154,8 +155,8 @@ module Langchain
     #
     # @param content [String] The content of the message
     # @return [Array<Langchain::Message>] The messages
-    def add_message_and_run!(content:)
-      add_message_and_run(content: content, auto_tool_execution: true)
+    def add_message_and_run!(content: nil, image_url: nil)
+      add_message_and_run(content: content, image_url: image_url, auto_tool_execution: true)
     end
 
     # Submit tool output
@@ -388,11 +389,12 @@ module Langchain
     #
     # @param role [String] The role of the message
     # @param content [String] The content of the message
+    # @param image_url [String] The URL of the image to include in the message
     # @param tool_calls [Array<Hash>] The tool calls to include in the message
     # @param tool_call_id [String] The ID of the tool call to include in the message
     # @return [Langchain::Message] The Message object
-    def build_message(role:, content: nil, tool_calls: [], tool_call_id: nil)
-      @llm_adapter.build_message(role: role, content: content, tool_calls: tool_calls, tool_call_id: tool_call_id)
+    def build_message(role:, content: nil, image_url: nil, tool_calls: [], tool_call_id: nil)
+      @llm_adapter.build_message(role: role, content: content, image_url: image_url, tool_calls: tool_calls, tool_call_id: tool_call_id)
     end
 
     # Increment the tokens count based on the last interaction with the LLM
@@ -443,7 +445,7 @@ module Langchain
             raise NotImplementedError, "Subclasses must implement extract_tool_call_args"
           end
 
-          def build_message(role:, content: nil, tool_calls: [], tool_call_id: nil)
+          def build_message(role:, content: nil, image_url: nil, tool_calls: [], tool_call_id: nil)
             raise NotImplementedError, "Subclasses must implement build_message"
           end
         end
@@ -457,7 +459,9 @@ module Langchain
             params
           end
 
-          def build_message(role:, content: nil, tool_calls: [], tool_call_id: nil)
+          def build_message(role:, content: nil, image_url: nil, tool_calls: [], tool_call_id: nil)
+            warn "Image URL is not supported by Ollama currently" if image_url
+
             Langchain::Messages::OllamaMessage.new(role: role, content: content, tool_calls: tool_calls, tool_call_id: tool_call_id)
           end
 
@@ -506,8 +510,8 @@ module Langchain
             params
           end
 
-          def build_message(role:, content: nil, tool_calls: [], tool_call_id: nil)
-            Langchain::Messages::OpenAIMessage.new(role: role, content: content, tool_calls: tool_calls, tool_call_id: tool_call_id)
+          def build_message(role:, content: nil, image_url: nil, tool_calls: [], tool_call_id: nil)
+            Langchain::Messages::OpenAIMessage.new(role: role, content: content, image_url: image_url, tool_calls: tool_calls, tool_call_id: tool_call_id)
           end
 
           # Extract the tool call information from the OpenAI tool call hash
@@ -564,8 +568,8 @@ module Langchain
             params
           end
 
-          def build_message(role:, content: nil, tool_calls: [], tool_call_id: nil)
-            Langchain::Messages::MistralAIMessage.new(role: role, content: content, tool_calls: tool_calls, tool_call_id: tool_call_id)
+          def build_message(role:, content: nil, image_url: nil, tool_calls: [], tool_call_id: nil)
+            Langchain::Messages::MistralAIMessage.new(role: role, content: content, image_url: image_url, tool_calls: tool_calls, tool_call_id: tool_call_id)
           end
 
           # Extract the tool call information from the OpenAI tool call hash
@@ -623,7 +627,9 @@ module Langchain
             params
           end
 
-          def build_message(role:, content: nil, tool_calls: [], tool_call_id: nil)
+          def build_message(role:, content: nil, image_url: nil, tool_calls: [], tool_call_id: nil)
+            warn "Image URL is not supported by Google Gemini" if image_url
+
             Langchain::Messages::GoogleGeminiMessage.new(role: role, content: content, tool_calls: tool_calls, tool_call_id: tool_call_id)
           end
 
@@ -676,7 +682,9 @@ module Langchain
             params
           end
 
-          def build_message(role:, content: nil, tool_calls: [], tool_call_id: nil)
+          def build_message(role:, content: nil, image_url: nil, tool_calls: [], tool_call_id: nil)
+            warn "Image URL is not supported by Anthropic currently" if image_url
+
             Langchain::Messages::AnthropicMessage.new(role: role, content: content, tool_calls: tool_calls, tool_call_id: tool_call_id)
           end
 

data/lib/langchain/assistants/messages/base.rb

@@ -3,7 +3,11 @@
 module Langchain
   module Messages
     class Base
-      attr_reader :role, :content, :tool_calls, :tool_call_id
+      attr_reader :role,
+        :content,
+        :image_url,
+        :tool_calls,
+        :tool_call_id
 
       # Check if the message came from a user
       #

data/lib/langchain/assistants/messages/mistral_ai_message.rb

@@ -15,17 +15,20 @@ module Langchain
 
       # Initialize a new MistralAI message
       #
-      # @param [String] The role of the message
-      # @param [String] The content of the message
-      # @param [Array<Hash>] The tool calls made in the message
-      # @param [String] The ID of the tool call
-      def initialize(role:, content: nil, tool_calls: [], tool_call_id: nil) # TODO: Implement image_file: reference (https://platform.openai.com/docs/api-reference/messages/object#messages/object-content)
+      # @param role [String] The role of the message
+      # @param content [String] The content of the message
+      # @param image_url [String] The URL of the image
+      # @param tool_calls [Array<Hash>] The tool calls made in the message
+      # @param tool_call_id [String] The ID of the tool call
+      def initialize(role:, content: nil, image_url: nil, tool_calls: [], tool_call_id: nil) # TODO: Implement image_file: reference (https://platform.openai.com/docs/api-reference/messages/object#messages/object-content)
         raise ArgumentError, "Role must be one of #{ROLES.join(", ")}" unless ROLES.include?(role)
         raise ArgumentError, "Tool calls must be an array of hashes" unless tool_calls.is_a?(Array) && tool_calls.all? { |tool_call| tool_call.is_a?(Hash) }
 
         @role = role
         # Some Tools return content as a JSON hence `.to_s`
         @content = content.to_s
+        # Make sure you're using the Pixtral model if you want to send image_url
+        @image_url = image_url
         @tool_calls = tool_calls
         @tool_call_id = tool_call_id
       end
@@ -43,9 +46,28 @@ module Langchain
       def to_hash
         {}.tap do |h|
           h[:role] = role
-          h[:content] = content if content # Content is nil for tool calls
-          h[:tool_calls] = tool_calls if tool_calls.any?
-          h[:tool_call_id] = tool_call_id if tool_call_id
+
+          if tool_calls.any?
+            h[:tool_calls] = tool_calls
+          else
+            h[:tool_call_id] = tool_call_id if tool_call_id
+
+            h[:content] = []
+
+            if content && !content.empty?
+              h[:content] << {
+                type: "text",
+                text: content
+              }
+            end
+
+            if image_url
+              h[:content] << {
+                type: "image_url",
+                image_url: image_url
+              }
+            end
+          end
         end
       end
 

data/lib/langchain/assistants/messages/openai_message.rb

@@ -15,17 +15,25 @@ module Langchain
 
       # Initialize a new OpenAI message
       #
-      # @param [String] The role of the message
-      # @param [String] The content of the message
-      # @param [Array<Hash>] The tool calls made in the message
-      # @param [String] The ID of the tool call
-      def initialize(role:, content: nil, tool_calls: [], tool_call_id: nil) # TODO: Implement image_file: reference (https://platform.openai.com/docs/api-reference/messages/object#messages/object-content)
+      # @param role [String] The role of the message
+      # @param content [String] The content of the message
+      # @param image_url [String] The URL of the image
+      # @param tool_calls [Array<Hash>] The tool calls made in the message
+      # @param tool_call_id [String] The ID of the tool call
+      def initialize(
+        role:,
+        content: nil,
+        image_url: nil,
+        tool_calls: [],
+        tool_call_id: nil
+      )
         raise ArgumentError, "Role must be one of #{ROLES.join(", ")}" unless ROLES.include?(role)
         raise ArgumentError, "Tool calls must be an array of hashes" unless tool_calls.is_a?(Array) && tool_calls.all? { |tool_call| tool_call.is_a?(Hash) }
 
         @role = role
         # Some Tools return content as a JSON hence `.to_s`
         @content = content.to_s
+        @image_url = image_url
         @tool_calls = tool_calls
         @tool_call_id = tool_call_id
       end
@@ -43,9 +51,30 @@ module Langchain
       def to_hash
         {}.tap do |h|
           h[:role] = role
-          h[:content] = content if content # Content is nil for tool calls
-          h[:tool_calls] = tool_calls if tool_calls.any?
-          h[:tool_call_id] = tool_call_id if tool_call_id
+
+          if tool_calls.any?
+            h[:tool_calls] = tool_calls
+          else
+            h[:tool_call_id] = tool_call_id if tool_call_id
+
+            h[:content] = []
+
+            if content && !content.empty?
+              h[:content] << {
+                type: "text",
+                text: content
+              }
+            end
+
+            if image_url
+              h[:content] << {
+                type: "image_url",
+                image_url: {
+                  url: image_url
+                }
+              }
+            end
+          end
         end
       end
 

data/lib/langchain/llm/base.rb

@@ -11,9 +11,8 @@ module Langchain::LLM
   # - {Langchain::LLM::Anthropic}
   # - {Langchain::LLM::Azure}
   # - {Langchain::LLM::Cohere}
-  # - {Langchain::LLM::GooglePalm}
-  # - {Langchain::LLM::GoogleVertexAI}
   # - {Langchain::LLM::GoogleGemini}
+  # - {Langchain::LLM::GoogleVertexAI}
   # - {Langchain::LLM::HuggingFace}
   # - {Langchain::LLM::LlamaCpp}
   # - {Langchain::LLM::OpenAI}

data/lib/langchain/vectorsearch/elasticsearch.rb

@@ -37,7 +37,7 @@ module Langchain::Vectorsearch
       @options = {
         url: url,
         request_timeout: 20,
-        log: false
+        logger: Langchain.logger
       }.merge(es_options)
 
       @es_client = ::Elasticsearch::Client.new(**options)

data/lib/langchain/vectorsearch/milvus.rb

@@ -6,16 +6,18 @@ module Langchain::Vectorsearch
     # Wrapper around Milvus REST APIs.
     #
     # Gem requirements:
-    #     gem "milvus", "~> 0.9.3"
+    #     gem "milvus", "~> 0.10.3"
     #
     # Usage:
-    # milvus = Langchain::Vectorsearch::Milvus.new(url:, index_name:, llm:, api_key:)
+    #     milvus = Langchain::Vectorsearch::Milvus.new(url:, index_name:, llm:, api_key:)
     #
-
     def initialize(url:, index_name:, llm:, api_key: nil)
       depends_on "milvus"
 
-      @client = ::Milvus::Client.new(url: url)
+      @client = ::Milvus::Client.new(
+        url: url,
+        logger: Langchain.logger
+      )
       @index_name = index_name
 
       super(llm: llm)
@@ -24,33 +26,24 @@ module Langchain::Vectorsearch
     def add_texts(texts:)
       client.entities.insert(
         collection_name: index_name,
-        num_rows: Array(texts).size,
-        fields_data: [
-          {
-            field_name: "content",
-            type: ::Milvus::DATA_TYPES["varchar"],
-            field: Array(texts)
-          }, {
-            field_name: "vectors",
-            type: ::Milvus::DATA_TYPES["float_vector"],
-            field: Array(texts).map { |text| llm.embed(text: text).embedding }
-          }
-        ]
+        data: texts.map do |text|
+          {content: text, vector: llm.embed(text: text).embedding}
+        end
       )
     end
 
+    # TODO: Add update_texts method
+
     # Deletes a list of texts in the index
     #
     # @param ids [Array<Integer>] The ids of texts to delete
     # @return [Boolean] The response from the server
     def remove_texts(ids:)
       raise ArgumentError, "ids must be an array" unless ids.is_a?(Array)
-      # Convert ids to integers if strings are passed
-      ids = ids.map(&:to_i)
 
       client.entities.delete(
         collection_name: index_name,
-        expression: "id in #{ids}"
+        filter: "id in #{ids}"
       )
     end
 
@@ -62,33 +55,25 @@ module Langchain::Vectorsearch
       client.collections.create(
         auto_id: true,
         collection_name: index_name,
-        description: "Default schema created by langchain.rb",
         fields: [
           {
-            name: "id",
-            is_primary_key: true,
-            autoID: true,
-            data_type: ::Milvus::DATA_TYPES["int64"]
+            fieldName: "id",
+            isPrimary: true,
+            dataType: "Int64"
           }, {
-            name: "content",
-            is_primary_key: false,
-            data_type: ::Milvus::DATA_TYPES["varchar"],
-            type_params: [
-              {
-                key: "max_length",
-                value: "32768" # Largest allowed value
-              }
-            ]
+            fieldName: "content",
+            isPrimary: false,
+            dataType: "VarChar",
+            elementTypeParams: {
+              max_length: "32768" # Largest allowed value
+            }
           }, {
-            name: "vectors",
-            data_type: ::Milvus::DATA_TYPES["float_vector"],
-            is_primary_key: false,
-            type_params: [
-              {
-                key: "dim",
-                value: llm.default_dimensions.to_s
-              }
-            ]
+            fieldName: "vector",
+            isPrimary: false,
+            dataType: "FloatVector",
+            elementTypeParams: {
+              dim: llm.default_dimensions.to_s
+            }
           }
         ]
       )
@@ -97,13 +82,17 @@ module Langchain::Vectorsearch
     # Create the default index
     # @return [Boolean] The response from the server
     def create_default_index
-      client.indices.create(
+      client.indexes.create(
         collection_name: index_name,
-        field_name: "vectors",
-        extra_params: [
-          {key: "metric_type", value: "L2"},
-          {key: "index_type", value: "IVF_FLAT"},
-          {key: "params", value: "{\"nlist\":1024}"}
+        index_params: [
+          {
+            metricType: "L2",
+            fieldName: "vector",
+            indexName: "vector_idx",
+            indexConfig: {
+              index_type: "AUTOINDEX"
+            }
+          }
         ]
       )
     end
@@ -111,13 +100,13 @@ module Langchain::Vectorsearch
     # Get the default schema
     # @return [Hash] The response from the server
     def get_default_schema
-      client.collections.get(collection_name: index_name)
+      client.collections.describe(collection_name: index_name)
     end
 
     # Delete default schema
     # @return [Hash] The response from the server
     def destroy_default_schema
-      client.collections.delete(collection_name: index_name)
+      client.collections.drop(collection_name: index_name)
     end
 
     # Load default schema into memory
@@ -138,16 +127,12 @@ module Langchain::Vectorsearch
     def similarity_search_by_vector(embedding:, k: 4)
       load_default_schema
 
-      client.search(
+      client.entities.search(
         collection_name: index_name,
-        output_fields: ["id", "content"], # Add "vectors" if need to have full vectors returned.
-        top_k: k.to_s,
-        vectors: [embedding],
-        dsl_type: 1,
-        params: "{\"nprobe\": 10}",
-        anns_field: "vectors",
-        metric_type: "L2",
-        vector_type: ::Milvus::DATA_TYPES["float_vector"]
+        anns_field: "vector",
+        data: [embedding],
+        limit: k,
+        output_fields: ["content", "id", "vector"]
       )
     end
 
@@ -159,8 +144,7 @@ module Langchain::Vectorsearch
     def ask(question:, k: 4, &block)
       search_results = similarity_search(query: question, k: k)
 
-      content_field = search_results.dig("results", "fields_data").select { |field| field.dig("field_name") == "content" }
-      content_data = content_field.first.dig("Field", "Scalars", "Data", "StringData", "data")
+      content_data = search_results.dig("data").map { |result| result.dig("content") }
 
       context = content_data.join("\n---\n")
 

data/lib/langchain/vectorsearch/qdrant.rb

@@ -6,7 +6,7 @@ module Langchain::Vectorsearch
     # Wrapper around Qdrant
     #
     # Gem requirements:
-    #     gem "qdrant-ruby", "~> 0.9.3"
+    #     gem "qdrant-ruby", "~> 0.9.8"
     #
     # Usage:
     #     qdrant = Langchain::Vectorsearch::Qdrant.new(url:, api_key:, index_name:, llm:)
@@ -22,7 +22,8 @@ module Langchain::Vectorsearch
 
       @client = ::Qdrant::Client.new(
         url: url,
-        api_key: api_key
+        api_key: api_key,
+        logger: Langchain.logger
       )
       @index_name = index_name
 

data/lib/langchain/vectorsearch/weaviate.rb

@@ -6,7 +6,7 @@ module Langchain::Vectorsearch
     # Wrapper around Weaviate
     #
     # Gem requirements:
-    #     gem "weaviate-ruby", "~> 0.9.0"
+    #     gem "weaviate-ruby", "~> 0.9.2"
     #
     # Usage:
     #     weaviate = Langchain::Vectorsearch::Weaviate.new(url: ENV["WEAVIATE_URL"], api_key: ENV["WEAVIATE_API_KEY"], index_name: "Docs", llm: llm)
@@ -22,7 +22,8 @@ module Langchain::Vectorsearch
 
       @client = ::Weaviate::Client.new(
         url: url,
-        api_key: api_key
+        api_key: api_key,
+        logger: Langchain.logger
       )
 
       # Weaviate requires the class name to be Capitalized: https://weaviate.io/developers/weaviate/configuration/schema-configuration#create-a-class

data/lib/langchain/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: langchainrb
 version: !ruby/object:Gem::Version
-  version: 0.16.1
+  version: 0.17.0
 platform: ruby
 authors:
 - Andrei Bondarev
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-09-30 00:00:00.000000000 Z
+date: 2024-10-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: baran
@@ -318,20 +318,6 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-- !ruby/object:Gem::Dependency
-  name: google_palm_api
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: 0.1.3
-  type: :development
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: 0.1.3
 - !ruby/object:Gem::Dependency
   name: google_search_results
   requirement: !ruby/object:Gem::Requirement
@@ -380,14 +366,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.9.3
+        version: 0.10.3
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.9.3
+        version: 0.10.3
 - !ruby/object:Gem::Dependency
   name: llama_cpp
   requirement: !ruby/object:Gem::Requirement
@@ -520,14 +506,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.9.4
+        version: 0.9.8
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.9.4
+        version: 0.9.8
 - !ruby/object:Gem::Dependency
   name: roo
   requirement: !ruby/object:Gem::Requirement
@@ -604,14 +590,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.8.10
+        version: 0.9.2
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.8.10
+        version: 0.9.2
 - !ruby/object:Gem::Dependency
   name: wikipedia-client
   requirement: !ruby/object:Gem::Requirement
@@ -683,7 +669,6 @@ files:
 - lib/langchain/llm/base.rb
 - lib/langchain/llm/cohere.rb
 - lib/langchain/llm/google_gemini.rb
-- lib/langchain/llm/google_palm.rb
 - lib/langchain/llm/google_vertex_ai.rb
 - lib/langchain/llm/hugging_face.rb
 - lib/langchain/llm/llama_cpp.rb
@@ -701,7 +686,6 @@ files:
 - lib/langchain/llm/response/base_response.rb
 - lib/langchain/llm/response/cohere_response.rb
 - lib/langchain/llm/response/google_gemini_response.rb
-- lib/langchain/llm/response/google_palm_response.rb
 - lib/langchain/llm/response/hugging_face_response.rb
 - lib/langchain/llm/response/llama_cpp_response.rb
 - lib/langchain/llm/response/mistral_ai_response.rb

data/lib/langchain/llm/google_palm.rb

@@ -1,177 +0,0 @@
-# frozen_string_literal: true
-
-module Langchain::LLM
-  #
-  # Wrapper around the Google PaLM (Pathways Language Model) APIs: https://ai.google/build/machine-learning/
-  #
-  # Gem requirements:
-  #     gem "google_palm_api", "~> 0.1.3"
-  #
-  # Usage:
-  #     google_palm = Langchain::LLM::GooglePalm.new(api_key: ENV["GOOGLE_PALM_API_KEY"])
-  #
-  class GooglePalm < Base
-    extend Gem::Deprecate
-
-    DEFAULTS = {
-      temperature: 0.0,
-      dimensions: 768, # This is what the `embedding-gecko-001` model generates
-      completion_model_name: "text-bison-001",
-      chat_completion_model_name: "chat-bison-001",
-      embeddings_model_name: "embedding-gecko-001"
-    }.freeze
-
-    ROLE_MAPPING = {
-      "assistant" => "ai"
-    }
-
-    attr_reader :defaults
-
-    # @deprecated Please use Langchain::LLM::GoogleGemini instead
-    #
-    # @param api_key [String] The API key for the Google PaLM API
-    def initialize(api_key:, default_options: {})
-      depends_on "google_palm_api"
-
-      @client = ::GooglePalmApi::Client.new(api_key: api_key)
-      @defaults = DEFAULTS.merge(default_options)
-    end
-    deprecate :initialize, "Langchain::LLM::GoogleGemini.new(api_key:)", 2024, 10
-
-    #
-    # Generate an embedding for a given text
-    #
-    # @param text [String] The text to generate an embedding for
-    # @return [Langchain::LLM::GooglePalmResponse] Response object
-    #
-    def embed(text:)
-      response = client.embed(text: text)
-
-      Langchain::LLM::GooglePalmResponse.new response,
-        model: @defaults[:embeddings_model_name]
-    end
-
-    #
-    # Generate a completion for a given prompt
-    #
-    # @param prompt [String] The prompt to generate a completion for
-    # @param params extra parameters passed to GooglePalmAPI::Client#generate_text
-    # @return [Langchain::LLM::GooglePalmResponse] Response object
-    #
-    def complete(prompt:, **params)
-      default_params = {
-        prompt: prompt,
-        temperature: @defaults[:temperature],
-        model: @defaults[:completion_model_name]
-      }
-
-      if params[:stop_sequences]
-        default_params[:stop_sequences] = params.delete(:stop_sequences)
-      end
-
-      if params[:max_tokens]
-        default_params[:max_output_tokens] = params.delete(:max_tokens)
-      end
-
-      default_params.merge!(params)
-
-      response = client.generate_text(**default_params)
-
-      Langchain::LLM::GooglePalmResponse.new response,
-        model: default_params[:model]
-    end
-
-    #
-    # Generate a chat completion for a given prompt
-    #
-    # @param prompt [String] The prompt to generate a chat completion for
-    # @param messages [Array<Hash>] The messages that have been sent in the conversation
-    # @param context [String] An initial context to provide as a system message, ie "You are RubyGPT, a helpful chat bot for helping people learn Ruby"
-    # @param examples [Array<Hash>] Examples of messages to provide to the model. Useful for Few-Shot Prompting
-    # @param options [Hash] extra parameters passed to GooglePalmAPI::Client#generate_chat_message
-    # @return [Langchain::LLM::GooglePalmResponse] Response object
-    #
-    def chat(prompt: "", messages: [], context: "", examples: [], **options)
-      raise ArgumentError.new(":prompt or :messages argument is expected") if prompt.empty? && messages.empty?
-
-      default_params = {
-        temperature: @defaults[:temperature],
-        model: @defaults[:chat_completion_model_name],
-        context: context,
-        messages: compose_chat_messages(prompt: prompt, messages: messages),
-        examples: compose_examples(examples)
-      }
-
-      if options[:stop_sequences]
-        default_params[:stop] = options.delete(:stop_sequences)
-      end
-
-      if options[:max_tokens]
-        default_params[:max_output_tokens] = options.delete(:max_tokens)
-      end
-
-      default_params.merge!(options)
-
-      response = client.generate_chat_message(**default_params)
-      raise "GooglePalm API returned an error: #{response}" if response.dig("error")
-
-      Langchain::LLM::GooglePalmResponse.new response,
-        model: default_params[:model]
-      # TODO: Pass in prompt_tokens: prompt_tokens
-    end
-
-    #
-    # Generate a summarization for a given text
-    #
-    # @param text [String] The text to generate a summarization for
-    # @return [String] The summarization
-    #
-    def summarize(text:)
-      prompt_template = Langchain::Prompt.load_from_path(
-        file_path: Langchain.root.join("langchain/llm/prompts/summarize_template.yaml")
-      )
-      prompt = prompt_template.format(text: text)
-
-      complete(
-        prompt: prompt,
-        temperature: @defaults[:temperature],
-        # Most models have a context length of 2048 tokens (except for the newest models, which support 4096).
-        max_tokens: 256
-      )
-    end
-
-    private
-
-    def compose_chat_messages(prompt:, messages:)
-      history = []
-      history.concat transform_messages(messages) unless messages.empty?
-
-      unless prompt.empty?
-        if history.last && history.last[:role] == "user"
-          history.last[:content] += "\n#{prompt}"
-        else
-          history.append({author: "user", content: prompt})
-        end
-      end
-      history
-    end
-
-    def compose_examples(examples)
-      examples.each_slice(2).map do |example|
-        {
-          input: {content: example.first[:content]},
-          output: {content: example.last[:content]}
-        }
-      end
-    end
-
-    def transform_messages(messages)
-      messages.map do |message|
-        {
-          author: ROLE_MAPPING.fetch(message[:role], message[:role]),
-          content: message[:content]
-        }
-      end
-    end
-  end
-end

data/lib/langchain/llm/response/google_palm_response.rb

@@ -1,40 +0,0 @@
-# frozen_string_literal: true
-
-module Langchain::LLM
-  class GooglePalmResponse < BaseResponse
-    attr_reader :prompt_tokens
-
-    def initialize(raw_response, model: nil, prompt_tokens: nil)
-      @prompt_tokens = prompt_tokens
-      super(raw_response, model: model)
-    end
-
-    def completion
-      completions&.dig(0, "output")
-    end
-
-    def embedding
-      embeddings.first
-    end
-
-    def completions
-      raw_response.dig("candidates")
-    end
-
-    def chat_completion
-      chat_completions&.dig(0, "content")
-    end
-
-    def chat_completions
-      raw_response.dig("candidates")
-    end
-
-    def embeddings
-      [raw_response.dig("embedding", "value")]
-    end
-
-    def role
-      "assistant"
-    end
-  end
-end