geminize 0.1.1 → 1.1.0

data/lib/geminize/request_builder.rb

@@ -52,6 +52,35 @@ module Geminize
       def build_streaming_endpoint(model_name)
         build_model_endpoint(model_name, "streamGenerateContent")
       end
+
+      # Build a models list request parameters
+      # @param page_size [Integer, nil] Number of models to return per page
+      # @param page_token [String, nil] Token for retrieving a specific page
+      # @return [Hash] The query parameters for models listing
+      def build_models_list_params(page_size = nil, page_token = nil)
+        params = {}
+        params[:pageSize] = page_size if page_size
+        params[:pageToken] = page_token if page_token
+        params
+      end
+
+      # Build a specific model endpoint for the get model info API
+      # @param model_name [String] The model name or ID to get info for
+      # @return [String] The complete API endpoint path for getting model info
+      def build_get_model_endpoint(model_name)
+        # Handle both formats: "models/gemini-1.5-pro" or just "gemini-1.5-pro"
+        unless model_name.start_with?("models/")
+          model_name = "models/#{model_name}"
+        end
+        model_name
+      end
+
+      # Build the endpoint for embedding generation
+      # @param model_name [String] The name of the model
+      # @return [String] The complete API endpoint path for embedding generation
+      def build_embedding_endpoint(model_name)
+        build_model_endpoint(model_name, "embedContent")
+      end
     end
   end
 end

data/lib/geminize/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Geminize
-  VERSION = "0.1.1"
+  VERSION = "1.1.0"
 end

data/lib/geminize.rb

@@ -37,9 +37,6 @@ require_relative "geminize/conversation_repository"
 require_relative "geminize/conversation_service"
 require_relative "geminize/model_info"
 
-# Conditionally load Rails integration if Rails is defined
-require_relative "geminize/rails" if defined?(::Rails)
-
 # Main module for the Geminize gem
 module Geminize
   class Error < StandardError; end
@@ -481,35 +478,92 @@ module Geminize
       )
     end
 
-    # Get a list of available models from the Gemini API
+    # List available models from the Gemini API
+    # @param page_size [Integer, nil] Number of models to return per page (max 1000)
+    # @param page_token [String, nil] Token for retrieving a specific page
     # @param force_refresh [Boolean] Force a refresh from the API instead of using cache
     # @param client_options [Hash] Options to pass to the client
     # @return [Geminize::Models::ModelList] List of available models
     # @raise [Geminize::GeminizeError] If the request fails
-    # @example Get a list of all available models
-    #   models = Geminize.list_models
-    # @example Get a fresh list bypassing cache
-    #   models = Geminize.list_models(force_refresh: true)
-    # @example Filter models by capability
-    #   vision_models = Geminize.list_models.vision_models
-    def list_models(force_refresh: false, client_options: {})
+    def list_models(page_size: nil, page_token: nil, force_refresh: false, client_options: {})
       validate_configuration!
       model_info = ModelInfo.new(nil, client_options)
-      model_info.list_models(force_refresh: force_refresh)
+      model_info.list_models(page_size: page_size, page_token: page_token, force_refresh: force_refresh)
+    end
+
+    # Get all available models, handling pagination automatically
+    # @param force_refresh [Boolean] Force a refresh from the API instead of using cache
+    # @param client_options [Hash] Options to pass to the client
+    # @return [Geminize::Models::ModelList] Complete list of all available models
+    # @raise [Geminize::GeminizeError] If the request fails
+    def list_all_models(force_refresh: false, client_options: {})
+      validate_configuration!
+      model_info = ModelInfo.new(nil, client_options)
+      model_info.list_all_models(force_refresh: force_refresh)
     end
 
     # Get information about a specific model
-    # @param model_id [String] The model ID to retrieve
+    # @param model_name [String] The model name to retrieve (can be "models/gemini-1.5-pro" or just "gemini-1.5-pro")
     # @param force_refresh [Boolean] Force a refresh from the API instead of using cache
     # @param client_options [Hash] Options to pass to the client
     # @return [Geminize::Models::Model] The model information
     # @raise [Geminize::GeminizeError] If the request fails or model is not found
