gemini_craft 0.1.0 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fb9a603d6cf3b6c09120f38cc52368bed245f3d65f5189984e14814b91e1c207
-  data.tar.gz: f04c2337fd5a66396a80f2b712269705e624b275e494ddacd0e38d40ecf72d03
+  metadata.gz: 272eb66f29c87c7681a0f703ac2d3d697ef6ce23a6f067565179b623adfa6d85
+  data.tar.gz: 6ccdc9d479c0fa4f2d28b15d9e45c72115ae841f28d2163cff367875b565dfb3
 SHA512:
-  metadata.gz: cc32823fac5b9369b87f645126d5845386777bc31db783d08b8961fbd3dcda9d2b7a6ace8648d6cfbeffb22ed54025fc6cea1e3e35f57e283d3fba78338c6293
-  data.tar.gz: a5cd1b48fe3f4ba7a1b40b4b62822e8975e172683b3f4af7ca5d056e1759f33b5d6df5c4624d5dc35af2645048baaeb8ccd5e892efed94288c95264e7389b40e
+  metadata.gz: c9344da16d0a82761e10bcff5717ea5ca0c4b9dc31386e7814668875abcc2d2e3d916a95c18ef80154feaf4a659d7667a8e97f0f99c4ab39f70fe78f8487fa27
+  data.tar.gz: bd822c34dc245184c0fc7c4843898d93fc20c4dae4a0771df7be3ef9885712ecd4c1874446646952984bc639de4e340856f3d5f60249634a354b3bf1f899ea5b

data/lib/gemini_craft/cache.rb

@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+
+require "digest"
+
+module GeminiCraft
+  # Simple in-memory cache for API responses
+  class Cache
+    # Initialize the cache
+    # @param config [GeminiCraft::Configuration] Configuration object
+    def initialize(config)
+      @config = config
+      @store = {}
+      @timestamps = {}
+    end
+
+    # Get a value from the cache
+    # @param key [String] Cache key
+    # @return [String, nil] Cached value or nil if not found/expired
+    def get(key)
+      return nil unless @store.key?(key)
+
+      # Check if the entry has expired
+      if Time.now.to_i - @timestamps[key] > @config.cache_ttl
+        @store.delete(key)
+        @timestamps.delete(key)
+        return nil
+      end
+
+      @store[key]
+    end
+
+    # Set a value in the cache
+    # @param key [String] Cache key
+    # @param value [String] Value to cache
+    def set(key, value)
+      @store[key] = value
+      @timestamps[key] = Time.now.to_i
+    end
+
+    # Clear the entire cache
+    def clear
+      @store.clear
+      @timestamps.clear
+    end
+
+    # Remove expired entries from the cache
+    def cleanup
+      current_time = Time.now.to_i
+      @timestamps.each do |key, timestamp|
+        if current_time - timestamp > @config.cache_ttl
+          @store.delete(key)
+          @timestamps.delete(key)
+        end
+      end
+    end
+  end
+end

data/lib/gemini_craft/client.rb

