google-genai 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: a18e13e385a81cf2ebab0a30af1bed3005fdb7f9d647f1e8fe216a34943337f4
+  data.tar.gz: 48f03f1a03de21d84480bfb9518f35bd13589f81418ad3dc7f74b5c73e331c7c
+SHA512:
+  metadata.gz: 84e01482f630c3d49b40970b1db16defc4b649b7844fe6e929923bfc569e4f957ddae89f56d17de843367c48f0d36eb210ae2ba85f613f429532426c45e888d6
+  data.tar.gz: 725c5071b50d49bbdb5515a9473629c9f97f34e8ec8c1c0371987dc0fc7b863c0a297b44bd88999b3255c30d193745618e3757c3a1bcd954bae9199e718904b6

data/LICENSE

@@ -0,0 +1,202 @@
+
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright [yyyy] [name of copyright owner]
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

data/README.md

@@ -0,0 +1,39 @@
+# Google GenAI Ruby SDK
+
+> **Note:** This is an unofficial Ruby port of the official [Google Python SDK](https://github.com/googleapis/python-genai) for the Gemini API. This port was primarily developed with the assistance of the Gemini CLI.
+
+Welcome to the Google GenAI Ruby SDK. This library allows you to integrate Google's generative AI models, including the Gemini family, into your Ruby applications.
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'google-genai'
+```
+
+And then execute:
+
+    $ bundle install
+
+Or install it yourself as:
+
+    $ gem install google-genai
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and the created tag, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/privatenote/ruby-genai.
+
+## License
+
+The gem is available as open source under the terms of the [Apache-2.0 License](https://www.apache.org/licenses/LICENSE-2.0).

data/lib/google/genai/api_client.rb

@@ -0,0 +1,116 @@
+# frozen_string_literal: true
+
+require "faraday"
+require "faraday/retry"
+require "googleauth"
+require "json"
+require "mimemagic"
+require_relative "errors"
+
+module Google
+  module Genai
+    class ApiClient
+      attr_reader :api_key, :vertexai
+
+      def initialize(api_key: nil, vertexai: nil, project: nil, location: nil, http_options: nil)
+        @api_key = api_key
+        @vertexai = vertexai
+        @http_options = http_options || {}
+
+        base_url = if @vertexai
+                     # TODO: Handle global location
+                     "https://#{location}-aiplatform.googleapis.com/"
+                   else
+                     "https://generativelanguage.googleapis.com/"
+                   end
+
+        @connection = Faraday.new(url: base_url) do |faraday|
+          faraday.request :retry
+          faraday.headers['Content-Type'] = 'application/json'
+          if @api_key
+            faraday.headers['x-goog-api-key'] = @api_key
+          elsif @vertexai
+            scopes = ['https://www.googleapis.com/auth/cloud-platform']
+            authorizer = Google::Auth.get_application_default(scopes)
+            faraday.request :authorization, 'Bearer', -> { authorizer.fetch_access_token!['access_token'] }
+          end
+        end
+      end
+
+      def upload_file(file_path, file_size, mime_type, display_name: nil)
+        # 1. Start resumable upload
+        start_upload_headers = {
+          "X-Goog-Upload-Protocol" => "resumable",
+          "X-Goog-Upload-Command" => "start",
+          "X-Goog-Upload-Header-Content-Length" => file_size.to_s,
+          "X-Goog-Upload-Header-Content-Type" => mime_type,
+          "Content-Type" => "application/json"
+        }
+        start_response = @connection.post("upload/v1beta/files") do |req|
+          req.headers.merge!(start_upload_headers)
+          req.body = { file: { display_name: display_name } }.to_json
+        end
+
+        handle_response(start_response)
+
+        upload_url = start_response.headers["x-goog-upload-url"]
+
+        # 2. Upload file content
+        File.open(file_path, "rb") do |file|
+          upload_connection = Faraday.new(url: upload_url)
+          upload_response = upload_connection.post do |req|
+            req.headers.merge!({
+              "X-Goog-Upload-Offset" => "0",
+              "X-Goog-Upload-Command" => "upload, finalize",
+              "Content-Length" => file_size.to_s,
+              "Content-Type" => mime_type
+            })
+            req.body = file.read
+          end
+          
+          handle_response(upload_response)
+          return JSON.parse(upload_response.body)
+        end
+      end
+
+      def request(http_method, path, body: nil)
+        response = @connection.send(http_method, path) do |req|
+          req.body = body.to_json if body
+        end
+        handle_response(response)
+      end
+
+      def get(path)
+        request(:get, path)
+      end
+
+      def delete(path)
+        request(:delete, path)
+      end
+
+      private
+
+      def handle_response(response)
+        return response if response.success?
+
+        begin
+          response_json = JSON.parse(response.body)
+        rescue JSON::ParserError
+          response_json = {
+            'message' => response.body,
+            'status' => response.reason_phrase
+          }
+        end
+
+        status_code = response.status
+        if (400...500).cover?(status_code)
+          raise ClientError.new(status_code, response_json, response)
+        elsif (500...600).cover?(status_code)
+          raise ServerError.new(status_code, response_json, response)
+        else
+          raise APIError.new(status_code, response_json, response)
+        end
+      end
+    end
+  end
+end

data/lib/google/genai/batches.rb

@@ -0,0 +1,71 @@
+# frozen_string_literal: true
+
+require_relative 'types'
+require_relative 'pagers'
+
+module Google
+  module Genai
+    class Batches
+      def initialize(api_client)
+        @api_client = api_client
+      end
+
+      def create(model:, src:, config: nil)
+        raise "Batches is only supported for Vertex AI" unless @api_client.vertexai
+
+        body = {
+          model: model,
+          inputConfig: {
+            instancesFormat: src.is_a?(String) && src.start_with?('bq://') ? 'bigquery' : 'jsonl',
+            gcsSource: { uris: [src] }
+          }
+        }
+        # TODO: Add other config options
+
+        response = @api_client.request(:post, "v1beta/batchPredictionJobs", body: body)
+        Types::BatchJob.new(JSON.parse(response.body))
+      end
+
+      def get(name:, config: nil)
+        raise "Batches is only supported for Vertex AI" unless @api_client.vertexai
+        response = @api_client.get("v1beta/#{name}")
+        Types::BatchJob.new(JSON.parse(response.body))
+      end
+
+      def list(config: nil)
+        raise "Batches is only supported for Vertex AI" unless @api_client.vertexai
+        
+        list_request = ->(options) do
+          path = "v1beta/batchPredictionJobs"
+          params = {}
+          params[:pageToken] = options[:page_token] if options&.key?(:page_token)
+          params[:pageSize] = options[:page_size] if options&.key?(:page_size)
+          path += "?#{URI.encode_www_form(params)}" unless params.empty?
+          @api_client.get(path)
+        end
+
+        response = list_request.call(config)
+
+        Pager.new(
+          name: :batchPredictionJobs,
+          request: list_request,
+          response: response,
+          config: config,
+          item_class: Types::BatchJob
+        )
+      end
+
+      def cancel(name:, config: nil)
+        raise "Batches is only supported for Vertex AI" unless @api_client.vertexai
+        @api_client.request(:post, "v1beta/#{name}:cancel", body: {})
+        nil
+      end
+
+      def delete(name:, config: nil)
+        raise "Batches is only supported for Vertex AI" unless @api_client.vertexai
+        @api_client.delete("v1beta/#{name}")
+        nil
+      end
+    end
+  end
+end

data/lib/google/genai/caches.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+require_relative 'types'
+require_relative 'pagers'
+
+module Google
+  module Genai
+    class Caches
+      def initialize(api_client)
+        @api_client = api_client
+      end
+
+      def create(model:, config:)
+        response = @api_client.request(:post, "v1beta/cachedContents", body: { model: model }.merge(config))
+        Types::CachedContent.new(JSON.parse(response.body))
+      end
+
+      def get(name:, config: nil)
+        response = @api_client.get("v1beta/#{name}")
+        Types::CachedContent.new(JSON.parse(response.body))
+      end
+
+      def list(config: nil)
+        list_request = ->(options) do
+          path = "v1beta/cachedContents"
+          params = {}
+          params[:pageToken] = options[:page_token] if options&.key?(:page_token)
+          params[:pageSize] = options[:page_size] if options&.key?(:page_size)
+          path += "?#{URI.encode_www_form(params)}" unless params.empty?
+          @api_client.get(path)
+        end
+
+        response = list_request.call(config)
+
+        Pager.new(
+          name: :cachedContents,
+          request: list_request,
+          response: response,
+          config: config,
+          item_class: Types::CachedContent
+        )
+      end
+
+      def update(name:, config:)
+        response = @api_client.request(:patch, "v1beta/#{name}", body: config)
+        Types::CachedContent.new(JSON.parse(response.body))
+      end
+
+      def delete(name:, config: nil)
+        @api_client.delete("v1beta/#{name}")
+        nil
+      end
+    end
+  end
+end

data/lib/google/genai/chats.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+module Google
+  module Genai
+    class Chats
+      def initialize(client)
+        @client = client
+      end
+
+      def create(model:, history: [], config: nil)
+        Chat.new(client: @client, model: model, history: history, config: config)
+      end
+    end
+
+    class Chat
+      attr_reader :history
+
+      def initialize(client:, model:, history: [], config: nil)
+        @client = client
+        @model = model
+        @history = history
+        @config = config
+      end
+
+      def send_message(message)
+        @history << { role: 'user', parts: [{ text: message }] }
+        response = @client.models.generate_content(
+          model: @model,
+          contents: @history,
+          config: @config
+        )
+        @history << response.candidates.first.content.to_h
+        response
+      end
+    end
+  end
+end

data/lib/google/genai/client.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+# Zeitwerk will autoload all the required files.
+
+module Google
+  module Genai
+    class Client
+      def initialize(api_key: nil, vertexai: nil, credentials: nil, project: nil, location: nil, http_options: nil)
+        @api_client = ApiClient.new(
+          api_key: api_key,
+          vertexai: vertexai,
+          project: project,
+          location: location,
+          http_options: http_options
+        )
+      end
+
+      def models
+        @models ||= Models.new(@api_client)
+      end
+
+      def chats
+        @chats ||= Chats.new(self)
+      end
+
+      def files
+        @files ||= Files.new(@api_client)
+      end
+
+      def tunings
+        @tunings ||= Tunings.new(@api_client)
+      end
+
+      def caches
+        @caches ||= Caches.new(@api_client)
+      end
+
+      def batches
+        @batches ||= Batches.new(@api_client)
+      end
+
+      def operations
+        @operations ||= Operations.new(@api_client)
+      end
+
+      def tokens
+        @tokens ||= Tokens.new(@api_client)
+      end
+
+      def live
+        @live ||= Live.new(@api_client)
+      end
+    end
+  end
+end

data/lib/google/genai/errors.rb

@@ -0,0 +1,43 @@
+
+# frozen_string_literal: true
+
+module Google
+  module Genai
+    class Error < StandardError; end
+
+    class APIError < Error
+      attr_reader :code, :status, :details, :response
+
+      def initialize(code, response_json, response)
+        @response = response
+        @details = response_json.is_a?(Array) && response_json.length == 1 ? response_json[0] : response_json
+        @message = get_message(@details)
+        @status = get_status(@details)
+        @code = code || get_code(@details)
+
+        super("#{@code} #{@status}. #{@details}")
+      end
+
+      private
+
+      def get_status(details)
+        details.is_a?(Hash) ? (details['status'] || details.dig('error', 'status')) : nil
+      end
+
+      def get_message(details)
+        details.is_a?(Hash) ? (details['message'] || details.dig('error', 'message')) : nil
+      end
+
+      def get_code(details)
+        details.is_a?(Hash) ? (details['code'] || details.dig('error', 'code')) : nil
+      end
+    end
+
+    class ClientError < APIError; end
+    class ServerError < APIError; end
+    class UnknownFunctionCallArgumentError < ArgumentError; end
+    class UnsupportedFunctionError < ArgumentError; end
+    class FunctionInvocationError < StandardError; end
+    class UnknownApiResponseError < StandardError; end
+  end
+end