-    # @example Get information about a specific model
-    #   model = Geminize.get_model("gemini-1.5-pro")
-    def get_model(model_id, force_refresh: false, client_options: {})
+    def get_model(model_name, force_refresh: false, client_options: {})
+      validate_configuration!
+      model_info = ModelInfo.new(nil, client_options)
+      model_info.get_model(model_name, force_refresh: force_refresh)
+    end
+
+    # Get models that support a specific generation method
+    # @param method [String] The generation method (e.g., "generateContent", "embedContent")
+    # @param force_refresh [Boolean] Force a refresh from the API instead of using cache
+    # @param client_options [Hash] Options to pass to the client
+    # @return [Geminize::Models::ModelList] List of models that support the method
+    # @raise [Geminize::GeminizeError] If the request fails
+    def get_models_by_method(method, force_refresh: false, client_options: {})
       validate_configuration!
       model_info = ModelInfo.new(nil, client_options)
-      model_info.get_model(model_id, force_refresh: force_refresh)
+      model_info.get_models_by_method(method, force_refresh: force_refresh)
+    end
+
+    # Get models that support content generation
+    # @param force_refresh [Boolean] Force a refresh from the API instead of using cache
+    # @param client_options [Hash] Options to pass to the client
+    # @return [Geminize::Models::ModelList] List of models that support content generation
+    # @raise [Geminize::GeminizeError] If the request fails
+    def get_content_generation_models(force_refresh: false, client_options: {})
+      models = list_all_models(force_refresh: force_refresh, client_options: client_options)
+      models.content_generation_models
+    end
+
+    # Get models that support embedding generation
+    # @param force_refresh [Boolean] Force a refresh from the API instead of using cache
+    # @param client_options [Hash] Options to pass to the client
+    # @return [Geminize::Models::ModelList] List of models that support embedding generation
+    # @raise [Geminize::GeminizeError] If the request fails
+    def get_embedding_models(force_refresh: false, client_options: {})
+      models = list_all_models(force_refresh: force_refresh, client_options: client_options)
+      models.embedding_models
+    end
+
+    # Get models that support chat generation
+    # @param force_refresh [Boolean] Force a refresh from the API instead of using cache
+    # @param client_options [Hash] Options to pass to the client
+    # @return [Geminize::Models::ModelList] List of models that support chat generation
+    # @raise [Geminize::GeminizeError] If the request fails
+    def get_chat_models(force_refresh: false, client_options: {})
+      models = list_all_models(force_refresh: force_refresh, client_options: client_options)
+      models.chat_models
+    end
+
+    # Get models that support streaming generation
+    # @param force_refresh [Boolean] Force a refresh from the API instead of using cache
+    # @param client_options [Hash] Options to pass to the client
+    # @return [Geminize::Models::ModelList] List of models that support streaming generation
+    # @raise [Geminize::GeminizeError] If the request fails
+    def get_streaming_models(force_refresh: false, client_options: {})
+      models = list_all_models(force_refresh: force_refresh, client_options: client_options)
+      models.streaming_models
     end
 
     # Update a conversation's system instruction

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: geminize
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 1.1.0
 platform: ruby
 authors:
 - Nhat Long Nguyen
@@ -123,16 +123,57 @@ dependencies:
         version: '2.8'
 description: A convenient and robust Ruby interface for the Google Gemini API, enabling
   easy integration of powerful generative AI models into your applications. Includes
-  support for text generation, chat conversations, embeddings, multimodal content,
-  and Rails integration.
+  support for text generation, chat conversations, embeddings, multimodal content.
 email:
 - nhatlongnguyen1992@gmail.com
 executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- ".cursor/mcp.json"
