langchainrb 0.17.1 → 0.19.0

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: langchainrb
 version: !ruby/object:Gem::Version
-  version: 0.17.1
+  version: 0.19.0
 platform: ruby
 authors:
 - Andrei Bondarev
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-10-07 00:00:00.000000000 Z
+date: 2024-10-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: baran
@@ -637,20 +637,20 @@ files:
 - LICENSE.txt
 - README.md
 - lib/langchain.rb
-- lib/langchain/assistants/assistant.rb
-- lib/langchain/assistants/llm/adapter.rb
-- lib/langchain/assistants/llm/adapters/_base.rb
-- lib/langchain/assistants/llm/adapters/anthropic.rb
-- lib/langchain/assistants/llm/adapters/google_gemini.rb
-- lib/langchain/assistants/llm/adapters/mistral_ai.rb
-- lib/langchain/assistants/llm/adapters/ollama.rb
-- lib/langchain/assistants/llm/adapters/openai.rb
-- lib/langchain/assistants/messages/anthropic_message.rb
-- lib/langchain/assistants/messages/base.rb
-- lib/langchain/assistants/messages/google_gemini_message.rb
-- lib/langchain/assistants/messages/mistral_ai_message.rb
-- lib/langchain/assistants/messages/ollama_message.rb
-- lib/langchain/assistants/messages/openai_message.rb
+- lib/langchain/assistant.rb
+- lib/langchain/assistant/llm/adapter.rb
+- lib/langchain/assistant/llm/adapters/anthropic.rb
+- lib/langchain/assistant/llm/adapters/base.rb
+- lib/langchain/assistant/llm/adapters/google_gemini.rb
+- lib/langchain/assistant/llm/adapters/mistral_ai.rb
+- lib/langchain/assistant/llm/adapters/ollama.rb
+- lib/langchain/assistant/llm/adapters/openai.rb
+- lib/langchain/assistant/messages/anthropic_message.rb
+- lib/langchain/assistant/messages/base.rb
+- lib/langchain/assistant/messages/google_gemini_message.rb
+- lib/langchain/assistant/messages/mistral_ai_message.rb
+- lib/langchain/assistant/messages/ollama_message.rb
+- lib/langchain/assistant/messages/openai_message.rb
 - lib/langchain/chunk.rb
 - lib/langchain/chunker/base.rb
 - lib/langchain/chunker/markdown.rb

data/lib/langchain/assistants/llm/adapters/_base.rb

@@ -1,21 +0,0 @@
-module Langchain
-  class Assistant
-    module LLM
-      module Adapters
-        class Base
-          def build_chat_params(tools:, instructions:, messages:, tool_choice:)
-            raise NotImplementedError, "Subclasses must implement build_chat_params"
-          end
-
-          def extract_tool_call_args(tool_call:)
-            raise NotImplementedError, "Subclasses must implement extract_tool_call_args"
-          end
-
-          def build_message(role:, content: nil, image_url: nil, tool_calls: [], tool_call_id: nil)
-            raise NotImplementedError, "Subclasses must implement build_message"
-          end
-        end
-      end
-    end
-  end
-end

data/lib/langchain/assistants/llm/adapters/anthropic.rb

