tiny_gemini 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 424a3c3b460be94e67b592b32673d707754b683e53f3bddf4be64f37b7a0b9b9
+  data.tar.gz: 70d89f17147279c1741c0a77dc9ddc724109c8957bd5211ca1455c51d7c7e06c
+SHA512:
+  metadata.gz: 3aab600148ed5a4bbad01709de20cf73bf905c04482100c4875d4316c2b3f612de646841356f5ca76a89341b1a830ff444b1dd0e404cb407c3999a9d31ebda06
+  data.tar.gz: 6fb67b948f0ea36b183dca45fc7433dec1ed469b347fb07bf8f451e0c74bfcd145c92b9bbf1a21d4a90f26fac67e433c2589d0d1e33e8a1a500727bfb10f3a04

data/lib/tiny_gemini.rb

@@ -0,0 +1,77 @@
+require 'net/http'
+require 'json'
+
+# tiny, not very robust, HTTP client the Google Gemini API
+# see also: https://ai.google.dev/api/rest
+class TinyGemini
+  # a good rule of thumb would be to have a .tiny_gemini.yml config in your
+  # project, to parse that as YAML, and pass the parsed reulst into here.
+  def initialize(
+    model: 'gemini-1.5-flash',
+    host: 'generativelanguage.googleapis.com',
+    path: '/v1beta/models',
+    action: 'generateContent',
+    api_key: ENV['GEMINI_KEY'],
+    system_instruction: nil
+  )
+
+    @model = model
+    @host = host
+    @path = path
+    @action = action
+    @api_key = api_key
+    @system_instruction = system_instruction
+  end
+
+  # sends a request to POST generateContent
+  #
+  # messages: an array of hashes in the following format:
+  # [
+  #     {
+  #         "parts": [
+  #             {
+  #                 "text": "hi, how are you?"
+  #             }
+  #         ],
+  #         "role": "user"
+  #     },
+  #     {
+  #         "parts": [
+  #             {
+  #                 "text": "I am an AI language model, so I don't have feelings or experiences like humans do. However, I am here to assist you with any questions or tasks you may have! How can I help you today??"
+  #             }
+  #         ],
+  #         "role": "model"
+  #     },
+  #     {
+  #         "parts": [
+  #             {
+  #                 "text": "oh, you don't have feelings? I guess I didn't realize that - you seem so ... real!"
+  #             }
+  #         ],
+  #         "role": "user"
+  #     },
+  # ]
+  #
+  # NOTE: if you want the model to impersonate a character (i.e. a talking cat)
+  # you need to have set the `system_instruction` parameter when initializing
+  # this class to make that work
+  def chat(messages)
+    body = { contents: messages }
+    body.merge!(system_instruction: { parts: { text: @system_instruction } }) if @system_instruction
+    request_body = body.to_json
+
+    uri = URI("https://#{@host}#{@path}/#{@model}:#{@action}?key=#{@api_key}")
+    headers = { 'Content-Type' => 'application/json; charset=UTF-8' }
+    response = Net::HTTP.post(uri, request_body, headers)
+
+    # Handle potential errors (e.g., non-200 responses)
+    unless response.is_a?(Net::HTTPSuccess)
+      raise TinyGeminiModelError, "Gemini API Error: #{response.code}\n#{JSON.pretty_generate(response.body)}"
+    end
+
+    JSON.parse(response.body)['candidates'].first.dig('content', 'parts').first['text']
+  end
+end
+
+class TinyGeminiModelError < StandardError; end

metadata

@@ -0,0 +1,43 @@
+--- !ruby/object:Gem::Specification
+name: tiny_gemini
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Jeff Lunt
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2024-07-09 00:00:00.000000000 Z
+dependencies: []
+description: a tiny gem for using Google Gemini's generative models for chat
+email: jefflunt@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/tiny_gemini.rb
+homepage: https://github.com/jefflunt/tiny_gemini
+licenses:
+- MIT
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.4.19
+signing_key:
+specification_version: 4
+summary: a tiny gem for using Google Gemini's generative models for chat
+test_files: []