+- ".cursor/rules/isolation_rules/Core/command-execution.mdc"
+- ".cursor/rules/isolation_rules/Core/complexity-decision-tree.mdc"
+- ".cursor/rules/isolation_rules/Core/creative-phase-enforcement.mdc"
+- ".cursor/rules/isolation_rules/Core/creative-phase-metrics.mdc"
+- ".cursor/rules/isolation_rules/Core/file-verification.mdc"
+- ".cursor/rules/isolation_rules/Core/platform-awareness.mdc"
+- ".cursor/rules/isolation_rules/Level3/planning-comprehensive.mdc"
+- ".cursor/rules/isolation_rules/Level3/task-tracking-intermediate.mdc"
+- ".cursor/rules/isolation_rules/Phases/CreativePhase/creative-phase-architecture.mdc"
+- ".cursor/rules/isolation_rules/main.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/archive-mode-map.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/creative-mode-map.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/implement-mode-map.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/plan-mode-map.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/qa-mode-map.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/reflect-mode-map.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/van-mode-map.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/van_mode_split/van-complexity-determination.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/van_mode_split/van-file-verification.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/van_mode_split/van-mode-map.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/van_mode_split/van-platform-detection.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/van_mode_split/van-qa-checks/build-test.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/van_mode_split/van-qa-checks/config-check.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/van_mode_split/van-qa-checks/dependency-check.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/van_mode_split/van-qa-checks/environment-check.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/van_mode_split/van-qa-checks/file-verification.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/van_mode_split/van-qa-main.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/van_mode_split/van-qa-utils/common-fixes.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/van_mode_split/van-qa-utils/mode-transitions.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/van_mode_split/van-qa-utils/reports.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/van_mode_split/van-qa-utils/rule-calling-guide.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/van_mode_split/van-qa-utils/rule-calling-help.mdc"
+- ".cursor/rules/isolation_rules/visual-maps/van_mode_split/van-qa-validation.md.old"
+- ".env.example"
+- ".memory_bank/activeContext.md"
+- ".memory_bank/progress.md"
+- ".memory_bank/projectbrief.md"
+- ".memory_bank/systemPatterns.md"
+- ".memory_bank/tasks.md"
+- ".memory_bank/techContext.md"
 - ".rspec"
 - ".standard.yml"
+- ".tool-versions"
 - ".yardopts"
 - CHANGELOG.md
 - CODE_OF_CONDUCT.md
@@ -143,15 +184,10 @@ files:
 - examples/README.md
 - examples/configuration.rb
 - examples/embeddings.rb
+- examples/models_api.rb
 - examples/multimodal.rb
-- examples/rails_chat/README.md
-- examples/rails_chat/app/controllers/chat_controller.rb
-- examples/rails_chat/app/views/chat/index.html.erb
-- examples/rails_chat/config/routes.rb
-- examples/rails_initializer.rb
 - examples/ruby.png
 - examples/system_instructions.rb
-- geminize-0.1.0.gem
 - lib/geminize.rb
 - lib/geminize/chat.rb
 - lib/geminize/client.rb
@@ -176,21 +212,11 @@ files:
 - lib/geminize/models/model.rb
 - lib/geminize/models/model_list.rb
 - lib/geminize/models/stream_response.rb
-- lib/geminize/rails.rb
-- lib/geminize/rails/app/controllers/concerns/geminize/controller.rb
-- lib/geminize/rails/app/helpers/geminize_helper.rb
-- lib/geminize/rails/controller_additions.rb
-- lib/geminize/rails/engine.rb
-- lib/geminize/rails/helper_additions.rb
-- lib/geminize/railtie.rb
 - lib/geminize/request_builder.rb
 - lib/geminize/text_generation.rb
 - lib/geminize/validators.rb
 - lib/geminize/vector_utils.rb
 - lib/geminize/version.rb
-- lib/generators/geminize/install_generator.rb
-- lib/generators/geminize/templates/README
-- lib/generators/geminize/templates/initializer.rb
 - sig/geminize.rbs
 homepage: https://github.com/nhlongnguyen/geminize
 licenses:
