langchainrb 0.12.1 → 0.13.0

data/lib/langchain/tool/wikipedia/wikipedia.rb

@@ -9,8 +9,8 @@ module Langchain::Tool
     #     gem "wikipedia-client", "~> 1.17.0"
     #
     # Usage:
-    #     weather = Langchain::Tool::Wikipedia.new
-    #     weather.execute(input: "The Roman Empire")
+    #     wikipedia = Langchain::Tool::Wikipedia.new
+    #     wikipedia.execute(input: "The Roman Empire")
     #
     NAME = "wikipedia"
     ANNOTATIONS_PATH = Langchain.root.join("./langchain/tool/#{NAME}/#{NAME}.json").to_path

data/lib/langchain/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Langchain
-  VERSION = "0.12.1"
+  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.1
+  version: 0.13.0
 platform: ruby
 authors:
 - Andrei Bondarev
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-05-13 00:00:00.000000000 Z
+date: 2024-05-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -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
@@ -751,8 +754,8 @@ 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
@@ -791,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