@@ -0,0 +1,173 @@
+# frozen_string_literal: true
+
+require "faraday"
+require "faraday/retry"
+require "json"
+
+module GeminiCraft
+  # Client for interacting with the Gemini API
+  class Client
+    attr_reader :config, :cache
+
+    # Initialize a new client
+    def initialize
+      @config = GeminiCraft.configuration
+      @config.validate!
+      @cache = Cache.new(@config)
+    end
+
+    # Generate content using Gemini
+    # @param text [String] The text prompt to send to Gemini
+    # @param system_instruction [String, nil] Optional system instruction to guide the model
+    # @param options [Hash] Additional options for the request
+    # @return [String] The generated content
+    def generate_content(text, system_instruction = nil, options = {})
+      # Create cache key from the request parameters
+      cache_key = generate_cache_key(text, system_instruction, options)
+
+      # Check cache if enabled
+      if @config.cache_enabled && (cached_response = @cache.get(cache_key))
+        return cached_response
+      end
+
+      # Prepare request payload
+      payload = build_payload(text, system_instruction, options)
+
+      # Send request to API
+      response = make_request("models/#{@config.model}:generateContent", payload)
+
+      # Process response
+      content = extract_content(response)
+
+      # Cache response if enabled
+      @cache.set(cache_key, content) if @config.cache_enabled
+
+      content
+    end
+
+    private
+
+    # Build the API request payload
+    # @param text [String] The text prompt
+    # @param system_instruction [String, nil] Optional system instruction
+    # @param options [Hash] Additional options
+    # @return [Hash] The request payload
+    def build_payload(text, system_instruction, options)
+      payload = {
+        contents: [
+          {
+            parts: [
+              {
+                text: text
+              }
+            ]
+          }
+        ]
+      }
+
+      # Add system instruction if provided
+      if system_instruction
+        payload[:system_instruction] = {
+          parts: [
+            {
+              text: system_instruction
+            }
+          ]
+        }
+      end
+
+      # Merge additional options if provided
+      payload.merge!(options) if options && !options.empty?
+
+      payload
+    end
+
+    # Make a request to the Gemini API
+    # @param endpoint [String] API endpoint
+    # @param payload [Hash] Request payload
+    # @return [Hash] Parsed response
+    # @raise [GeminiCraft::APIError] If the API returns an error
+    def make_request(endpoint, payload)
+      response = connection.post(endpoint) do |req|
+        req.params["key"] = @config.api_key
+        req.headers["Content-Type"] = "application/json"
+        req.body = JSON.generate(payload)
+      end
+
+      handle_response(response)
+    rescue Faraday::Error => e
+      raise APIError, "API request failed: #{e.message}"
+    end
+
+    # Set up a Faraday connection with retry logic
+    # @return [Faraday::Connection] Configured connection
+    def connection
+      Faraday.new(url: @config.api_base_url) do |faraday|
+        faraday.options.timeout = @config.timeout
+        faraday.request :retry, max: @config.max_retries, interval: 0.5,
+                                interval_randomness: 0.5, backoff_factor: 2,
+                                exceptions: [Faraday::TimeoutError, Faraday::ConnectionFailed]
+        faraday.adapter Faraday.default_adapter
+      end
+    end
+
+    # Handle and parse the API response
+    # @param response [Faraday::Response] The API response
+    # @return [Hash] Parsed response body
+    # @raise [GeminiCraft::APIError] If the API returns an error
+    def handle_response(response)
+      case response.status
+      when 200
+        JSON.parse(response.body)
+      when 400..499
+        error_body = begin
+          JSON.parse(response.body)
+        rescue StandardError
+          { "error" => response.body }
+        end
+        raise APIError, "API client error (#{response.status}): #{error_body["error"]["message"] || "Unknown error"}"
+      when 500..599
+        raise APIError, "API server error (#{response.status}): The server encountered an error"
+      else
+        raise APIError, "Unknown API error (#{response.status})"
+      end
+    end
+
+    # Extract content from the response
+    # @param response [Hash] Parsed API response
+    # @return [String] The extracted content
+    # @raise [GeminiCraft::ResponseError] If the response format is unexpected
+    def extract_content(response)
+      candidates = response["candidates"]
+      return "" if candidates.nil? || candidates.empty?
+
+      content = candidates[0]["content"]
+      return "" if content.nil?
+
+      parts = content["parts"]
+      return "" if parts.nil? || parts.empty?
+
+      text = parts[0]["text"]
+      text || ""
+    rescue StandardError => e
+      raise ResponseError, "Failed to extract content from response: #{e.message}"
+    end
+
+    # Generate a cache key from request parameters
+    # @param text [String] The text prompt
+    # @param system_instruction [String, nil] Optional system instruction
+    # @param options [Hash] Additional options
+    # @return [String] A unique cache key
+    def generate_cache_key(text, system_instruction, options)
+      key_parts = [
+        @config.model,
+        text,
+        system_instruction,
+        options.to_s
+      ]
+
+      # Create a deterministic string from the key parts
+      Digest::SHA256.hexdigest(key_parts.join("--"))
+    end
+  end
+end