@@ -214,7 +240,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.8
+rubygems_version: 3.6.7
 specification_version: 4
 summary: Ruby interface for Google's Gemini AI API
 test_files: []

data/examples/rails_chat/README.md

@@ -1,69 +0,0 @@
-# Rails Chat Example
-
-This is a basic example of how to implement a chat application with the Geminize gem in a Rails application.
-
-## Features
-
-- Chat interface with Google's Gemini AI
-- Conversation memory between requests using Rails session
-- Support for Turbo and JSON responses
-- Reset conversation functionality
-
-## Implementation
-
-This example demonstrates:
-
-1. How to include Geminize controller concerns
-2. How to use Geminize view helpers
-3. How to maintain conversation state in a Rails application
-
-## File Structure
-
-- `app/controllers/chat_controller.rb` - Controller with Geminize integration
-- `app/views/chat/index.html.erb` - Chat interface view
-- `config/routes.rb` - Routes configuration
-
-## Setup in Your Application
-
-1. Add Geminize to your Gemfile:
-
-```ruby
-gem 'geminize'
-```
-
-2. Run the installer:
-
-```bash
-rails generate geminize:install
-```
-
-3. Configure your API key in `config/initializers/geminize.rb`
-
-4. Include the Geminize helpers in your `ApplicationHelper`
-
-```ruby
-module ApplicationHelper
-  geminize_helper
-end
-```
-
-5. Create your controller with Geminize support
-
-```ruby
-class YourChatController < ApplicationController
-  geminize_controller
-
-  # ... your controller actions
-end
-```
-
-6. Use the helper methods in your views
-
-```erb
-<%= render_gemini_conversation %>
-<%= gemini_chat_form %>
-```
-
-## Styling
-
-This example includes basic CSS to style the chat interface. Feel free to customize it to match your application's design.

data/examples/rails_chat/app/controllers/chat_controller.rb

@@ -1,26 +0,0 @@
-# frozen_string_literal: true
-
-class ChatController < ApplicationController
-  # Include Geminize controller functionality
-  geminize_controller
-
-  def index
-    # Reset conversation if requested
-    reset_gemini_conversation("Chat with Gemini") if params[:reset]
-  end
-
-  def create
-    @response = send_gemini_message(params[:message])
-
-    respond_to do |format|
-      format.html { redirect_to chat_path }
-      format.turbo_stream
-      format.json { render json: {message: @response.text} }
-    end
-  end
-
-  def reset
-    reset_gemini_conversation("New Chat Session")
-    redirect_to chat_path, notice: "Started a new conversation"
-  end
-end

data/examples/rails_chat/app/views/chat/index.html.erb