@@ -1,62 +0,0 @@
-module Langchain
-  class Assistant
-    module LLM
-      module Adapters
-        class Anthropic < Base
-          def build_chat_params(tools:, instructions:, messages:, tool_choice:)
-            params = {messages: messages}
-            if tools.any?
-              params[:tools] = build_tools(tools)
-              params[:tool_choice] = build_tool_choice(tool_choice)
-            end
-            params[:system] = instructions if instructions
-            params
-          end
-
-          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
-
-          # Extract the tool call information from the Anthropic tool call hash
-          #
-          # @param tool_call [Hash] The tool call hash, format: {"type"=>"tool_use", "id"=>"toolu_01TjusbFApEbwKPRWTRwzadR", "name"=>"news_retriever__get_top_headlines", "input"=>{"country"=>"us", "page_size"=>10}}], "stop_reason"=>"tool_use"}
-          # @return [Array] The tool call information
-          def extract_tool_call_args(tool_call:)
-            tool_call_id = tool_call.dig("id")
-            function_name = tool_call.dig("name")
-            tool_name, method_name = function_name.split("__")
-            tool_arguments = tool_call.dig("input").transform_keys(&:to_sym)
-            [tool_call_id, tool_name, method_name, tool_arguments]
-          end
-
-          def build_tools(tools)
-            tools.map { |tool| tool.class.function_schemas.to_anthropic_format }.flatten
-          end
-
-          def allowed_tool_choices
-            ["auto", "any"]
-          end
-
-          def available_tool_names(tools)
-            build_tools(tools).map { |tool| tool.dig(:name) }
-          end
-
-          private
-
-          def build_tool_choice(choice)
-            case choice
-            when "auto"
-              {type: "auto"}
-            when "any"
-              {type: "any"}
-            else
-              {type: "tool", name: choice}
-            end
-          end
-        end
-      end
-    end
-  end
-end

data/lib/langchain/assistants/llm/adapters/ollama.rb

@@ -1,57 +0,0 @@
-module Langchain
-  class Assistant
-    module LLM
-      module Adapters
-        class Ollama < Base
-          def build_chat_params(tools:, instructions:, messages:, tool_choice:)
-            params = {messages: messages}
-            if tools.any?
-              params[:tools] = build_tools(tools)
-            end
-            params
-          end
-
-          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
-
-          # Extract the tool call information from the OpenAI tool call hash
-          #
-          # @param tool_call [Hash] The tool call hash
-          # @return [Array] The tool call information
-          def extract_tool_call_args(tool_call:)
-            tool_call_id = tool_call.dig("id")
-
-            function_name = tool_call.dig("function", "name")
-            tool_name, method_name = function_name.split("__")
-
-            tool_arguments = tool_call.dig("function", "arguments")
-            tool_arguments = if tool_arguments.is_a?(Hash)
-              Langchain::Utils::HashTransformer.symbolize_keys(tool_arguments)
-            else
-              JSON.parse(tool_arguments, symbolize_names: true)
-            end
-
-            [tool_call_id, tool_name, method_name, tool_arguments]
-          end
-
-          def available_tool_names(tools)
-            build_tools(tools).map { |tool| tool.dig(:function, :name) }
-          end
-
-          def allowed_tool_choices
-            ["auto", "none"]
-          end
-
-          private
-
-          def build_tools(tools)
-            tools.map { |tool| tool.class.function_schemas.to_openai_format }.flatten
-          end
-        end
-      end
-    end
-  end
-end

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

