langchainrb 0.16.1 → 0.17.1

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