@@ -1,112 +0,0 @@
-<%# Chat index view %>
-<div class="chat-container">
-  <div class="chat-header">
-    <h1>Chat with Gemini</h1>
-    <%= link_to "Start New Chat", reset_chat_path, class: "new-chat-button", data: { turbo_method: :post } %>
-  </div>
-
-  <div class="conversation-container">
-    <div class="conversation">
-      <%= render_gemini_conversation(nil, include_timestamps: true) %>
-    </div>
-  </div>
-
-  <div class="chat-form-container">
-    <%= gemini_chat_form(
-      placeholder: "Ask me anything...",
-      submit_text: "Send",
-      form_class: "chat-form",
-      input_class: "chat-input",
-      submit_class: "chat-submit"
-    ) %>
-  </div>
-</div>
-
-<% content_for :styles do %>
-<style>
-  .chat-container {
-    max-width: 800px;
-    margin: 0 auto;
-    padding: 20px;
-    font-family: system-ui, -apple-system, sans-serif;
-  }
-
-  .chat-header {
-    display: flex;
-    justify-content: space-between;
-    align-items: center;
-    margin-bottom: 20px;
-  }
-
-  .new-chat-button {
-    background: #6200ee;
-    color: white;
-    padding: 8px 16px;
-    border-radius: 4px;
-    text-decoration: none;
-  }
-
-  .conversation-container {
-    height: 400px;
-    overflow-y: auto;
-    border: 1px solid #e0e0e0;
-    border-radius: 8px;
-    margin-bottom: 20px;
-    padding: 16px;
-    background: #f8f9fa;
-  }
-
-  .gemini-message {
-    margin-bottom: 16px;
-    padding: 12px;
-    border-radius: 8px;
-  }
-
-  .gemini-user-message {
-    background: #e3f2fd;
-    margin-left: 20px;
-  }
-
-  .gemini-ai-message {
-    background: #f1f3f4;
-    margin-right: 20px;
-  }
-
-  .gemini-message-role {
-    font-weight: bold;
-    margin-bottom: 8px;
-  }
-
-  .gemini-message-timestamp {
-    font-size: 0.8em;
-    color: #666;
-    margin-top: 8px;
-    text-align: right;
-  }
-
-  .chat-form {
-    display: flex;
-    gap: 8px;
-  }
-
-  .chat-input {
-    flex: 1;
-    padding: 12px;
-    border: 1px solid #e0e0e0;
-    border-radius: 4px;
-    font-family: inherit;
-    resize: none;
-    height: 80px;
-  }
-
-  .chat-submit {
-    background: #6200ee;
-    color: white;
-    border: none;
-    padding: 12px 24px;
-    border-radius: 4px;
-    cursor: pointer;
-    align-self: flex-end;
-  }
-</style>
-<% end %>

data/examples/rails_chat/config/routes.rb

@@ -1,8 +0,0 @@
-# frozen_string_literal: true
-
-Rails.application.routes.draw do
-  # Chat routes
-  get "chat", to: "chat#index"
-  post "chat", to: "chat#create"
-  post "chat/reset", to: "chat#reset", as: :reset_chat
-end

data/examples/rails_initializer.rb

@@ -1,46 +0,0 @@
-# frozen_string_literal: true
-
-# config/initializers/geminize.rb
-
-# This is an example of how to configure the Geminize gem in a Rails application.
-# Place this file in your config/initializers directory.
-
-# Configure the Geminize gem with your API key and any other settings
-Geminize.configure do |config|
-  # Required settings
-
-  # Get the API key from environment variables (recommended approach)
-  # config.api_key = ENV["GEMINI_API_KEY"]
-
-  # Or set it directly (not recommended for production)
-  config.api_key = "your-api-key-here"
-
-  # Optional settings with defaults
-
-  # API version to use
-  config.api_version = "v1beta"
-
-  # Default model to use when not specified in requests
-  config.default_model = "gemini-1.5-pro-latest"
-
-  # Request timeout in seconds
-  config.timeout = 30
-
-  # Connection open timeout in seconds
-  config.open_timeout = 10
-
-  # Enable request logging for development/debugging
-  # In production, keep this false unless debugging an issue
-  config.log_requests = Rails.env.development?
-end
-
-# Optionally validate the configuration during app initialization
-# This will raise Geminize::ConfigurationError if the configuration is invalid
-begin
-  Geminize.validate_configuration!
-  Rails.logger.info "Geminize configured successfully"
-rescue Geminize::ConfigurationError => e
-  Rails.logger.error "Geminize configuration error: #{e.message}"
-  # You might want to raise the error in development but not in production
-  raise e if Rails.env.development?
-end

data/lib/geminize/rails/app/controllers/concerns/geminize/controller.rb