data/lib/gemini_craft/configuration.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module GeminiCraft
+  # Configuration for the GeminiCraft gem
+  class Configuration
+    attr_accessor :api_key, :api_base_url, :model, :timeout, :cache_enabled, :cache_ttl, :max_retries
+
+    # Initialize a new configuration with default values
+    def initialize
+      @api_key = ENV.fetch("GEMINI_API_KEY", nil)
+      @api_base_url = "https://generativelanguage.googleapis.com/v1beta"
+      @model = "gemini-2.0-flash"
+      @timeout = 30
+      @cache_enabled = false
+      @cache_ttl = 3600 # 1 hour in seconds
+      @max_retries = 3
+    end
+
+    # Validate that the configuration has required parameters
+    # @raise [GeminiCraft::ConfigurationError] if the configuration is invalid
+    def validate!
+      raise ConfigurationError, "API key must be configured" unless api_key
+      raise ConfigurationError, "Model must be configured" unless model
+    end
+  end
+
+  # Error raised when configuration is invalid
+  class ConfigurationError < Error; end
+end

data/lib/gemini_craft/error.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module GeminiCraft
+  # Base error class for all GeminiCraft errors
+  class Error < StandardError; end
+
+  # Error raised when there's an issue with API requests
+  class APIError < Error; end
+
+  # Error raised when the response cannot be processed
+  class ResponseError < Error; end
+end

data/lib/gemini_craft/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module GeminiCraft
-  VERSION = "0.1.0"
+  VERSION = "0.1.2"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: gemini_craft
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.2
 platform: ruby
 authors:
 - Shobhit Jain
@@ -158,19 +158,16 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- ".rspec"
-- ".rubocop.yml"
 - CHANGELOG.md
 - CODE_OF_CONDUCT.md
-- Gemfile
 - LICENSE.txt
 - README.md
-- Rakefile
-- bin/console
-- bin/setup
 - lib/gemini_craft.rb
+- lib/gemini_craft/cache.rb
+- lib/gemini_craft/client.rb
+- lib/gemini_craft/configuration.rb
+- lib/gemini_craft/error.rb
 - lib/gemini_craft/version.rb
-- sig/gemini_craft.rbs
 homepage: https://github.com/shobhits7/gemini_craft
 licenses:
 - MIT
@@ -178,6 +175,7 @@ metadata:
   homepage_uri: https://github.com/shobhits7/gemini_craft
   source_code_uri: https://github.com/shobhits7/gemini_craft
   changelog_uri: https://github.com/shobhits7/gemini_craft/blob/main/CHANGELOG.md
+  rubygems_mfa_required: 'true'
 post_install_message:
 rdoc_options: []
 require_paths:

data/.rspec

@@ -1,3 +0,0 @@
---format documentation
---color
---require spec_helper

data/.rubocop.yml

@@ -1,8 +0,0 @@
-AllCops:
-  TargetRubyVersion: 2.6.0
-
-Style/StringLiterals:
-  EnforcedStyle: double_quotes
-
-Style/StringLiteralsInInterpolation:
-  EnforcedStyle: double_quotes

data/Gemfile

@@ -1,13 +0,0 @@
-# frozen_string_literal: true
-
-source "https://rubygems.org"
-
-# Specify your gem's dependencies in gemini_craft.gemspec
-gemspec
-
-gem "irb"
-gem "rake", "~> 13.0"
-
-gem "rspec", "~> 3.0"
-
-gem "rubocop", "~> 1.21"

data/Rakefile

@@ -1,12 +0,0 @@
-# frozen_string_literal: true
-
-require "bundler/gem_tasks"
-require "rspec/core/rake_task"
-
-RSpec::Core::RakeTask.new(:spec)
-
-require "rubocop/rake_task"
-
-RuboCop::RakeTask.new
-
-task default: %i[spec rubocop]

data/bin/console

@@ -1,11 +0,0 @@
-#!/usr/bin/env ruby
-# frozen_string_literal: true
-
-require "bundler/setup"
-require "gemini_craft"
-
-# 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.
-
-require "irb"
-IRB.start(__FILE__)

data/bin/setup

@@ -1,8 +0,0 @@
-#!/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/sig/gemini_craft.rbs

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