langchainrb 0.17.1 → 0.18.0

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