gemini_client 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: fa79c9c23c71cae2d43dc94db1c8ca92faec83c7458ca3fa99c4b985a3017ce3
+  data.tar.gz: 04ae660a3b939ae5867ffc1ba8fe217cc587b92acd0c62d86fb79cf9927a802e
+SHA512:
+  metadata.gz: bb659e7159c0b9d72cc9a21f6fbd57e36d5b6ae70773f727a5b51a0a96fc45a15bfce5efc41c8a65bcc62271080b5721ceb9c10aa92ecb2f4eeda49b16f3d5ce
+  data.tar.gz: 40e46bd1a1e8b4dcd325542ccd7a5994c76189732a34f8abf60088b93517b3e996461cbaafcdc979d928f9b1135cd27cd956ec80962268bac275e3a2def7724b

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2025 i2bskn
+
+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,54 @@
+# GeminiClient
+
+Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/gemini_client`. To experiment with that code, run `bin/console` for an interactive prompt.
+
+## Installation
+
+Install the gem and add to the application's Gemfile by executing:
+
+    $ bundle add gemini_client
+
+If bundler is not being used to manage dependencies, install the gem by executing:
+
+    $ gem install gemini_client
+
+## Usage
+
+```
+client = GeminiClient.new(api_key: GEMINI_API_KEY)
+payload = {
+  contents: [
+    {
+      "role": "user",
+      "parts": [
+        {
+          "text": "hello"
+        }
+      ]
+    }
+  ]
+}
+res = client.generate_content(payload: payload)
+data = JSON.parse(res.body)
+pp data
+```
+
+```
+{"candidates"=>[{"content"=>{"parts"=>[{"text"=>"Hello there! How can I help you today?\n"}], "role"=>"model"}, "finishReason"=>"STOP", "avgLogprobs"=>-0.0006325314752757549}],
+ "usageMetadata"=>{"promptTokenCount"=>2, "candidatesTokenCount"=>11, "totalTokenCount"=>13},
+ "modelVersion"=>"gemini-1.5-flash"}
+```
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. 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/i2bskn/gemini_client.
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).

data/Rakefile

@@ -0,0 +1,4 @@
+# frozen_string_literal: true
+
+require "bundler/gem_tasks"
+task default: %i[]

data/lib/gemini_client/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+class GeminiClient
+  VERSION = "0.1.0"
+end

data/lib/gemini_client.rb

@@ -0,0 +1,83 @@
+# frozen_string_literal: true
+
+require "net/http"
+require "json"
+
+require_relative "gemini_client/version"
+
+class GeminiClient
+  class Error < StandardError; end
+
+  module Model
+    GEMINI_1_5_FLASH = "gemini-1.5-flash"
+  end
+
+  module Method
+    GENERATE_CONTENT = "generateContent"
+  end
+
+  API_VERSION = "v1beta".freeze
+
+  def initialize(api_key:, version: API_VERSION, model: Model::GEMINI_1_5_FLASH)
+    @api_key = api_key
+    @version = version
+    @model = model
+  end
+
+  def generate_content(payload:, model: nil)
+    url = api_url(api_key: @api_key, version: @version, model: model || @model, method_name: Method::GENERATE_CONTENT)
+    request(:post, url, payload: JSON.dump(payload), headers: { "Content-Type" => "application/json" })
+  end
+
+  def stream_generate_content
+    # TODO
+  end
+
+  def count_tokens
+    # TODO
+  end
+
+  def embed_content
+    # TODO
+  end
+
+  def models
+    # TODO
+  end
+
+  def model_info(model)
+    # TODO
+  end
+
+  private
+
+    def api_url(api_key:, version:, model: nil, method_name: nil)
+      model_and_method = [model, method_name].compact.join(":")
+      "https://generativelanguage.googleapis.com/#{version}/models/#{model_and_method}?key=#{api_key}"
+    end
+
+    def request(meth, url, options = {})
+      url = URI.parse(url)
+      # query strings
+      url.query = URI.encode_www_form(options.fetch(:query)) if options.key?(:query)
+      req = Net::HTTP.const_get(meth.to_s.capitalize).new(url.request_uri)
+      # request http headers
+      (options[:headers] || {}).each { |k, v| req[k.to_s] = v }
+      # Authorization
+      req["Authorization"] = options.fetch(:auth) if options.key?(:auth)
+      # Basic Authentication
+      req.basic_auth(options.fetch(:user), options.fetch(:password)) if options.key?(:user) && options.key?(:password)
+      case
+      when options.key?(:form)
+        # form data
+        req.set_form_data(options.fetch(:form))
+      when options.key?(:payload)
+        # payload
+        req.body = options.fetch(:payload)
+      end
+
+      http = Net::HTTP.new(url.host, url.port)
+      http.use_ssl = url.is_a?(URI::HTTPS)
+      http.request(req)
+    end
+end

data/sig/gemini_client.rbs

@@ -0,0 +1,4 @@
+class GeminiClient
+  VERSION: String
+  # See the writing guide of rbs: https://github.com/ruby/rbs#guides
+end

metadata

@@ -0,0 +1,66 @@
+--- !ruby/object:Gem::Specification
+name: gemini_client
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- i2bskn
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2025-01-03 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.15.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.15.0
+description: Gemini client is a client library for Gemini API.
+email:
+- i2bskn@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- LICENSE.txt
+- README.md
+- Rakefile
+- lib/gemini_client.rb
+- lib/gemini_client/version.rb
+- sig/gemini_client.rbs
+homepage: https://github.com/i2bskn/gemini_client
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/i2bskn/gemini_client
+  source_code_uri: https://github.com/i2bskn/gemini_client
+  changelog_uri: https://github.com/i2bskn/gemini_client/blob/master/CHANGELOG.md
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.0.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.5.9
+signing_key:
+specification_version: 4
+summary: Gemini client is a client library for Gemini API.
+test_files: []