@@ -1,75 +0,0 @@
-# frozen_string_literal: true
-
-module Langchain
-  module Messages
-    class AnthropicMessage < Base
-      ROLES = [
-        "assistant",
-        "user",
-        "tool_result"
-      ].freeze
-
-      TOOL_ROLE = "tool_result"
-
-      def initialize(role:, content: 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
-        @tool_calls = tool_calls
-        @tool_call_id = tool_call_id
-      end
-
-      # Convert the message to an Anthropic API-compatible hash
-      #
-      # @return [Hash] The message as an Anthropic API-compatible hash
-      def to_hash
-        {}.tap do |h|
-          h[:role] = tool? ? "user" : role
-
-          h[:content] = if tool?
-            [
-              {
-                type: "tool_result",
-                tool_use_id: tool_call_id,
-                content: content
-              }
-            ]
-          elsif tool_calls.any?
-            tool_calls
-          else
-            content
-          end
-        end
-      end
-
-      # Check if the message is a tool call
-      #
-      # @return [Boolean] true/false whether this message is a tool call
-      def tool?
-        role == "tool_result"
-      end
-
-      # Anthropic does not implement system prompts
-      def system?
-        false
-      end
-
-      # Check if the message came from an LLM
-      #
-      # @return [Boolean] true/false whether this message was produced by an LLM
-      def assistant?
-        role == "assistant"
-      end
-
-      # Check if the message came from an LLM
-      #
-      # @return [Boolean] true/false whether this message was produced by an LLM
-      def llm?
-        assistant?
-      end
-    end
-  end
-end

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

@@ -1,54 +0,0 @@
-# frozen_string_literal: true
-
-module Langchain
-  module Messages
-    class Base
-      attr_reader :role,
-        :content,
-        :image_url,
-        :tool_calls,
-        :tool_call_id
-
-      # Check if the message came from a user
-      #
-      # @return [Boolean] true/false whether the message came from a user
-      def user?
-        role == "user"
-      end
-
-      # Check if the message came from an LLM
-      #
-      # @raise NotImplementedError if the subclass does not implement this method
-      def llm?
-        raise NotImplementedError, "Class #{self.class.name} must implement the method 'llm?'"
-      end
-
-      # Check if the message is a tool call
-      #
-      # @raise NotImplementedError if the subclass does not implement this method
-      def tool?
-        raise NotImplementedError, "Class #{self.class.name} must implement the method 'tool?'"
-      end
-
-      # Check if the message is a system prompt
-      #
-      # @raise NotImplementedError if the subclass does not implement this method
-      def system?
-        raise NotImplementedError, "Class #{self.class.name} must implement the method 'system?'"
-      end
-
-      # Returns the standardized role symbol based on the specific role methods
-      #
-      # @return [Symbol] the standardized role symbol (:system, :llm, :tool, :user, or :unknown)
-      def standard_role
-        return :user if user?
-        return :llm if llm?
-        return :tool if tool?
-        return :system if system?
-
-        # TODO: Should we return :unknown or raise an error?
-        :unknown
-      end
-    end
-  end
-end

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

@@ -1,90 +0,0 @@
-# frozen_string_literal: true
-
-module Langchain
-  module Messages
-    class GoogleGeminiMessage < Base
-      # Google Gemini uses the following roles:
-      ROLES = [
-        "user",
-        "model",
-        "function"
-      ].freeze
-
-      TOOL_ROLE = "function"
-
-      # Initialize a new Google Gemini 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)
-        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
-        @tool_calls = tool_calls
-        @tool_call_id = tool_call_id
-      end
-
-      # Check if the message came from an LLM
-      #
-      # @return [Boolean] true/false whether this message was produced by an LLM
-      def llm?
-        model?
-      end
-
-      # Convert the message to a Google Gemini API-compatible hash
-      #
-      # @return [Hash] The message as a Google Gemini API-compatible hash
-      def to_hash
-        {}.tap do |h|
-          h[:role] = role
-          h[:parts] = if function?
-            [{
-              functionResponse: {
-                name: tool_call_id,
-                response: {
-                  name: tool_call_id,
-                  content: content
-                }
-              }
-            }]
-          elsif tool_calls.any?
-            tool_calls
-          else
-            [{text: content}]
-          end
-        end
-      end
-
-      # Google Gemini does not implement system prompts
-      def system?
-        false
-      end
-
-      # Check if the message is a tool call
-      #
-      # @return [Boolean] true/false whether this message is a tool call
-      def tool?
-        function?
-      end
-
-      # Check if the message is a tool call
-      #
-      # @return [Boolean] true/false whether this message is a tool call
-      def function?
-        role == "function"
-      end
-
-      # Check if the message came from an LLM
-      #
-      # @return [Boolean] true/false whether this message was produced by an LLM
-      def model?
-        role == "model"
-      end
-    end
-  end
-end

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

@@ -1,96 +0,0 @@
-# frozen_string_literal: true
-
-module Langchain
-  module Messages
-    class MistralAIMessage < Base
-      # MistralAI uses the following roles:
-      ROLES = [
-        "system",
-        "assistant",
-        "user",
-        "tool"
-      ].freeze
-
-      TOOL_ROLE = "tool"
-
-      # Initialize a new MistralAI message
-      #
-      # @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
-
-      # Check if the message came from an LLM
-      #
-      # @return [Boolean] true/false whether this message was produced by an LLM
-      def llm?
-        assistant?
-      end
-
-      # Convert the message to an MistralAI API-compatible hash
-      #
-      # @return [Hash] The message as an MistralAI API-compatible hash
-      def to_hash
-        {}.tap do |h|
-          h[:role] = role
-
-          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
-
-      # Check if the message came from an LLM
-      #
-      # @return [Boolean] true/false whether this message was produced by an LLM
-      def assistant?
-        role == "assistant"
-      end
-
-      # Check if the message are system instructions
-      #
-      # @return [Boolean] true/false whether this message are system instructions
-      def system?
-        role == "system"
-      end
-
-      # Check if the message is a tool call
-      #
-      # @return [Boolean] true/false whether this message is a tool call
-      def tool?
-        role == "tool"
-      end
-    end
-  end
-end

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

@@ -1,74 +0,0 @@
-# frozen_string_literal: true
-
-module Langchain
-  module Messages
-    class OllamaMessage < Base
-      # OpenAI uses the following roles:
-      ROLES = [
-        "system",
-        "assistant",
-        "user",
-        "tool"
-      ].freeze
-
-      TOOL_ROLE = "tool"
-
-      # 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)
-        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
-        @tool_calls = tool_calls
-        @tool_call_id = tool_call_id
-      end
-
-      # Convert the message to an OpenAI API-compatible hash
-      #
-      # @return [Hash] The message as an OpenAI API-compatible hash
-      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
-        end
-      end
-
-      # Check if the message came from an LLM
-      #
-      # @return [Boolean] true/false whether this message was produced by an LLM
-      def llm?
-        assistant?
-      end
-
-      # Check if the message came from an LLM
-      #
-      # @return [Boolean] true/false whether this message was produced by an LLM
-      def assistant?
-        role == "assistant"
-      end
-
-      # Check if the message are system instructions
-      #
-      # @return [Boolean] true/false whether this message are system instructions
-      def system?
-        role == "system"
-      end
-
-      # Check if the message is a tool call
-      #
-      # @return [Boolean] true/false whether this message is a tool call
-      def tool?
-        role == "tool"
-      end
-    end
-  end
-end

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

@@ -1,103 +0,0 @@
-# frozen_string_literal: true
-
-module Langchain
-  module Messages
-    class OpenAIMessage < Base
-      # OpenAI uses the following roles:
-      ROLES = [
-        "system",
-        "assistant",
-        "user",
-        "tool"
-      ].freeze
-
-      TOOL_ROLE = "tool"
-
-      # Initialize a new OpenAI message
-      #
-      # @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
-
-      # Check if the message came from an LLM
-      #
-      # @return [Boolean] true/false whether this message was produced by an LLM
-      def llm?
-        assistant?
-      end
-
-      # Convert the message to an OpenAI API-compatible hash
-      #
-      # @return [Hash] The message as an OpenAI API-compatible hash
-      def to_hash
-        {}.tap do |h|
-          h[:role] = role
-
-          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
-
-      # Check if the message came from an LLM
-      #
-      # @return [Boolean] true/false whether this message was produced by an LLM
-      def assistant?
-        role == "assistant"
-      end
-
-      # Check if the message are system instructions
-      #
-      # @return [Boolean] true/false whether this message are system instructions
-      def system?
-        role == "system"
-      end
-
-      # Check if the message is a tool call
-      #
-      # @return [Boolean] true/false whether this message is a tool call
-      def tool?
-        role == "tool"
-      end
-    end
-  end
-end