langchainrb 0.12.0 → 0.13.0

data/lib/langchain/utils/token_length/openai_validator.rb

@@ -28,10 +28,11 @@ module Langchain
           "text-embedding-3-large" => 8191,
           "text-embedding-3-small" => 8191,
           "text-embedding-ada-002" => 8191,
-          "gpt-3.5-turbo" => 4096,
+          "gpt-3.5-turbo" => 16385,
           "gpt-3.5-turbo-0301" => 4096,
           "gpt-3.5-turbo-0613" => 4096,
           "gpt-3.5-turbo-1106" => 16385,
+          "gpt-3.5-turbo-0125" => 16385,
           "gpt-3.5-turbo-16k" => 16384,
           "gpt-3.5-turbo-16k-0613" => 16384,
           "text-davinci-003" => 4097,
@@ -44,9 +45,13 @@ module Langchain
           "gpt-4-32k-0314" => 32768,
           "gpt-4-32k-0613" => 32768,
           "gpt-4-1106-preview" => 128000,
+          "gpt-4-turbo" => 128000,
+          "gpt-4-turbo-2024-04-09" => 128000,
           "gpt-4-turbo-preview" => 128000,
           "gpt-4-0125-preview" => 128000,
           "gpt-4-vision-preview" => 128000,
+          "gpt-4o" => 128000,
+          "gpt-4o-2024-05-13" => 128000,
           "text-curie-001" => 2049,
           "text-babbage-001" => 2049,
           "text-ada-001" => 2049,

data/lib/langchain/version.rb

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

data/lib/langchain.rb

