gemini_craft 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fb9a603d6cf3b6c09120f38cc52368bed245f3d65f5189984e14814b91e1c207
-  data.tar.gz: f04c2337fd5a66396a80f2b712269705e624b275e494ddacd0e38d40ecf72d03
+  metadata.gz: 3cb129b8cc115605e729b88b19abac03e7809dc4d7f652dbd2c33a3afa14c602
+  data.tar.gz: 3da67f63ceca11564f2f2a11715adf3eeae491bc93d150111ad8768065c33edd
 SHA512:
-  metadata.gz: cc32823fac5b9369b87f645126d5845386777bc31db783d08b8961fbd3dcda9d2b7a6ace8648d6cfbeffb22ed54025fc6cea1e3e35f57e283d3fba78338c6293
-  data.tar.gz: a5cd1b48fe3f4ba7a1b40b4b62822e8975e172683b3f4af7ca5d056e1759f33b5d6df5c4624d5dc35af2645048baaeb8ccd5e892efed94288c95264e7389b40e
+  metadata.gz: 3012eda7aa4d40600343f993ad397ed7fe9474f790f1e3a56f74b6c1b3e5be2da3541e51d874a0e4a4e7e7e688b6e8e47b84fb12655ed138566133a8c8867513
+  data.tar.gz: f1b10bf9dd7cf3098b8f65893af8bfd81c3c1b8ded920028fcb895f6f354ca93b4d553cc9ea9046ce2594915bff592157e0c9ecd9e0b692e9a7357ef8059e634

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["GEMINI_API_KEY"]
+      @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.1"
 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.1
 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

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