ruby-gemini-ai 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: d7b4f9c65b80cb5b10a2a1440b29b78b40ca6161ae7297642ef1f33b4dfb3501
+  data.tar.gz: 3580107b3720e3299cca98c5f9f039b61bc4c891915dd94aca25d648452d333f
+SHA512:
+  metadata.gz: c46abb842e6cefdbe2a59bf6e646f5136b6d68eac4f8de6f12bb224483fa66eb1709a6935daf447927ec0cf75cad763376357e37cfbb528d75db813fac010a4d
+  data.tar.gz: 58c436b5d93829a68e996a7f152478c5fca9170850346ce1ce4614ad8be52b258f724005de7b02d19f49c22ca71effdeb1762be525b41fd1634e1b74f88eba72

data/.github/workflows/gem-push.yml

@@ -0,0 +1,48 @@
+name: Ruby Gem
+
+on:
+  push:
+    branches: [ "master" ]
+  pull_request:
+    branches: [ "master" ]
+
+jobs:
+  build:
+    name: Build + Publish
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+      packages: write
+
+    steps:
+    - uses: actions/checkout@v3
+    - name: Set up Ruby 2.6.7
+    # To automatically get bug fixes and new Ruby versions for ruby/setup-ruby,
+    # change this to (see https://github.com/ruby/setup-ruby#versioning):
+    # uses: ruby/setup-ruby@v1
+      uses: ruby/setup-ruby@55283cc23133118229fd3f97f9336ee23a179fcf # v1.146.0
+      with:
+        ruby-version: 2.6.7
+
+    - name: Publish to GPR
+      run: |
+        mkdir -p $HOME/.gem
+        touch $HOME/.gem/credentials
+        chmod 0600 $HOME/.gem/credentials
+        printf -- "---\n:github: ${GEM_HOST_API_KEY}\n" > $HOME/.gem/credentials
+        gem build *.gemspec
+        gem push --KEY github --host https://rubygems.pkg.github.com/${OWNER} *.gem
+      env:
+        GEM_HOST_API_KEY: "Bearer ${{secrets.GITHUB_TOKEN}}"
+        OWNER: ${{ github.repository_owner }}
+
+    - name: Publish to RubyGems
+      run: |
+        mkdir -p $HOME/.gem
+        touch $HOME/.gem/credentials
+        chmod 0600 $HOME/.gem/credentials
+        printf -- "---\n:rubygems_api_key: ${GEM_HOST_API_KEY}\n" > $HOME/.gem/credentials
+        gem build *.gemspec
+        gem push *.gem
+      env:
+        GEM_HOST_API_KEY: "${{secrets.RUBYGEMS_AUTH_TOKEN}}"

data/.gitignore