@@ -1,105 +0,0 @@
-# frozen_string_literal: true
-
-module Geminize
-  # Controller concern for including Gemini functionality in Rails controllers
-  module Controller
-    extend ActiveSupport::Concern
-
-    included do
-      # Helper methods for controllers using Geminize
-      helper_method :current_gemini_conversation if respond_to?(:helper_method)
-    end
-
-    # Current conversation for this controller context
-    # Uses session to maintain conversation state between requests
-    # @return [Geminize::Models::Conversation]
-    def current_gemini_conversation
-      return @current_gemini_conversation if defined?(@current_gemini_conversation)
-
-      if session[:gemini_conversation_id]
-        # Try to load existing conversation
-        begin
-          @current_gemini_conversation = Geminize.load_conversation(session[:gemini_conversation_id])
-        rescue => e
-          Rails.logger.error("Failed to load Gemini conversation: #{e.message}")
-          # Create a new conversation if loading fails
-          create_new_gemini_conversation
-        end
-      else
-        # Create a new conversation if one doesn't exist
-        create_new_gemini_conversation
-      end
-
-      @current_gemini_conversation
-    end
-
-    # Send a message in the current conversation
-    # @param message [String] The message to send
-    # @param model_name [String, nil] Optional model name override
-    # @param params [Hash] Additional parameters for the message
-    # @return [Geminize::Models::ChatResponse] The response
-    def send_gemini_message(message, model_name = nil, params = {})
-      response = Geminize.chat(message, current_gemini_conversation, model_name, params)
-      Geminize.save_conversation(current_gemini_conversation)
-      response
-    end
-
-    # Generate text using Gemini
-    # @param prompt [String] The prompt to send
-    # @param model_name [String, nil] Optional model name override
-    # @param params [Hash] Additional parameters for generation
-    # @return [Geminize::Models::ContentResponse] The response
-    def generate_gemini_text(prompt, model_name = nil, params = {})
-      Geminize.generate_text(prompt, model_name, params)
-    end
-
-    # Generate text with images using Gemini
-    # @param prompt [String] The prompt to send
-    # @param images [Array<Hash>] Array of image data
-    # @param model_name [String, nil] Optional model name override
-    # @param params [Hash] Additional parameters for generation
-    # @return [Geminize::Models::ContentResponse] The response
-    def generate_gemini_multimodal(prompt, images, model_name = nil, params = {})
-      Geminize.generate_text_multimodal(prompt, images, model_name, params)
-    end
-
-    # Generate embeddings for text using Gemini
-    # @param text [String, Array<String>] The text to embed
-    # @param model_name [String, nil] Optional model name override
-    # @param params [Hash] Additional parameters for embedding
-    # @return [Geminize::Models::EmbeddingResponse] The embedding response
-    def generate_gemini_embedding(text, model_name = nil, params = {})
-      Geminize.generate_embedding(text, model_name, params)
-    end
-
-    # Start a new conversation and store its ID in the session
-    # @param title [String, nil] Optional title for the conversation
-    # @param system_instruction [String, nil] Optional system instruction to guide model behavior
-    # @return [Geminize::Models::Conversation] The new conversation
-    def reset_gemini_conversation(title = nil, system_instruction = nil)
-      create_new_gemini_conversation(title, system_instruction)
-    end
-
-    # Set the system instruction for the current conversation
-    # @param system_instruction [String] The system instruction to set
-    # @return [Geminize::Models::Conversation] The updated conversation
-    def set_gemini_system_instruction(system_instruction)
-      current_gemini_conversation.system_instruction = system_instruction
-      Geminize.save_conversation(current_gemini_conversation)
-      current_gemini_conversation
-    end
-
-    private
-
-    # Create a new conversation and store its ID in the session
-    # @param title [String, nil] Optional title for the conversation
-    # @param system_instruction [String, nil] Optional system instruction to guide model behavior
-    # @return [Geminize::Models::Conversation] The new conversation
-    def create_new_gemini_conversation(title = nil, system_instruction = nil)
-      @current_gemini_conversation = Geminize.create_chat(title, system_instruction)
-      session[:gemini_conversation_id] = @current_gemini_conversation.id
-      Geminize.save_conversation(@current_gemini_conversation)
-      @current_gemini_conversation
-    end
-  end
-end