@@ -12,6 +12,7 @@ loader.inflector.inflect(
   "ai21_response" => "AI21Response",
   "ai21_validator" => "AI21Validator",
   "csv" => "CSV",
+  "google_vertex_ai" => "GoogleVertexAI",
   "html" => "HTML",
   "json" => "JSON",
   "jsonl" => "JSONL",
@@ -21,6 +22,7 @@ loader.inflector.inflect(
   "openai" => "OpenAI",
   "openai_validator" => "OpenAIValidator",
   "openai_response" => "OpenAIResponse",
+  "openai_message" => "OpenAIMessage",
   "pdf" => "PDF"
 )
 loader.collapse("#{__dir__}/langchain/llm/response")
@@ -31,6 +33,7 @@ loader.collapse("#{__dir__}/langchain/tool/database")
 loader.collapse("#{__dir__}/langchain/tool/file_system")
 loader.collapse("#{__dir__}/langchain/tool/google_search")
 loader.collapse("#{__dir__}/langchain/tool/ruby_code_interpreter")
+loader.collapse("#{__dir__}/langchain/tool/news_retriever")
 loader.collapse("#{__dir__}/langchain/tool/vectorsearch")
 loader.collapse("#{__dir__}/langchain/tool/weather")
 loader.collapse("#{__dir__}/langchain/tool/wikipedia")

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: langchainrb
 version: !ruby/object:Gem::Version
-  version: 0.12.0
+  version: 0.13.0
 platform: ruby
 authors:
 - Andrei Bondarev
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-04-22 00:00:00.000000000 Z
+date: 2024-05-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -238,16 +238,16 @@ dependencies:
   name: anthropic
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '0.2'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '0.2'
 - !ruby/object:Gem::Dependency
   name: aws-sdk-bedrockruntime
   requirement: !ruby/object:Gem::Requirement
@@ -282,14 +282,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.9.8
+        version: 0.9.10
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.9.8
+        version: 0.9.10
 - !ruby/object:Gem::Dependency
   name: docx
   requirement: !ruby/object:Gem::Requirement
@@ -347,19 +347,19 @@ dependencies:
       - !ruby/object:Gem::Version
         version: 1.6.5
 - !ruby/object:Gem::Dependency
-  name: google-apis-aiplatform_v1
+  name: googleauth
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '0.7'
+        version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '0.7'
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: google_palm_api
   requirement: !ruby/object:Gem::Requirement
@@ -708,7 +708,9 @@ files:
 - README.md
 - lib/langchain.rb
 - lib/langchain/assistants/assistant.rb
-- lib/langchain/assistants/message.rb
+- lib/langchain/assistants/messages/base.rb
+- lib/langchain/assistants/messages/google_gemini_message.rb
+- lib/langchain/assistants/messages/openai_message.rb
 - lib/langchain/assistants/thread.rb
 - lib/langchain/chunk.rb
 - lib/langchain/chunker/base.rb
@@ -735,6 +737,7 @@ files:
 - lib/langchain/llm/azure.rb
 - 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
@@ -742,6 +745,7 @@ files:
 - lib/langchain/llm/mistral_ai.rb
 - lib/langchain/llm/ollama.rb
 - lib/langchain/llm/openai.rb
+- lib/langchain/llm/parameters/chat.rb
 - lib/langchain/llm/prompts/ollama/summarize_template.yaml
 - lib/langchain/llm/prompts/summarize_template.yaml
 - lib/langchain/llm/replicate.rb
@@ -750,14 +754,15 @@ files:
 - lib/langchain/llm/response/aws_titan_response.rb
 - 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/google_vertex_ai_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
 - lib/langchain/llm/response/ollama_response.rb
 - lib/langchain/llm/response/openai_response.rb
 - lib/langchain/llm/response/replicate_response.rb
+- lib/langchain/llm/unified_parameters.rb
 - lib/langchain/loader.rb
 - lib/langchain/output_parsers/base.rb
 - lib/langchain/output_parsers/output_fixing_parser.rb
@@ -789,6 +794,8 @@ files:
 - lib/langchain/tool/file_system/file_system.rb
 - lib/langchain/tool/google_search/google_search.json
 - lib/langchain/tool/google_search/google_search.rb
+- lib/langchain/tool/news_retriever/news_retriever.json
+- lib/langchain/tool/news_retriever/news_retriever.rb
 - lib/langchain/tool/ruby_code_interpreter/ruby_code_interpreter.json
 - lib/langchain/tool/ruby_code_interpreter/ruby_code_interpreter.rb
 - lib/langchain/tool/vectorsearch/vectorsearch.json

data/lib/langchain/assistants/message.rb

@@ -1,58 +0,0 @@
-# frozen_string_literal: true
-
-module Langchain
-  # Langchain::Message are the messages that are sent to LLM chat methods
-  class Message
-    attr_reader :role, :content, :tool_calls, :tool_call_id
-
-    ROLES = %w[
-      system
-      assistant
-      user
-      tool
-    ].freeze
-
-    # @param role [String] The role of the message
-    # @param content [String] The content of the message
-    # @param tool_calls [Array<Hash>] Tool calls to be made
-    # @param tool_call_id [String] The ID of the tool call to be made
-    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)
-      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_openai_format
-      {}.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
-
-    def assistant?
-      role == "assistant"
-    end
-
-    def system?
-      role == "system"
-    end
-
-    def user?
-      role == "user"
-    end
-
-    def tool?
-      role == "tool"
-    end
-  end
-end

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

@@ -1,33 +0,0 @@
-# frozen_string_literal: true
-
-module Langchain::LLM
-  class GoogleVertexAiResponse < BaseResponse
-    attr_reader :prompt_tokens
-
-    def initialize(raw_response, model: nil)
-      @prompt_tokens = prompt_tokens
-      super(raw_response, model: model)
-    end
-
-    def completion
-      # completions&.dig(0, "output")
-      raw_response.predictions[0]["content"]
-    end
-
-    def embedding
-      embeddings.first
-    end
-
-    def completions
-      raw_response.predictions.map { |p| p["content"] }
-    end
-
-    def total_tokens
-      raw_response.dig(:predictions, 0, :embeddings, :statistics, :token_count)
-    end
-
-    def embeddings
-      [raw_response.dig(:predictions, 0, :embeddings, :values)]
-    end
-  end
-end