@@ -0,0 +1,56 @@
+# Ignore bundler config.
+/.bundle
+
+# Ignore local development configuration.
+/config/*.yml
+
+# Ignore test and spec directories.
+/spec/tmp
+/spec/fixtures
+/spec/reports
+/spec/examples
+
+# Ignore coverage report generated by SimpleCov.
+/coverage
+
+# Ignore environment-specific files.
+.env
+.env.local
+.env.development.local
+.env.test.local
+.env.production.local
+
+# Ignore local database files.
+/db/*.sqlite3
+/db/*.sqlite3-journal
+
+# Ignore the output of 'gem build'.
+/*.gem
+
+# rspec failure tracking
+.rspec_status
+
+# IDE
+.idea
+.idea/
+.idea/*
+.vscode
+.vs/
+*.iml
+*.sublime-project
+*.sublime-workspace
+/.vscode
+
+# Mac
+.DS_Store
+
+# Ignore system and application-specific files.
+*.log
+*.swp
+*.bak
+Thumbs.db
+
+# Ignore gem artifacts.
+/pkg
+*.gem
+

data/.rspec

@@ -0,0 +1 @@
+--require spec_helper

data/Gemfile

@@ -0,0 +1,11 @@
+source "https://rubygems.org"
+
+# Include gem dependencies from ruby-geminiai.gemspec
+gemspec
+
+group :test, :development do
+  gem "byebug", "~> 11.1.3"
+  gem "dotenv", "~> 2.8.1"
+  gem "rake", "~> 13.1"
+  gem "rspec", "~> 3.12"
+end

data/Gemfile.lock

@@ -0,0 +1,69 @@
+PATH
+  remote: .
+  specs:
+    ruby-gemini-ai (0.1.0)
+      event_stream_parser (>= 0.3.0, < 2.0.0)
+      faraday (>= 1)
+      googleauth (~> 1.8)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    addressable (2.8.6)
+      public_suffix (>= 2.0.2, < 6.0)
+    byebug (11.1.3)
+    diff-lcs (1.5.1)
+    dotenv (2.8.1)
+    event_stream_parser (1.0.0)
+    faraday (2.9.0)
+      faraday-net_http (>= 2.0, < 3.2)
+    faraday-net_http (3.1.0)
+      net-http
+    google-cloud-env (2.1.1)
+      faraday (>= 1.0, < 3.a)
+    googleauth (1.11.0)
+      faraday (>= 1.0, < 3.a)
+      google-cloud-env (~> 2.1)
+      jwt (>= 1.4, < 3.0)
+      multi_json (~> 1.11)
+      os (>= 0.9, < 2.0)
+      signet (>= 0.16, < 2.a)
+    jwt (2.7.1)
+    multi_json (1.15.0)
+    net-http (0.4.1)
+      uri
+    os (1.1.4)
+    public_suffix (5.0.4)
+    rake (13.1.0)
+    rspec (3.13.0)
+      rspec-core (~> 3.13.0)
+      rspec-expectations (~> 3.13.0)
+      rspec-mocks (~> 3.13.0)
+    rspec-core (3.13.0)
+      rspec-support (~> 3.13.0)
+    rspec-expectations (3.13.0)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.13.0)
+    rspec-mocks (3.13.0)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.13.0)
+    rspec-support (3.13.0)
+    signet (0.19.0)
+      addressable (~> 2.8)
+      faraday (>= 0.17.5, < 3.a)
+      jwt (>= 1.5, < 3.0)
+      multi_json (~> 1.10)
+    uri (0.13.0)
+
+PLATFORMS
+  arm64-darwin-22
+
+DEPENDENCIES
+  byebug (~> 11.1.3)
+  dotenv (~> 2.8.1)
+  rake (~> 13.1)
+  rspec (~> 3.12)
+  ruby-gemini-ai!
+
+BUNDLED WITH
+   2.3.22

data/LICENSE

@@ -0,0 +1,22 @@
+MIT License
+
+Copyright (c) 2024 GeminiAI
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.
+

data/README.md

@@ -0,0 +1,381 @@
+# Ruby GeminiAi
+
+This guide explains how to seamlessly integrate the powerful Gemini AI API into your Ruby projects. Utilize Gemini's cutting-edge language capabilities for generating text, translating languages, and more.
+
+# Table of Contents
+
+- [Installation](#installation)
+  - [Bundler](#bundler)
+  - [Gem Install](#gem-install)
+- [Credentials](#credentials)
+  - [Option 1: API Key (Generative Language API)](#option-1-api-key-generative-language-api)
+  - [Option 2: Service Account Credentials File (Vertex AI API)](#option-2-service-account-credentials-file-vertex-ai-api)
+  - [Option 3: Application Default Credentials (Vertex AI API)](#option-3-application-default-credentials-vertex-ai-api)
+- [Usage](#usage)
+  - [Quickstart](#quickstart)
+  - [With Config](#with-config)
+  - [Verbose Logging](#verbose-logging)
+- [Methods](#methods)
+  - [stream_generate_content](#stream_generate_content)
+  - [generate_content](#generate_content)
+- [Development](#development)
+- [Compatibility](#compatibility)
+- [License](#license)
+- [Resources and References](#resources-and-references)
+- [Additional Notes](#additional-notes)
+
+## Installation
+
+### Bundler
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem "ruby-gemini-ai"
+```
+
+And then execute:
+
+```bash
+$ bundle install
+```
+
+### Gem install
+
+Or install with:
+
+```bash
+$ gem install ruby-gemini-ai
+```
+
+and require with:
+
+```ruby
+require "gemini-ai"
+```
+
+## Credentials
+
+- [Option 1: API Key (Generative Language API)](#option-1-api-key-generative-language-api)
+- [Option 2: Service Account Credentials File (Vertex AI API)](#option-2-service-account-credentials-file-vertex-ai-api)
+- [Option 3: Application Default Credentials (Vertex AI API)](#option-3-application-default-credentials-vertex-ai-api)
+
+#### Option 1: API Key (Generative Language API):
+
+Obtain an API Key from your Google Cloud project: [Google Cloud](https://console.cloud.google.com) through the Google Cloud Console: [https://console.cloud.google.com/apis/credentials](https://console.cloud.google.com/apis/credentials).
+
+Enable the Generative Language API service in your Google Cloud Console. which can be done [here](https://console.cloud.google.com/apis/library/generativelanguage.googleapis.com).
+
+Alternatively, you can generate an API Key through Google AI Studio [here](https://makersuite.google.com/app/apikey), which will automatically create a project for you.
+
+#### Option 2: Service Account Credentials File (Vertex AI API)
+
+For the Vertex AI API, create a [Google Cloud](https://console.cloud.google.com) Project and a [_Service Account_](https://cloud.google.com/iam/docs/service-account-overview). Enable the [Vertex AI] (https://cloud.google.com/vertex-ai) API for your project [here](https://console.cloud.google.com/apis/library/aiplatform.googleapis.com).
+
+Generate credentials for your Service Account [here](https://console.cloud.google.com/apis/credentials) and download a JSON file named google-credentials.json. 
+
+```json
+{
+  "type": "service_account",
+  "project_id": "YOUR_PROJECT_ID",
+  "private_key_id": "a00...",
+  "private_key": "-----BEGIN PRIVATE KEY-----\n...\n-----END PRIVATE KEY-----\n",
+  "client_email": "PROJECT_ID@PROJECT_ID.iam.gserviceaccount.com",
+  "client_id": "000...",
+  "auth_uri": "https://accounts.google.com/o/oauth2/auth",
+  "token_uri": "https://oauth2.googleapis.com/token",
+  "auth_provider_x509_cert_url": "https://www.googleapis.com/oauth2/v1/certs",
+  "client_x509_cert_url": "https://www.googleapis.com/robot/v1/metadata/x509/..."
+}
+```
+
+Ensure the necessary [policies](https://cloud.google.com/iam/docs/policies) (`roles/aiplatform.user` and possibly `roles/ml.admin`) are in place use the Vertex AI API.
+
+You can add them by navigating to the [IAM Console](https://console.cloud.google.com/iam-admin/iam) and clicking on the _"Edit principal"_ (✏️ pencil icon) next to your _Service Account_.
+
+Alternatively, you can add them through the [gcloud CLI](https://cloud.google.com/sdk/gcloud) as follows:
+
+```sh
+gcloud projects add-iam-policy-binding PROJECT_ID \
+  --member='serviceAccount:PROJECT_ID@PROJECT_ID.iam.gserviceaccount.com' \
+  --role='roles/aiplatform.user'
+```
+
+Some people reported having trouble accessing the API, and adding the role `roles/ml.admin` fixed it:
+
+```sh
+gcloud projects add-iam-policy-binding PROJECT_ID \
+  --member='serviceAccount:PROJECT_ID@PROJECT_ID.iam.gserviceaccount.com' \
+  --role='roles/ml.admin'
+```
+
+If you are not using a _Service Account_:
+```sh
+gcloud projects add-iam-policy-binding PROJECT_ID \
+  --member='user:YOUR@MAIL.COM' \
+  --role='roles/aiplatform.user'
+
+gcloud projects add-iam-policy-binding PROJECT_ID \
+  --member='user:YOUR@MAIL.COM' \
+  --role='roles/ml.admin'
+```
+
+#### Option 3: Application Default Credentials (Vertex AI API)
+
+
+Similar to [Option 2](#option-2-service-account-credentials-file-vertex-ai-api), but you don't need to download a `google-credentials.json`. These automatically find credentials based on your environment. [_Application Default Credentials_](https://cloud.google.com/docs/authentication/application-default-credentials).
+
+Generate them using the gcloud CLI before local development. [gcloud CLI](https://cloud.google.com/sdk/gcloud):
+
+```sh
+gcloud auth application-default login 
+```
+
+For more details about alternative methods and different environments, check the official documentation:
+[Set up Application Default Credentials](https://cloud.google.com/docs/authentication/provide-credentials-adc)
+
+
+## Usage 
+
+### Quickstart
+
+For a quick test you can pass your token directly to a new client:
+
+```ruby
+client = GeminiAi::Client.new(api_key: "gemini_api_key")
+```
+
+### With Config
+
+We can configure Gemini with Ruby using three options.
+
+**Option 1**, API KEY
+
+For a more robust setup, you can configure the gem with your API keys, for example in an `gemini.rb` initializer file. Never hardcode secrets into your codebase - instead use something like [dotenv](https://github.com/motdotla/dotenv) to pass the keys safely into your environments.
+
+```ruby
+GeminiAi.configure do |config|
+  config.api_key = ENV.fetch("GEMINI_API_KEY")
+  config.service = ENV.fetch("GEMINI_API_SERVICE")
+end
+```
+
+**Option 2**, Service Account
+
+For the Service Account, provide a `google-credentials.json` file and a `REGION`:
+
+```ruby
+GeminiAi.configure do |config|
+  config.service = 'vertex-ai-api'
+  config.region = 'us-east4'
+  config.file_path = 'google-credentials.json'
+end
+```
+
+**Option 3**, Default Credentials
+
+For _Application Default Credentials_, omit both the `api_key` and the `file_path`:
+
+```ruby
+GeminiAi.configure do |config|
+  config.region = 'us-east4'
+  config.service = 'vertex-ai-api'
+end
+```
+
+Then you can create a client like this:
+
+```ruby
+client = GeminiAi::Client.new
+```
+
+
+## Methods
+
+### stream_generate_content(contents, model):
+
+ - Streams generated text in real-time.
+ - contents (hash): User input and role information.
+ - model (string): Optional model name (e.g., gemini-pro).
+ - Returns an array of candidates objects with generated text and safety ratings.
+
+```ruby
+client = GeminiAi::Client.new
+# Assuming you configured with your API key or credentials
+
+contents = {
+  contents: {
+    role: 'user',
+    parts: {
+      text: 'Write a poem about the ocean.'
+    }
+  }
+}
+
+stream = client.stream_generate_content(contents, model: 'gemini-pro')
+```
+
+In this case, the result will be an array with all the received events:
+
+```ruby
+[{ 'candidates' =>
+   [{ 'content' => {
+        'role' => 'model',
+        'parts' => [{ 'text' => 'exmaple poem content.......' }]
+      },
+      'finishReason' => 'STOP',
+      'safetyRatings' =>
+      [{ 'category' => 'HARM_CATEGORY_HARASSMENT', 'probability' => 'NEGLIGIBLE' },
+       { 'category' => 'HARM_CATEGORY_HATE_SPEECH', 'probability' => 'NEGLIGIBLE' },
+       { 'category' => 'HARM_CATEGORY_SEXUALLY_EXPLICIT', 'probability' => 'NEGLIGIBLE' },
+       { 'category' => 'HARM_CATEGORY_DANGEROUS_CONTENT', 'probability' => 'NEGLIGIBLE' }] }],
+   'usageMetadata' => {
+     'promptTokenCount' => 2,
+     'candidatesTokenCount' => 8,
+     'totalTokenCount' => 10
+   } }]
+```
+
+#### with stream
+
+```ruby
+client = GeminiAi::Client.new
+# Assuming you configured with your API key or credentials
+
+contents = {
+  contents: {
+    role: 'user',
+    parts: {
+      text: 'Write a poem about the ocean.'
+    }
+  }
+}
+
+client.stream_generate_content(contents, model: 'gemini-pro', stream: true) do |part_text, event, parsed, raw|
+  puts text
+end
+```
+OR
+
+```ruby
+client = GeminiAi::Client.new
+# Assuming you configured with your API key or credentials
+
+contents = {
+  contents: {
+    role: 'user',
+    parts: {
+      text: 'Write a poem about the ocean.'
+    }
+  }
+}
+
+# Assuming you have a block or procedure (proc) defined
+stream_proc = Proc.new do |part_text, event, parsed, raw|
+  puts part_text 
+end
+
+client.stream_generate_content(contents, model: 'gemini-pro', stream: true, &stream_proc)
+```
+
+
+In this case, the result will be an array with all the received events:
+
+```ruby
+'exmaple poem content.......'
+```
+
+### generate_content(contents, model)
+
+ - Generates text in a single request.
+ - contents (hash): User input and role information.
+ - model (string): Optional model name (e.g., gemini-pro).
+ - Returns a hash with generated text, safety ratings, and prompt feedback
+
+```ruby
+result = client.generate_content(
+  { contents: { role: 'user', parts: { text: 'hi!' } } }, model: 'gemini-pro'
+)
+```
+
+```ruby
+client = GeminiAi::Client.new
+# Assuming you configured with your API key or credentials
+
+contents = {
+  contents: {
+    role: 'user',
+    parts: {
+      text: 'Write a poem about the ocean.'
+    }
+  }
+}
+
+stream = client.generate_content(contents, model: 'gemini-pro')
+```
+
+
+Result:
+```ruby
+{ 'candidates' =>
+  [{ 'content' => { 'parts' => [{ 'text' => 'exampled poem.......' }], 'role' => 'model' },
+     'finishReason' => 'STOP',
+     'index' => 0,
+     'safetyRatings' =>
+     [{ 'category' => 'HARM_CATEGORY_SEXUALLY_EXPLICIT', 'probability' => 'NEGLIGIBLE' },
+      { 'category' => 'HARM_CATEGORY_HATE_SPEECH', 'probability' => 'NEGLIGIBLE' },
+      { 'category' => 'HARM_CATEGORY_HARASSMENT', 'probability' => 'NEGLIGIBLE' },
+      { 'category' => 'HARM_CATEGORY_DANGEROUS_CONTENT', 'probability' => 'NEGLIGIBLE' }] }],
+  'promptFeedback' =>
+  { 'safetyRatings' =>
+    [{ 'category' => 'HARM_CATEGORY_SEXUALLY_EXPLICIT', 'probability' => 'NEGLIGIBLE' },
+     { 'category' => 'HARM_CATEGORY_HATE_SPEECH', 'probability' => 'NEGLIGIBLE' },
+     { 'category' => 'HARM_CATEGORY_HARASSMENT', 'probability' => 'NEGLIGIBLE' },
+     { 'category' => 'HARM_CATEGORY_DANGEROUS_CONTENT', 'probability' => 'NEGLIGIBLE' }] } }
+```
+
+#### Verbose Logging
+
+You can pass [Faraday middleware](https://lostisland.github.io/faraday/#/middleware/index) to the client in a block, eg. to enable verbose logging with Ruby's [Logger](https://ruby-doc.org/3.2.2/stdlibs/logger/Logger.html):
+
+```ruby
+  client = GeminiAi::Client.new do |f|
+    f.response :logger, Logger.new($stdout), bodies: true
+  end
+```
+
+## Development
+
+ 1) Clone the repository.
+ 2) Run **bin/setup** to install dependencies.
+ 3) Use **bin/console** for interactive exploration.
+ 4) Run **bundle exec** rake install to install the gem locally.
+
+## Compatibility
+
+ruby-gemini-ai gem is compatible with Ruby versions 2.6.7 and higher.
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).
+
+## Resources and References
+
+Explore the following curated list of resources and references to enhance your understanding throughout the learning process:
+
+- [Google AI for Developers](https://ai.google.dev): Stay updated on the latest developments and resources in the field of Artificial Intelligence by visiting Google AI for Developers.
+- [Get started with the Gemini API](https://ai.google.dev/docs): Initiate your journey into the Gemini API with comprehensive guides and documentation provided by Google.
+- [Getting Started with the Vertex AI Gemini API with cURL](https://github.com/GoogleCloudPlatform/generative-ai/blob/main/gemini/getting-started/intro_gemini_curl.ipynb): Explore hands-on examples and tutorials using cURL to kickstart your experience with the Vertex AI Gemini API.
+- [Gemini API Documentation](https://cloud.google.com/vertex-ai/docs/generative-ai/model-reference/gemini): Refer to the official Gemini API documentation for detailed information on model references and implementation guidelines.
+- [Vertex AI API Documentation](https://cloud.google.com/vertex-ai/docs/reference): Dive into the Vertex AI API documentation to gain a comprehensive understanding of Vertex AI services.
+- [REST Documentation](https://cloud.google.com/vertex-ai/docs/reference/rest): Explore the RESTful API documentation for Vertex AI to facilitate seamless integration with your applications.
+- [Google DeepMind Gemini](https://deepmind.google/technologies/gemini/): Gain insights into Google DeepMind's Gemini technology, a cutting-edge advancement in the field of Artificial Intelligence.
+- [Stream responses from Generative AI models](https://cloud.google.com/vertex-ai/docs/generative-ai/learn/streaming): Learn how to effectively stream responses from Generative AI models by consulting this specific guide within the Vertex AI documentation.
+- [Function calling](https://cloud.google.com/vertex-ai/docs/generative-ai/multimodal/function-calling): Understand the intricacies of function calling in the context of Generative AI models with this guide from the Vertex AI documentation.
+
+These resources collectively provide a comprehensive foundation for your exploration of the Gemini API and Vertex AI services.
+
+## Additional Notes
+
+- As of now, only generate_content is supported with the `vertex-ai-api` service.
+- For detailed API documentation and advanced usage, refer to the official Gemini AI documentation Consider adding examples and error handling for a more user-friendly experience.

data/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "gemini-ai"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start(__FILE__)

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/lib/gemini-ai/client.rb

@@ -0,0 +1,76 @@
+# frozen_string_literal: true
+
+require 'event_stream_parser'
+require 'faraday'
+require 'json'
+require 'googleauth'
+require 'uri'
+
+require_relative 'errors'
+
+module GeminiAi
+  class Client
+    include GeminiAi::HTTP
+
+    CONFIG_KEYS = %i[ api_key region file_path version service ].freeze
+    attr_accessor :authentication, :authorizer, :project_id, :service_version
+    attr_reader *CONFIG_KEYS, :faraday_middleware
+
+    def initialize(config = {}, &faraday_middleware)
+      CONFIG_KEYS.each do |key|
+        # Set instance variables like service authentication etc
+        instance_variable_set("@#{key}", config[key] || GeminiAi.configuration.send(key))
+      end
+      @service_version = @version || DEFAULT_SERVICE_VERSION
+      case
+      when @api_key
+        @authentication = :api_key
+      when @file_path
+        @authentication = :service_account
+        @authorizer = Google::Auth::ServiceAccountCredentials.make_creds(
+          json_key_io: File.open(@file_path),
+          scope: 'https://www.googleapis.com/auth/cloud-platform'
+        )
+      else
+        @authentication = :default_credentials
+        @authorizer = Google::Auth.get_application_default
+      end
+
+      if @authentication == :service_account || @authentication == :default_credentials
+        @project_id = @authorizer.project_id || @authorizer.quota_project_id
+        raise Errors::MissingProjectIdError, 'Could not determine project_id, which is required.' if @project_id.nil?
+      end
+      @faraday_middleware = faraday_middleware
+    end
+
+    def stream_generate_content(payload, model: nil, stream: nil, &callback)
+      path = build_request_url('streamGenerateContent', model, stream)
+      json_post(path: path, parameters: payload, &callback)
+    end
+
+    def generate_content(payload, model: nil, stream: nil, &callback)
+      path = build_request_url('generateContent', model, stream)
+      json_post(path: path, parameters: payload, &callback)
+    end
+
+    private
+
+    def build_request_url(path, model, stream)
+      base_url = case @service
+                 when 'vertex-ai-api'
+                   "https://#{@region}-aiplatform.googleapis.com/#{service_version}/projects/#{@project_id}/locations/#{@region}/publishers/google/models/#{model}"
+                 when 'generative-language-api'
+                   "https://generativelanguage.googleapis.com/#{service_version}/models/#{model}"
+                 end
+
+      params = {}
+      params[:alt] = 'sse' if stream
+      params[:key] = @api_key if @authentication == :api_key
+
+      uri = URI("#{base_url}:#{path}")
+      uri.query = URI.encode_www_form(params) if params.present?
+      uri.to_s
+    end
+
+  end
+end

data/lib/gemini-ai/compatibility.rb

@@ -0,0 +1,10 @@
+module Ruby
+  module GeminiAI
+    VERSION = ::GeminiAI::VERSION
+
+    Error = ::GeminiAI::Error
+    ConfigurationError = ::GeminiAI::ConfigurationError
+    Configuration = ::GeminiAI::Configuration
+    MiddlewareErrors = ::GeminiAI::MiddlewareErrors
+  end
+end

data/lib/gemini-ai/errors.rb

@@ -0,0 +1,18 @@
+module GeminiAi
+  module Errors
+    class GeminiAiError < StandardError; end
+    class MissingProjectIdError < GeminiAiError; end
+    class UnsupportedServiceError < GeminiAiError; end
+    class BlockWithoutServerSentEventsError < GeminiAiError; end
+
+    class RequestError < GeminiAiError
+      attr_reader :request, :payload
+
+      def initialize(message = nil, request: nil, payload: nil)
+        @request = request
+        @payload = payload
+        super(message)
+      end
+    end
+  end
+end

data/lib/gemini-ai/http.rb

@@ -0,0 +1,117 @@
+require "event_stream_parser"
+
+require_relative "http_headers"
+
+module GeminiAi
+  module HTTP
+    include HTTPHeaders
+
+    def json_post(path:, parameters:, &callback)
+      conn.post(path) do |req|
+        configure_json_post_request(req, parameters, &callback)
+      end&.body
+    end
+
+    private
+
+    def parse_json(response)
+      return unless response
+      return response unless response.is_a?(String)
+
+      # Convert a multiline string of JSON objects to a JSON array.
+      response = response.gsub("}\n{", "},{").prepend("[").concat("]")
+
+      JSON.parse(response)
+    end
+
+    #===================================================================================================
+
+    def to_json_stream(&callback)
+      partial_json = ''
+      parser = EventStreamParser::Parser.new
+
+      proc do |chunk, bytes, env|
+        if env && env.status != 200
+          raise_error = Faraday::Response::RaiseError.new
+          raise_error.on_complete(env.merge(body: chunk))
+        end
+
+        parser.feed(chunk) do |type, data, id, reconnection_time|
+          partial_json += data
+          parsed_json = parse_partial_json(partial_json)
+
+          if parsed_json
+            parsed_json_text = extract_text_from_json(parsed_json)
+            result = {
+              text: parsed_json_text,
+              event: parsed_json,
+              parsed: { type: type, data: data, id: id, reconnection_time: reconnection_time },
+              raw: { chunk: chunk, bytes: bytes, env: env }
+            }
+
+            callback.call(result[:text], result[:event], result[:parsed], result[:raw]) unless callback.nil?
+
+            partial_json = ''
+
+            if parsed_json['candidates']
+              parsed_json['candidates'].find do |candidate|
+                !candidate['finishReason'].nil? && candidate['finishReason'] != ''
+              end
+            end
+
+          end
+        end
+      end
+    end
+
+    def extract_text_from_json(parsed_json)
+      parsed_json["candidates"].map do |candidate|
+        candidate["content"]["parts"].map { |part| part["text"] }
+      end.join(" ")
+    end
+
+    def parse_partial_json(response)
+      return unless response
+      return response unless response.is_a?(String)
+      response.to_s.lstrip.start_with?('{', '[') ? JSON.parse(response) : nil
+    rescue JSON::ParserError
+      nil
+    end
+
+    #===================================================================================================
+
+    def conn(multipart: false)
+      connection = Faraday.new do |f|
+        f.options[:timeout] = @request_timeout
+        f.request(:multipart) if multipart
+        f.use MiddlewareErrors
+        f.response :raise_error
+        f.response :json
+      end
+
+      @faraday_middleware&.call(connection)
+
+      connection
+    end
+
+    def configure_json_post_request(req, parameters, &callback)
+      req_parameters = parameters.dup
+      if req.params["alt"].eql?("sse")
+        req.options.on_data = to_json_stream(&callback)
+      elsif req.params["alt"]
+        raise ArgumentError, "The alt parameter value must be a sse"
+      end
+
+      req.headers = headers
+      req.body = req_parameters.to_json
+    end
+
+    def try_parse_json(maybe_json)
+      JSON.parse(maybe_json)
+    rescue JSON::ParserError
+      maybe_json
+    end
+  end
+end
+
+

data/lib/gemini-ai/http_headers.rb

@@ -0,0 +1,16 @@
+module GeminiAi
+  module HTTPHeaders
+
+    private
+
+    def headers
+      default_headers = {
+        "Content-Type" => "application/json"
+      }
+      if @authentication == :service_account || @authentication == :default_credentials
+        default_headers.merge!("Authorization": "Bearer #{@authorizer.fetch_access_token!['access_token']}")
+      end
+      default_headers
+    end
+  end
+end

data/lib/gemini-ai/version.rb

@@ -0,0 +1,3 @@
+module GeminiAi
+  VERSION = "0.1.0".freeze
+end

data/lib/gemini-ai.rb

@@ -0,0 +1,87 @@
+require "faraday"
+
+require_relative "gemini-ai/http"
+require_relative "gemini-ai/client"
+require_relative "gemini-ai/errors"
+require_relative "gemini-ai/version"
+
+module GeminiAi
+  class Error < StandardError; end
+
+  class ConfigurationError < Error; end
+
+  class MiddlewareErrors < Faraday::Middleware
+    def call(env)
+      @app.call(env)
+    rescue Faraday::Error => e
+      raise e unless e.response.is_a?(Hash)
+
+      logger = Logger.new($stdout)
+      logger.formatter = proc do |_severity, _datetime, _progname, msg|
+        "\033[31mGemini AI HTTP Error (spotted in ruby-gemini-ai #{VERSION}): #{msg}\n\033[0m"
+      end
+      logger.error(e.response[:body])
+
+      raise e
+    end
+  end
+
+  class Configuration
+    attr_writer :api_key, :region, :file_path, :version, :service
+    DEFAULT_SERVICE_VERSION = 'v1'.freeze
+    DEFAULT_SERVICE = 'generative-language-api'.freeze
+
+    def initialize
+      @api_key = nil
+      @region = nil
+      @file_path = nil
+      @version = DEFAULT_SERVICE_VERSION
+      @service = DEFAULT_SERVICE
+    end
+
+    def api_key
+      @api_key
+    end
+
+    def service
+      unless %w[vertex-ai-api generative-language-api].include?(@service)
+        raise Errors::UnsupportedServiceError, "Unsupported service: #{@service}"
+      end
+      @service
+    end
+
+    def version
+      @version
+    end
+
+    def file_path
+      @file_path
+    end
+
+    def region
+      @region
+    end
+
+  end
+
+  class << self
+    attr_writer :configuration
+  end
+
+  def self.configuration
+    @configuration ||= GeminiAi::Configuration.new
+  end
+
+  def self.configure
+    yield(configuration)
+  end
+
+  def self.parsed_response(response, join_val: " ")
+    response.flat_map do |entry|
+      entry["candidates"].map do |candidate|
+        candidate.dig("content", "parts").map { |part| part["text"] }.join(join_val)
+      end
+    end.join(join_val)
+  end
+
+end

data/lib/ruby/gemini-ai.rb

@@ -0,0 +1,2 @@
+require_relative "../gemini-ai"
+require_relative "../gemini-ai/compatibility"

data/ruby-gemini-ai.gemspec

@@ -0,0 +1,31 @@
+require_relative "lib/gemini-ai/version"
+
+Gem::Specification.new do |spec|
+  spec.name          = "ruby-gemini-ai"
+  spec.version       = GeminiAi::VERSION
+  spec.authors       = ["Saurav Saini"]
+  spec.email         = ["sainisaurav019@gmai.com"]
+
+  spec.summary       = "Gemini API"
+  spec.homepage      = "https://github.com/sauravsaini98/ruby-gemini-ai"
+  spec.license       = "MIT"
+  spec.required_ruby_version = Gem::Requirement.new(">= 2.6.0")
+
+  spec.metadata["homepage_uri"] = spec.homepage
+  spec.metadata["source_code_uri"] = "https://github.com/sauravsaini98/ruby-gemini-ai"
+  spec.metadata["rubygems_mfa_required"] = "true"
+
+  # Specify which files should be added to the gem when it is released.
+  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
+  spec.files = Dir.chdir(File.expand_path(__dir__)) do
+    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  end
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "event_stream_parser", ">= 0.3.0", "< 2.0.0"
+  spec.add_dependency "faraday", ">= 1"
+  spec.add_dependency 'googleauth', '~> 1.8'
+
+end

metadata

@@ -0,0 +1,112 @@
+--- !ruby/object:Gem::Specification
+name: ruby-gemini-ai
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Saurav Saini
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2024-02-16 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: event_stream_parser
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.3.0
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: 2.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.3.0
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: 2.0.0
+- !ruby/object:Gem::Dependency
+  name: faraday
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1'
+- !ruby/object:Gem::Dependency
+  name: googleauth
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.8'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.8'
+description:
+email:
+- sainisaurav019@gmai.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".github/workflows/gem-push.yml"
+- ".gitignore"
+- ".rspec"
+- Gemfile
+- Gemfile.lock
+- LICENSE
+- README.md
+- bin/console
+- bin/setup
+- lib/gemini-ai.rb
+- lib/gemini-ai/client.rb
+- lib/gemini-ai/compatibility.rb
+- lib/gemini-ai/errors.rb
+- lib/gemini-ai/http.rb
+- lib/gemini-ai/http_headers.rb
+- lib/gemini-ai/version.rb
+- lib/ruby/gemini-ai.rb
+- ruby-gemini-ai.gemspec
+homepage: https://github.com/sauravsaini98/ruby-gemini-ai
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/sauravsaini98/ruby-gemini-ai
+  source_code_uri: https://github.com/sauravsaini98/ruby-gemini-ai
+  rubygems_mfa_required: 'true'
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.6.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.2.3
+signing_key:
+specification_version: 4
+summary: Gemini API
+test_files: []